eee

src/Base/SessionMiddleware.php

@@ -0,0 +1,45 @@
+<?php
+declare(strict_types=1);
+
+namespace Kiri\Router\Base;
+
+use Kiri\Router\Session;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Server\MiddlewareInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+
+/**
+ * Session 中间件
+ * 自动启动和保存 Session
+ */
+class SessionMiddleware implements MiddlewareInterface
+{
+
+    /**
+     * @param ServerRequestInterface $request
+     * @param RequestHandlerInterface $handler
+     * @return ResponseInterface
+     * @throws
+     */
+    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
+    {
+        // 启动 Session
+        Session::start($request);
+        
+        try {
+            // 处理请求
+            $response = $handler->handle($request);
+            
+            // 保存 Session
+            Session::save();
+            
+            return $response;
+        } catch (\Throwable $e) {
+            // 即使出错也保存 Session
+            Session::save();
+            throw $e;
+        }
+    }
+}
+