llm-content-creator 0.1.0

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

@@ -0,0 +1,41 @@
+/**
+ * Result Repository
+ *
+ * 负责任务结果的数据库操作
+ */
+import { Pool } from 'pg';
+import type { IResultRepository } from '../../domain/repositories/ResultRepository.js';
+/**
+ * PostgreSQL Result Repository 实现
+ */
+export declare class PostgresResultRepository implements IResultRepository {
+    private pool;
+    constructor(pool: Pool);
+    /**
+     * 创建结果记录
+     */
+    create(params: {
+        taskId: string;
+        resultType: 'article' | 'image' | 'text';
+        content?: string;
+        filePath?: string;
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * 根据任务 ID 查询结果
+     */
+    findByTaskId(taskId: string): Promise<Array<{
+        id: number;
+        taskId: string;
+        resultType: string;
+        content: string | null;
+        filePath: string | null;
+        metadata: any;
+        createdAt: Date;
+    }>>;
+    /**
+     * 删除任务的所有结果
+     */
+    deleteByTaskId(taskId: string): Promise<void>;
+}
+//# sourceMappingURL=ResultRepository.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ResultRepository.d.ts","sourceRoot":"","sources":["../../../src/infrastructure/database/ResultRepository.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAE1B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+CAA+C,CAAC;AAIvF;;GAEG;AACH,qBAAa,wBAAyB,YAAW,iBAAiB;IAChE,OAAO,CAAC,IAAI,CAAO;gBAEP,IAAI,EAAE,IAAI;IAItB;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,SAAS,GAAG,OAAO,GAAG,MAAM,CAAC;QACzC,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBjB;;OAEG;IAEG,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAChD,EAAE,EAAE,MAAM,CAAC;QACX,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,QAAQ,EAAE,GAAG,CAAC;QACd,SAAS,EAAE,IAAI,CAAC;KACjB,CAAC,CAAC;IA6BH;;OAEG;IACG,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAgBpD"}

package/dist/infrastructure/database/ResultRepository.js

@@ -0,0 +1,86 @@
+/**
+ * Result Repository
+ *
+ * 负责任务结果的数据库操作
+ */
+import { Pool } from 'pg';
+import { createLogger } from '../logging/logger.js';
+const logger = createLogger('ResultRepository');
+/**
+ * PostgreSQL Result Repository 实现
+ */
+export class PostgresResultRepository {
+    pool;
+    constructor(pool) {
+        this.pool = pool;
+    }
+    /**
+     * 创建结果记录
+     */
+    async create(params) {
+        const client = await this.pool.connect();
+        try {
+            await client.query(`INSERT INTO results (task_id, result_type, content, file_path, metadata)
+         VALUES ($1, $2, $3, $4, $5)`, [params.taskId, params.resultType, params.content, params.filePath, JSON.stringify(params.metadata || {})]);
+            logger.info('Result created', {
+                taskId: params.taskId,
+                resultType: params.resultType,
+            });
+        }
+        catch (error) {
+            logger.error('Failed to create result', error);
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * 根据任务 ID 查询结果
+     */
+    // @ts-ignore - Database returns string for resultType, interface expects union type
+    async findByTaskId(taskId) {
+        const client = await this.pool.connect();
+        try {
+            const result = await client.query(`SELECT id, task_id, result_type, content, file_path, metadata, created_at
+         FROM results
+         WHERE task_id = $1
+         ORDER BY result_type`, [taskId]);
+            return result.rows.map((row) => ({
+                id: row.id,
+                taskId: row.task_id,
+                resultType: row.result_type,
+                content: row.content,
+                filePath: row.file_path,
+                // jsonb 类型已被 pg driver 自动解析，无需 JSON.parse
+                metadata: row.metadata || null,
+                createdAt: row.created_at,
+            }));
+        }
+        catch (error) {
+            logger.error('Failed to query results', error);
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /**
+     * 删除任务的所有结果
+     */
+    async deleteByTaskId(taskId) {
+        const client = await this.pool.connect();
+        try {
+            await client.query(`DELETE FROM results WHERE task_id = $1`, [taskId]);
+            logger.info('Results deleted', { taskId });
+        }
+        catch (error) {
+            logger.error('Failed to delete results', error);
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+}
+//# sourceMappingURL=ResultRepository.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ResultRepository.js","sourceRoot":"","sources":["../../../src/infrastructure/database/ResultRepository.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAGpD,MAAM,MAAM,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC;AAEhD;;GAEG;AACH,MAAM,OAAO,wBAAwB;IAC3B,IAAI,CAAO;IAEnB,YAAY,IAAU;QACpB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,MAMZ;QACC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAChB;qCAC6B,EAC7B,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAC3G,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAC5B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,oFAAoF;IACpF,KAAK,CAAC,YAAY,CAAC,MAAc;QAS/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;8BAGsB,EACtB,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,MAAM,EAAE,GAAG,CAAC,OAAO;gBACnB,UAAU,EAAE,GAAG,CAAC,WAAW;gBAC3B,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,QAAQ,EAAE,GAAG,CAAC,SAAS;gBACvB,0CAA0C;gBAC1C,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;gBAC9B,SAAS,EAAE,GAAG,CAAC,UAAU;aAC1B,CAAC,CAAQ,CAAC;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAChB,wCAAwC,EACxC,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAc,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,101 @@
+/**
+ * SQLite Task Repository
+ *
+ * 使用 better-sqlite3 实现的轻量级数据库访问层
+ * 适合开发和测试环境
+ */
+import type { Task, TaskCreateParams, TaskUpdateParams, TaskListFilters, PaginatedResult } from '../../domain/repositories/TaskRepository.js';
+import { TaskStatus } from '../../domain/entities/Task.js';
+/**
+ * SQLite Task Repository 实现
+ */
+export declare class SQLiteTaskRepository {
+    private db;
+    private dbPath;
+    constructor(dbPath?: string);
+    /**
+     * 初始化数据库表
+     */
+    private initializeTables;
+    /**
+     * 创建任务
+     */
+    create(params: TaskCreateParams): Promise<Task>;
+    /**
+     * 根据 ID 查找任务
+     */
+    findById(id: string): Promise<Task | null>;
+    /**
+     * 更新任务
+     */
+    update(id: string, params: TaskUpdateParams): Promise<Task>;
+    /**
+     * 抢占任务（Worker 抢占机制）
+     */
+    claimForProcessing(workerId: string, limit?: number): Promise<Task[]>;
+    /**
+     * 列出任务（带过滤和分页）
+     */
+    list(filters?: TaskListFilters): Promise<PaginatedResult<Task>>;
+    /**
+     * 更新任务状态（带乐观锁）
+     */
+    updateStatus(taskId: string, status: TaskStatus, version: number): Promise<boolean>;
+    /**
+     * 更新当前步骤（带乐观锁）
+     */
+    updateCurrentStep(taskId: string, step: string, version: number): Promise<boolean>;
+    /**
+     * 抢占任务（Worker 抢占机制）
+     */
+    claimTask(taskId: string, workerId: string, version: number): Promise<boolean>;
+    /**
+     * 增加重试计数（带乐观锁）
+     */
+    incrementRetryCount(taskId: string, type: 'text' | 'image', version: number): Promise<boolean>;
+    /**
+     * 保存状态快照（带乐观锁）
+     */
+    saveStateSnapshot(taskId: string, snapshot: object, _version: number): Promise<boolean>;
+    /**
+     * 标记任务完成（带乐观锁）
+     */
+    markAsCompleted(taskId: string, version: number): Promise<boolean>;
+    /**
+     * 标记任务失败（带乐观锁）
+     */
+    markAsFailed(taskId: string, errorMessage: string, version: number): Promise<boolean>;
+    /**
+     * 释放 Worker 占用（用于任务崩溃时）
+     */
+    releaseWorker(taskId: string, workerId: string, version: number): Promise<boolean>;
+    /**
+     * 软删除任务
+     */
+    softDelete(taskId: string): Promise<boolean>;
+    /**
+     * 永久删除任务
+     */
+    delete(taskId: string): Promise<boolean>;
+    /**
+     * 获取待处理任务队列（用于 Worker 获取任务）
+     */
+    getPendingTasks(limit?: number): Promise<Task[]>;
+    /**
+     * 获取 Worker 的活跃任务
+     */
+    getActiveTasksByWorker(workerId: string): Promise<Task[]>;
+    /**
+     * 健康检查
+     */
+    healthCheck(): Promise<boolean>;
+    /**
+     * 关闭数据库连接
+     */
+    close(): void;
+    /**
+     * 将数据库行映射为 Task 对象
+     */
+    private mapRowToTask;
+}
+//# sourceMappingURL=SQLiteTaskRepository.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"SQLiteTaskRepository.d.ts","sourceRoot":"","sources":["../../../src/infrastructure/database/SQLiteTaskRepository.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,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,EAAE,CAAoB;IAC9B,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,CAAC,EAAE,MAAM;IAa3B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAwExB;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IA2CrD;;OAEG;IACG,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAWhD;;OAEG;IACG,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IA6DjE;;OAEG;IACG,kBAAkB,CACtB,QAAQ,EAAE,MAAM,EAChB,KAAK,GAAE,MAAU,GAChB,OAAO,CAAC,IAAI,EAAE,CAAC;IA+ClB;;OAEG;IACG,IAAI,CAAC,OAAO,GAAE,eAAoB,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAyDzE;;OAEG;IACG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAsBzF;;OAEG;IACG,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAsBxF;;OAEG;IACG,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAwBpF;;OAEG;IACG,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAsBpG;;OAEG;IACG,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAiB7F;;OAEG;IACG,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAuBxE;;OAEG;IACG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAwB3F;;OAEG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAuBxF;;OAEG;IACG,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKlD;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAO9C;;OAEG;IACG,eAAe,CAAC,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAY1D;;OAEG;IACG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAU/D;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAWrC;;OAEG;IACH,KAAK,IAAI,IAAI;IAKb;;OAEG;IACH,OAAO,CAAC,YAAY;CA0BrB"}