@dangao/bun-server 1.2.0 → 1.4.0

package/src/microservice/monitoring/types.ts

@@ -0,0 +1,118 @@
+/**
+ * 服务调用指标
+ */
+export interface ServiceCallMetrics {
+  /**
+   * 服务名
+   */
+  serviceName: string;
+
+  /**
+   * 实例标识（IP:Port）
+   */
+  instance: string;
+
+  /**
+   * 总请求数
+   */
+  totalRequests: number;
+
+  /**
+   * 成功请求数
+   */
+  successRequests: number;
+
+  /**
+   * 失败请求数
+   */
+  failureRequests: number;
+
+  /**
+   * 平均延迟（毫秒）
+   */
+  averageLatency: number;
+
+  /**
+   * 最小延迟（毫秒）
+   */
+  minLatency: number;
+
+  /**
+   * 最大延迟（毫秒）
+   */
+  maxLatency: number;
+
+  /**
+   * 错误率（0-1）
+   */
+  errorRate: number;
+
+  /**
+   * 最后更新时间（时间戳）
+   */
+  lastUpdateTime: number;
+}
+
+/**
+ * 服务实例健康状态
+ */
+export interface ServiceInstanceHealth {
+  /**
+   * 服务名
+   */
+  serviceName: string;
+
+  /**
+   * 实例标识（IP:Port）
+   */
+  instance: string;
+
+  /**
+   * 是否健康
+   */
+  healthy: boolean;
+
+  /**
+   * 健康检查时间（时间戳）
+   */
+  checkTime: number;
+
+  /**
+   * 连续失败次数
+   */
+  consecutiveFailures: number;
+
+  /**
+   * 最后成功时间（时间戳）
+   */
+  lastSuccessTime?: number;
+
+  /**
+   * 最后失败时间（时间戳）
+   */
+  lastFailureTime?: number;
+}
+
+/**
+ * 监控配置选项
+ */
+export interface MonitoringOptions {
+  /**
+   * 是否启用监控
+   * @default true
+   */
+  enabled?: boolean;
+
+  /**
+   * 指标收集间隔（毫秒）
+   * @default 60000
+   */
+  collectionInterval?: number;
+
+  /**
+   * 是否自动上报到 MetricsModule
+   * @default true
+   */
+  autoReportToMetrics?: boolean;
+}
+

package/src/microservice/service-client/call-decorators.ts

@@ -0,0 +1,107 @@
+import 'reflect-metadata';
+import type { ServiceCallOptions, ServiceCallResponse } from './types';
+import type { Constructor } from '../../core/types';
+
+/**
+ * 服务调用装饰器元数据键
+ */
+const SERVICE_CALL_METADATA_KEY = Symbol('service-call:metadata');
+
+/**
+ * 服务调用元数据
+ */
+export interface ServiceCallMetadata {
+  /**
+   * 服务名
+   */
+  serviceName: string;
+
+  /**
+   * HTTP 方法
+   */
+  method: 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+
+  /**
+   * 请求路径
+   */
+  path: string;
+
+  /**
+   * 服务调用选项
+   */
+  options?: Partial<ServiceCallOptions>;
+}
+
+/**
+ * ServiceCall 装饰器
+ * 用于标记服务调用方法，自动注入 ServiceClient 调用逻辑
+ *
+ * @example
+ * ```typescript
+ * @Injectable()
+ * class MyService {
+ *   public constructor(
+ *     @ServiceClient() private readonly serviceClient: ServiceClient,
+ *   ) {}
+ *
+ *   @ServiceCall('user-service', 'GET', '/api/users/:id')
+ *   public async getUser(id: string): Promise<User> {
+ *     // 自动调用 user-service 的 /api/users/:id 接口
+ *     // 需要手动调用 serviceClient.call()，装饰器只提供元数据
+ *   }
+ * }
+ * ```
+ *
+ * 注意：此装饰器主要用于标记和提供元数据，实际调用仍需要通过 ServiceClient
+ */
+export function ServiceCall(
+  serviceName: string,
+  method: 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH',
+  path: string,
+  options?: Partial<ServiceCallOptions>,
+): MethodDecorator {
+  return function (
+    target: object,
+    propertyKey: string | symbol,
+    descriptor: PropertyDescriptor,
+  ) {
+    const metadata: ServiceCallMetadata = {
+      serviceName,
+      method,
+      path,
+      options,
+    };
+
+    // 保存元数据
+    const existingMetadata: Map<string | symbol, ServiceCallMetadata> =
+      Reflect.getMetadata(SERVICE_CALL_METADATA_KEY, target.constructor) ||
+      new Map();
+    existingMetadata.set(propertyKey, metadata);
+    Reflect.defineMetadata(
+      SERVICE_CALL_METADATA_KEY,
+      existingMetadata,
+      target.constructor,
+    );
+
+    // 保存原始方法
+    const originalMethod = descriptor.value;
+
+    // 包装方法，添加服务调用元数据
+    descriptor.value = async function (...args: any[]) {
+      // 元数据可以通过 Reflect.getMetadata 获取
+      // 实际调用逻辑需要结合 ServiceClient 实现
+      // 这里只提供元数据标记，不自动执行调用
+      return originalMethod.apply(this, args);
+    };
+  };
+}
+
+/**
+ * 获取方法的服务调用元数据
+ */
+export function getServiceCallMetadata(
+  target: Constructor<unknown>,
+): Map<string | symbol, ServiceCallMetadata> {
+  return Reflect.getMetadata(SERVICE_CALL_METADATA_KEY, target) || new Map();
+}
+

package/src/microservice/service-client/decorators.ts

@@ -0,0 +1,87 @@
+import 'reflect-metadata';
+import { ServiceClient as ServiceClientImpl } from './service-client';
+import type { ServiceRegistry } from '../service-registry/types';
+import { SERVICE_REGISTRY_TOKEN } from '../service-registry/types';
+import { ControllerRegistry } from '../../controller/controller';
+import type { Constructor } from '../../core/types';
+
+/**
+ * ServiceClient 注入装饰器元数据键
+ */
+const SERVICE_CLIENT_METADATA_KEY = Symbol('service-client:inject');
+
+/**
+ * ServiceClient 装饰器
+ * 用于在类中注入 ServiceClient 实例
+ *
+ * @example
+ * ```typescript
+ * @Injectable()
+ * class MyService {
+ *   public constructor(
+ *     @ServiceClient() private readonly serviceClient: ServiceClient,
+ *   ) {}
+ *
+ *   public async callOtherService() {
+ *     return await this.serviceClient.call({
+ *       serviceName: 'other-service',
+ *       path: '/api/data',
+ *       method: 'GET',
+ *     });
+ *   }
+ * }
+ * ```
+ */
+export function ServiceClient(): ParameterDecorator {
+  return function (
+    target: unknown,
+    _propertyKey: string | symbol | undefined,
+    parameterIndex: number,
+  ) {
+    const constructor = typeof target === 'function'
+      ? (target as Constructor<unknown>)
+      : (target as any)?.constructor;
+
+    if (!constructor) {
+      return;
+    }
+
+    // 保存元数据，标记该参数需要注入 ServiceClient
+    const existingMetadata: number[] =
+      Reflect.getMetadata(SERVICE_CLIENT_METADATA_KEY, constructor) || [];
+    existingMetadata.push(parameterIndex);
+    Reflect.defineMetadata(
+      SERVICE_CLIENT_METADATA_KEY,
+      existingMetadata,
+      constructor,
+    );
+  };
+}
+
+/**
+ * 创建 ServiceClient 实例
+ */
+export function createServiceClient(): ServiceClientImpl {
+  const container = ControllerRegistry.getInstance().getContainer();
+  const serviceRegistry = container.resolve<ServiceRegistry>(
+    SERVICE_REGISTRY_TOKEN,
+  );
+
+  if (!serviceRegistry) {
+    throw new Error(
+      'ServiceRegistry not found. Please register ServiceRegistryModule first.',
+    );
+  }
+
+  return new ServiceClientImpl(serviceRegistry);
+}
+
+/**
+ * 获取需要注入 ServiceClient 的参数索引列表
+ */
+export function getServiceClientParameterIndices(
+  target: Constructor<unknown>,
+): number[] {
+  return Reflect.getMetadata(SERVICE_CLIENT_METADATA_KEY, target) || [];
+}
+

package/src/microservice/service-client/index.ts

@@ -0,0 +1,37 @@
+export { ServiceClient } from './service-client';
+export {
+  LoadBalancerFactory,
+  RandomLoadBalancer,
+  RoundRobinLoadBalancer,
+  WeightedRoundRobinLoadBalancer,
+  ConsistentHashLoadBalancer,
+  LeastActiveLoadBalancer,
+} from './load-balancer';
+export {
+  TraceIdRequestInterceptor,
+  UserInfoRequestInterceptor,
+  RequestLogInterceptor,
+  ResponseLogInterceptor,
+  ResponseTransformInterceptor,
+  ErrorHandlerInterceptor,
+} from './interceptors';
+export {
+  ServiceClient as ServiceClientDecorator,
+  createServiceClient,
+  getServiceClientParameterIndices,
+} from './decorators';
+export {
+  ServiceCall,
+  getServiceCallMetadata,
+  type ServiceCallMetadata,
+} from './call-decorators';
+export type {
+  LoadBalanceStrategy,
+  LoadBalancer,
+  ServiceCallOptions,
+  ServiceCallResponse,
+  ServiceRequestInterceptor,
+  ServiceResponseInterceptor,
+  ServiceCallError,
+} from './types';
+

package/src/microservice/service-client/interceptors.ts

@@ -0,0 +1,182 @@
+import type {
+  ServiceCallOptions,
+  ServiceCallResponse,
+  ServiceRequestInterceptor,
+  ServiceResponseInterceptor,
+} from './types';
+
+/**
+ * 追踪 ID 请求拦截器
+ * 自动在请求头中添加追踪 ID
+ */
+export class TraceIdRequestInterceptor implements ServiceRequestInterceptor {
+  private readonly headerName: string;
+  private readonly traceIdGenerator: () => string;
+
+  public constructor(options?: {
+    /**
+     * 追踪 ID 请求头名称
+     * @default 'X-Trace-Id'
+     */
+    headerName?: string;
+
+    /**
+     * 追踪 ID 生成器
+     * @default 生成 UUID v4
+     */
+    traceIdGenerator?: () => string;
+  }) {
+    this.headerName = options?.headerName ?? 'X-Trace-Id';
+    this.traceIdGenerator = options?.traceIdGenerator ?? this.generateTraceId;
+  }
+
+  public intercept(options: ServiceCallOptions): ServiceCallOptions {
+    const traceId = this.traceIdGenerator();
+    return {
+      ...options,
+      headers: {
+        ...options.headers,
+        [this.headerName]: traceId,
+      },
+    };
+  }
+
+  /**
+   * 生成追踪 ID（UUID v4 简化版）
+   */
+  private generateTraceId(): string {
+    return `${Date.now()}-${Math.random().toString(36).substring(2, 15)}`;
+  }
+}
+
+/**
+ * 用户信息请求拦截器
+ * 自动在请求头中添加用户信息
+ */
+export class UserInfoRequestInterceptor implements ServiceRequestInterceptor {
+  private readonly headerName: string;
+  private readonly userInfoProvider: () => string | undefined;
+
+  public constructor(options: {
+    /**
+     * 用户信息请求头名称
+     * @default 'X-User-Info'
+     */
+    headerName?: string;
+
+    /**
+     * 用户信息提供者
+     */
+    userInfoProvider: () => string | undefined;
+  }) {
+    this.headerName = options.headerName ?? 'X-User-Info';
+    this.userInfoProvider = options.userInfoProvider;
+  }
+
+  public intercept(options: ServiceCallOptions): ServiceCallOptions {
+    const userInfo = this.userInfoProvider();
+    if (!userInfo) {
+      return options;
+    }
+
+    return {
+      ...options,
+      headers: {
+        ...options.headers,
+        [this.headerName]: userInfo,
+      },
+    };
+  }
+}
+
+/**
+ * 请求日志拦截器
+ * 记录请求日志
+ */
+export class RequestLogInterceptor implements ServiceRequestInterceptor {
+  private readonly logger?: (message: string) => void;
+
+  public constructor(options?: {
+    /**
+     * 日志记录器
+     */
+    logger?: (message: string) => void;
+  }) {
+    this.logger = options?.logger ?? console.log;
+  }
+
+  public intercept(options: ServiceCallOptions): ServiceCallOptions {
+    if (this.logger) {
+      this.logger(
+        `[ServiceClient] Request: ${options.method} ${options.serviceName}${options.path}`,
+      );
+    }
+    return options;
+  }
+}
+
+/**
+ * 响应日志拦截器
+ * 记录响应日志
+ */
+export class ResponseLogInterceptor implements ServiceResponseInterceptor {
+  private readonly logger?: (message: string) => void;
+
+  public constructor(options?: {
+    /**
+     * 日志记录器
+     */
+    logger?: (message: string) => void;
+  }) {
+    this.logger = options?.logger ?? console.log;
+  }
+
+  public intercept<T>(response: ServiceCallResponse<T>): ServiceCallResponse<T> {
+    if (this.logger) {
+      this.logger(
+        `[ServiceClient] Response: ${response.status} from ${response.instance.ip}:${response.instance.port}`,
+      );
+    }
+    return response;
+  }
+}
+
+/**
+ * 响应数据转换拦截器
+ * 转换响应数据格式
+ * 
+ * 注意：这个拦截器会改变响应数据的类型，因此需要类型断言
+ */
+export class ResponseTransformInterceptor<TInput = unknown, TOutput = unknown> {
+  private readonly transformer: (data: TInput) => TOutput;
+
+  public constructor(transformer: (data: TInput) => TOutput) {
+    this.transformer = transformer;
+  }
+
+  /**
+   * 拦截响应并转换数据
+   */
+  public intercept<T>(response: ServiceCallResponse<T>): ServiceCallResponse<TOutput> {
+    return {
+      ...response,
+      data: this.transformer(response.data as unknown as TInput),
+    } as ServiceCallResponse<TOutput>;
+  }
+}
+
+/**
+ * 错误处理拦截器
+ * 统一处理错误响应
+ */
+export class ErrorHandlerInterceptor implements ServiceResponseInterceptor {
+  public intercept<T>(response: ServiceCallResponse<T>): ServiceCallResponse<T> {
+    // 如果状态码表示错误，可以在这里进行统一处理
+    // 例如：记录错误日志、转换错误格式等
+    if (response.status >= 400) {
+      // 可以在这里添加错误处理逻辑
+    }
+    return response;
+  }
+}
+