@dangao/bun-server 1.3.0 → 1.5.0

package/src/core/server.ts

@@ -27,6 +27,12 @@ export interface ServerOptions {
    * WebSocket 网关注册表
    */
   websocketRegistry?: WebSocketGatewayRegistry;
+
+  /**
+   * 优雅停机超时时间（毫秒）
+   * 默认 30 秒
+   */
+  gracefulShutdownTimeout?: number;
 }
 
 /**
@@ -36,6 +42,10 @@ export interface ServerOptions {
 export class BunServer {
   private server?: Server<WebSocketConnectionData>;
   private readonly options: ServerOptions;
+  private activeRequests: number = 0;
+  private isShuttingDown: boolean = false;
+  private shutdownPromise?: Promise<void>;
+  private shutdownResolve?: () => void;
 
   public constructor(options: ServerOptions) {
     this.options = options;
@@ -51,6 +61,12 @@ export class BunServer {
 
     const logger = LoggerManager.getLogger();
 
+    // 重置状态
+    this.activeRequests = 0;
+    this.isShuttingDown = false;
+    this.shutdownPromise = undefined;
+    this.shutdownResolve = undefined;
+
     this.server = Bun.serve({
       port: this.options.port ?? 3000,
       hostname: this.options.hostname,
@@ -58,6 +74,11 @@ export class BunServer {
         request: Request,
         server: Server<WebSocketConnectionData>,
       ): Response | Promise<Response> | undefined => {
+        // 如果正在关闭，拒绝新请求
+        if (this.isShuttingDown) {
+          return new Response("Server is shutting down", { status: 503 });
+        }
+
         const upgradeHeader = request.headers.get("upgrade");
         if (
           this.options.websocketRegistry &&
@@ -77,8 +98,34 @@ export class BunServer {
           return new Response("WebSocket upgrade failed", { status: 400 });
         }
 
+        // 增加活跃请求计数
+        this.activeRequests++;
+
         const context = new Context(request);
-        return this.options.fetch(context);
+        const responsePromise = this.options.fetch(context);
+
+        // 处理响应完成后的清理
+        if (responsePromise instanceof Promise) {
+          responsePromise
+            .finally(() => {
+              this.activeRequests--;
+              // 如果正在关闭且没有活跃请求，触发关闭完成
+              if (this.isShuttingDown && this.activeRequests === 0 && this.shutdownResolve) {
+                this.shutdownResolve();
+              }
+            })
+            .catch(() => {
+              // 错误已在中间件中处理，这里只负责计数
+            });
+        } else {
+          // 同步响应
+          this.activeRequests--;
+          if (this.isShuttingDown && this.activeRequests === 0 && this.shutdownResolve) {
+            this.shutdownResolve();
+          }
+        }
+
+        return responsePromise;
       },
       websocket: {
         open: (ws) => {
@@ -99,17 +146,82 @@ export class BunServer {
   }
 
   /**
-   * 停止服务器
+   * 停止服务器（立即停止，不等待请求完成）
    */
   public stop(): void {
     if (this.server) {
       const logger = LoggerManager.getLogger();
       this.server.stop();
       this.server = undefined;
+      this.isShuttingDown = false;
+      this.activeRequests = 0;
       logger.info("Server stopped");
     }
   }
 
+  /**
+   * 优雅停机
+   * 停止接受新请求，等待正在处理的请求完成
+   * @param timeout - 超时时间（毫秒），默认使用配置的 gracefulShutdownTimeout 或 30000
+   * @returns Promise，在停机完成时 resolve
+   */
+  public async gracefulShutdown(timeout?: number): Promise<void> {
+    if (!this.server || this.isShuttingDown) {
+      return;
+    }
+
+    const logger = LoggerManager.getLogger();
+    const shutdownTimeout = timeout ?? this.options.gracefulShutdownTimeout ?? 30000;
+
+    logger.info(`Starting graceful shutdown (timeout: ${shutdownTimeout}ms, active requests: ${this.activeRequests})`);
+
+    // 标记为正在关闭，停止接受新请求
+    this.isShuttingDown = true;
+
+    // 如果没有活跃请求，立即关闭
+    if (this.activeRequests === 0) {
+      this.stop();
+      logger.info("Graceful shutdown completed (no active requests)");
+      return;
+    }
+
+    // 创建关闭 Promise
+    if (!this.shutdownPromise) {
+      this.shutdownPromise = new Promise<void>((resolve) => {
+        this.shutdownResolve = resolve;
+      });
+    }
+
+    // 设置超时
+    const timeoutPromise = new Promise<void>((resolve) => {
+      setTimeout(() => {
+        logger.warn(`Graceful shutdown timeout (${shutdownTimeout}ms), forcing shutdown`);
+        resolve();
+      }, shutdownTimeout);
+    });
+
+    // 等待所有请求完成或超时
+    await Promise.race([this.shutdownPromise, timeoutPromise]);
+
+    // 停止服务器
+    this.stop();
+    logger.info(`Graceful shutdown completed (remaining active requests: ${this.activeRequests})`);
+  }
+
+  /**
+   * 获取当前活跃请求数
+   */
+  public getActiveRequests(): number {
+    return this.activeRequests;
+  }
+
+  /**
+   * 检查是否正在关闭
+   */
+  public isShuttingDownState(): boolean {
+    return this.isShuttingDown;
+  }
+
   /**
    * 获取服务器实例
    * @returns Bun Server 实例
@@ -125,4 +237,18 @@ export class BunServer {
   public isRunning(): boolean {
     return this.server !== undefined;
   }
+
+  /**
+   * 获取服务器端口
+   */
+  public getPort(): number {
+    return this.options.port ?? 3000;
+  }
+
+  /**
+   * 获取服务器主机名
+   */
+  public getHostname(): string | undefined {
+    return this.options.hostname;
+  }
 }

package/src/index.ts

@@ -306,4 +306,85 @@ export type {
   SessionData,
   RedisSessionStoreOptions,
 } from './session';
+// Microservice 模块
+export {
+  ConfigCenterModule,
+  NacosConfigCenter,
+  CONFIG_CENTER_TOKEN,
+  type ConfigCenterModuleOptions,
+  type ConfigCenterProvider,
+  type NacosConfigCenterOptions,
+  type ConfigCenter,
+  type ConfigChangeListener,
+  type ConfigResult,
+} from './microservice';
+export {
+  ServiceRegistryModule,
+  NacosServiceRegistry,
+  SERVICE_REGISTRY_TOKEN,
+  type ServiceRegistryModuleOptions,
+  type ServiceRegistryProvider,
+  type NacosServiceRegistryOptions,
+  type ServiceRegistry,
+  type GetInstancesOptions,
+  type InstancesChangeListener,
+  type ServiceInstance,
+} from './microservice';
+// Service Client 模块
+export {
+  ServiceClient,
+  LoadBalancerFactory,
+  RandomLoadBalancer,
+  RoundRobinLoadBalancer,
+  WeightedRoundRobinLoadBalancer,
+  ConsistentHashLoadBalancer,
+  LeastActiveLoadBalancer,
+  TraceIdRequestInterceptor,
+  UserInfoRequestInterceptor,
+  RequestLogInterceptor,
+  ResponseLogInterceptor,
+  ResponseTransformInterceptor,
+  ErrorHandlerInterceptor,
+  type LoadBalanceStrategy,
+  type LoadBalancer,
+  type ServiceCallOptions,
+  type ServiceCallResponse,
+  type ServiceRequestInterceptor,
+  type ServiceResponseInterceptor,
+  type ServiceCallError,
+} from './microservice';
+// Governance 模块
+export {
+  CircuitBreaker,
+  CircuitBreakerState,
+  RateLimiter,
+  RetryStrategyImpl,
+  type CircuitBreakerOptions,
+  type CircuitBreakerStats,
+  type RateLimiterOptions,
+  type RetryStrategy,
+} from './microservice';
+// Tracing 模块
+export {
+  Tracer,
+  ConsoleTraceCollector,
+  MemoryTraceCollector,
+  SpanStatus,
+  SpanKind,
+  type TraceId,
+  type SpanId,
+  type SpanContext,
+  type Span,
+  type SpanTags,
+  type SpanEvent,
+  type TracingOptions,
+  type TraceCollector,
+} from './microservice';
+// Monitoring 模块
+export {
+  ServiceMetricsCollector,
+  type ServiceCallMetrics,
+  type ServiceInstanceHealth,
+  type MonitoringOptions,
+} from './microservice';
 

package/src/microservice/config-center/config-center-module.ts

@@ -0,0 +1,98 @@
+import { Module, MODULE_METADATA_KEY, type ModuleProvider } from '../../di/module';
+import { NacosClient } from '@dangao/nacos-client';
+import type { NacosClientOptions } from '@dangao/nacos-client';
+import { NacosConfigCenter } from './nacos-config-center';
+import { CONFIG_CENTER_TOKEN, type ConfigCenter } from './types';
+
+/**
+ * 配置中心 Provider 类型
+ */
+export type ConfigCenterProvider = 'nacos' | 'consul' | 'etcd' | 'apollo';
+
+/**
+ * Nacos 配置中心选项
+ */
+export interface NacosConfigCenterOptions {
+  /**
+   * Nacos 客户端配置
+   */
+  client: NacosClientOptions;
+
+  /**
+   * 配置监听轮询间隔（毫秒）
+   * @default 3000
+   */
+  watchInterval?: number;
+}
+
+/**
+ * 配置中心模块选项
+ */
+export interface ConfigCenterModuleOptions {
+  /**
+   * Provider 类型
+   */
+  provider: ConfigCenterProvider;
+
+  /**
+   * Nacos 配置（当 provider 为 'nacos' 时使用）
+   */
+  nacos?: NacosConfigCenterOptions;
+
+  // 未来可以添加其他 provider 的配置
+  // consul?: ConsulConfigCenterOptions;
+  // etcd?: EtcdConfigCenterOptions;
+}
+
+/**
+ * 配置中心模块
+ */
+@Module({
+  providers: [],
+})
+export class ConfigCenterModule {
+  /**
+   * 创建配置中心模块
+   * @param options - 模块配置
+   */
+  public static forRoot(options: ConfigCenterModuleOptions): typeof ConfigCenterModule {
+    const providers: ModuleProvider[] = [];
+
+    let configCenter: ConfigCenter;
+
+    switch (options.provider) {
+      case 'nacos':
+        if (!options.nacos) {
+          throw new Error('Nacos configuration is required when provider is "nacos"');
+        }
+
+        const nacosClient = new NacosClient(options.nacos.client);
+        configCenter = new NacosConfigCenter(nacosClient, {
+          watchInterval: options.nacos.watchInterval,
+        });
+        break;
+
+      default:
+        throw new Error(`Unsupported config center provider: ${options.provider}`);
+    }
+
+    providers.push({
+      provide: CONFIG_CENTER_TOKEN,
+      useValue: configCenter,
+    });
+
+    const existingMetadata = Reflect.getMetadata(MODULE_METADATA_KEY, ConfigCenterModule) || {};
+    const metadata = {
+      ...existingMetadata,
+      providers: [...(existingMetadata.providers || []), ...providers],
+      exports: [
+        ...(existingMetadata.exports || []),
+        CONFIG_CENTER_TOKEN,
+      ],
+    };
+    Reflect.defineMetadata(MODULE_METADATA_KEY, metadata, ConfigCenterModule);
+
+    return ConfigCenterModule;
+  }
+}
+

package/src/microservice/config-center/decorators.ts

@@ -0,0 +1,159 @@
+import 'reflect-metadata';
+import type { ConfigCenter } from './types';
+import { CONFIG_CENTER_TOKEN } from './types';
+import { ControllerRegistry } from '../../controller/controller';
+import type { Constructor } from '../../core/types';
+
+/**
+ * 配置中心值装饰器元数据键
+ */
+const CONFIG_CENTER_VALUE_METADATA_KEY = Symbol('config-center:value');
+
+/**
+ * 配置中心值元数据
+ */
+export interface ConfigCenterValueMetadata {
+  /**
+   * 配置 dataId
+   */
+  dataId: string;
+
+  /**
+   * 配置分组
+   */
+  groupName: string;
+
+  /**
+   * 命名空间（可选）
+   */
+  namespaceId?: string;
+
+  /**
+   * 默认值（可选）
+   */
+  defaultValue?: string;
+
+  /**
+   * 是否监听配置变更
+   */
+  watch?: boolean;
+}
+
+/**
+ * ConfigCenterValue 装饰器
+ * 用于自动注入配置中心的值
+ *
+ * @example
+ * ```typescript
+ * @Injectable()
+ * class MyService {
+ *   @ConfigCenterValue('my-config', 'DEFAULT_GROUP')
+ *   public configValue: string = '';
+ *
+ *   @ConfigCenterValue('app-name', 'DEFAULT_GROUP', { defaultValue: 'MyApp' })
+ *   public appName: string = '';
+ * }
+ * ```
+ */
+export function ConfigCenterValue(
+  dataId: string,
+  groupName: string = 'DEFAULT_GROUP',
+  options?: {
+    namespaceId?: string;
+    defaultValue?: string;
+    watch?: boolean;
+  },
+): PropertyDecorator {
+  return function (target: object, propertyKey: string | symbol) {
+    const metadata: ConfigCenterValueMetadata = {
+      dataId,
+      groupName,
+      namespaceId: options?.namespaceId,
+      defaultValue: options?.defaultValue,
+      watch: options?.watch ?? false,
+    };
+
+    // 保存元数据
+    const existingMetadata: Map<string | symbol, ConfigCenterValueMetadata> =
+      Reflect.getMetadata(CONFIG_CENTER_VALUE_METADATA_KEY, target.constructor) ||
+      new Map();
+    existingMetadata.set(propertyKey, metadata);
+    Reflect.defineMetadata(
+      CONFIG_CENTER_VALUE_METADATA_KEY,
+      existingMetadata,
+      target.constructor,
+    );
+
+    // 定义属性描述符，实现自动注入
+    // 注意：属性 getter 必须是同步的，不能是 async
+    // 实际值通过 initializeConfigCenterValues 在实例化时预加载
+    let value: string | undefined = options?.defaultValue;
+
+    Object.defineProperty(target, propertyKey, {
+      get: () => {
+        // 同步返回已加载的值（如果还未加载，返回默认值）
+        return value ?? options?.defaultValue ?? '';
+      },
+      set: (newValue: string) => {
+        value = newValue;
+      },
+      enumerable: true,
+      configurable: true,
+    });
+  };
+}
+
+/**
+ * 获取类的所有配置中心值元数据
+ */
+export function getConfigCenterValueMetadata(
+  target: Constructor<unknown>,
+): Map<string | symbol, ConfigCenterValueMetadata> {
+  return (
+    Reflect.getMetadata(CONFIG_CENTER_VALUE_METADATA_KEY, target) || new Map()
+  );
+}
+
+/**
+ * 初始化类的配置中心值
+ * 在类实例化后调用，自动加载配置值
+ */
+export async function initializeConfigCenterValues(
+  instance: object,
+  target: Constructor<unknown>,
+): Promise<void> {
+  const metadata = getConfigCenterValueMetadata(target);
+  const container = ControllerRegistry.getInstance().getContainer();
+  const configCenter = container.resolve<ConfigCenter>(CONFIG_CENTER_TOKEN);
+
+  if (!configCenter) {
+    return;
+  }
+
+  for (const [propertyKey, meta] of metadata.entries()) {
+    try {
+      const result = await configCenter.getConfig(
+        meta.dataId,
+        meta.groupName,
+        meta.namespaceId,
+      );
+      (instance as any)[propertyKey] = result.content;
+
+      // 如果启用监听，设置监听器
+      if (meta.watch) {
+        configCenter.watchConfig(meta.dataId, meta.groupName, (newResult: any) => {
+          (instance as any)[propertyKey] = newResult.content;
+        });
+      }
+    } catch (error) {
+      console.error(
+        `[ConfigCenterValue] Failed to initialize config ${meta.dataId}:`,
+        error,
+      );
+      if (meta.defaultValue !== undefined) {
+        (instance as any)[propertyKey] = meta.defaultValue;
+      }
+    }
+  }
+}
+

package/src/microservice/config-center/index.ts

@@ -0,0 +1,13 @@
+export { ConfigCenterModule } from './config-center-module';
+export type { ConfigCenterModuleOptions, ConfigCenterProvider, NacosConfigCenterOptions } from './config-center-module';
+export { NacosConfigCenter } from './nacos-config-center';
+export { CONFIG_CENTER_TOKEN } from './types';
+export type { ConfigCenter, ConfigChangeListener, ConfigResult } from './types';
+export {
+  ConfigCenterValue,
+  getConfigCenterValueMetadata,
+  initializeConfigCenterValues,
+} from './decorators';
+export { NacosValue } from './nacos-decorators';
+export type { ConfigCenterValueMetadata } from './decorators';
+

package/src/microservice/config-center/nacos-config-center.ts

@@ -0,0 +1,126 @@
+import { NacosClient, NacosConfigClient } from '@dangao/nacos-client';
+import type { ConfigResult, ConfigChangeListener, ConfigCenter } from './types';
+
+/**
+ * Nacos 配置中心实现
+ * 实现 ConfigCenter 接口，内部使用 @dangao/nacos-client
+ */
+export class NacosConfigCenter implements ConfigCenter {
+  private readonly client: NacosClient;
+  private readonly configClient: NacosConfigClient;
+  private readonly watchers: Map<string, { listener: ConfigChangeListener; intervalId: number }> = new Map();
+  private readonly watchInterval: number = 3000; // 默认 3 秒轮询一次
+
+  public constructor(
+    client: NacosClient,
+    options?: {
+      /**
+       * 配置监听轮询间隔（毫秒）
+       * @default 3000
+       */
+      watchInterval?: number;
+    },
+  ) {
+    this.client = client;
+    this.configClient = new NacosConfigClient(client);
+    this.watchInterval = options?.watchInterval ?? 3000;
+  }
+
+  /**
+   * 获取配置
+   */
+  public async getConfig(
+    dataId: string,
+    groupName: string,
+    namespaceId?: string,
+  ): Promise<ConfigResult> {
+    const result = await this.configClient.getConfig({
+      dataId,
+      groupName,
+      namespaceId,
+    });
+
+    return {
+      content: result.content,
+      md5: result.md5,
+      lastModified: result.lastModified,
+      contentType: result.contentType,
+    };
+  }
+
+  /**
+   * 监听配置变更
+   * 通过轮询和 md5 比对实现配置热更新
+   */
+  public watchConfig(
+    dataId: string,
+    groupName: string,
+    listener: ConfigChangeListener,
+    namespaceId?: string,
+  ): () => void {
+    const key = this.getWatchKey(dataId, groupName, namespaceId);
+
+    // 如果已经存在监听器，先取消
+    const existing = this.watchers.get(key);
+    if (existing) {
+      clearInterval(existing.intervalId);
+    }
+
+    let lastMd5: string | undefined;
+
+    const intervalId = setInterval(async () => {
+      try {
+        const result = await this.getConfig(dataId, groupName, namespaceId);
+
+        // 通过 md5 判断配置是否变更
+        if (lastMd5 === undefined || lastMd5 !== result.md5) {
+          lastMd5 = result.md5;
+          listener(result);
+        }
+      } catch (error) {
+        // 监听错误不抛出，避免影响其他监听器
+        console.error(`[NacosConfigCenter] Failed to watch config ${dataId}:`, error);
+      }
+    }, this.watchInterval) as unknown as number;
+
+    // 立即获取一次配置
+    this.getConfig(dataId, groupName, namespaceId)
+      .then((result) => {
+        lastMd5 = result.md5;
+        listener(result);
+      })
+      .catch((error) => {
+        console.error(`[NacosConfigCenter] Failed to get initial config ${dataId}:`, error);
+      });
+
+    this.watchers.set(key, { listener, intervalId });
+
+    // 返回取消监听的函数
+    return () => {
+      const watcher = this.watchers.get(key);
+      if (watcher) {
+        clearInterval(watcher.intervalId);
+        this.watchers.delete(key);
+      }
+    };
+  }
+
+  /**
+   * 关闭配置中心连接
+   */
+  public async close(): Promise<void> {
+    // 清除所有监听器
+    for (const watcher of this.watchers.values()) {
+      clearInterval(watcher.intervalId);
+    }
+    this.watchers.clear();
+  }
+
+  /**
+   * 生成监听器 key
+   */
+  private getWatchKey(dataId: string, groupName: string, namespaceId?: string): string {
+    return `${namespaceId ?? 'default'}:${groupName}:${dataId}`;
+  }
+}
+

package/src/microservice/config-center/nacos-decorators.ts

@@ -0,0 +1,34 @@
+import 'reflect-metadata';
+import { ConfigCenterValue } from './decorators';
+import type { Constructor } from '../../core/types';
+
+/**
+ * NacosValue 装饰器
+ * Nacos 特定的配置值注入装饰器
+ * 这是 @ConfigCenterValue 的便捷别名，专门用于 Nacos
+ *
+ * @example
+ * ```typescript
+ * @Injectable()
+ * class MyService {
+ *   @NacosValue('my-config', 'DEFAULT_GROUP')
+ *   public configValue: string = '';
+ *
+ *   @NacosValue('app-name', 'DEFAULT_GROUP', { defaultValue: 'MyApp' })
+ *   public appName: string = '';
+ * }
+ * ```
+ */
+export function NacosValue(
+  dataId: string,
+  groupName: string = 'DEFAULT_GROUP',
+  options?: {
+    namespaceId?: string;
+    defaultValue?: string;
+    watch?: boolean;
+  },
+): PropertyDecorator {
+  // 直接使用 ConfigCenterValue，因为 NacosConfigCenter 实现了 ConfigCenter 接口
+  return ConfigCenterValue(dataId, groupName, options);
+}
+