kb-server 0.0.1-beta.22 → 0.0.1-beta.24

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

@@ -14,10 +14,6 @@ operations: {
      * 向客户端推送消息
      */
     send: (event: string, message: string) => void;
-    /**
-     * 推送完毕回调
-     */
-    done: () => void;
 }) => Promise<R>;
 export type AnySseExecution = SseExecution<any, any, any>;
 export type SseHandlers = Record<string, AnySseExecution>;

package/dist/common/sse-middleware.js

@@ -6,18 +6,20 @@ 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 sseMap = new Map(Object.entries(sseHandlers).map(([action, execution]) => [
-        action,
-        execution,
-    ]));
     const { authFn, log = true, route = "/sse" } = options || {};
     return async (req, res, next) => {
+        // 生成SSE API映射
+        const sseMap = new Map(Object.entries(sseHandlers).map(([action, execution]) => [
+            action,
+            execution,
+        ]));
+        // 检查SSE路由
         const isValidRoute = route.startsWith("/");
         if (!isValidRoute) {
             throw new create_errors_1.CommonErrors.InvalidParameter.RouteError();
         }
         const { path } = req;
-        // SSE的路径直接往后走
+        // 非SSE的请求直接放过
         if (path !== route) {
             next();
             return;
@@ -60,30 +62,18 @@ const packSSE = (sseHandlers, options) => {
             if (typeof execution !== "function") {
                 throw new create_errors_1.CommonErrors.ResourceNotFound.APINotFound();
             }
-            /**
-             * 开始处理SSE请求就要统计
-             *
-             * 请求处理完毕的时间
-             *
-             * 不要让 API 随意改写 req res 里面的东西
-             *
-             * 统一管理请求头
-             *
-             */
+            // 写请求头
             res.writeHead(200, {
                 "Content-Type": "text/event-stream",
                 "Cache-Control": "no-cache",
                 Connection: "keep-alive",
             });
-            await new Promise((resolve) => {
-                const send = (event, message) => {
-                    const response = (0, exports.createSSEMessage)(event, message);
-                    res.write(response);
-                    logger_1.logger.log("info", `发送消息：\n${response}`);
-                };
-                const done = () => resolve(true);
-                execution(params, ctx, { send, done });
-            });
+            const send = (event, message) => {
+                const response = (0, exports.createSSEMessage)(event, message);
+                res.write(response);
+                logger_1.logger.log("info", `发送消息：\n${response}`);
+            };
+            await execution(params, ctx, { send });
             // 完成响应
             took = Date.now() - start;
             logger_1.logger.log("info", `耗时：${took} ms - RequestId: ${requestId}`);

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export { createServer } from "./common/create-server";
 export { createErrors } from "./common/create-errors";
 export { implementAPI, ServerContext } from "./common/create-api";
+export { implementSseAPI } from "./common/create-sse";
 export { callService } from "./common/call-service";

package/dist/index.js

@@ -1,11 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.callService = exports.implementAPI = exports.createErrors = exports.createServer = void 0;
+exports.callService = exports.implementSseAPI = exports.implementAPI = exports.createErrors = exports.createServer = void 0;
 var create_server_1 = require("./common/create-server");
 Object.defineProperty(exports, "createServer", { enumerable: true, get: function () { return create_server_1.createServer; } });
 var create_errors_1 = require("./common/create-errors");
 Object.defineProperty(exports, "createErrors", { enumerable: true, get: function () { return create_errors_1.createErrors; } });
 var create_api_1 = require("./common/create-api");
 Object.defineProperty(exports, "implementAPI", { enumerable: true, get: function () { return create_api_1.implementAPI; } });
+var create_sse_1 = require("./common/create-sse");
+Object.defineProperty(exports, "implementSseAPI", { enumerable: true, get: function () { return create_sse_1.implementSseAPI; } });
 var call_service_1 = require("./common/call-service");
 Object.defineProperty(exports, "callService", { enumerable: true, get: function () { return call_service_1.callService; } });

package/package.json

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