imean-service-engine 1.5.0 → 1.7.0

package/dist/mod.cjs

@@ -7,6 +7,7 @@ var nodeServer = require('@hono/node-server');
 var etcd3 = require('etcd3');
 var fs = require('fs-extra');
 var hono = require('hono');
+var timing = require('hono/timing');
 var api = require('@opentelemetry/api');
 var winston = require('winston');
 var prettier = require('prettier');
@@ -17,6 +18,8 @@ var mcp_js = require('@modelcontextprotocol/sdk/server/mcp.js');
 var types_js = require('@modelcontextprotocol/sdk/types.js');
 var streaming = require('hono/streaming');
 var ulid = require('ulid');
+var html = require('hono/html');
+var jsxRuntime = require('hono/jsx/jsx-runtime');
 var dayjs = require('dayjs');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
@@ -595,6 +598,62 @@ var ActionHandler = class {
 function isAsyncIterable(obj) {
   return obj != null && typeof obj[Symbol.asyncIterator] === "function";
 }
+
+// decorators/page.ts
+var PAGE_METADATA = Symbol("page:metadata");
+function Page(options) {
+  return function(_target, context) {
+    const methodName = context.name;
+    context.addInitializer(function() {
+      const prototype = this.constructor.prototype;
+      const existingMetadata = prototype[PAGE_METADATA] || {};
+      existingMetadata[methodName] = {
+        name: methodName,
+        description: options.description || "",
+        method: options.method,
+        path: options.path
+      };
+      prototype[PAGE_METADATA] = existingMetadata;
+    });
+  };
+}
+function getPageMetadata(target) {
+  return target.constructor.prototype[PAGE_METADATA] ?? {};
+}
+var tracer3 = api.trace.getTracer("page-handler");
+var PageHandler = class {
+  constructor(moduleInstance, options, moduleName) {
+    this.moduleInstance = moduleInstance;
+    this.options = options;
+    this.moduleName = moduleName;
+  }
+  async handle(ctx) {
+    return await tracer3.startActiveSpan(
+      `handle ${this.moduleName}.${this.options.name}`,
+      async (span) => {
+        span.setAttribute("module", this.moduleName);
+        span.setAttribute("page", this.options.name);
+        span.setAttribute("path", this.options.path);
+        try {
+          const result = await this.moduleInstance[this.options.name].apply(
+            this.moduleInstance,
+            [ctx]
+          );
+          return ctx.html(result);
+        } catch (error) {
+          span.recordException(error);
+          span.setStatus({
+            code: api.SpanStatusCode.ERROR,
+            message: error.message
+          });
+          throw error;
+        } finally {
+          span.end();
+        }
+      }
+    );
+  }
+};
 var WebSocketHandler = class {
   constructor(microservice, options) {
     this.microservice = microservice;
@@ -779,10 +838,12 @@ var Microservice = class {
   lease;
   scheduler;
   abortController;
-  isShuttingDown = false;
   statisticsTimer;
   wsHandler;
   actionHandlers = /* @__PURE__ */ new Map();
+  pageHandlers = /* @__PURE__ */ new Map();
+  activeRequests = /* @__PURE__ */ new Map();
+  status = "running";
   modules = /* @__PURE__ */ new Map();
   fetch;
   options;
@@ -790,6 +851,7 @@ var Microservice = class {
   serviceId;
   constructor(options) {
     this.app = new hono.Hono();
+    this.app.use(timing.timing());
     this.nodeWebSocket = nodeWs.createNodeWebSocket({ app: this.app });
     this.serviceId = crypto.randomUUID();
     this.options = {
@@ -805,6 +867,10 @@ var Microservice = class {
       websocket: { enabled: false },
       cacheAdapter: new MemoryCacheAdapter(),
       plugins: [],
+      gracefulShutdown: {
+        timeout: 3e4,
+        cleanupHooks: []
+      },
       ...options
     };
     this.cache = this.options.cacheAdapter;
@@ -824,6 +890,11 @@ var Microservice = class {
     }
     await this.registerService(true);
     await this.initPlugins();
+    this.app.get(this.options.prefix, (ctx) => {
+      const name = this.options.name ?? "Microservice";
+      const version = this.options.version ?? "1.0.0";
+      return ctx.text(`${name} is ${this.status}. version: ${version}`);
+    });
   }
   async initModules() {
     for (const ModuleClass of this.options.modules) {
@@ -838,6 +909,7 @@ var Microservice = class {
       logger_default.info(`[ \u6CE8\u518C\u6A21\u5757 ] ${moduleName} ${metadata.options.description}`);
       this.modules.set(moduleName, moduleInstance);
       const actions = getActionMetadata(ModuleClass.prototype);
+      const pages = getPageMetadata(ModuleClass.prototype);
       for (const [actionName, actionMetadata] of Object.entries(actions)) {
         const handler = new ActionHandler(
           moduleInstance,
@@ -851,6 +923,18 @@ var Microservice = class {
           `[ \u6CE8\u518C\u52A8\u4F5C ] ${moduleName}.${actionName} ${actionMetadata.description} ${actionMetadata.mcp ? "MCP:" + actionMetadata.mcp?.type : ""}`
         );
       }
+      for (const [_, page] of Object.entries(pages)) {
+        const handler = new PageHandler(
+          moduleInstance,
+          page,
+          moduleName
+        );
+        this.pageHandlers.set(`${moduleName}.${page.name}`, handler);
+        this.app[page.method](`${this.options.prefix}${page.path}`, (ctx) => handler.handle(ctx));
+        logger_default.info(
+          `[ \u6CE8\u518C\u9875\u9762 ] ${moduleName}.${page.name} ${page.method.toUpperCase()} ${page.path} ${page.description}`
+        );
+      }
       const schedules = getScheduleMetadata(ModuleClass.prototype);
       if (schedules && Object.keys(schedules).length > 0) {
         if (!this.scheduler && this.etcdClient) {
@@ -880,11 +964,6 @@ var Microservice = class {
   initRoutes() {
     const startTime = Date.now();
     const prefix = this.options.prefix || "/api";
-    this.app.get(prefix, (ctx) => {
-      const name = this.options.name ?? "Microservice";
-      const version = this.options.version ?? "1.0.0";
-      return ctx.text(`${name} is running. version: ${version}`);
-    });
     this.app.get(`${prefix}/health`, (ctx) => {
       return ctx.json({
         status: "ok",
@@ -899,7 +978,9 @@ var Microservice = class {
         version: this.options.version,
         env: this.options.env,
         modules: this.getModules(),
-        stats: Object.fromEntries(this.statsMap)
+        stats: Object.fromEntries(this.statsMap),
+        activeRequests: this.getActiveRequestCount(),
+        status: this.status
       });
     });
     this.app.get(`${prefix}/client.ts`, async (ctx) => {
@@ -1096,7 +1177,12 @@ var Microservice = class {
     process.on("SIGINT", () => {
       logger_default.info(`
 Received SIGINT signal`);
-      this.gracefulShutdown();
+      this.shutdown();
+    });
+    process.on("SIGTERM", () => {
+      logger_default.info(`
+Received SIGTERM signal`);
+      this.shutdown();
     });
     process.on("unhandledrejection", (event) => {
       logger_default.error("Unhandled rejection:", event.reason);
@@ -1108,14 +1194,16 @@ Received SIGINT signal`);
   /**
    * 优雅停机
    */
-  async gracefulShutdown() {
-    if (this.isShuttingDown) return;
-    this.isShuttingDown = true;
-    logger_default.info("\nGraceful shutdown initiated...");
+  async shutdown() {
+    if (this.status === "shutting_down") return;
+    this.status = "shutting_down";
+    logger_default.info("\nshutdown initiated...");
     if (this.statisticsTimer) clearInterval(this.statisticsTimer);
     try {
+      await this.waitForActiveRequests();
+      await this.executeCleanupHooks();
       await this.stop();
-      logger_default.info("Graceful shutdown completed");
+      logger_default.info("shutdown completed");
     } catch (error) {
       logger_default.error("Error during shutdown:", error);
       process.exit(1);
@@ -1123,6 +1211,57 @@ Received SIGINT signal`);
       process.exit(0);
     }
   }
+  /**
+   * 等待所有活跃请求完成
+   */
+  async waitForActiveRequests() {
+    const timeout = this.options.gracefulShutdown?.timeout || 3e4;
+    const startTime = Date.now();
+    logger_default.info(`Waiting for ${this.activeRequests.size} active requests to complete...`);
+    return new Promise((resolve) => {
+      const checkInterval = setInterval(() => {
+        const elapsed = Date.now() - startTime;
+        if (this.activeRequests.size === 0) {
+          clearInterval(checkInterval);
+          logger_default.info("All active requests completed");
+          resolve();
+        } else if (elapsed >= timeout) {
+          clearInterval(checkInterval);
+          logger_default.warn(`Timeout waiting for requests to complete. ${this.activeRequests.size} requests still active`);
+          resolve();
+        } else {
+          logger_default.info(`Still waiting for ${this.activeRequests.size} requests... (${elapsed}ms elapsed)`);
+        }
+      }, 1e3);
+    });
+  }
+  /**
+   * 执行清理hook
+   */
+  async executeCleanupHooks() {
+    const hooks = this.options.gracefulShutdown?.cleanupHooks || [];
+    if (hooks.length === 0) {
+      logger_default.info("No cleanup hooks configured");
+      return;
+    }
+    logger_default.info(`Executing ${hooks.length} cleanup hooks...`);
+    for (const hook of hooks) {
+      try {
+        const timeout = hook.timeout || 5e3;
+        logger_default.info(`Executing cleanup hook: ${hook.name}`);
+        await Promise.race([
+          Promise.resolve(hook.cleanup()),
+          new Promise(
+            (_, reject) => setTimeout(() => reject(new Error(`Cleanup hook ${hook.name} timeout`)), timeout)
+          )
+        ]);
+        logger_default.info(`Cleanup hook ${hook.name} completed successfully`);
+      } catch (error) {
+        logger_default.error(`Cleanup hook ${hook.name} failed:`, error);
+      }
+    }
+    logger_default.info("All cleanup hooks completed");
+  }
   initStatsEventManager() {
     this.statisticsTimer = setInterval(async () => {
       await this.updateStats();
@@ -1164,14 +1303,54 @@ Received SIGINT signal`);
     }
     return handler;
   }
+  /**
+   * 添加活跃请求跟踪
+   */
+  addActiveRequest(requestId, requestInfo) {
+    this.activeRequests.set(requestId, requestInfo);
+  }
+  /**
+   * 移除活跃请求跟踪
+   */
+  removeActiveRequest(requestId) {
+    this.activeRequests.delete(requestId);
+  }
+  /**
+   * 获取当前活跃请求数量
+   */
+  getActiveRequestCount() {
+    return this.activeRequests.size;
+  }
+  /**
+   * 获取当前活跃请求信息
+   */
+  getActiveRequests() {
+    return Array.from(this.activeRequests.values());
+  }
   handleRequest = async (ctx) => {
     const { moduleName, actionName } = ctx.req.param();
     const handler = this.getActionHandler(moduleName, actionName);
+    const requestId = crypto.randomUUID();
+    const startTime = Date.now();
+    if (this.status === "shutting_down") {
+      return ctx.json({
+        success: false,
+        error: "Service is shutting down"
+      }, 503);
+    }
     try {
       const paramsText = await ctx.req.text();
+      this.addActiveRequest(requestId, {
+        id: requestId,
+        moduleName,
+        actionName,
+        startTime,
+        params: paramsText
+      });
       const result = await handler.handle(paramsText);
       if (handler.metadata.stream) {
         const encoder = new TextEncoder();
+        const microservice = this;
         const stream = new ReadableStream({
           async start(controller) {
             try {
@@ -1195,6 +1374,8 @@ Received SIGINT signal`);
                 encoder.encode(ejson4__default.default.stringify(response) + "\n")
               );
               controller.close();
+            } finally {
+              microservice.removeActiveRequest(requestId);
             }
           }
         });
@@ -1206,8 +1387,10 @@ Received SIGINT signal`);
           }
         });
       }
+      this.removeActiveRequest(requestId);
       return ctx.text(ejson4__default.default.stringify({ success: true, data: result }));
     } catch (error) {
+      this.removeActiveRequest(requestId);
       return ctx.json({ success: false, error: error.message });
     }
   };
@@ -1223,28 +1406,6 @@ Received SIGINT signal`);
     await this.waitingInitialization;
   }
 };
-
-// utils/checker.ts
-async function startCheck(checkers, pass) {
-  logger_default.info("[ \u9884\u68C0\u5F00\u59CB ]");
-  for (const [index, checker] of checkers.entries()) {
-    const seq = index + 1;
-    logger_default.info(`${seq}. ${checker.name}`);
-    try {
-      if (checker.skip) {
-        logger_default.warn(`${seq}. ${checker.name} [\u8DF3\u8FC7]`);
-        continue;
-      }
-      await checker.check();
-      logger_default.info(`${seq}. ${checker.name} [\u6210\u529F]`);
-    } catch (error) {
-      logger_default.error(`${seq}. ${checker.name} [\u5931\u8D25]`);
-      throw error;
-    }
-  }
-  logger_default.info("[ \u9884\u68C0\u5B8C\u6210 ]");
-  if (pass) await pass();
-}
 var HonoTransport = class {
   constructor(url, stream, closeStream) {
     this.url = url;
@@ -1360,22 +1521,221 @@ var ModelContextProtocolPlugin = class extends Plugin {
     );
   };
 };
+var DEFAULT_FAVICON = /* @__PURE__ */ jsxRuntime.jsx(
+  "link",
+  {
+    rel: "icon",
+    href: "data:image/svg+xml;utf8,<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 100 100' width='100' height='100'><defs><linearGradient id='nodeGradient' x1='0%' y1='0%' x2='100%' y2='100%'><stop offset='0%' stop-color='%233498db'/><stop offset='100%' stop-color='%232980b9'/></linearGradient><linearGradient id='centerNodeGradient' x1='0%' y1='0%' x2='100%' y2='100%'><stop offset='0%' stop-color='%232ecc71'/><stop offset='100%' stop-color='%2327ae60'/></linearGradient></defs><circle cx='50' cy='50' r='45' fill='%23f5f7fa'/><path d='M30,30 L50,50' stroke='%23bdc3c7' stroke-width='2' stroke-linecap='round'/><path d='M70,30 L50,50' stroke='%23bdc3c7' stroke-width='2' stroke-linecap='round'/><path d='M30,70 L50,50' stroke='%23bdc3c7' stroke-width='2' stroke-linecap='round'/><path d='M70,70 L50,50' stroke='%23bdc3c7' stroke-width='2' stroke-linecap='round'/><polygon points='30,15 45,25 45,45 30,55 15,45 15,25' fill='url(%23nodeGradient)' stroke='%232980b9' stroke-width='1.5'/><polygon points='70,15 85,25 85,45 70,55 55,45 55,25' fill='url(%23nodeGradient)' stroke='%232980b9' stroke-width='1.5'/><polygon points='30,45 45,55 45,75 30,85 15,75 15,55' fill='url(%23nodeGradient)' stroke='%232980b9' stroke-width='1.5'/><polygon points='70,45 85,55 85,75 70,85 55,75 55,55' fill='url(%23nodeGradient)' stroke='%232980b9' stroke-width='1.5'/><polygon points='50,30 65,40 65,60 50,70 35,60 35,40' fill='url(%23centerNodeGradient)' stroke='%2327ae60' stroke-width='2'/><circle cx='30' cy='30' r='3' fill='%23ffffff'/><circle cx='70' cy='30' r='3' fill='%23ffffff'/><circle cx='30' cy='70' r='3' fill='%23ffffff'/><circle cx='70' cy='70' r='3' fill='%23ffffff'/><circle cx='50' cy='50' r='4' fill='%23ffffff'/></svg>",
+    type: "image/svg+xml"
+  }
+);
+var BaseLayout = (props = {
+  title: "Microservice Template"
+}) => html.html`<!DOCTYPE html>
+    <html>
+      <head>
+        <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+        <title>${props.title}</title>
+        ${props.heads}
+      </head>
+      <body>
+        ${props.children}
+      </body>
+    </html>`;
+var HtmxLayout = (props = {
+  title: "Microservice Template"
+}) => BaseLayout({
+  title: props.title,
+  heads: html.html`
+      <script src="https://unpkg.com/htmx.org@latest"></script>
+      <script src="https://unpkg.com/hyperscript.org@latest"></script>
+      <script src="https://cdn.tailwindcss.com"></script>
+      ${props.favicon || DEFAULT_FAVICON}
+    `,
+  children: props.children
+});
+var InfoCard = ({
+  icon,
+  iconColor,
+  bgColor,
+  label,
+  value
+}) => /* @__PURE__ */ jsxRuntime.jsxs("div", { className: `${bgColor} p-4 rounded-lg`, children: [
+  /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center mb-2", children: [
+    /* @__PURE__ */ jsxRuntime.jsx(
+      "svg",
+      {
+        className: `w-5 h-5 ${iconColor} mr-2`,
+        fill: "none",
+        stroke: "currentColor",
+        viewBox: "0 0 24 24",
+        children: /* @__PURE__ */ jsxRuntime.jsx(
+          "path",
+          {
+            strokeLinecap: "round",
+            strokeLinejoin: "round",
+            strokeWidth: 2,
+            d: icon
+          }
+        )
+      }
+    ),
+    /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-sm font-medium text-gray-600", children: label })
+  ] }),
+  /* @__PURE__ */ jsxRuntime.jsx("p", { className: `text-xl font-semibold text-gray-900`, children: value })
+] });
+var getEnvironmentBadgeClass = (env) => {
+  switch (env) {
+    case "prod":
+      return "bg-red-100 text-red-800";
+    case "stg":
+      return "bg-yellow-100 text-yellow-800";
+    case "dev":
+    default:
+      return "bg-blue-100 text-blue-800";
+  }
+};
+var ServiceInfoCards = ({
+  serviceInfo
+}) => {
+  const infoCards = [
+    {
+      icon: "M7 4V2a1 1 0 011-1h8a1 1 0 011 1v2m-9 0h10m-10 0a2 2 0 00-2 2v14a2 2 0 002 2h10a2 2 0 002-2V6a2 2 0 00-2-2",
+      iconColor: "text-blue-600",
+      bgColor: "bg-blue-50",
+      label: "\u670D\u52A1\u540D\u79F0",
+      value: serviceInfo.name
+    },
+    {
+      icon: "M13.828 10.172a4 4 0 00-5.656 0l-4 4a4 4 0 105.656 5.656l1.102-1.101m-.758-4.899a4 4 0 005.656 0l4-4a4 4 0 00-5.656-5.656l-1.1 1.1",
+      iconColor: "text-orange-600",
+      bgColor: "bg-orange-50",
+      label: "\u670D\u52A1\u8DEF\u5F84",
+      value: serviceInfo.prefix || "/",
+      isMonospace: true
+    },
+    {
+      icon: "M9 12l2 2 4-4m6 2a9 9 0 11-18 0 9 9 0 0118 0z",
+      iconColor: "text-green-600",
+      bgColor: "bg-green-50",
+      label: "\u8FD0\u884C\u73AF\u5883",
+      value: /* @__PURE__ */ jsxRuntime.jsx(
+        "span",
+        {
+          className: `px-2 py-1 rounded-full text-sm ${getEnvironmentBadgeClass(serviceInfo.env ?? "dev")}`,
+          children: serviceInfo.env ?? "dev"
+        }
+      )
+    },
+    {
+      icon: "M7 7h.01M7 3h5c.512 0 1.024.195 1.414.586l7 7a2 2 0 010 2.828l-7 7a2 2 0 01-2.828 0l-7-7A1.994 1.994 0 013 12V7a4 4 0 014-4z",
+      iconColor: "text-purple-600",
+      bgColor: "bg-purple-50",
+      label: "\u7248\u672C\u53F7",
+      value: serviceInfo.version || "unknown"
+    }
+  ];
+  return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "bg-white rounded-lg shadow-md p-6 mb-8", children: [
+    /* @__PURE__ */ jsxRuntime.jsxs("h2", { className: "text-2xl font-semibold text-gray-800 mb-6 flex items-center", children: [
+      /* @__PURE__ */ jsxRuntime.jsx(
+        "svg",
+        {
+          className: "w-6 h-6 mr-2 text-blue-600",
+          fill: "none",
+          stroke: "currentColor",
+          viewBox: "0 0 24 24",
+          children: /* @__PURE__ */ jsxRuntime.jsx(
+            "path",
+            {
+              strokeLinecap: "round",
+              strokeLinejoin: "round",
+              strokeWidth: 2,
+              d: "M13 10V3L4 14h7v7l9-11h-7z"
+            }
+          )
+        }
+      ),
+      "\u670D\u52A1\u57FA\u672C\u4FE1\u606F"
+    ] }),
+    /* @__PURE__ */ jsxRuntime.jsx("div", { className: "grid grid-cols-1 md:grid-cols-2 lg:grid-cols-2 gap-6", children: infoCards.map((card, index) => /* @__PURE__ */ jsxRuntime.jsx(InfoCard, { ...card }, index)) })
+  ] });
+};
+var ServiceStatusPage = ({
+  serviceInfo
+}) => {
+  return /* @__PURE__ */ jsxRuntime.jsx("div", { className: "min-h-screen bg-gray-50 py-8", children: /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "max-w-6xl mx-auto px-4", children: [
+    /* @__PURE__ */ jsxRuntime.jsx("div", { className: "mb-8", children: /* @__PURE__ */ jsxRuntime.jsx("h1", { className: "text-4xl font-bold text-gray-900 mb-2", children: "Service Status" }) }),
+    /* @__PURE__ */ jsxRuntime.jsx(ServiceInfoCards, { serviceInfo })
+  ] }) });
+};
+var ServiceStatusPage_default = ServiceStatusPage;
+
+// core/plugins/page/mod.ts
+var PageRenderPlugin = class extends Plugin {
+  initialize = async (engine) => {
+    const app = engine.getApp();
+    app.get(`${engine.options.prefix}`, async (ctx) => {
+      return ctx.html(HtmxLayout({
+        title: engine.options.name,
+        children: ServiceStatusPage_default({
+          serviceInfo: {
+            name: engine.options.name,
+            prefix: engine.options.prefix,
+            version: engine.options.version,
+            env: engine.options.env,
+            id: engine.serviceId,
+            modules: engine.getModules(false)
+          }
+        })
+      }));
+    });
+    logger_default.info(`PageRenderPlugin enabled`);
+  };
+};
+
+// utils/checker.ts
+async function startCheck(checkers, pass) {
+  logger_default.info("[ \u9884\u68C0\u5F00\u59CB ]");
+  for (const [index, checker] of checkers.entries()) {
+    const seq = index + 1;
+    logger_default.info(`${seq}. ${checker.name}`);
+    try {
+      if (checker.skip) {
+        logger_default.warn(`${seq}. ${checker.name} [\u8DF3\u8FC7]`);
+        continue;
+      }
+      await checker.check();
+      logger_default.info(`${seq}. ${checker.name} [\u6210\u529F]`);
+    } catch (error) {
+      logger_default.error(`${seq}. ${checker.name} [\u5931\u8D25]`);
+      throw error;
+    }
+  }
+  logger_default.info("[ \u9884\u68C0\u5B8C\u6210 ]");
+  if (pass) await pass();
+}
 
 Object.defineProperty(exports, "dayjs", {
   enumerable: true,
   get: function () { return dayjs__default.default; }
 });
 exports.Action = Action;
+exports.BaseLayout = BaseLayout;
 exports.CacheAdapter = CacheAdapter;
+exports.HtmxLayout = HtmxLayout;
 exports.MemoryCacheAdapter = MemoryCacheAdapter;
 exports.Microservice = Microservice;
 exports.ModelContextProtocolPlugin = ModelContextProtocolPlugin;
 exports.Module = Module;
+exports.Page = Page;
+exports.PageRenderPlugin = PageRenderPlugin;
 exports.Plugin = Plugin;
 exports.RedisCacheAdapter = RedisCacheAdapter;
 exports.Schedule = Schedule;
 exports.ScheduleMode = ScheduleMode;
 exports.ServiceContext = ServiceContext;
+exports.ServiceInfoCards = ServiceInfoCards;
+exports.ServiceStatusPage = ServiceStatusPage;
 exports.logger = logger_default;
 exports.startCheck = startCheck;
 Object.keys(zod).forEach(function (k) {