@thinkbun/plugin 1.0.3 → 1.0.5

package/README.md

@@ -17,6 +17,7 @@ Plugin 包依赖于 ThinkBun Core 包，为其提供扩展功能。这些插件
 增强应用程序的日志功能，支持彩色输出、作用域标识和不同日志级别。
 
 #### 功能特性
+
 - 彩色日志输出，提高可读性
 - 支持日志作用域标识
 - 自定义日志级别
@@ -68,6 +69,7 @@ app.logger.warn('警告信息');
 - `app.logger.warn(message, scope)`: 输出警告级别的日志
 
 **参数**：
+
 - `message`: 日志消息内容
 - `scope`: 可选，日志作用域，可以是字符串或字符串数组
 
@@ -76,6 +78,7 @@ app.logger.warn('警告信息');
 提供服务器启动和集群支持功能，优化应用程序的性能和可靠性。
 
 #### 功能特性
+
 - 基于 Bun.serve 的高性能服务器
 - 自动集群管理，充分利用多核 CPU
 - 工作进程自动重启
@@ -134,6 +137,7 @@ export default {
 #### 集群模式
 
 ServerPlugin 自动支持集群模式：
+
 - 主进程管理工作进程的创建和重启
 - 工作进程处理实际的 HTTP 请求
 - 当工作进程意外退出时，自动创建新的工作进程
@@ -143,19 +147,19 @@ ServerPlugin 自动支持集群模式：
 
 ### LoggerPlugin 生命周期
 
-| 阶段 | 执行顺序 | 功能 |
-|------|----------|------|
-| `setup` | `pre` | 初始化日志功能，替换默认日志器 |
-| `run` | - | 无 |
-| `close` | - | 无 |
+| 阶段    | 执行顺序 | 功能                           |
+| ------- | -------- | ------------------------------ |
+| `setup` | `pre`    | 初始化日志功能，替换默认日志器 |
+| `run`   | -        | 无                             |
+| `close` | -        | 无                             |
 
 ### ServerPlugin 生命周期
 
-| 阶段 | 执行顺序 | 功能 |
-|------|----------|------|
-| `setup` | - | 无 |
-| `run` | `post` | 启动服务器，根据配置创建工作进程 |
-| `close` | - | 优雅关闭服务器和工作进程 |
+| 阶段    | 执行顺序 | 功能                             |
+| ------- | -------- | -------------------------------- |
+| `setup` | -        | 无                               |
+| `run`   | `post`   | 启动服务器，根据配置创建工作进程 |
+| `close` | -        | 优雅关闭服务器和工作进程         |
 
 ## 独立运行/测试
 
@@ -207,6 +211,7 @@ export const CustomPlugin: Plugin = {
 ## 插件执行顺序
 
 插件的执行顺序由 `enforce` 属性决定：
+
 - `pre`: 在核心功能之前执行
 - `post`: 在核心功能之后执行
 - 省略: 默认执行顺序
@@ -302,4 +307,4 @@ export default {
 
 ---
 
-**ThinkBun Plugin** - 为 ThinkBun 框架提供强大的扩展功能 🧩
+**ThinkBun Plugin** - 为 ThinkBun 框架提供强大的扩展功能 🧩

package/dist/index.d.mts

@@ -1,9 +1,18 @@
 import { Plugin } from "@thinkbun/core";
 
 //#region src/logger.d.ts
+
+/**
+ * 日志插件，为应用添加统一的日志功能
+ * 支持不同作用域的彩色日志输出
+ */
 declare const LoggerPlugin: Plugin;
 //#endregion
 //#region src/server.d.ts
+/**
+ * 服务器插件，提供集群管理和 HTTP 服务功能
+ * 支持多进程模式和工作进程自动重启
+ */
 declare const ServerPlugin: Plugin;
 //#endregion
 export { LoggerPlugin, ServerPlugin };

package/dist/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","names":[],"sources":["../src/logger.ts","../src/server.ts"],"sourcesContent":[],"mappings":";;;cAOa,cAAc;;;cCEd,cAAc"}
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/logger.ts","../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;AAoDA;;cAAa,cAAc;;;;;AAA3B;;cC4Ha,cAAc"}

package/dist/index.mjs

@@ -1,86 +1,229 @@
-import { invariant } from "@thinkbun/core";
+import { LOG_COLORS, invariant } from "@thinkbun/core";
 import pc from "picocolors";
 import cluster from "node:cluster";
 import os from "node:os";
 
 //#region src/logger.ts
-const colorMap = {
-	app: "yellow",
-	PluginManager: "yellow"
+/**
+* 生成作用域前缀
+* @param scope - 日志作用域
+* @returns 格式化的作用域前缀
+*/
+const generateScopePrefix = (scope) => {
+	if (Array.isArray(scope)) return scope.map((item) => `[${item}]`).join("");
+	return `[${scope}]`;
 };
+/**
+* 获取日志颜色函数
+* @param scope - 日志作用域
+* @returns 颜色函数
+*/
+const getLogColor = (scope) => {
+	return pc[LOG_COLORS[scope] || "yellow"];
+};
+/**
+* 创建统一的日志方法
+* @param level - 日志级别
+* @returns 日志方法函数
+*/
+const createLogMethod = (level) => {
+	return (msg, scope = "app") => {
+		const consoleMethod = level === "info" ? console.log : level === "error" ? console.error : level === "debug" ? console.debug : console.warn;
+		const scopePrefix = generateScopePrefix(scope);
+		consoleMethod(getLogColor(scope)(scopePrefix), msg);
+	};
+};
+/**
+* 日志插件，为应用添加统一的日志功能
+* 支持不同作用域的彩色日志输出
+*/
 const LoggerPlugin = {
 	name: "LoggerPlugin",
 	enforce: "pre",
 	setup(app) {
-		const genScope = (scope) => Array.isArray(scope) ? scope.map((el) => `[${el}]`) : `[${scope}]`;
 		app.logger = {
-			info: (msg, scope = "app") => console.log(pc[colorMap[scope] || "yellow"](genScope(scope)), msg),
-			error: (msg, scope = "app") => console.error(pc[colorMap[scope] || "yellow"](genScope(scope)), msg),
-			debug: (msg, scope = "app") => console.debug(pc[colorMap[scope] || "yellow"](genScope(scope)), msg),
-			warn: (msg, scope = "app") => console.warn(pc[colorMap[scope] || "yellow"](genScope(scope)), msg)
+			info: createLogMethod("info"),
+			error: createLogMethod("error"),
+			debug: createLogMethod("debug"),
+			warn: createLogMethod("warn")
 		};
 	}
 };
 
 //#endregion
 //#region src/server.ts
-const isTest = typeof Bun !== "undefined" && Bun.env.NODE_ENV === "test";
-let shuttingDown = false;
-let isPrimary = false;
+/**
+* 集群关闭超时时间（毫秒）
+*/
+const CLUSTER_SHUTDOWN_TIMEOUT = 5e3;
+/**
+* 检查是否为测试环境
+*/
+const isTestEnvironment = () => {
+	return typeof Bun !== "undefined" && Bun.env.NODE_ENV === "test";
+};
+/**
+* 计算合适的工作进程数量
+* @param configWorkers - 配置中的工作进程数量
+* @returns 实际使用的工作进程数量
+*/
+const calculateWorkerCount = (configWorkers) => {
+	const cpuCount = os.cpus().length;
+	return configWorkers === 0 ? cpuCount : Math.max(1, Math.min(cpuCount, configWorkers));
+};
+/**
+* 启动工作进程
+* @param count - 工作进程数量
+*/
+const startWorkers = (count) => {
+	for (let i = 0; i < count; i++) try {
+		cluster.fork();
+	} catch (error) {
+		console.error(`Failed to fork worker ${i}:`, error);
+	}
+};
+/**
+* 设置工作进程重启逻辑
+* @param shuttingDownRef - 关闭状态引用
+*/
+const setupWorkerRestart = (shuttingDownRef) => {
+	cluster.on("exit", (worker, code, signal) => {
+		if (shuttingDownRef.value) return;
+		console.log(`Worker ${worker.process.pid} died (${code}, ${signal}). Restarting...`);
+		try {
+			cluster.fork();
+		} catch (error) {
+			console.error("Failed to restart worker:", error);
+		}
+	});
+};
+/**
+* 启动 HTTP 服务器
+* @param app - 应用实例
+*/
+const startHttpServer = (app) => {
+	try {
+		app.server = Bun.serve({
+			port: app.loader.config.port,
+			hostname: app.loader.config.host,
+			reusePort: true,
+			routes: app.routes
+		});
+	} catch (error) {
+		console.error("Failed to start server:", error);
+		process.exit(1);
+	}
+};
+/**
+* 设置工作进程的消息处理
+* @param app - 应用实例
+*/
+const setupWorkerMessageHandler = (app) => {
+	process.on("message", async (msg) => {
+		if (typeof msg === "object" && msg !== null && msg.type === "shutdown") try {
+			await app.server?.stop?.();
+			process.exit(0);
+		} catch (error) {
+			console.error("Error during worker shutdown:", error);
+			process.exit(1);
+		}
+	});
+};
+/**
+* 优雅关闭工作进程
+*/
+const gracefulShutdownWorkers = () => {
+	return new Promise((resolve) => {
+		const workers = cluster.workers || {};
+		const workerCount = Object.keys(workers).length;
+		if (workerCount === 0) {
+			resolve();
+			return;
+		}
+		let exitedWorkers = 0;
+		const forceKillTimer = setTimeout(() => {
+			console.log("Force killing remaining workers...");
+			for (const id in workers) try {
+				workers[id]?.kill("SIGKILL");
+			} catch (error) {
+				console.error(`Failed to kill worker ${id}:`, error);
+			}
+			resolve();
+		}, CLUSTER_SHUTDOWN_TIMEOUT);
+		const onWorkerExit = () => {
+			exitedWorkers++;
+			if (exitedWorkers >= workerCount) {
+				clearTimeout(forceKillTimer);
+				resolve();
+			}
+		};
+		cluster.on("exit", onWorkerExit);
+		for (const id in workers) try {
+			workers[id]?.send({ type: "shutdown" });
+		} catch (error) {
+			console.error(`Failed to send shutdown message to worker ${id}:`, error);
+			exitedWorkers++;
+		}
+		if (exitedWorkers >= workerCount) {
+			clearTimeout(forceKillTimer);
+			cluster.removeListener("exit", onWorkerExit);
+			resolve();
+		}
+	});
+};
+/**
+* 服务器插件，提供集群管理和 HTTP 服务功能
+* 支持多进程模式和工作进程自动重启
+*/
 const ServerPlugin = {
 	name: "ServerPlugin",
 	enforce: "post",
 	run(app) {
-		isPrimary = cluster.isPrimary;
-		const { workers: configWorkers } = app.loader.config;
-		invariant(typeof configWorkers === "number", "config.workers must be a number");
-		const cpuCount = os.cpus().length;
-		const workers = configWorkers === 0 ? cpuCount : Math.max(1, Math.min(cpuCount, configWorkers));
-		if (isPrimary && workers > 1) {
-			for (let i = 0; i < workers; i++) cluster.fork();
-			cluster.on("exit", () => {
-				if (shuttingDown) return;
-				cluster.fork();
-			});
+		const isPrimaryProcess = cluster.isPrimary;
+		const config = app.loader.config;
+		invariant(typeof config.workers === "number", "config.workers must be a number");
+		const workerCount = calculateWorkerCount(config.workers);
+		const shuttingDownRef = { value: false };
+		if (isPrimaryProcess && workerCount > 1) {
+			console.log(`Starting ${workerCount} workers...`);
+			startWorkers(workerCount);
+			setupWorkerRestart(shuttingDownRef);
 		} else {
-			app.server = Bun.serve({
-				port: app.loader.config.port,
-				hostname: app.loader.config.host,
-				reusePort: true,
-				routes: app.routes
-			});
-			process.on("message", async (msg) => {
-				if (typeof msg === "object" && msg !== null && msg.type === "shutdown") {
-					await app.server?.stop?.();
-					process.exit(0);
-				}
-			});
+			startHttpServer(app);
+			setupWorkerMessageHandler(app);
 		}
+		this.isPrimary = isPrimaryProcess;
+		this.shuttingDownRef = shuttingDownRef;
 	},
 	async close(app) {
+		const isTest = isTestEnvironment();
+		const isPrimary = this.isPrimary;
+		const shuttingDownRef = this.shuttingDownRef || { value: false };
 		if (isTest) {
-			await app.server?.stop?.();
+			try {
+				await app.server?.stop?.();
+			} catch (error) {
+				console.error("Error stopping server in test mode:", error);
+			}
 			return;
 		}
 		if (!isPrimary) {
-			await app.server?.stop?.();
+			try {
+				await app.server?.stop?.();
+			} catch (error) {
+				console.error("Error stopping worker server:", error);
+			}
 			process.exit(0);
 		}
-		if (shuttingDown) return;
-		shuttingDown = true;
-		for (const id in cluster.workers) cluster.workers[id]?.send({ type: "shutdown" });
-		await new Promise((resolve) => {
-			const timer = setTimeout(() => {
-				for (const id in cluster.workers) cluster.workers[id]?.kill("SIGKILL");
-				resolve();
-			}, 5e3);
-			cluster.on("exit", () => {
-				if (Object.keys(cluster?.workers || {}).length === 0) {
-					clearTimeout(timer);
-					resolve();
-				}
-			});
-		});
+		if (shuttingDownRef.value) return;
+		shuttingDownRef.value = true;
+		console.log("Shutting down server cluster...");
+		try {
+			await gracefulShutdownWorkers();
+			console.log("All workers shut down successfully");
+		} catch (error) {
+			console.error("Error during cluster shutdown:", error);
+		}
 		process.exit(0);
 	}
 };

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","names":["LoggerPlugin: Plugin","ServerPlugin: Plugin"],"sources":["../src/logger.ts","../src/server.ts"],"sourcesContent":["import { type Plugin } from '@thinkbun/core';\nimport pc from 'picocolors';\n\nconst colorMap = {\n  app: 'yellow',\n  PluginManager: 'yellow',\n};\nexport const LoggerPlugin: Plugin = {\n  name: 'LoggerPlugin',\n  enforce: 'pre',\n  setup(app) {\n    const genScope = (scope: string | string[]) => (Array.isArray(scope) ? scope.map((el) => `[${el}]`) : `[${scope}]`);\n    app.logger = {\n      // @ts-ignore\n      info: (msg: string, scope = 'app') => console.log(pc[colorMap[scope] || 'yellow'](genScope(scope)), msg),\n      // @ts-ignore\n      error: (msg: string, scope = 'app') => console.error(pc[colorMap[scope] || 'yellow'](genScope(scope)), msg),\n      // @ts-ignore\n      debug: (msg: string, scope = 'app') => console.debug(pc[colorMap[scope] || 'yellow'](genScope(scope)), msg),\n      // @ts-ignore\n      warn: (msg: string, scope = 'app') => console.warn(pc[colorMap[scope] || 'yellow'](genScope(scope)), msg),\n    };\n  },\n};\n","import cluster from 'node:cluster';\nimport os from 'node:os';\n\nimport { invariant, type Plugin } from '@thinkbun/core';\n\nconst isTest = typeof Bun !== 'undefined' && Bun.env.NODE_ENV === 'test';\n\nlet shuttingDown = false;\nlet isPrimary = false;\nexport const ServerPlugin: Plugin = {\n  name: 'ServerPlugin',\n  enforce: 'post',\n  run(app) {\n    isPrimary = cluster.isPrimary;\n    const { workers: configWorkers } = app.loader.config;\n    invariant(typeof configWorkers === 'number', 'config.workers must be a number');\n    const cpuCount = os.cpus().length;\n    const workers = configWorkers === 0 ? cpuCount : Math.max(1, Math.min(cpuCount, configWorkers));\n    if (isPrimary && workers > 1) {\n      for (let i = 0; i < workers; i++) {\n        cluster.fork();\n      }\n\n      (cluster as any).on('exit', () => {\n        if (shuttingDown) return;\n        cluster.fork();\n      });\n    } else {\n      app.server = Bun.serve({\n        port: app.loader.config.port,\n        hostname: app.loader.config.host,\n        reusePort: true,\n        routes: app.routes,\n      });\n\n      process.on('message', async (msg: { type: string }) => {\n        if (typeof msg === 'object' && msg !== null && msg.type === 'shutdown') {\n          await app.server?.stop?.();\n          process.exit(0);\n        }\n      });\n    }\n  },\n  async close(app) {\n    if (isTest) {\n      // 只关闭 server，不退出进程\n      await app.server?.stop?.();\n      return;\n    }\n    // 只允许 primary 主动关闭 cluster\n    if (!isPrimary) {\n      await app.server?.stop?.();\n      process.exit(0);\n    }\n\n    if (shuttingDown) return;\n    shuttingDown = true;\n\n    // 通知所有 worker\n    for (const id in cluster.workers) {\n      cluster.workers[id]?.send({ type: 'shutdown' });\n    }\n\n    // 等待 worker 退出 or 强杀\n    await new Promise<void>((resolve) => {\n      const timer = setTimeout(() => {\n        for (const id in cluster.workers) {\n          cluster.workers[id]?.kill('SIGKILL');\n        }\n        resolve();\n      }, 5000);\n      (cluster as any).on('exit', () => {\n        if (Object.keys(cluster?.workers || {}).length === 0) {\n          clearTimeout(timer);\n          resolve();\n        }\n      });\n    });\n\n    process.exit(0);\n  },\n};\n"],"mappings":";;;;;;AAGA,MAAM,WAAW;CACf,KAAK;CACL,eAAe;CAChB;AACD,MAAaA,eAAuB;CAClC,MAAM;CACN,SAAS;CACT,MAAM,KAAK;EACT,MAAM,YAAY,UAA8B,MAAM,QAAQ,MAAM,GAAG,MAAM,KAAK,OAAO,IAAI,GAAG,GAAG,GAAG,IAAI,MAAM;AAChH,MAAI,SAAS;GAEX,OAAO,KAAa,QAAQ,UAAU,QAAQ,IAAI,GAAG,SAAS,UAAU,UAAU,SAAS,MAAM,CAAC,EAAE,IAAI;GAExG,QAAQ,KAAa,QAAQ,UAAU,QAAQ,MAAM,GAAG,SAAS,UAAU,UAAU,SAAS,MAAM,CAAC,EAAE,IAAI;GAE3G,QAAQ,KAAa,QAAQ,UAAU,QAAQ,MAAM,GAAG,SAAS,UAAU,UAAU,SAAS,MAAM,CAAC,EAAE,IAAI;GAE3G,OAAO,KAAa,QAAQ,UAAU,QAAQ,KAAK,GAAG,SAAS,UAAU,UAAU,SAAS,MAAM,CAAC,EAAE,IAAI;GAC1G;;CAEJ;;;;AClBD,MAAM,SAAS,OAAO,QAAQ,eAAe,IAAI,IAAI,aAAa;AAElE,IAAI,eAAe;AACnB,IAAI,YAAY;AAChB,MAAaC,eAAuB;CAClC,MAAM;CACN,SAAS;CACT,IAAI,KAAK;AACP,cAAY,QAAQ;EACpB,MAAM,EAAE,SAAS,kBAAkB,IAAI,OAAO;AAC9C,YAAU,OAAO,kBAAkB,UAAU,kCAAkC;EAC/E,MAAM,WAAW,GAAG,MAAM,CAAC;EAC3B,MAAM,UAAU,kBAAkB,IAAI,WAAW,KAAK,IAAI,GAAG,KAAK,IAAI,UAAU,cAAc,CAAC;AAC/F,MAAI,aAAa,UAAU,GAAG;AAC5B,QAAK,IAAI,IAAI,GAAG,IAAI,SAAS,IAC3B,SAAQ,MAAM;AAGhB,GAAC,QAAgB,GAAG,cAAc;AAChC,QAAI,aAAc;AAClB,YAAQ,MAAM;KACd;SACG;AACL,OAAI,SAAS,IAAI,MAAM;IACrB,MAAM,IAAI,OAAO,OAAO;IACxB,UAAU,IAAI,OAAO,OAAO;IAC5B,WAAW;IACX,QAAQ,IAAI;IACb,CAAC;AAEF,WAAQ,GAAG,WAAW,OAAO,QAA0B;AACrD,QAAI,OAAO,QAAQ,YAAY,QAAQ,QAAQ,IAAI,SAAS,YAAY;AACtE,WAAM,IAAI,QAAQ,QAAQ;AAC1B,aAAQ,KAAK,EAAE;;KAEjB;;;CAGN,MAAM,MAAM,KAAK;AACf,MAAI,QAAQ;AAEV,SAAM,IAAI,QAAQ,QAAQ;AAC1B;;AAGF,MAAI,CAAC,WAAW;AACd,SAAM,IAAI,QAAQ,QAAQ;AAC1B,WAAQ,KAAK,EAAE;;AAGjB,MAAI,aAAc;AAClB,iBAAe;AAGf,OAAK,MAAM,MAAM,QAAQ,QACvB,SAAQ,QAAQ,KAAK,KAAK,EAAE,MAAM,YAAY,CAAC;AAIjD,QAAM,IAAI,SAAe,YAAY;GACnC,MAAM,QAAQ,iBAAiB;AAC7B,SAAK,MAAM,MAAM,QAAQ,QACvB,SAAQ,QAAQ,KAAK,KAAK,UAAU;AAEtC,aAAS;MACR,IAAK;AACR,GAAC,QAAgB,GAAG,cAAc;AAChC,QAAI,OAAO,KAAK,SAAS,WAAW,EAAE,CAAC,CAAC,WAAW,GAAG;AACpD,kBAAa,MAAM;AACnB,cAAS;;KAEX;IACF;AAEF,UAAQ,KAAK,EAAE;;CAElB"}
+{"version":3,"file":"index.mjs","names":["LoggerPlugin: Plugin","ServerPlugin: Plugin"],"sources":["../src/logger.ts","../src/server.ts"],"sourcesContent":["import { type Plugin, type Application, type LogScope, LOG_COLORS, type LogLevel } from '@thinkbun/core';\nimport pc from 'picocolors';\n\n/**\n * 生成作用域前缀\n * @param scope - 日志作用域\n * @returns 格式化的作用域前缀\n */\nconst generateScopePrefix = (scope: LogScope | LogScope[]): string => {\n  if (Array.isArray(scope)) {\n    return scope.map((item) => `[${item}]`).join('');\n  }\n  return `[${scope}]`;\n};\n\n/**\n * 获取日志颜色函数\n * @param scope - 日志作用域\n * @returns 颜色函数\n */\nconst getLogColor = (scope: LogScope) => {\n  const colorName = LOG_COLORS[scope as keyof typeof LOG_COLORS] || 'yellow';\n  return pc[colorName];\n};\n\n/**\n * 创建统一的日志方法\n * @param level - 日志级别\n * @returns 日志方法函数\n */\nconst createLogMethod = (level: LogLevel) => {\n  return (msg: string, scope: LogScope = 'app'): void => {\n    const consoleMethod =\n      level === 'info'\n        ? console.log\n        : level === 'error'\n          ? console.error\n          : level === 'debug'\n            ? console.debug\n            : console.warn;\n\n    const scopePrefix = generateScopePrefix(scope);\n    const colorFn = getLogColor(scope);\n\n    consoleMethod(colorFn(scopePrefix), msg);\n  };\n};\n\n/**\n * 日志插件，为应用添加统一的日志功能\n * 支持不同作用域的彩色日志输出\n */\nexport const LoggerPlugin: Plugin = {\n  name: 'LoggerPlugin',\n  enforce: 'pre',\n  setup(app: Application) {\n    app.logger = {\n      info: createLogMethod('info'),\n      error: createLogMethod('error'),\n      debug: createLogMethod('debug'),\n      warn: createLogMethod('warn'),\n    };\n  },\n};\n","import cluster from 'node:cluster';\nimport os from 'node:os';\n\nimport { invariant, type Plugin, type Application } from '@thinkbun/core';\n\n/**\n * 集群关闭超时时间（毫秒）\n */\nconst CLUSTER_SHUTDOWN_TIMEOUT = 5000;\n\n/**\n * 集群消息类型\n */\ninterface ClusterMessage {\n  type: 'shutdown';\n}\n\n/**\n * 服务器配置接口\n */\ninterface ServerConfig {\n  workers: number;\n  port: number;\n  host: string;\n}\n\n/**\n * 检查是否为测试环境\n */\nconst isTestEnvironment = (): boolean => {\n  return typeof Bun !== 'undefined' && Bun.env.NODE_ENV === 'test';\n};\n\n/**\n * 计算合适的工作进程数量\n * @param configWorkers - 配置中的工作进程数量\n * @returns 实际使用的工作进程数量\n */\nconst calculateWorkerCount = (configWorkers: number): number => {\n  const cpuCount = os.cpus().length;\n  return configWorkers === 0 ? cpuCount : Math.max(1, Math.min(cpuCount, configWorkers));\n};\n\n/**\n * 启动工作进程\n * @param count - 工作进程数量\n */\nconst startWorkers = (count: number): void => {\n  for (let i = 0; i < count; i++) {\n    try {\n      cluster.fork();\n    } catch (error) {\n      console.error(`Failed to fork worker ${i}:`, error);\n    }\n  }\n};\n\n/**\n * 设置工作进程重启逻辑\n * @param shuttingDownRef - 关闭状态引用\n */\nconst setupWorkerRestart = (shuttingDownRef: { value: boolean }): void => {\n  (cluster as any).on('exit', (worker: any, code: number, signal: string) => {\n    if (shuttingDownRef.value) {\n      return;\n    }\n\n    console.log(`Worker ${worker.process.pid} died (${code}, ${signal}). Restarting...`);\n\n    try {\n      cluster.fork();\n    } catch (error) {\n      console.error('Failed to restart worker:', error);\n    }\n  });\n};\n\n/**\n * 启动 HTTP 服务器\n * @param app - 应用实例\n */\nconst startHttpServer = (app: Application): void => {\n  try {\n    app.server = Bun.serve({\n      port: app.loader.config.port,\n      hostname: app.loader.config.host,\n      reusePort: true,\n      routes: app.routes,\n    });\n  } catch (error) {\n    console.error('Failed to start server:', error);\n    process.exit(1);\n  }\n};\n\n/**\n * 设置工作进程的消息处理\n * @param app - 应用实例\n */\nconst setupWorkerMessageHandler = (app: Application): void => {\n  process.on('message', async (msg: ClusterMessage) => {\n    if (typeof msg === 'object' && msg !== null && msg.type === 'shutdown') {\n      try {\n        await app.server?.stop?.();\n        process.exit(0);\n      } catch (error) {\n        console.error('Error during worker shutdown:', error);\n        process.exit(1);\n      }\n    }\n  });\n};\n\n/**\n * 优雅关闭工作进程\n */\nconst gracefulShutdownWorkers = (): Promise<void> => {\n  return new Promise<void>((resolve) => {\n    const workers = cluster.workers || {};\n    const workerCount = Object.keys(workers).length;\n\n    if (workerCount === 0) {\n      resolve();\n      return;\n    }\n\n    let exitedWorkers = 0;\n\n    // 设置强制关闭定时器\n    const forceKillTimer = setTimeout(() => {\n      console.log('Force killing remaining workers...');\n      for (const id in workers) {\n        try {\n          workers[id]?.kill('SIGKILL');\n        } catch (error) {\n          console.error(`Failed to kill worker ${id}:`, error);\n        }\n      }\n      resolve();\n    }, CLUSTER_SHUTDOWN_TIMEOUT);\n\n    // 监听工作进程退出\n    const onWorkerExit = () => {\n      exitedWorkers++;\n      if (exitedWorkers >= workerCount) {\n        clearTimeout(forceKillTimer);\n        resolve();\n      }\n    };\n\n    (cluster as any).on('exit', onWorkerExit);\n\n    // 发送关闭信号\n    for (const id in workers) {\n      try {\n        workers[id]?.send({ type: 'shutdown' });\n      } catch (error) {\n        console.error(`Failed to send shutdown message to worker ${id}:`, error);\n        // 如果发送失败，直接视为已退出\n        exitedWorkers++;\n      }\n    }\n\n    // 检查是否所有工作进程都已经退出\n    if (exitedWorkers >= workerCount) {\n      clearTimeout(forceKillTimer);\n      (cluster as any).removeListener('exit', onWorkerExit);\n      resolve();\n    }\n  });\n};\n\n/**\n * 服务器插件，提供集群管理和 HTTP 服务功能\n * 支持多进程模式和工作进程自动重启\n */\nexport const ServerPlugin: Plugin = {\n  name: 'ServerPlugin',\n  enforce: 'post',\n\n  /**\n   * 插件运行逻辑\n   * @param app - 应用实例\n   */\n  run(app: Application) {\n    const isPrimaryProcess = cluster.isPrimary;\n    const config = app.loader.config as ServerConfig;\n\n    // 验证配置\n    invariant(typeof config.workers === 'number', 'config.workers must be a number');\n\n    const workerCount = calculateWorkerCount(config.workers);\n    const shuttingDownRef = { value: false };\n\n    if (isPrimaryProcess && workerCount > 1) {\n      // 主进程：启动工作进程\n      console.log(`Starting ${workerCount} workers...`);\n      startWorkers(workerCount);\n      setupWorkerRestart(shuttingDownRef);\n    } else {\n      // 工作进程或单进程模式：启动 HTTP 服务器\n      startHttpServer(app);\n      setupWorkerMessageHandler(app);\n    }\n\n    // 保存状态供关闭时使用\n    (this as any).isPrimary = isPrimaryProcess;\n    (this as any).shuttingDownRef = shuttingDownRef;\n  },\n\n  /**\n   * 插件关闭逻辑\n   * @param app - 应用实例\n   */\n  async close(app: Application) {\n    const isTest = isTestEnvironment();\n    const isPrimary = (this as any).isPrimary;\n    const shuttingDownRef = (this as any).shuttingDownRef || { value: false };\n\n    if (isTest) {\n      // 测试环境：只关闭服务器，不退出进程\n      try {\n        await app.server?.stop?.();\n      } catch (error) {\n        console.error('Error stopping server in test mode:', error);\n      }\n      return;\n    }\n\n    // 非主进程工作节点：关闭服务器并退出\n    if (!isPrimary) {\n      try {\n        await app.server?.stop?.();\n      } catch (error) {\n        console.error('Error stopping worker server:', error);\n      }\n      process.exit(0);\n    }\n\n    // 主进程：优雅关闭所有工作进程\n    if (shuttingDownRef.value) {\n      return;\n    }\n\n    shuttingDownRef.value = true;\n    console.log('Shutting down server cluster...');\n\n    try {\n      await gracefulShutdownWorkers();\n      console.log('All workers shut down successfully');\n    } catch (error) {\n      console.error('Error during cluster shutdown:', error);\n    }\n\n    process.exit(0);\n  },\n};\n"],"mappings":";;;;;;;;;;;AAQA,MAAM,uBAAuB,UAAyC;AACpE,KAAI,MAAM,QAAQ,MAAM,CACtB,QAAO,MAAM,KAAK,SAAS,IAAI,KAAK,GAAG,CAAC,KAAK,GAAG;AAElD,QAAO,IAAI,MAAM;;;;;;;AAQnB,MAAM,eAAe,UAAoB;AAEvC,QAAO,GADW,WAAW,UAAqC;;;;;;;AASpE,MAAM,mBAAmB,UAAoB;AAC3C,SAAQ,KAAa,QAAkB,UAAgB;EACrD,MAAM,gBACJ,UAAU,SACN,QAAQ,MACR,UAAU,UACR,QAAQ,QACR,UAAU,UACR,QAAQ,QACR,QAAQ;EAElB,MAAM,cAAc,oBAAoB,MAAM;AAG9C,gBAFgB,YAAY,MAAM,CAEZ,YAAY,EAAE,IAAI;;;;;;;AAQ5C,MAAaA,eAAuB;CAClC,MAAM;CACN,SAAS;CACT,MAAM,KAAkB;AACtB,MAAI,SAAS;GACX,MAAM,gBAAgB,OAAO;GAC7B,OAAO,gBAAgB,QAAQ;GAC/B,OAAO,gBAAgB,QAAQ;GAC/B,MAAM,gBAAgB,OAAO;GAC9B;;CAEJ;;;;;;;ACvDD,MAAM,2BAA2B;;;;AAqBjC,MAAM,0BAAmC;AACvC,QAAO,OAAO,QAAQ,eAAe,IAAI,IAAI,aAAa;;;;;;;AAQ5D,MAAM,wBAAwB,kBAAkC;CAC9D,MAAM,WAAW,GAAG,MAAM,CAAC;AAC3B,QAAO,kBAAkB,IAAI,WAAW,KAAK,IAAI,GAAG,KAAK,IAAI,UAAU,cAAc,CAAC;;;;;;AAOxF,MAAM,gBAAgB,UAAwB;AAC5C,MAAK,IAAI,IAAI,GAAG,IAAI,OAAO,IACzB,KAAI;AACF,UAAQ,MAAM;UACP,OAAO;AACd,UAAQ,MAAM,yBAAyB,EAAE,IAAI,MAAM;;;;;;;AASzD,MAAM,sBAAsB,oBAA8C;AACxE,CAAC,QAAgB,GAAG,SAAS,QAAa,MAAc,WAAmB;AACzE,MAAI,gBAAgB,MAClB;AAGF,UAAQ,IAAI,UAAU,OAAO,QAAQ,IAAI,SAAS,KAAK,IAAI,OAAO,kBAAkB;AAEpF,MAAI;AACF,WAAQ,MAAM;WACP,OAAO;AACd,WAAQ,MAAM,6BAA6B,MAAM;;GAEnD;;;;;;AAOJ,MAAM,mBAAmB,QAA2B;AAClD,KAAI;AACF,MAAI,SAAS,IAAI,MAAM;GACrB,MAAM,IAAI,OAAO,OAAO;GACxB,UAAU,IAAI,OAAO,OAAO;GAC5B,WAAW;GACX,QAAQ,IAAI;GACb,CAAC;UACK,OAAO;AACd,UAAQ,MAAM,2BAA2B,MAAM;AAC/C,UAAQ,KAAK,EAAE;;;;;;;AAQnB,MAAM,6BAA6B,QAA2B;AAC5D,SAAQ,GAAG,WAAW,OAAO,QAAwB;AACnD,MAAI,OAAO,QAAQ,YAAY,QAAQ,QAAQ,IAAI,SAAS,WAC1D,KAAI;AACF,SAAM,IAAI,QAAQ,QAAQ;AAC1B,WAAQ,KAAK,EAAE;WACR,OAAO;AACd,WAAQ,MAAM,iCAAiC,MAAM;AACrD,WAAQ,KAAK,EAAE;;GAGnB;;;;;AAMJ,MAAM,gCAA+C;AACnD,QAAO,IAAI,SAAe,YAAY;EACpC,MAAM,UAAU,QAAQ,WAAW,EAAE;EACrC,MAAM,cAAc,OAAO,KAAK,QAAQ,CAAC;AAEzC,MAAI,gBAAgB,GAAG;AACrB,YAAS;AACT;;EAGF,IAAI,gBAAgB;EAGpB,MAAM,iBAAiB,iBAAiB;AACtC,WAAQ,IAAI,qCAAqC;AACjD,QAAK,MAAM,MAAM,QACf,KAAI;AACF,YAAQ,KAAK,KAAK,UAAU;YACrB,OAAO;AACd,YAAQ,MAAM,yBAAyB,GAAG,IAAI,MAAM;;AAGxD,YAAS;KACR,yBAAyB;EAG5B,MAAM,qBAAqB;AACzB;AACA,OAAI,iBAAiB,aAAa;AAChC,iBAAa,eAAe;AAC5B,aAAS;;;AAIb,EAAC,QAAgB,GAAG,QAAQ,aAAa;AAGzC,OAAK,MAAM,MAAM,QACf,KAAI;AACF,WAAQ,KAAK,KAAK,EAAE,MAAM,YAAY,CAAC;WAChC,OAAO;AACd,WAAQ,MAAM,6CAA6C,GAAG,IAAI,MAAM;AAExE;;AAKJ,MAAI,iBAAiB,aAAa;AAChC,gBAAa,eAAe;AAC5B,GAAC,QAAgB,eAAe,QAAQ,aAAa;AACrD,YAAS;;GAEX;;;;;;AAOJ,MAAaC,eAAuB;CAClC,MAAM;CACN,SAAS;CAMT,IAAI,KAAkB;EACpB,MAAM,mBAAmB,QAAQ;EACjC,MAAM,SAAS,IAAI,OAAO;AAG1B,YAAU,OAAO,OAAO,YAAY,UAAU,kCAAkC;EAEhF,MAAM,cAAc,qBAAqB,OAAO,QAAQ;EACxD,MAAM,kBAAkB,EAAE,OAAO,OAAO;AAExC,MAAI,oBAAoB,cAAc,GAAG;AAEvC,WAAQ,IAAI,YAAY,YAAY,aAAa;AACjD,gBAAa,YAAY;AACzB,sBAAmB,gBAAgB;SAC9B;AAEL,mBAAgB,IAAI;AACpB,6BAA0B,IAAI;;AAIhC,EAAC,KAAa,YAAY;AAC1B,EAAC,KAAa,kBAAkB;;CAOlC,MAAM,MAAM,KAAkB;EAC5B,MAAM,SAAS,mBAAmB;EAClC,MAAM,YAAa,KAAa;EAChC,MAAM,kBAAmB,KAAa,mBAAmB,EAAE,OAAO,OAAO;AAEzE,MAAI,QAAQ;AAEV,OAAI;AACF,UAAM,IAAI,QAAQ,QAAQ;YACnB,OAAO;AACd,YAAQ,MAAM,uCAAuC,MAAM;;AAE7D;;AAIF,MAAI,CAAC,WAAW;AACd,OAAI;AACF,UAAM,IAAI,QAAQ,QAAQ;YACnB,OAAO;AACd,YAAQ,MAAM,iCAAiC,MAAM;;AAEvD,WAAQ,KAAK,EAAE;;AAIjB,MAAI,gBAAgB,MAClB;AAGF,kBAAgB,QAAQ;AACxB,UAAQ,IAAI,kCAAkC;AAE9C,MAAI;AACF,SAAM,yBAAyB;AAC/B,WAAQ,IAAI,qCAAqC;WAC1C,OAAO;AACd,WAAQ,MAAM,kCAAkC,MAAM;;AAGxD,UAAQ,KAAK,EAAE;;CAElB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@thinkbun/plugin",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "type": "module",
   "exports": {
     ".": "./dist/index.mjs",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "picocolors": "^1.1.1",
-    "@thinkbun/core": "^1.0.0"
+    "@thinkbun/core": "^1.0.4"
   },
   "files": [
     "dist"