kb-server 0.0.1-beta.41 → 0.0.1-beta.42

package/dist/common/create-server.d.ts

@@ -23,6 +23,12 @@ export interface ICreateServerParams {
          * 默认为：`/sse`
          */
         route?: string;
+        /**
+         * 超时时间（单位：毫秒）
+         *
+         * @description 每次推送消息会重置超时时间
+         */
+        timeout?: number;
     };
     /**
      * 鉴权函数

package/dist/common/create-server.js

@@ -33,7 +33,7 @@ const sse_middleware_1 = require("./sse-middleware");
 function createServer(params, options) {
     const { apis, sse, authFn, log, middlewares = [] } = params || {};
     const { limit = "10mb" } = options || {};
-    const { handlers, route } = sse || {};
+    const { handlers, ...resetSSE } = sse || {};
     const app = (0, express_1.default)();
     // POST 参数获取
     app.use((0, express_1.urlencoded)({ extended: true, limit }));
@@ -41,7 +41,7 @@ function createServer(params, options) {
     // 使用自定义中间件
     middlewares.forEach((middleware) => app.use(middleware));
     // 注入SSE
-    handlers && app.use((0, sse_middleware_1.packSSE)(handlers, { route, log }));
+    handlers && app.use((0, sse_middleware_1.packSSE)(handlers, { log, ...resetSSE }));
     // 注入API
     app.use((0, api_middleware_1.packAPI)(apis, { authFn, log }));
     return app;

package/dist/common/sse-middleware.d.ts

@@ -5,6 +5,7 @@ interface IPackSSEOptions {
     authFn?: AuthFunction;
     route?: string;
     log?: boolean;
+    timeout?: number;
 }
 export declare const packSSE: (sseHandlers: SseHandlers, options?: IPackSSEOptions) => express.RequestHandler;
 export declare const createSSEMsg: (event: string, message: string) => string;

package/dist/common/sse-middleware.js

@@ -6,7 +6,7 @@ const logger_1 = require("../helper/logger");
 const create_errors_1 = require("./create-errors");
 const short_id_1 = require("../helper/short-id");
 const packSSE = (sseHandlers, options) => {
-    const { authFn, log = true, route = "/sse" } = options || {};
+    const { authFn, log = true, route = "/sse", timeout = 1000 * 30, } = options || {};
     return async (req, res, next) => {
         // 生成SSE API映射
         const sseMap = new Map(Object.entries(sseHandlers).map(([action, execution]) => [
@@ -63,9 +63,12 @@ const packSSE = (sseHandlers, options) => {
                 throw new create_errors_1.CommonErrors.ResourceNotFound.APINotFound();
             }
             const abortController = new AbortController();
+            let timeoutId;
             // 连接断开
             let isConnected = true;
             res.on("close", () => {
+                if (timeoutId)
+                    clearTimeout(timeoutId);
                 isConnected = false;
                 abortController.abort();
                 // 完成响应
@@ -79,8 +82,18 @@ const packSSE = (sseHandlers, options) => {
                 "Cache-Control": "no-cache",
                 Connection: "keep-alive",
             });
+            // 初始化超时定时器
+            const resetTimeout = () => {
+                if (timeoutId)
+                    clearTimeout(timeoutId);
+                timeoutId = setTimeout(() => {
+                    logger_1.logger.info(`超时自动关闭：${requestId}`);
+                    close();
+                }, timeout);
+            };
             // 推送消息
             const push = (data) => {
+                resetTimeout();
                 if (!isConnected) {
                     logger_1.logger.warning(`连接已关闭： ${requestId}`);
                     return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kb-server",
-  "version": "0.0.1-beta.41",
+  "version": "0.0.1-beta.42",
   "description": "A fast server for Node.JS,made by express.",
   "main": "dist/index.js",
   "scripts": {