llm-content-creator 0.1.0

package/src/infrastructure/database/index.ts

@@ -0,0 +1,83 @@
+/**
+ * 数据库层导出
+ *
+ * 根据环境自动选择合适的实现
+ */
+
+import { config } from '../../config/index.js';
+import { MemoryTaskRepository } from './MemoryTaskRepository.js';
+import { SQLiteTaskRepository } from './SQLiteTaskRepository.js';
+import { createLogger } from '../logging/logger.js';
+
+const logger = createLogger('Database:Factory');
+
+// 导出类型
+export type { Task, TaskCreateParams, TaskUpdateParams, TaskListFilters, PaginatedResult } from '../../domain/repositories/TaskRepository.js';
+
+/**
+ * 创建 Task Repository
+ *
+ * 根据配置自动选择合适的实现：
+ * - development: SQLite (默认，适合开发)
+ * - production: PostgreSQL (需要数据库，连接失败时 fallback 到 SQLite)
+ * - test: Memory (默认，适合测试)
+ *
+ * 支持三种类型：
+ * - 'memory': 内存存储，适合测试
+ * - 'sqlite': SQLite 文件数据库，适合开发
+ * - 'postgres': PostgreSQL 数据库，适合生产
+ *
+ * PostgreSQL Fallback 策略：
+ * - 如果配置了 DATABASE_TYPE=postgres 但连接失败，自动 fallback 到 SQLite
+ * - 如果未配置 DATABASE_TYPE，根据环境自动选择
+ *
+ * @param pool - 可选的 PostgreSQL 连接池（仅 postgres 模式使用）
+ * @param dbPath - 可选的 SQLite 数据库路径（仅 sqlite 模式使用）
+ */
+export function createTaskRepository(pool?: any, dbPath?: string) {
+  const dbType = config.database.type;
+
+  logger.info('Creating Task Repository', { databaseType: dbType });
+
+  if (dbType === 'memory') {
+    // 内存版本（适合快速测试）
+    // 不创建任何数据库连接
+    logger.info('Using MemoryTaskRepository');
+    return new MemoryTaskRepository();
+  }
+
+  if (dbType === 'sqlite') {
+    // SQLite 版本（适合开发和本地开发）
+    logger.info('Using SQLiteTaskRepository', { dbPath: dbPath || './data/content-creator.db' });
+    return new SQLiteTaskRepository(dbPath);
+  }
+
+  if (dbType === 'postgres') {
+    // PostgreSQL 版本（需要数据库连接）
+    // 动态导入以避免在不需要时加载 pg
+    try {
+      const { PostgresTaskRepository } = require('./PostgresTaskRepository.js');
+      // 如果提供了 pool，使用提供的；否则由 BaseRepository 自动创建
+      logger.info('Using PostgresTaskRepository');
+      return new PostgresTaskRepository(pool);
+    } catch (error) {
+      const errorMsg = error instanceof Error ? error.message : String(error);
+      logger.warn('PostgreSQL initialization failed, falling back to SQLite', { error: errorMsg });
+      console.warn('⚠️  PostgreSQL not available, falling back to SQLite repository');
+      logger.info('Fallback: Using SQLiteTaskRepository', { dbPath: dbPath || './data/content-creator.db' });
+      return new SQLiteTaskRepository(dbPath);
+    }
+  }
+
+  // 不应该到这里，但作为最后的 fallback，使用 SQLite
+  logger.warn(`Unknown database type: ${dbType}, falling back to SQLite`);
+  return new SQLiteTaskRepository(dbPath);
+}
+
+// 导出具体的 Repository 类（可选使用）
+export { MemoryTaskRepository } from './MemoryTaskRepository.js';
+export { PostgresTaskRepository } from './PostgresTaskRepository.js';
+export { SQLiteTaskRepository } from './SQLiteTaskRepository.js';
+
+// 导出单例（全局使用）
+export const taskRepository = createTaskRepository();

package/src/infrastructure/logging/logger.ts

@@ -0,0 +1,231 @@
+/**
+ * 日志系统
+ *
+ * 基于 Winston 的结构化日志系统
+ */
+
+import winston from 'winston';
+import { config } from '../../config/index.js';
+
+/**
+ * 日志元数据
+ */
+export interface LogMetadata {
+  [key: string]: any;
+  context?: string;
+  userId?: string;
+  taskId?: string;
+  workerId?: string;
+  error?: string | {
+    code?: string;
+    stack?: string;
+    [key: string]: any;
+  };
+}
+
+/**
+ * 创建日志格式
+ */
+function createFormat() {
+  if (config.isDevelopment) {
+    // 开发环境：彩色、易读的格式
+    return winston.format.combine(
+      winston.format.timestamp({ format: 'YYYY-MM-DD HH:mm:ss' }),
+      winston.format.colorize(),
+      winston.format.printf(({ level, message, timestamp, ...metadata }) => {
+        let msg = `${timestamp} [${level}]: ${message}`;
+        if (Object.keys(metadata).length > 0) {
+          msg += ` ${JSON.stringify(metadata)}`;
+        }
+        return msg;
+      })
+    );
+  } else {
+    // 生产环境：JSON 格式
+    return winston.format.combine(
+      winston.format.timestamp(),
+      winston.format.errors({ stack: true }),
+      winston.format.json()
+    );
+  }
+}
+
+/**
+ * 创建日志传输器
+ */
+function createTransports(): winston.transport[] {
+  const transports: winston.transport[] = [];
+
+  // 控制台输出
+  transports.push(
+    new winston.transports.Console({
+      level: config.logging.level,
+      format: createFormat(),
+    })
+  );
+
+  // 文件输出（所有日志）
+  transports.push(
+    new winston.transports.File({
+      filename: config.logging.file,
+      level: config.logging.level,
+      format: winston.format.combine(
+        winston.format.timestamp(),
+        winston.format.errors({ stack: true }),
+        winston.format.json()
+      ),
+      maxsize: 10 * 1024 * 1024, // 10MB
+      maxFiles: 5,
+    })
+  );
+
+  // 错误日志单独记录
+  transports.push(
+    new winston.transports.File({
+      filename: config.logging.file.replace('.log', '.error.log'),
+      level: 'error',
+      format: winston.format.combine(
+        winston.format.timestamp(),
+        winston.format.errors({ stack: true }),
+        winston.format.json()
+      ),
+      maxsize: 10 * 1024 * 1024, // 10MB
+      maxFiles: 5,
+    })
+  );
+
+  return transports;
+}
+
+/**
+ * Winston Logger 实例
+ */
+const winstonLogger = winston.createLogger({
+  level: config.logging.level,
+  format: createFormat(),
+  transports: createTransports(),
+  // 异常处理
+  exceptionHandlers: [
+    new winston.transports.File({
+      filename: config.logging.file.replace('.log', '.exceptions.log'),
+    }),
+  ],
+  // 未处理的 Promise 拒绝
+  rejectionHandlers: [
+    new winston.transports.File({
+      filename: config.logging.file.replace('.log', '.rejections.log'),
+    }),
+  ],
+});
+
+/**
+ * 日志类
+ * 提供类型安全的日志接口
+ */
+export class Logger {
+  private context?: string;
+
+  constructor(context?: string) {
+    this.context = context;
+  }
+
+  /**
+   * 格式化日志消息
+   */
+  private formatMessage(message: string, _metadata?: LogMetadata): string {
+    let formattedMessage = message;
+
+    if (this.context) {
+      formattedMessage = `[${this.context}] ${message}`;
+    }
+
+    return formattedMessage;
+  }
+
+  /**
+   * 格式化日志元数据
+   */
+  private formatMetadata(metadata?: LogMetadata): any {
+    const meta: any = { ...metadata };
+
+    // 添加 worker ID
+    if (config.worker.id) {
+      meta.workerId = config.worker.id;
+    }
+
+    // 添加 context
+    if (this.context && !meta.context) {
+      meta.context = this.context;
+    }
+
+    return meta;
+  }
+
+  /**
+   * 记录调试日志
+   */
+  debug(message: string, metadata?: LogMetadata): void {
+    winstonLogger.debug(this.formatMessage(message), this.formatMetadata(metadata));
+  }
+
+  /**
+   * 记录信息日志
+   */
+  info(message: string, metadata?: LogMetadata): void {
+    winstonLogger.info(this.formatMessage(message), this.formatMetadata(metadata));
+  }
+
+  /**
+   * 记录警告日志
+   */
+  warn(message: string, metadata?: LogMetadata): void {
+    winstonLogger.warn(this.formatMessage(message), this.formatMetadata(metadata));
+  }
+
+  /**
+   * 记录错误日志
+   */
+  error(message: string, error?: Error | LogMetadata, metadata?: LogMetadata): void {
+    let meta = this.formatMetadata(metadata);
+
+    if (error instanceof Error) {
+      meta.error = {
+        message: error.message,
+        name: error.name,
+        stack: error.stack,
+        ...meta.error,
+      };
+    } else if (error) {
+      meta = { ...meta, ...error };
+    }
+
+    winstonLogger.error(this.formatMessage(message), meta);
+  }
+
+  /**
+   * 创建子日志器（带上下文）
+   */
+  child(context: string): Logger {
+    return new Logger(this.context ? `${this.context}:${context}` : context);
+  }
+}
+
+/**
+ * 默认日志器实例
+ */
+export const logger = new Logger();
+
+/**
+ * 关闭所有日志传输（用于应用退出时清理资源）
+ */
+export async function closeLogger(): Promise<void> {
+  await winstonLogger.close();
+  winstonLogger.info('Logger closed');
+}
+
+/**
+ * 创建带上下文的日志器
+ */
+export function createLogger(context: string): Logger {
+  return new Logger(context);
+}

package/src/infrastructure/monitoring/LoggingService.ts

@@ -0,0 +1,292 @@
+/**
+ * 增强日志服务
+ *
+ * 基于 Winston 的增强日志服务
+ * 提供日志轮转、结构化日志、日志查询等功能
+ */
+
+import winston from 'winston';
+import DailyRotateFile from 'winston-daily-rotate-file';
+import { createLogger } from '../logging/logger.js';
+
+const baseLogger = createLogger('LoggingService');
+
+/**
+ * 日志级别
+ */
+export type LogLevel = 'error' | 'warn' | 'info' | 'http' | 'verbose' | 'debug' | 'silly';
+
+/**
+ * 日志条目
+ */
+export interface LogEntry {
+  level: LogLevel;
+  message: string;
+  timestamp?: string;
+  context?: string;
+  [key: string]: any;
+}
+
+/**
+ * 日志查询选项
+ */
+export interface LogQueryOptions {
+  level?: LogLevel;
+  context?: string;
+  startTime?: Date;
+  endTime?: Date;
+  limit?: number;
+}
+
+/**
+ * 日志服务配置
+ */
+export interface LoggingServiceOptions {
+  logDir?: string;
+  datePattern?: string;
+  maxSize?: string;
+  maxFiles?: string;
+  level?: LogLevel;
+  format?: 'json' | 'text';
+}
+
+/**
+ * 默认配置
+ */
+const defaultOptions: LoggingServiceOptions = {
+  logDir: 'logs',
+  datePattern: 'YYYY-MM-DD',
+  maxSize: '20m',
+  maxFiles: '30d',
+  level: 'info',
+  format: 'json',
+};
+
+/**
+ * 日志服务类
+ */
+export class LoggingService {
+  private winstonLogger: winston.Logger;
+  private options: LoggingServiceOptions;
+
+  constructor(options?: Partial<LoggingServiceOptions>) {
+    this.options = { ...defaultOptions, ...options };
+    this.winstonLogger = this.createLogger();
+  }
+
+  /**
+   * 创建日志器
+   */
+  private createLogger(): winston.Logger {
+    const transports: winston.transport[] = [];
+
+    // 控制台输出
+    transports.push(
+      new winston.transports.Console({
+        format: winston.format.combine(
+          winston.format.timestamp({ format: 'YYYY-MM-DD HH:mm:ss' }),
+          winston.format.colorize(),
+          winston.format.printf(({ level, message, timestamp, ...metadata }) => {
+            let msg = `${timestamp} [${level}]: ${message}`;
+            if (Object.keys(metadata).length > 0) {
+              msg += ` ${JSON.stringify(metadata)}`;
+            }
+            return msg;
+          })
+        ),
+      })
+    );
+
+    // 综合日志文件（按日轮转）
+    transports.push(
+      new DailyRotateFile({
+        dirname: this.options.logDir,
+        filename: 'combined-%DATE%.log',
+        datePattern: this.options.datePattern,
+        maxSize: this.options.maxSize,
+        maxFiles: this.options.maxFiles,
+        format: winston.format.combine(
+          winston.format.timestamp(),
+          winston.format.errors({ stack: true }),
+          winston.format.json()
+        ),
+      })
+    );
+
+    // 错误日志文件（按日轮转）
+    transports.push(
+      new DailyRotateFile({
+        dirname: this.options.logDir,
+        filename: 'error-%DATE%.log',
+        datePattern: this.options.datePattern,
+        level: 'error',
+        maxSize: this.options.maxSize,
+        maxFiles: this.options.maxFiles,
+        format: winston.format.combine(
+          winston.format.timestamp(),
+          winston.format.errors({ stack: true }),
+          winston.format.json()
+        ),
+      })
+    );
+
+    // 性能日志文件（按日轮转）
+    transports.push(
+      new DailyRotateFile({
+        dirname: this.options.logDir,
+        filename: 'performance-%DATE%.log',
+        datePattern: this.options.datePattern,
+        maxSize: this.options.maxSize,
+        maxFiles: this.options.maxFiles,
+        format: winston.format.combine(
+          winston.format.timestamp(),
+          winston.format.json()
+        ),
+      })
+    );
+
+    const logger = winston.createLogger({
+      level: this.options.level || 'info',
+      transports,
+      // 异常处理
+      exceptionHandlers: [
+        new winston.transports.File({
+          filename: `${this.options.logDir}/exceptions.log`,
+        }),
+      ],
+      // 未处理的 Promise 拒绝
+      rejectionHandlers: [
+        new winston.transports.File({
+          filename: `${this.options.logDir}/rejections.log`,
+        }),
+      ],
+    });
+
+    baseLogger.info('Logging service created', {
+      logDir: this.options.logDir,
+      level: this.options.level,
+    });
+
+    return logger;
+  }
+
+  /**
+   * 记录日志
+   */
+  log(level: LogLevel, message: string, metadata?: Record<string, any>): void {
+    this.winstonLogger.log(level, message, metadata);
+  }
+
+  /**
+   * 记录调试日志
+   */
+  debug(message: string, metadata?: Record<string, any>): void {
+    this.winstonLogger.debug(message, metadata);
+  }
+
+  /**
+   * 记录信息日志
+   */
+  info(message: string, metadata?: Record<string, any>): void {
+    this.winstonLogger.info(message, metadata);
+  }
+
+  /**
+   * 记录警告日志
+   */
+  warn(message: string, metadata?: Record<string, any>): void {
+    this.winstonLogger.warn(message, metadata);
+  }
+
+  /**
+   * 记录错误日志
+   */
+  error(message: string, error?: Error | Record<string, any>, metadata?: Record<string, any>): void {
+    let meta = { ...metadata };
+
+    if (error instanceof Error) {
+      meta = {
+        ...meta,
+        error: {
+          message: error.message,
+          name: error.name,
+          stack: error.stack,
+        },
+      };
+    } else if (error) {
+      meta = { ...meta, ...error };
+    }
+
+    this.winstonLogger.error(message, meta);
+  }
+
+  /**
+   * 记录性能日志
+   */
+  performance(operation: string, duration: number, metadata?: Record<string, any>): void {
+    this.winstonLogger.info('performance', {
+      operation,
+      duration,
+      unit: 'ms',
+      ...metadata,
+    });
+  }
+
+  /**
+   * 记录 HTTP 请求日志
+   */
+  http(req: {
+    method: string;
+    url: string;
+    status?: number;
+    duration?: number;
+    userAgent?: string;
+    ip?: string;
+  }): void {
+    this.winstonLogger.http('HTTP request', {
+      method: req.method,
+      url: req.url,
+      status: req.status,
+      duration: req.duration,
+      userAgent: req.userAgent,
+      ip: req.ip,
+    });
+  }
+
+  /**
+   * 创建子日志器（带上下文）
+   */
+  child(context: string): LoggingService {
+    const childService = new LoggingService(this.options);
+    childService.winstonLogger = this.winstonLogger.child({ context });
+    return childService;
+  }
+
+  /**
+   * 设置日志级别
+   */
+  setLevel(level: LogLevel): void {
+    this.winstonLogger.level = level;
+    this.winstonLogger.info('Log level changed', { level });
+  }
+
+  /**
+   * 获取日志级别
+   */
+  getLevel(): string {
+    return this.winstonLogger.level;
+  }
+
+  /**
+   * 关闭日志器
+   */
+  async close(): Promise<void> {
+    this.winstonLogger.close();
+    baseLogger.info('Logger closed');
+  }
+}
+
+/**
+ * 日志服务单例
+ */
+export const loggingService = new LoggingService();