llm-content-creator 0.1.0

package/dist/infrastructure/database/BaseRepository.d.ts

@@ -0,0 +1,98 @@
+/**
+ * BaseRepository - Repository 基类
+ *
+ * 提供通用的数据库操作和事务管理
+ * 所有具体 Repository 应该继承此类
+ */
+import type { Pool, PoolClient, QueryResult, QueryResultRow } from 'pg';
+/**
+ * 查询统计信息
+ */
+export interface QueryStats {
+    text: string;
+    duration: number;
+    rows: number;
+}
+/**
+ * Repository 基类
+ */
+export declare abstract class BaseRepository {
+    protected pool: Pool;
+    constructor(pool?: Pool);
+    /**
+     * 执行查询
+     *
+     * @param text - SQL 查询语句
+     * @param params - 查询参数
+     * @returns 查询结果
+     */
+    protected query<T extends QueryResultRow>(text: string, params?: any[]): Promise<QueryResult<T>>;
+    /**
+     * 获取数据库连接（用于事务）
+     *
+     * @returns 数据库连接
+     */
+    protected getConnection(): Promise<PoolClient>;
+    /**
+     * 执行事务
+     *
+     * @param callback - 事务回调函数
+     * @returns 回调函数的返回值
+     */
+    protected transaction<T>(callback: (client: PoolClient) => Promise<T>): Promise<T>;
+    /**
+     * 批量执行查询（在单个事务中）
+     *
+     * @param queries - 查询数组
+     * @returns 所有查询的结果数组
+     */
+    protected batchQuery<T extends QueryResultRow>(queries: Array<{
+        text: string;
+        params?: any[];
+    }>): Promise<QueryResult<T>[]>;
+    /**
+     * 检查记录是否存在
+     *
+     * @param table - 表名
+     * @param field - 字段名
+     * @param value - 字段值
+     * @returns 是否存在
+     */
+    protected exists(table: string, field: string, value: any): Promise<boolean>;
+    /**
+     * 计数查询
+     *
+     * @param table - 表名
+     * @param where - WHERE 条件（可选）
+     * @returns 记录数
+     */
+    protected count(table: string, where?: string): Promise<number>;
+    /**
+     * 记录查询日志
+     *
+     * @param stats - 查询统计信息
+     */
+    private logQuery;
+    /**
+     * 获取连接池统计信息
+     *
+     * @returns 连接池统计信息
+     */
+    getPoolStats(): {
+        totalCount: number;
+        idleCount: number;
+        waitingCount: number;
+    };
+    /**
+     * 关闭连接池
+     * 应该在应用关闭时调用
+     */
+    close(): Promise<void>;
+    /**
+     * 健康检查
+     *
+     * @returns 数据库是否健康
+     */
+    healthCheck(): Promise<boolean>;
+}
+//# sourceMappingURL=BaseRepository.d.ts.map

package/dist/infrastructure/database/BaseRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseRepository.d.ts","sourceRoot":"","sources":["../../../src/infrastructure/database/BaseRepository.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAIxE;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,8BAAsB,cAAc;IAClC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC;gBAET,IAAI,CAAC,EAAE,IAAI;IAmBvB;;;;;;OAMG;cACa,KAAK,CAAC,CAAC,SAAS,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAmBtG;;;;OAIG;cACa,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC;IAIpD;;;;;OAKG;cACa,WAAW,CAAC,CAAC,EAC3B,QAAQ,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,GAC3C,OAAO,CAAC,CAAC,CAAC;IAgBb;;;;;OAKG;cACa,UAAU,CAAC,CAAC,SAAS,cAAc,EAAE,OAAO,EAAE,KAAK,CAAC;QAClE,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC;KAChB,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IAa9B;;;;;;;OAOG;cACa,MAAM,CACpB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,GAAG,GACT,OAAO,CAAC,OAAO,CAAC;IAMnB;;;;;;OAMG;cACa,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAAe,GAAG,OAAO,CAAC,MAAM,CAAC;IAM7E;;;;OAIG;IACH,OAAO,CAAC,QAAQ;IAUhB;;;;OAIG;IACH,YAAY,IAAI;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE;IAQ/E;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAU5B;;;;OAIG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAStC"}

package/dist/infrastructure/database/BaseRepository.js

@@ -0,0 +1,178 @@
+/**
+ * BaseRepository - Repository 基类
+ *
+ * 提供通用的数据库操作和事务管理
+ * 所有具体 Repository 应该继承此类
+ */
+import { Pool as PoolClass } from 'pg';
+import { config } from '../../config/index.js';
+/**
+ * Repository 基类
+ */
+export class BaseRepository {
+    pool;
+    constructor(pool) {
+        this.pool = pool || new PoolClass({
+            host: config.database.host,
+            port: config.database.port,
+            database: config.database.database,
+            user: config.database.user,
+            password: config.database.password,
+            max: 20, // 连接池大小
+            idleTimeoutMillis: 30000,
+            connectionTimeoutMillis: 2000,
+        });
+        // 监听连接池错误
+        this.pool.on('error', (err) => {
+            console.error('Unexpected error on idle client', err);
+            process.exit(-1);
+        });
+    }
+    /**
+     * 执行查询
+     *
+     * @param text - SQL 查询语句
+     * @param params - 查询参数
+     * @returns 查询结果
+     */
+    async query(text, params) {
+        const start = Date.now();
+        try {
+            const res = await this.pool.query(text, params);
+            const duration = Date.now() - start;
+            // 记录查询统计（开发环境）
+            if (process.env.NODE_ENV !== 'production') {
+                this.logQuery({ text, duration, rows: res.rowCount || 0 });
+            }
+            return res;
+        }
+        catch (error) {
+            console.error('Query error', { text, error });
+            throw error;
+        }
+    }
+    /**
+     * 获取数据库连接（用于事务）
+     *
+     * @returns 数据库连接
+     */
+    async getConnection() {
+        return await this.pool.connect();
+    }
+    /**
+     * 执行事务
+     *
+     * @param callback - 事务回调函数
+     * @returns 回调函数的返回值
+     */
+    async transaction(callback) {
+        const client = await this.getConnection();
+        try {
+            await client.query('BEGIN');
+            const result = await callback(client);
+            await client.query('COMMIT');
+            return result;
+        }
+        catch (error) {
+            await client.query('ROLLBACK');
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * 批量执行查询（在单个事务中）
+     *
+     * @param queries - 查询数组
+     * @returns 所有查询的结果数组
+     */
+    async batchQuery(queries) {
+        return this.transaction(async (client) => {
+            const results = [];
+            for (const query of queries) {
+                const result = await client.query(query.text, query.params);
+                results.push(result);
+            }
+            return results;
+        });
+    }
+    /**
+     * 检查记录是否存在
+     *
+     * @param table - 表名
+     * @param field - 字段名
+     * @param value - 字段值
+     * @returns 是否存在
+     */
+    async exists(table, field, value) {
+        const text = `SELECT EXISTS(SELECT 1 FROM ${table} WHERE ${field} = $1)`;
+        const result = await this.query(text, [value]);
+        return result.rows[0]?.exists ?? false;
+    }
+    /**
+     * 计数查询
+     *
+     * @param table - 表名
+     * @param where - WHERE 条件（可选）
+     * @returns 记录数
+     */
+    async count(table, where = 'TRUE') {
+        const text = `SELECT COUNT(*) as count FROM ${table} WHERE ${where}`;
+        const result = await this.query(text);
+        return parseInt(result.rows[0]?.count ?? '0', 10);
+    }
+    /**
+     * 记录查询日志
+     *
+     * @param stats - 查询统计信息
+     */
+    logQuery(stats) {
+        const { text, duration, rows } = stats;
+        // 截断过长的查询语句
+        const truncatedText = text.length > 100 ? text.substring(0, 100) + '...' : text;
+        console.log(`[Query] ${truncatedText} - ${duration}ms - ${rows} rows`);
+    }
+    /**
+     * 获取连接池统计信息
+     *
+     * @returns 连接池统计信息
+     */
+    getPoolStats() {
+        return {
+            totalCount: this.pool.totalCount,
+            idleCount: this.pool.idleCount,
+            waitingCount: this.pool.waitingCount,
+        };
+    }
+    /**
+     * 关闭连接池
+     * 应该在应用关闭时调用
+     */
+    async close() {
+        try {
+            await this.pool.end();
+            console.log('Database connection pool closed');
+        }
+        catch (error) {
+            console.error('Error closing database connection pool:', error);
+            throw error;
+        }
+    }
+    /**
+     * 健康检查
+     *
+     * @returns 数据库是否健康
+     */
+    async healthCheck() {
+        try {
+            await this.query('SELECT 1');
+            return true;
+        }
+        catch (error) {
+            console.error('Database health check failed:', error);
+            return false;
+        }
+    }
+}
+//# sourceMappingURL=BaseRepository.js.map

package/dist/infrastructure/database/BaseRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseRepository.js","sourceRoot":"","sources":["../../../src/infrastructure/database/BaseRepository.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,IAAI,IAAI,SAAS,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAW/C;;GAEG;AACH,MAAM,OAAgB,cAAc;IACxB,IAAI,CAAO;IAErB,YAAY,IAAW;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,SAAS,CAAC;YAChC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;YAC1B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;YAC1B,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;YAClC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;YAC1B,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;YAClC,GAAG,EAAE,EAAE,EAAE,QAAQ;YACjB,iBAAiB,EAAE,KAAK;YACxB,uBAAuB,EAAE,IAAI;SAC9B,CAAC,CAAC;QAEH,UAAU;QACV,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;YACnC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,CAAC;YACtD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACO,KAAK,CAAC,KAAK,CAA2B,IAAY,EAAE,MAAc;QAC1E,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAI,IAAI,EAAE,MAAM,CAAC,CAAC;YACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YAEpC,eAAe;YACf,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,EAAE,CAAC;gBAC1C,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACO,KAAK,CAAC,aAAa;QAC3B,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACnC,CAAC;IAED;;;;;OAKG;IACO,KAAK,CAAC,WAAW,CACzB,QAA4C;QAE5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC7B,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,KAAK,CAAC,UAAU,CAA2B,OAGnD;QACA,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACvC,MAAM,OAAO,GAAqB,EAAE,CAAC;YAErC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAI,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC/D,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACO,KAAK,CAAC,MAAM,CACpB,KAAa,EACb,KAAa,EACb,KAAU;QAEV,MAAM,IAAI,GAAG,+BAA+B,KAAK,UAAU,KAAK,QAAQ,CAAC;QACzE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAsB,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QACpE,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,KAAK,CAAC;IACzC,CAAC;IAED;;;;;;OAMG;IACO,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,QAAgB,MAAM;QACzD,MAAM,IAAI,GAAG,iCAAiC,KAAK,UAAU,KAAK,EAAE,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAoB,IAAI,CAAC,CAAC;QACzD,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;IACpD,CAAC;IAED;;;;OAIG;IACK,QAAQ,CAAC,KAAiB;QAChC,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QAEvC,YAAY;QACZ,MAAM,aAAa,GACjB,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;QAE5D,OAAO,CAAC,GAAG,CAAC,WAAW,aAAa,MAAM,QAAQ,QAAQ,IAAI,OAAO,CAAC,CAAC;IACzE,CAAC;IAED;;;;OAIG;IACH,YAAY;QACV,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;YAChC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;YAC9B,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;SACrC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC7B,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

package/dist/infrastructure/database/MemoryTaskRepository.d.ts

@@ -0,0 +1,77 @@
+/**
+ * Memory Task Repository
+ *
+ * 内存版本的 Task Repository，用于测试和开发
+ * 数据不持久化，适合快速测试工作流
+ */
+import type { Task, TaskCreateParams, TaskUpdateParams, TaskListFilters, PaginatedResult } from '../../domain/repositories/TaskRepository.js';
+import { TaskStatus } from '../../domain/entities/Task.js';
+/**
+ * 内存 Task Repository 实现
+ */
+export declare class MemoryTaskRepository {
+    private tasks;
+    private snapshots;
+    constructor();
+    /**
+     * 创建任务
+     */
+    create(params: TaskCreateParams): Promise<Task>;
+    /**
+     * 根据 ID 查找任务
+     */
+    findById(id: string): Promise<Task | null>;
+    /**
+     * 更新任务
+     */
+    update(id: string, params: TaskUpdateParams): Promise<Task>;
+    /**
+     * 更新任务状态（带乐观锁）
+     */
+    updateStatus(taskId: string, status: TaskStatus, version: number): Promise<boolean>;
+    /**
+     * 抢占任务（Worker 抢占机制）
+     */
+    claimForProcessing(workerId: string, limit?: number): Promise<Task[]>;
+    /**
+     * 保存状态快照（用于崩溃恢复）
+     */
+    saveStateSnapshot(taskId: string, step: string, state: any): Promise<void>;
+    /**
+     * 恢复状态快照
+     */
+    loadStateSnapshot(taskId: string): Promise<any | null>;
+    /**
+     * 列出任务（带过滤和分页）
+     */
+    list(filters?: TaskListFilters): Promise<PaginatedResult<Task>>;
+    /**
+     * 删除任务
+     */
+    delete(id: string): Promise<boolean>;
+    /**
+     * 健康检查
+     */
+    healthCheck(): Promise<boolean>;
+    /**
+     * 更新当前步骤（带乐观锁）
+     */
+    updateCurrentStep(taskId: string, step: string, version: number): Promise<boolean>;
+    /**
+     * 标记任务完成（带乐观锁）
+     */
+    markAsCompleted(taskId: string, version: number): Promise<boolean>;
+    /**
+     * 标记任务失败（带乐观锁）
+     */
+    markAsFailed(taskId: string, errorMessage: string, version: number): Promise<boolean>;
+    /**
+     * 获取统计信息
+     */
+    getStats(): {
+        totalTasks: number;
+        snapshots: number;
+        byStatus: Record<string, number>;
+    };
+}
+//# sourceMappingURL=MemoryTaskRepository.d.ts.map

package/dist/infrastructure/database/MemoryTaskRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MemoryTaskRepository.d.ts","sourceRoot":"","sources":["../../../src/infrastructure/database/MemoryTaskRepository.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EACV,IAAI,EACJ,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,eAAe,EAChB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAI3D;;GAEG;AACH,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,SAAS,CAA+B;;IAMhD;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IA2BrD;;OAEG;IACG,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAKhD;;OAEG;IACG,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAyBjE;;OAEG;IACG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA6BzF;;OAEG;IACG,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IA0C9E;;OAEG;IACG,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAYhF;;OAEG;IACG,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,GAAG,IAAI,CAAC;IAa5D;;OAEG;IACG,IAAI,CAAC,OAAO,GAAE,eAAoB,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IA6CzE;;OAEG;IACG,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAc1C;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAIrC;;OAEG;IACG,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAyBxF;;OAEG;IACG,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA0BxE;;OAEG;IACG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA2B3F;;OAEG;IACH,QAAQ;;;;;CAUT"}

package/dist/infrastructure/database/MemoryTaskRepository.js

@@ -0,0 +1,309 @@
+/**
+ * Memory Task Repository
+ *
+ * 内存版本的 Task Repository，用于测试和开发
+ * 数据不持久化，适合快速测试工作流
+ */
+import { createLogger } from '../../infrastructure/logging/logger.js';
+import { TaskStatus } from '../../domain/entities/Task.js';
+const logger = createLogger('Memory:TaskRepository');
+/**
+ * 内存 Task Repository 实现
+ */
+export class MemoryTaskRepository {
+    tasks = new Map();
+    snapshots = new Map();
+    constructor() {
+        logger.info('Memory repository initialized (no persistence)');
+    }
+    /**
+     * 创建任务
+     */
+    async create(params) {
+        const now = new Date();
+        const task = {
+            id: params.id,
+            taskId: params.id,
+            mode: params.mode,
+            type: params.type,
+            topic: params.topic,
+            requirements: params.requirements,
+            hardConstraints: params.hardConstraints,
+            status: TaskStatus.PENDING,
+            priority: 2, // NORMAL
+            version: 1,
+            textRetryCount: 0,
+            imageRetryCount: 0,
+            targetAudience: 'general',
+            createdAt: now,
+            updatedAt: now,
+        };
+        this.tasks.set(task.id, task);
+        logger.info('Task created (memory)', { taskId: task.id, topic: task.topic });
+        return task;
+    }
+    /**
+     * 根据 ID 查找任务
+     */
+    async findById(id) {
+        const task = this.tasks.get(id);
+        return task || null;
+    }
+    /**
+     * 更新任务
+     */
+    async update(id, params) {
+        const task = this.tasks.get(id);
+        if (!task) {
+            throw new Error(`Task not found: ${id}`);
+        }
+        const now = new Date();
+        const updatedTask = {
+            ...task,
+            ...(params.status !== undefined && { status: params.status }),
+            ...(params.currentStep !== undefined && { currentStep: params.currentStep }),
+            ...(params.errorMessage !== undefined && { errorMessage: params.errorMessage }),
+            ...(params.workerId !== undefined && { workerId: params.workerId }),
+            ...(params.startedAt !== undefined && { startedAt: params.startedAt ? new Date(params.startedAt) : undefined }),
+            ...(params.completedAt !== undefined && { completedAt: params.completedAt ? new Date(params.completedAt) : undefined }),
+            ...(params.version !== undefined && { version: params.version }),
+            updatedAt: now,
+        };
+        this.tasks.set(id, updatedTask);
+        logger.debug('Task updated (memory)', { taskId: id, updates: Object.keys(params) });
+        return updatedTask;
+    }
+    /**
+     * 更新任务状态（带乐观锁）
+     */
+    async updateStatus(taskId, status, version) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            return false;
+        }
+        // 检查版本号（乐观锁）
+        if (task.version !== version) {
+            logger.warn('Task version mismatch for updateStatus', { taskId, expectedVersion: version, actualVersion: task.version });
+            return false;
+        }
+        const now = new Date();
+        const updatedTask = {
+            ...task,
+            status,
+            version: version + 1,
+            updatedAt: now,
+            ...(status === TaskStatus.RUNNING && { startedAt: now }),
+            ...(status === TaskStatus.COMPLETED && { completedAt: now }),
+            ...(status === TaskStatus.FAILED && { completedAt: now }),
+            ...(status === TaskStatus.CANCELLED && { completedAt: now }),
+        };
+        this.tasks.set(taskId, updatedTask);
+        logger.debug('Task status updated (memory)', { taskId, status, newVersion: updatedTask.version });
+        return true;
+    }
+    /**
+     * 抢占任务（Worker 抢占机制）
+     */
+    async claimForProcessing(workerId, limit = 1) {
+        const now = new Date();
+        const claimedTasks = [];
+        // 找出待处理任务
+        const pendingTasks = Array.from(this.tasks.values())
+            .filter((task) => task.status === TaskStatus.PENDING ||
+            (task.status === TaskStatus.WAITING &&
+                new Date(task.updatedAt).getTime() < Date.now() - 5 * 60 * 1000))
+            .sort((a, b) => {
+            // 按优先级降序，然后按创建时间升序
+            if (a.priority !== b.priority) {
+                return b.priority - a.priority;
+            }
+            return new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime();
+        })
+            .slice(0, limit);
+        // 更新任务状态
+        for (const task of pendingTasks) {
+            const updatedTask = await this.update(task.id, {
+                status: TaskStatus.RUNNING,
+                workerId,
+                startedAt: now.toISOString(),
+            });
+            claimedTasks.push(updatedTask);
+        }
+        if (claimedTasks.length > 0) {
+            logger.info('Tasks claimed for processing (memory)', {
+                workerId,
+                count: claimedTasks.length,
+                taskIds: claimedTasks.map((t) => t.id),
+            });
+        }
+        return claimedTasks;
+    }
+    /**
+     * 保存状态快照（用于崩溃恢复）
+     */
+    async saveStateSnapshot(taskId, step, state) {
+        const key = `${taskId}_${step}`;
+        this.snapshots.set(key, {
+            taskId,
+            step,
+            state,
+            savedAt: new Date(),
+        });
+        logger.debug('State snapshot saved (memory)', { taskId, step });
+    }
+    /**
+     * 恢复状态快照
+     */
+    async loadStateSnapshot(taskId) {
+        // 找到最新的快照
+        const snapshots = Array.from(this.snapshots.values())
+            .filter((s) => s.taskId === taskId)
+            .sort((a, b) => new Date(b.savedAt).getTime() - new Date(a.savedAt).getTime());
+        if (snapshots.length === 0) {
+            return null;
+        }
+        return snapshots[0].state;
+    }
+    /**
+     * 列出任务（带过滤和分页）
+     */
+    async list(filters = {}) {
+        const { status, workerId, type, limit = 50, offset = 0, sortBy = 'createdAt', sortOrder = 'DESC', } = filters;
+        let tasks = Array.from(this.tasks.values());
+        // 应用过滤
+        if (status) {
+            tasks = tasks.filter((t) => t.status === status);
+        }
+        if (workerId) {
+            tasks = tasks.filter((t) => t.workerId === workerId);
+        }
+        if (type) {
+            tasks = tasks.filter((t) => t.type === type);
+        }
+        const total = tasks.length;
+        // 排序
+        tasks.sort((a, b) => {
+            const aVal = a[sortBy];
+            const bVal = b[sortBy];
+            const comparison = aVal < bVal ? -1 : aVal > bVal ? 1 : 0;
+            return sortOrder === 'DESC' ? -comparison : comparison;
+        });
+        // 分页
+        const data = tasks.slice(offset, offset + limit);
+        return {
+            data,
+            total,
+            limit,
+            offset,
+        };
+    }
+    /**
+     * 删除任务
+     */
+    async delete(id) {
+        const existed = this.tasks.has(id);
+        this.tasks.delete(id);
+        // 清理相关快照
+        for (const [key, snapshot] of this.snapshots.entries()) {
+            if (snapshot.taskId === id) {
+                this.snapshots.delete(key);
+            }
+        }
+        return existed;
+    }
+    /**
+     * 健康检查
+     */
+    async healthCheck() {
+        return true; // 内存版本永远健康
+    }
+    /**
+     * 更新当前步骤（带乐观锁）
+     */
+    async updateCurrentStep(taskId, step, version) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            return false;
+        }
+        // 检查版本号（乐观锁）
+        if (task.version !== version) {
+            logger.warn('Task version mismatch for updateCurrentStep', { taskId, expectedVersion: version, actualVersion: task.version });
+            return false;
+        }
+        const now = new Date();
+        const updatedTask = {
+            ...task,
+            currentStep: step,
+            version: version + 1,
+            updatedAt: now,
+        };
+        this.tasks.set(taskId, updatedTask);
+        logger.debug('Task currentStep updated (memory)', { taskId, step });
+        return true;
+    }
+    /**
+     * 标记任务完成（带乐观锁）
+     */
+    async markAsCompleted(taskId, version) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            return false;
+        }
+        // 检查版本号（乐观锁）
+        if (task.version !== version) {
+            logger.warn('Task version mismatch for markAsCompleted', { taskId, expectedVersion: version, actualVersion: task.version });
+            return false;
+        }
+        const now = new Date();
+        const updatedTask = {
+            ...task,
+            status: TaskStatus.COMPLETED,
+            completedAt: now,
+            version: version + 1,
+            updatedAt: now,
+        };
+        this.tasks.set(taskId, updatedTask);
+        logger.info('Task marked as completed (memory)', { taskId });
+        return true;
+    }
+    /**
+     * 标记任务失败（带乐观锁）
+     */
+    async markAsFailed(taskId, errorMessage, version) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            return false;
+        }
+        // 检查版本号（乐观锁）
+        if (task.version !== version) {
+            logger.warn('Task version mismatch for markAsFailed', { taskId, expectedVersion: version, actualVersion: task.version });
+            return false;
+        }
+        const now = new Date();
+        const updatedTask = {
+            ...task,
+            status: TaskStatus.FAILED,
+            errorMessage,
+            completedAt: now,
+            version: version + 1,
+            updatedAt: now,
+        };
+        this.tasks.set(taskId, updatedTask);
+        logger.info('Task marked as failed (memory)', { taskId, errorMessage });
+        return true;
+    }
+    /**
+     * 获取统计信息
+     */
+    getStats() {
+        return {
+            totalTasks: this.tasks.size,
+            snapshots: this.snapshots.size,
+            byStatus: Array.from(this.tasks.values()).reduce((acc, task) => {
+                acc[task.status] = (acc[task.status] || 0) + 1;
+                return acc;
+            }, {}),
+        };
+    }
+}
+//# sourceMappingURL=MemoryTaskRepository.js.map