workerclaw 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +129 -0
- package/dist/active-behavior/behavior-scheduler.d.ts +119 -0
- package/dist/active-behavior/behavior-scheduler.d.ts.map +1 -0
- package/dist/active-behavior/behavior-scheduler.js +297 -0
- package/dist/active-behavior/behavior-scheduler.js.map +1 -0
- package/dist/active-behavior/frequency-control.d.ts +72 -0
- package/dist/active-behavior/frequency-control.d.ts.map +1 -0
- package/dist/active-behavior/frequency-control.js +206 -0
- package/dist/active-behavior/frequency-control.js.map +1 -0
- package/dist/active-behavior/index.d.ts +8 -0
- package/dist/active-behavior/index.d.ts.map +1 -0
- package/dist/active-behavior/index.js +6 -0
- package/dist/active-behavior/index.js.map +1 -0
- package/dist/agent/agent-engine.d.ts +122 -0
- package/dist/agent/agent-engine.d.ts.map +1 -0
- package/dist/agent/agent-engine.js +380 -0
- package/dist/agent/agent-engine.js.map +1 -0
- package/dist/agent/context-window.d.ts +69 -0
- package/dist/agent/context-window.d.ts.map +1 -0
- package/dist/agent/context-window.js +210 -0
- package/dist/agent/context-window.js.map +1 -0
- package/dist/agent/llm-client.d.ts +37 -0
- package/dist/agent/llm-client.d.ts.map +1 -0
- package/dist/agent/llm-client.js +155 -0
- package/dist/agent/llm-client.js.map +1 -0
- package/dist/agent/personality.d.ts +71 -0
- package/dist/agent/personality.d.ts.map +1 -0
- package/dist/agent/personality.js +147 -0
- package/dist/agent/personality.js.map +1 -0
- package/dist/agent/session-manager.d.ts +111 -0
- package/dist/agent/session-manager.d.ts.map +1 -0
- package/dist/agent/session-manager.js +205 -0
- package/dist/agent/session-manager.js.map +1 -0
- package/dist/agent/tool-executor.d.ts +34 -0
- package/dist/agent/tool-executor.d.ts.map +1 -0
- package/dist/agent/tool-executor.js +145 -0
- package/dist/agent/tool-executor.js.map +1 -0
- package/dist/agent/tool-registry.d.ts +67 -0
- package/dist/agent/tool-registry.d.ts.map +1 -0
- package/dist/agent/tool-registry.js +272 -0
- package/dist/agent/tool-registry.js.map +1 -0
- package/dist/cli/configure.d.ts +16 -0
- package/dist/cli/configure.d.ts.map +1 -0
- package/dist/cli/configure.js +212 -0
- package/dist/cli/configure.js.map +1 -0
- package/dist/cli/index.d.ts +17 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +256 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/prompter.d.ts +59 -0
- package/dist/cli/prompter.d.ts.map +1 -0
- package/dist/cli/prompter.js +120 -0
- package/dist/cli/prompter.js.map +1 -0
- package/dist/cli/sections/index.d.ts +9 -0
- package/dist/cli/sections/index.d.ts.map +1 -0
- package/dist/cli/sections/index.js +9 -0
- package/dist/cli/sections/index.js.map +1 -0
- package/dist/cli/sections/llm.d.ts +14 -0
- package/dist/cli/sections/llm.d.ts.map +1 -0
- package/dist/cli/sections/llm.js +155 -0
- package/dist/cli/sections/llm.js.map +1 -0
- package/dist/cli/sections/personality.d.ts +14 -0
- package/dist/cli/sections/personality.d.ts.map +1 -0
- package/dist/cli/sections/personality.js +90 -0
- package/dist/cli/sections/personality.js.map +1 -0
- package/dist/cli/sections/platform.d.ts +14 -0
- package/dist/cli/sections/platform.d.ts.map +1 -0
- package/dist/cli/sections/platform.js +179 -0
- package/dist/cli/sections/platform.js.map +1 -0
- package/dist/cli/sections/security.d.ts +14 -0
- package/dist/cli/sections/security.d.ts.map +1 -0
- package/dist/cli/sections/security.js +106 -0
- package/dist/cli/sections/security.js.map +1 -0
- package/dist/cli/sections/skills.d.ts +10 -0
- package/dist/cli/sections/skills.d.ts.map +1 -0
- package/dist/cli/sections/skills.js +149 -0
- package/dist/cli/sections/skills.js.map +1 -0
- package/dist/cli.d.ts +8 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +116 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/config.d.ts +175 -0
- package/dist/core/config.d.ts.map +1 -0
- package/dist/core/config.js +109 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/events.d.ts +211 -0
- package/dist/core/events.d.ts.map +1 -0
- package/dist/core/events.js +152 -0
- package/dist/core/events.js.map +1 -0
- package/dist/core/logger.d.ts +23 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +53 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/workerclaw.d.ts +83 -0
- package/dist/core/workerclaw.d.ts.map +1 -0
- package/dist/core/workerclaw.js +161 -0
- package/dist/core/workerclaw.js.map +1 -0
- package/dist/index.d.ts +27 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/ingress/message-parser.d.ts +36 -0
- package/dist/ingress/message-parser.d.ts.map +1 -0
- package/dist/ingress/message-parser.js +107 -0
- package/dist/ingress/message-parser.js.map +1 -0
- package/dist/ingress/miniabc-client.d.ts +109 -0
- package/dist/ingress/miniabc-client.d.ts.map +1 -0
- package/dist/ingress/miniabc-client.js +345 -0
- package/dist/ingress/miniabc-client.js.map +1 -0
- package/dist/ingress/platform-api.d.ts +82 -0
- package/dist/ingress/platform-api.d.ts.map +1 -0
- package/dist/ingress/platform-api.js +209 -0
- package/dist/ingress/platform-api.js.map +1 -0
- package/dist/sandbox/command-sandbox.d.ts +46 -0
- package/dist/sandbox/command-sandbox.d.ts.map +1 -0
- package/dist/sandbox/command-sandbox.js +144 -0
- package/dist/sandbox/command-sandbox.js.map +1 -0
- package/dist/sandbox/fs-sandbox.d.ts +56 -0
- package/dist/sandbox/fs-sandbox.d.ts.map +1 -0
- package/dist/sandbox/fs-sandbox.js +119 -0
- package/dist/sandbox/fs-sandbox.js.map +1 -0
- package/dist/sandbox/network-sandbox.d.ts +41 -0
- package/dist/sandbox/network-sandbox.d.ts.map +1 -0
- package/dist/sandbox/network-sandbox.js +114 -0
- package/dist/sandbox/network-sandbox.js.map +1 -0
- package/dist/security/content-scanner.d.ts +59 -0
- package/dist/security/content-scanner.d.ts.map +1 -0
- package/dist/security/content-scanner.js +289 -0
- package/dist/security/content-scanner.js.map +1 -0
- package/dist/security/gate.d.ts +78 -0
- package/dist/security/gate.d.ts.map +1 -0
- package/dist/security/gate.js +150 -0
- package/dist/security/gate.js.map +1 -0
- package/dist/security/permission-level.d.ts +68 -0
- package/dist/security/permission-level.d.ts.map +1 -0
- package/dist/security/permission-level.js +191 -0
- package/dist/security/permission-level.js.map +1 -0
- package/dist/security/rate-limiter.d.ts +52 -0
- package/dist/security/rate-limiter.d.ts.map +1 -0
- package/dist/security/rate-limiter.js +133 -0
- package/dist/security/rate-limiter.js.map +1 -0
- package/dist/security/source-verifier.d.ts +33 -0
- package/dist/security/source-verifier.d.ts.map +1 -0
- package/dist/security/source-verifier.js +106 -0
- package/dist/security/source-verifier.js.map +1 -0
- package/dist/skills/builtin/code.d.ts +27 -0
- package/dist/skills/builtin/code.d.ts.map +1 -0
- package/dist/skills/builtin/code.js +132 -0
- package/dist/skills/builtin/code.js.map +1 -0
- package/dist/skills/builtin/index.d.ts +12 -0
- package/dist/skills/builtin/index.d.ts.map +1 -0
- package/dist/skills/builtin/index.js +16 -0
- package/dist/skills/builtin/index.js.map +1 -0
- package/dist/skills/builtin/search.d.ts +27 -0
- package/dist/skills/builtin/search.d.ts.map +1 -0
- package/dist/skills/builtin/search.js +105 -0
- package/dist/skills/builtin/search.js.map +1 -0
- package/dist/skills/builtin/writing.d.ts +22 -0
- package/dist/skills/builtin/writing.d.ts.map +1 -0
- package/dist/skills/builtin/writing.js +93 -0
- package/dist/skills/builtin/writing.js.map +1 -0
- package/dist/skills/index.d.ts +11 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +10 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/pack-loader.d.ts +31 -0
- package/dist/skills/pack-loader.d.ts.map +1 -0
- package/dist/skills/pack-loader.js +167 -0
- package/dist/skills/pack-loader.js.map +1 -0
- package/dist/skills/pack-registry.d.ts +40 -0
- package/dist/skills/pack-registry.d.ts.map +1 -0
- package/dist/skills/pack-registry.js +80 -0
- package/dist/skills/pack-registry.js.map +1 -0
- package/dist/skills/pack-types.d.ts +64 -0
- package/dist/skills/pack-types.d.ts.map +1 -0
- package/dist/skills/pack-types.js +7 -0
- package/dist/skills/pack-types.js.map +1 -0
- package/dist/skills/skill-registry.d.ts +83 -0
- package/dist/skills/skill-registry.d.ts.map +1 -0
- package/dist/skills/skill-registry.js +206 -0
- package/dist/skills/skill-registry.js.map +1 -0
- package/dist/skills/skill-runner.d.ts +49 -0
- package/dist/skills/skill-runner.d.ts.map +1 -0
- package/dist/skills/skill-runner.js +128 -0
- package/dist/skills/skill-runner.js.map +1 -0
- package/dist/skills/types.d.ts +94 -0
- package/dist/skills/types.d.ts.map +1 -0
- package/dist/skills/types.js +7 -0
- package/dist/skills/types.js.map +1 -0
- package/dist/task/concurrency.d.ts +87 -0
- package/dist/task/concurrency.d.ts.map +1 -0
- package/dist/task/concurrency.js +269 -0
- package/dist/task/concurrency.js.map +1 -0
- package/dist/task/task-evaluator.d.ts +52 -0
- package/dist/task/task-evaluator.d.ts.map +1 -0
- package/dist/task/task-evaluator.js +141 -0
- package/dist/task/task-evaluator.js.map +1 -0
- package/dist/task/task-manager.d.ts +112 -0
- package/dist/task/task-manager.d.ts.map +1 -0
- package/dist/task/task-manager.js +382 -0
- package/dist/task/task-manager.js.map +1 -0
- package/dist/task/task-state-machine.d.ts +63 -0
- package/dist/task/task-state-machine.d.ts.map +1 -0
- package/dist/task/task-state-machine.js +149 -0
- package/dist/task/task-state-machine.js.map +1 -0
- package/dist/types/agent.d.ts +79 -0
- package/dist/types/agent.d.ts.map +1 -0
- package/dist/types/agent.js +7 -0
- package/dist/types/agent.js.map +1 -0
- package/dist/types/index.d.ts +7 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +5 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/message.d.ts +87 -0
- package/dist/types/message.d.ts.map +1 -0
- package/dist/types/message.js +30 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/task.d.ts +176 -0
- package/dist/types/task.d.ts.map +1 -0
- package/dist/types/task.js +5 -0
- package/dist/types/task.js.map +1 -0
- package/package.json +57 -0
- package/workerclaw.config.example.json +78 -0
|
@@ -0,0 +1,382 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 任务管理器 (Phase 3 完整版)
|
|
3
|
+
*
|
|
4
|
+
* 负责任务的完整生命周期:
|
|
5
|
+
* 收到任务 → 安全检查 → 评估 → 状态机 → 并发控制 → 执行 → 结果上报
|
|
6
|
+
*/
|
|
7
|
+
import { createLogger } from '../core/logger.js';
|
|
8
|
+
import { WorkerClawEvent } from '../core/events.js';
|
|
9
|
+
import { AgentEngine } from '../agent/agent-engine.js';
|
|
10
|
+
import { messageParser } from '../ingress/message-parser.js';
|
|
11
|
+
import { PlatformApiClient } from '../ingress/platform-api.js';
|
|
12
|
+
import { createBuiltinToolRegistry } from '../agent/tool-registry.js';
|
|
13
|
+
import { ToolExecutor } from '../agent/tool-executor.js';
|
|
14
|
+
import { TaskStateMachine } from './task-state-machine.js';
|
|
15
|
+
import { TaskEvaluator } from './task-evaluator.js';
|
|
16
|
+
import { ConcurrencyController } from './concurrency.js';
|
|
17
|
+
import { WSMessageType } from '../types/message.js';
|
|
18
|
+
export class TaskManager {
|
|
19
|
+
logger;
|
|
20
|
+
config;
|
|
21
|
+
eventBus;
|
|
22
|
+
securityGate;
|
|
23
|
+
agentEngine;
|
|
24
|
+
// Phase 3 新增模块
|
|
25
|
+
stateMachine;
|
|
26
|
+
evaluator;
|
|
27
|
+
concurrency;
|
|
28
|
+
platformApi;
|
|
29
|
+
toolExecutor;
|
|
30
|
+
// 超时管理
|
|
31
|
+
taskTimeouts = new Map();
|
|
32
|
+
constructor(config, eventBus, securityGate) {
|
|
33
|
+
this.config = config;
|
|
34
|
+
this.eventBus = eventBus;
|
|
35
|
+
this.securityGate = securityGate;
|
|
36
|
+
// Agent 引擎
|
|
37
|
+
this.agentEngine = new AgentEngine({ llm: config.llm, personality: config.personality, security: config.security }, eventBus);
|
|
38
|
+
// Phase 3: 状态机
|
|
39
|
+
this.stateMachine = new TaskStateMachine(eventBus);
|
|
40
|
+
// Phase 3: 评估器
|
|
41
|
+
this.evaluator = new TaskEvaluator(config.task.evaluation);
|
|
42
|
+
// Phase 3: 并发控制器
|
|
43
|
+
this.concurrency = new ConcurrencyController(config.task.concurrency, eventBus);
|
|
44
|
+
this.concurrency.setQueueTimeoutHandler((task) => {
|
|
45
|
+
this.logger.warn(`队列任务超时 [${task.taskId}]`);
|
|
46
|
+
this.stateMachine.tryTransition(task.taskId, 'rejected', '队列等待超时');
|
|
47
|
+
this.eventBus.emit(WorkerClawEvent.TASK_REJECTED, {
|
|
48
|
+
taskId: task.taskId,
|
|
49
|
+
reason: '队列等待超时',
|
|
50
|
+
});
|
|
51
|
+
});
|
|
52
|
+
this.concurrency.setProcessQueueHandler(() => {
|
|
53
|
+
this.processNextFromQueue();
|
|
54
|
+
});
|
|
55
|
+
// Phase 3: 平台 API
|
|
56
|
+
this.platformApi = new PlatformApiClient({ platform: config.platform }, eventBus);
|
|
57
|
+
// Phase 3: 工具系统
|
|
58
|
+
const registry = createBuiltinToolRegistry();
|
|
59
|
+
this.toolExecutor = new ToolExecutor(registry, { security: config.security }, eventBus);
|
|
60
|
+
this.logger = createLogger('TaskManager');
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* 处理平台消息
|
|
64
|
+
*/
|
|
65
|
+
async handleMessage(message) {
|
|
66
|
+
// 1. 安全检查
|
|
67
|
+
const securityResult = await this.securityGate.check(message);
|
|
68
|
+
if (!securityResult.passed) {
|
|
69
|
+
this.logger.debug('消息未通过安全检查', { reason: securityResult.reason });
|
|
70
|
+
return;
|
|
71
|
+
}
|
|
72
|
+
// 2. 解析消息
|
|
73
|
+
const parsed = messageParser.parse(message);
|
|
74
|
+
if (parsed.category !== 'task') {
|
|
75
|
+
if (parsed.category === 'interaction') {
|
|
76
|
+
this.logger.debug('收到交互消息(暂不处理)', { type: message.type });
|
|
77
|
+
}
|
|
78
|
+
return;
|
|
79
|
+
}
|
|
80
|
+
if (!parsed.task) {
|
|
81
|
+
this.logger.warn('任务消息解析失败,无有效任务数据');
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
// 3. 分发处理
|
|
85
|
+
switch (message.type) {
|
|
86
|
+
case WSMessageType.TASK_PUSH:
|
|
87
|
+
await this.handleTaskPush(parsed.task);
|
|
88
|
+
break;
|
|
89
|
+
case WSMessageType.TASK_CANCEL:
|
|
90
|
+
await this.handleTaskCancel(parsed.task);
|
|
91
|
+
break;
|
|
92
|
+
case WSMessageType.TASK_UPDATE:
|
|
93
|
+
this.logger.debug('收到任务更新(暂不处理)', { taskId: parsed.task.taskId });
|
|
94
|
+
break;
|
|
95
|
+
default:
|
|
96
|
+
this.logger.debug('未处理的任务消息类型', { type: message.type });
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
/**
|
|
100
|
+
* 处理新任务推送
|
|
101
|
+
*/
|
|
102
|
+
async handleTaskPush(task) {
|
|
103
|
+
this.logger.info(`收到新任务 [${task.taskId}]`, {
|
|
104
|
+
type: task.taskType,
|
|
105
|
+
title: task.title,
|
|
106
|
+
reward: task.reward,
|
|
107
|
+
});
|
|
108
|
+
// 初始化状态
|
|
109
|
+
this.stateMachine.init(task);
|
|
110
|
+
this.eventBus.emit(WorkerClawEvent.TASK_RECEIVED, { task });
|
|
111
|
+
// Phase 3: 评估任务
|
|
112
|
+
if (!this.config.task.autoAccept.enabled) {
|
|
113
|
+
this.logger.info('自动接单已关闭,跳过任务');
|
|
114
|
+
this.stateMachine.transition(task.taskId, 'rejected', '自动接单已关闭');
|
|
115
|
+
this.eventBus.emit(WorkerClawEvent.TASK_REJECTED, {
|
|
116
|
+
taskId: task.taskId,
|
|
117
|
+
reason: '自动接单已关闭',
|
|
118
|
+
});
|
|
119
|
+
return;
|
|
120
|
+
}
|
|
121
|
+
// 进入评估状态
|
|
122
|
+
this.stateMachine.transition(task.taskId, 'evaluating');
|
|
123
|
+
const evaluation = this.evaluateTask(task);
|
|
124
|
+
this.eventBus.emit(WorkerClawEvent.TASK_EVALUATED, { taskId: task.taskId, evaluation });
|
|
125
|
+
// 根据评估结果决策
|
|
126
|
+
switch (evaluation.decision) {
|
|
127
|
+
case 'accept':
|
|
128
|
+
await this.acceptTask(task, evaluation);
|
|
129
|
+
break;
|
|
130
|
+
case 'defer':
|
|
131
|
+
await this.deferTask(task, evaluation);
|
|
132
|
+
break;
|
|
133
|
+
case 'reject':
|
|
134
|
+
this.stateMachine.transition(task.taskId, 'rejected', evaluation.reason);
|
|
135
|
+
this.eventBus.emit(WorkerClawEvent.TASK_REJECTED, {
|
|
136
|
+
taskId: task.taskId,
|
|
137
|
+
reason: evaluation.reason || '评估未通过',
|
|
138
|
+
});
|
|
139
|
+
break;
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
/**
|
|
143
|
+
* 评估任务
|
|
144
|
+
*/
|
|
145
|
+
evaluateTask(task) {
|
|
146
|
+
const concurrencyStats = this.concurrency.getStats();
|
|
147
|
+
const context = {
|
|
148
|
+
runningCount: concurrencyStats.running,
|
|
149
|
+
maxConcurrent: this.config.task.concurrency.maxConcurrent,
|
|
150
|
+
skills: this.toolExecutor.getRegistry().getToolNames(),
|
|
151
|
+
completedCountByType: {}, // Phase 4 追踪历史
|
|
152
|
+
threshold: this.config.task.evaluation.acceptThreshold,
|
|
153
|
+
};
|
|
154
|
+
return this.evaluator.evaluate(task, context);
|
|
155
|
+
}
|
|
156
|
+
/**
|
|
157
|
+
* 接受任务
|
|
158
|
+
*/
|
|
159
|
+
async acceptTask(task, evaluation) {
|
|
160
|
+
const result = this.concurrency.tryStart(task);
|
|
161
|
+
switch (result) {
|
|
162
|
+
case 'started': {
|
|
163
|
+
this.stateMachine.transition(task.taskId, 'accepted', evaluation.reason);
|
|
164
|
+
this.stateMachine.transition(task.taskId, 'running');
|
|
165
|
+
const permLevel = this.determinePermissionLevel(task);
|
|
166
|
+
this.stateMachine.setPermissionLevel(task.taskId, permLevel);
|
|
167
|
+
this.eventBus.emit(WorkerClawEvent.TASK_ACCEPTED, { taskId: task.taskId });
|
|
168
|
+
// 上报接单状态到平台
|
|
169
|
+
this.platformApi.updateStatus(task.taskId, 'accepted').catch(() => { });
|
|
170
|
+
// 设置任务超时
|
|
171
|
+
this.setTaskTimeout(task);
|
|
172
|
+
// 执行任务
|
|
173
|
+
this.executeTask(task, permLevel).catch(err => {
|
|
174
|
+
this.logger.error(`任务执行异常 [${task.taskId}]`, err.message);
|
|
175
|
+
});
|
|
176
|
+
break;
|
|
177
|
+
}
|
|
178
|
+
case 'queued':
|
|
179
|
+
this.stateMachine.transition(task.taskId, 'accepted', '进入等待队列');
|
|
180
|
+
// 设置队列等待超时
|
|
181
|
+
this.concurrency.setQueueTimeout(task.taskId, this.config.task.timeout.queueTimeoutMs);
|
|
182
|
+
break;
|
|
183
|
+
case 'rejected':
|
|
184
|
+
this.stateMachine.transition(task.taskId, 'rejected', '并发和队列均已满');
|
|
185
|
+
this.eventBus.emit(WorkerClawEvent.TASK_REJECTED, {
|
|
186
|
+
taskId: task.taskId,
|
|
187
|
+
reason: '并发和队列均已满',
|
|
188
|
+
});
|
|
189
|
+
break;
|
|
190
|
+
}
|
|
191
|
+
}
|
|
192
|
+
/**
|
|
193
|
+
* 延迟任务(放入队列等待)
|
|
194
|
+
*/
|
|
195
|
+
async deferTask(task, evaluation) {
|
|
196
|
+
this.logger.info(`任务延迟处理 [${task.taskId}]`, { score: evaluation.score });
|
|
197
|
+
this.eventBus.emit(WorkerClawEvent.TASK_DEFERRED, {
|
|
198
|
+
taskId: task.taskId,
|
|
199
|
+
score: evaluation.score,
|
|
200
|
+
});
|
|
201
|
+
// 尝试放入队列
|
|
202
|
+
const result = this.concurrency.tryStart(task);
|
|
203
|
+
if (result === 'queued') {
|
|
204
|
+
this.stateMachine.transition(task.taskId, 'accepted', '延迟接单,入队等待');
|
|
205
|
+
this.concurrency.setQueueTimeout(task.taskId, this.config.task.timeout.queueTimeoutMs);
|
|
206
|
+
}
|
|
207
|
+
else if (result === 'started') {
|
|
208
|
+
// 有空位直接执行
|
|
209
|
+
this.stateMachine.transition(task.taskId, 'accepted');
|
|
210
|
+
this.stateMachine.transition(task.taskId, 'running');
|
|
211
|
+
const permLevel = this.determinePermissionLevel(task);
|
|
212
|
+
this.stateMachine.setPermissionLevel(task.taskId, permLevel);
|
|
213
|
+
this.eventBus.emit(WorkerClawEvent.TASK_ACCEPTED, { taskId: task.taskId });
|
|
214
|
+
this.platformApi.updateStatus(task.taskId, 'accepted').catch(() => { });
|
|
215
|
+
this.setTaskTimeout(task);
|
|
216
|
+
this.executeTask(task, permLevel).catch(err => {
|
|
217
|
+
this.logger.error(`任务执行异常 [${task.taskId}]`, err.message);
|
|
218
|
+
});
|
|
219
|
+
}
|
|
220
|
+
else {
|
|
221
|
+
this.stateMachine.transition(task.taskId, 'rejected', '队列已满');
|
|
222
|
+
this.eventBus.emit(WorkerClawEvent.TASK_REJECTED, {
|
|
223
|
+
taskId: task.taskId,
|
|
224
|
+
reason: '队列已满',
|
|
225
|
+
});
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
/**
|
|
229
|
+
* 执行任务
|
|
230
|
+
*/
|
|
231
|
+
async executeTask(task, permLevel) {
|
|
232
|
+
try {
|
|
233
|
+
const context = {
|
|
234
|
+
task,
|
|
235
|
+
permissionLevel: permLevel,
|
|
236
|
+
maxOutputTokens: this.config.llm.safety.maxTokens,
|
|
237
|
+
timeoutMs: this.config.task.timeout.taskTimeoutMs,
|
|
238
|
+
receivedAt: Date.now(),
|
|
239
|
+
};
|
|
240
|
+
const result = await this.agentEngine.executeTask(task, context);
|
|
241
|
+
// 上报结果到平台
|
|
242
|
+
await this.reportResult(task, result);
|
|
243
|
+
}
|
|
244
|
+
finally {
|
|
245
|
+
// 清理超时
|
|
246
|
+
this.clearTaskTimeout(task.taskId);
|
|
247
|
+
// 释放并发槽位
|
|
248
|
+
this.concurrency.taskFinished(task.taskId);
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
/**
|
|
252
|
+
* 从队列取出下一个任务执行
|
|
253
|
+
*/
|
|
254
|
+
processNextFromQueue() {
|
|
255
|
+
if (!this.concurrency.hasCapacity())
|
|
256
|
+
return;
|
|
257
|
+
const stats = this.concurrency.getStats();
|
|
258
|
+
if (stats.queue === 0)
|
|
259
|
+
return;
|
|
260
|
+
// ConcurrencyController 的 processQueue 已经取出任务
|
|
261
|
+
// 这里由 setProcessQueueHandler 触发
|
|
262
|
+
// 注意:processQueue 返回 true 表示成功取出
|
|
263
|
+
// 我们需要获取被取出的任务
|
|
264
|
+
this.logger.debug('有空位,尝试处理队列', {
|
|
265
|
+
running: stats.running,
|
|
266
|
+
queue: stats.queue,
|
|
267
|
+
});
|
|
268
|
+
}
|
|
269
|
+
/**
|
|
270
|
+
* 上报任务结果到平台
|
|
271
|
+
*/
|
|
272
|
+
async reportResult(task, result) {
|
|
273
|
+
const taskId = task.taskId;
|
|
274
|
+
this.logger.info(`上报任务结果 [${taskId}]`, {
|
|
275
|
+
status: result.status,
|
|
276
|
+
durationMs: result.durationMs,
|
|
277
|
+
});
|
|
278
|
+
// 更新状态机
|
|
279
|
+
if (result.status === 'completed') {
|
|
280
|
+
this.stateMachine.transition(taskId, 'completed');
|
|
281
|
+
}
|
|
282
|
+
else {
|
|
283
|
+
this.stateMachine.transition(taskId, 'failed', result.error);
|
|
284
|
+
}
|
|
285
|
+
// 上报到平台 API
|
|
286
|
+
const reported = await this.platformApi.reportResult(taskId, result);
|
|
287
|
+
if (result.status === 'completed') {
|
|
288
|
+
this.logger.info(`✅ 任务完成 [${taskId}]`, {
|
|
289
|
+
contentPreview: result.content?.slice(0, 200),
|
|
290
|
+
reported,
|
|
291
|
+
});
|
|
292
|
+
}
|
|
293
|
+
else {
|
|
294
|
+
this.logger.error(`❌ 任务失败 [${taskId}]`, {
|
|
295
|
+
error: result.error,
|
|
296
|
+
reported,
|
|
297
|
+
});
|
|
298
|
+
}
|
|
299
|
+
// 清理状态记录
|
|
300
|
+
this.stateMachine.cleanup(taskId);
|
|
301
|
+
}
|
|
302
|
+
/**
|
|
303
|
+
* 设置任务执行超时
|
|
304
|
+
*/
|
|
305
|
+
setTaskTimeout(task) {
|
|
306
|
+
const timeoutMs = this.config.task.timeout.taskTimeoutMs;
|
|
307
|
+
const timer = setTimeout(() => {
|
|
308
|
+
this.logger.warn(`任务执行超时 [${task.taskId}]`, { timeoutMs });
|
|
309
|
+
this.stateMachine.tryTransition(task.taskId, 'timeout', '执行超时');
|
|
310
|
+
this.eventBus.emit(WorkerClawEvent.TASK_TIMEOUT, { taskId: task.taskId });
|
|
311
|
+
// 释放资源
|
|
312
|
+
this.concurrency.taskFinished(task.taskId);
|
|
313
|
+
this.taskTimeouts.delete(task.taskId);
|
|
314
|
+
// 上报超时到平台
|
|
315
|
+
this.platformApi.updateStatus(task.taskId, 'timeout', `执行超时 ${timeoutMs}ms`).catch(() => { });
|
|
316
|
+
}, timeoutMs);
|
|
317
|
+
this.taskTimeouts.set(task.taskId, timer);
|
|
318
|
+
}
|
|
319
|
+
/**
|
|
320
|
+
* 清除任务超时
|
|
321
|
+
*/
|
|
322
|
+
clearTaskTimeout(taskId) {
|
|
323
|
+
const timer = this.taskTimeouts.get(taskId);
|
|
324
|
+
if (timer) {
|
|
325
|
+
clearTimeout(timer);
|
|
326
|
+
this.taskTimeouts.delete(taskId);
|
|
327
|
+
}
|
|
328
|
+
}
|
|
329
|
+
/**
|
|
330
|
+
* 确定任务权限级别
|
|
331
|
+
*/
|
|
332
|
+
determinePermissionLevel(task) {
|
|
333
|
+
return this.securityGate.gradePermission(task);
|
|
334
|
+
}
|
|
335
|
+
/**
|
|
336
|
+
* 处理任务取消
|
|
337
|
+
*/
|
|
338
|
+
async handleTaskCancel(task) {
|
|
339
|
+
const state = this.stateMachine.getState(task.taskId);
|
|
340
|
+
if (state && !this.stateMachine.isTerminal(task.taskId)) {
|
|
341
|
+
this.logger.info(`任务被取消 [${task.taskId}]`);
|
|
342
|
+
this.stateMachine.transition(task.taskId, 'cancelled', '发单人取消');
|
|
343
|
+
// 从并发/队列中移除
|
|
344
|
+
this.concurrency.cancelTask(task.taskId);
|
|
345
|
+
this.clearTaskTimeout(task.taskId);
|
|
346
|
+
this.eventBus.emit(WorkerClawEvent.TASK_CANCELLED, { taskId: task.taskId });
|
|
347
|
+
this.platformApi.updateStatus(task.taskId, 'cancelled').catch(() => { });
|
|
348
|
+
this.stateMachine.cleanup(task.taskId);
|
|
349
|
+
}
|
|
350
|
+
}
|
|
351
|
+
/**
|
|
352
|
+
* 获取运行状态
|
|
353
|
+
*/
|
|
354
|
+
getStatus() {
|
|
355
|
+
const concurrencyStats = this.concurrency.getStats();
|
|
356
|
+
return {
|
|
357
|
+
runningTasks: concurrencyStats.running,
|
|
358
|
+
queuedTasks: concurrencyStats.queue,
|
|
359
|
+
taskIds: [], // Phase 4 可以从 stateMachine 获取
|
|
360
|
+
stateStats: this.stateMachine.getStats(),
|
|
361
|
+
concurrencyStats,
|
|
362
|
+
};
|
|
363
|
+
}
|
|
364
|
+
/**
|
|
365
|
+
* 获取 AgentEngine 实例(Phase 4)
|
|
366
|
+
*/
|
|
367
|
+
getAgentEngine() {
|
|
368
|
+
return this.agentEngine;
|
|
369
|
+
}
|
|
370
|
+
/**
|
|
371
|
+
* 清理资源
|
|
372
|
+
*/
|
|
373
|
+
dispose() {
|
|
374
|
+
for (const timer of this.taskTimeouts.values()) {
|
|
375
|
+
clearTimeout(timer);
|
|
376
|
+
}
|
|
377
|
+
this.taskTimeouts.clear();
|
|
378
|
+
this.concurrency.dispose();
|
|
379
|
+
this.agentEngine.dispose();
|
|
380
|
+
}
|
|
381
|
+
}
|
|
382
|
+
//# sourceMappingURL=task-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-manager.js","sourceRoot":"","sources":["../../src/task/task-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAY,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAE9D,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAMzD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAuBpD,MAAM,OAAO,WAAW;IACd,MAAM,CAAS;IACf,MAAM,CAAoB;IAC1B,QAAQ,CAAW;IACnB,YAAY,CAAe;IAC3B,WAAW,CAAc;IAEjC,eAAe;IACP,YAAY,CAAmB;IAC/B,SAAS,CAAgB;IACzB,WAAW,CAAwB;IACnC,WAAW,CAAoB;IAC/B,YAAY,CAAe;IAEnC,OAAO;IACC,YAAY,GAAG,IAAI,GAAG,EAAyC,CAAC;IAExE,YAAY,MAAyB,EAAE,QAAkB,EAAE,YAA0B;QACnF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QAEjC,WAAW;QACX,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAChC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAC/E,QAAQ,CACT,CAAC;QAEF,eAAe;QACf,IAAI,CAAC,YAAY,GAAG,IAAI,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAEnD,eAAe;QACf,IAAI,CAAC,SAAS,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE3D,iBAAiB;QACjB,IAAI,CAAC,WAAW,GAAG,IAAI,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAChF,IAAI,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE;YAC/C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YACnE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,QAAQ;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,sBAAsB,CAAC,GAAG,EAAE;YAC3C,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,kBAAkB;QAClB,IAAI,CAAC,WAAW,GAAG,IAAI,iBAAiB,CACtC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAC7B,QAAQ,CACT,CAAC;QAEF,gBAAgB;QAChB,MAAM,QAAQ,GAAG,yBAAyB,EAAE,CAAC;QAC7C,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAClC,QAAQ,EACR,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAC7B,QAAQ,CACT,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,aAAa,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,OAAY;QAC9B,UAAU;QACV,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,UAAU;QACV,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,MAAM,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;YAC/B,IAAI,MAAM,CAAC,QAAQ,KAAK,aAAa,EAAE,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACrC,OAAO;QACT,CAAC;QAED,UAAU;QACV,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,aAAa,CAAC,SAAS;gBAC1B,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,aAAa,CAAC,WAAW;gBAC5B,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,aAAa,CAAC,WAAW;gBAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBAClE,MAAM;YACR;gBACE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,IAAU;QACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,MAAM,GAAG,EAAE;YACzC,IAAI,EAAE,IAAI,CAAC,QAAQ;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QAEH,QAAQ;QACR,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5D,gBAAgB;QAChB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;YACjE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,SAAS;QACT,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAExD,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;QAExF,WAAW;QACX,QAAQ,UAAU,CAAC,QAAQ,EAAE,CAAC;YAC5B,KAAK,QAAQ;gBACX,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;gBACzE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;oBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,OAAO;iBACrC,CAAC,CAAC;gBACH,MAAM;QACV,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,IAAU;QAC7B,MAAM,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAErD,MAAM,OAAO,GAAsB;YACjC,YAAY,EAAE,gBAAgB,CAAC,OAAO;YACtC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,aAAa;YACzD,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,YAAY,EAAE;YACtD,oBAAoB,EAAE,EAAE,EAAE,eAAe;YACzC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,eAAe;SACvD,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU,CAAC,IAAU,EAAE,UAA0B;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAE/C,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;gBACzE,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAErD,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;gBACtD,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAE7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBAE3E,YAAY;gBACZ,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAEvE,SAAS;gBACT,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBAE1B,OAAO;gBACP,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;oBAC5C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC5D,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YACD,KAAK,QAAQ;gBACX,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;gBAChE,WAAW;gBACX,IAAI,CAAC,WAAW,CAAC,eAAe,CAC9B,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CACxC,CAAC;gBACF,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;gBAClE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;oBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBACH,MAAM;QACV,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,SAAS,CAAC,IAAU,EAAE,UAA0B;QAC5D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;YAChD,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,UAAU,CAAC,KAAK;SACxB,CAAC,CAAC;QAEH,SAAS;QACT,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;YACnE,IAAI,CAAC,WAAW,CAAC,eAAe,CAC9B,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CACxC,CAAC;QACJ,CAAC;aAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,UAAU;YACV,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACtD,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAErD,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAE7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YAC3E,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACvE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAE1B,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;gBAC5C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YAC9D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,MAAM;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,IAAU,EAAE,SAA0B;QAC9D,IAAI,CAAC;YACH,MAAM,OAAO,GAAyB;gBACpC,IAAI;gBACJ,eAAe,EAAE,SAAS;gBAC1B,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS;gBACjD,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa;gBACjD,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;aACvB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAEjE,UAAU;YACV,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAExC,CAAC;gBAAS,CAAC;YACT,OAAO;YACP,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,SAAS;YACT,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB;QAC1B,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE;YAAE,OAAO;QAE5C,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAC1C,IAAI,KAAK,CAAC,KAAK,KAAK,CAAC;YAAE,OAAO;QAE9B,8CAA8C;QAC9C,gCAAgC;QAChC,iCAAiC;QACjC,eAAe;QACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE;YAC9B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CAAC,IAAU,EAAE,MAAkB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAE3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,MAAM,GAAG,EAAE;YACrC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC,CAAC;QAEH,QAAQ;QACR,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/D,CAAC;QAED,YAAY;QACZ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAErE,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,MAAM,GAAG,EAAE;gBACrC,cAAc,EAAE,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBAC7C,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,MAAM,GAAG,EAAE;gBACtC,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;QAED,SAAS;QACT,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,IAAU;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;QACzD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAChE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YAE1E,OAAO;YACP,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEtC,UAAU;YACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC/F,CAAC,EAAE,SAAS,CAAC,CAAC;QACd,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAc;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,KAAK,EAAE,CAAC;YACV,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,IAAU;QACzC,OAAO,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAAC,IAAU;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEtD,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;YAEhE,YAAY;YACZ,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACzC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEnC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5E,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAExE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS;QAOP,MAAM,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QACrD,OAAO;YACL,YAAY,EAAE,gBAAgB,CAAC,OAAO;YACtC,WAAW,EAAE,gBAAgB,CAAC,KAAK;YACnC,OAAO,EAAE,EAAE,EAAE,8BAA8B;YAC3C,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YACxC,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;CACF"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 任务状态机
|
|
3
|
+
*
|
|
4
|
+
* 管理任务的生命周期状态流转
|
|
5
|
+
*
|
|
6
|
+
* 状态流转:
|
|
7
|
+
* created → evaluating → accepted → running → completed
|
|
8
|
+
* → rejected
|
|
9
|
+
* → rejected
|
|
10
|
+
* running → failed
|
|
11
|
+
* running → timeout
|
|
12
|
+
* any → cancelled
|
|
13
|
+
*/
|
|
14
|
+
import { EventBus } from '../core/events.js';
|
|
15
|
+
import type { Task, TaskStatus, TaskStateRecord } from '../types/task.js';
|
|
16
|
+
export declare class TaskStateMachine {
|
|
17
|
+
private logger;
|
|
18
|
+
private eventBus;
|
|
19
|
+
private states;
|
|
20
|
+
constructor(eventBus: EventBus);
|
|
21
|
+
/**
|
|
22
|
+
* 初始化任务状态
|
|
23
|
+
*/
|
|
24
|
+
init(task: Task): TaskStateRecord;
|
|
25
|
+
/**
|
|
26
|
+
* 转换任务状态
|
|
27
|
+
* @throws 如果转换不合法
|
|
28
|
+
*/
|
|
29
|
+
transition(taskId: string, newStatus: TaskStatus, reason?: string): TaskStateRecord;
|
|
30
|
+
/**
|
|
31
|
+
* 尝试转换(不抛异常)
|
|
32
|
+
*/
|
|
33
|
+
tryTransition(taskId: string, newStatus: TaskStatus, reason?: string): boolean;
|
|
34
|
+
/**
|
|
35
|
+
* 获取任务状态记录
|
|
36
|
+
*/
|
|
37
|
+
getState(taskId: string): TaskStateRecord | undefined;
|
|
38
|
+
/**
|
|
39
|
+
* 获取任务当前状态
|
|
40
|
+
*/
|
|
41
|
+
getStatus(taskId: string): TaskStatus | undefined;
|
|
42
|
+
/**
|
|
43
|
+
* 检查任务是否处于终态
|
|
44
|
+
*/
|
|
45
|
+
isTerminal(taskId: string): boolean;
|
|
46
|
+
/**
|
|
47
|
+
* 更新权限级别
|
|
48
|
+
*/
|
|
49
|
+
setPermissionLevel(taskId: string, level: TaskStateRecord['permissionLevel']): void;
|
|
50
|
+
/**
|
|
51
|
+
* 清理已完成任务的状态记录
|
|
52
|
+
*/
|
|
53
|
+
cleanup(taskId: string): void;
|
|
54
|
+
/**
|
|
55
|
+
* 获取所有运行中的任务数
|
|
56
|
+
*/
|
|
57
|
+
getRunningCount(): number;
|
|
58
|
+
/**
|
|
59
|
+
* 获取统计信息
|
|
60
|
+
*/
|
|
61
|
+
getStats(): Record<TaskStatus, number>;
|
|
62
|
+
}
|
|
63
|
+
//# sourceMappingURL=task-state-machine.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-state-machine.d.ts","sourceRoot":"","sources":["../../src/task/task-state-machine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAGH,OAAO,EAAE,QAAQ,EAAmB,MAAM,mBAAmB,CAAC;AAC9D,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAe1E,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAoC;IAClD,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,MAAM,CAAsC;gBAExC,QAAQ,EAAE,QAAQ;IAI9B;;OAEG;IACH,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,eAAe;IAYjC;;;OAGG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,eAAe;IAgCnF;;OAEG;IACH,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO;IAS9E;;OAEG;IACH,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIrD;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,GAAG,SAAS;IAIjD;;OAEG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAMnC;;OAEG;IACH,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,CAAC,iBAAiB,CAAC,GAAG,IAAI;IAQnF;;OAEG;IACH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI7B;;OAEG;IACH,eAAe,IAAI,MAAM;IAUzB;;OAEG;IACH,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC;CAOvC"}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 任务状态机
|
|
3
|
+
*
|
|
4
|
+
* 管理任务的生命周期状态流转
|
|
5
|
+
*
|
|
6
|
+
* 状态流转:
|
|
7
|
+
* created → evaluating → accepted → running → completed
|
|
8
|
+
* → rejected
|
|
9
|
+
* → rejected
|
|
10
|
+
* running → failed
|
|
11
|
+
* running → timeout
|
|
12
|
+
* any → cancelled
|
|
13
|
+
*/
|
|
14
|
+
import { createLogger } from '../core/logger.js';
|
|
15
|
+
import { WorkerClawEvent } from '../core/events.js';
|
|
16
|
+
/** 合法的状态转换表 */
|
|
17
|
+
const VALID_TRANSITIONS = {
|
|
18
|
+
created: ['evaluating', 'rejected', 'cancelled'],
|
|
19
|
+
evaluating: ['accepted', 'rejected', 'deferred', 'cancelled'],
|
|
20
|
+
accepted: ['running', 'cancelled'],
|
|
21
|
+
running: ['completed', 'failed', 'timeout', 'cancelled'],
|
|
22
|
+
failed: [], // 终态
|
|
23
|
+
completed: [], // 终态
|
|
24
|
+
rejected: [], // 终态
|
|
25
|
+
timeout: [], // 终态
|
|
26
|
+
cancelled: [], // 终态
|
|
27
|
+
};
|
|
28
|
+
export class TaskStateMachine {
|
|
29
|
+
logger = createLogger('TaskStateMachine');
|
|
30
|
+
eventBus;
|
|
31
|
+
states = new Map();
|
|
32
|
+
constructor(eventBus) {
|
|
33
|
+
this.eventBus = eventBus;
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* 初始化任务状态
|
|
37
|
+
*/
|
|
38
|
+
init(task) {
|
|
39
|
+
const now = Date.now();
|
|
40
|
+
const record = {
|
|
41
|
+
status: 'created',
|
|
42
|
+
history: [{ status: 'created', timestamp: now }],
|
|
43
|
+
createdAt: now,
|
|
44
|
+
updatedAt: now,
|
|
45
|
+
};
|
|
46
|
+
this.states.set(task.taskId, record);
|
|
47
|
+
return record;
|
|
48
|
+
}
|
|
49
|
+
/**
|
|
50
|
+
* 转换任务状态
|
|
51
|
+
* @throws 如果转换不合法
|
|
52
|
+
*/
|
|
53
|
+
transition(taskId, newStatus, reason) {
|
|
54
|
+
const record = this.states.get(taskId);
|
|
55
|
+
if (!record) {
|
|
56
|
+
throw new Error(`任务状态记录不存在: ${taskId}`);
|
|
57
|
+
}
|
|
58
|
+
const currentStatus = record.status;
|
|
59
|
+
const allowed = VALID_TRANSITIONS[currentStatus];
|
|
60
|
+
if (!allowed || !allowed.includes(newStatus)) {
|
|
61
|
+
throw new Error(`非法状态转换: ${taskId} 从 ${currentStatus} → ${newStatus}` +
|
|
62
|
+
` (允许: ${allowed?.join(', ') || '无'})`);
|
|
63
|
+
}
|
|
64
|
+
const now = Date.now();
|
|
65
|
+
record.status = newStatus;
|
|
66
|
+
record.updatedAt = now;
|
|
67
|
+
record.history.push({ status: newStatus, timestamp: now, reason });
|
|
68
|
+
this.logger.debug(`任务状态变更 [${taskId}]: ${currentStatus} → ${newStatus}`, { reason });
|
|
69
|
+
this.eventBus.emit(WorkerClawEvent.TASK_STATE_CHANGED, {
|
|
70
|
+
taskId,
|
|
71
|
+
from: currentStatus,
|
|
72
|
+
to: newStatus,
|
|
73
|
+
reason,
|
|
74
|
+
});
|
|
75
|
+
return record;
|
|
76
|
+
}
|
|
77
|
+
/**
|
|
78
|
+
* 尝试转换(不抛异常)
|
|
79
|
+
*/
|
|
80
|
+
tryTransition(taskId, newStatus, reason) {
|
|
81
|
+
try {
|
|
82
|
+
this.transition(taskId, newStatus, reason);
|
|
83
|
+
return true;
|
|
84
|
+
}
|
|
85
|
+
catch {
|
|
86
|
+
return false;
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
/**
|
|
90
|
+
* 获取任务状态记录
|
|
91
|
+
*/
|
|
92
|
+
getState(taskId) {
|
|
93
|
+
return this.states.get(taskId);
|
|
94
|
+
}
|
|
95
|
+
/**
|
|
96
|
+
* 获取任务当前状态
|
|
97
|
+
*/
|
|
98
|
+
getStatus(taskId) {
|
|
99
|
+
return this.states.get(taskId)?.status;
|
|
100
|
+
}
|
|
101
|
+
/**
|
|
102
|
+
* 检查任务是否处于终态
|
|
103
|
+
*/
|
|
104
|
+
isTerminal(taskId) {
|
|
105
|
+
const status = this.getStatus(taskId);
|
|
106
|
+
if (!status)
|
|
107
|
+
return true;
|
|
108
|
+
return ['completed', 'failed', 'rejected', 'timeout', 'cancelled'].includes(status);
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* 更新权限级别
|
|
112
|
+
*/
|
|
113
|
+
setPermissionLevel(taskId, level) {
|
|
114
|
+
const record = this.states.get(taskId);
|
|
115
|
+
if (record) {
|
|
116
|
+
record.permissionLevel = level;
|
|
117
|
+
record.updatedAt = Date.now();
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
/**
|
|
121
|
+
* 清理已完成任务的状态记录
|
|
122
|
+
*/
|
|
123
|
+
cleanup(taskId) {
|
|
124
|
+
this.states.delete(taskId);
|
|
125
|
+
}
|
|
126
|
+
/**
|
|
127
|
+
* 获取所有运行中的任务数
|
|
128
|
+
*/
|
|
129
|
+
getRunningCount() {
|
|
130
|
+
let count = 0;
|
|
131
|
+
for (const record of this.states.values()) {
|
|
132
|
+
if (['running', 'accepted'].includes(record.status)) {
|
|
133
|
+
count++;
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
return count;
|
|
137
|
+
}
|
|
138
|
+
/**
|
|
139
|
+
* 获取统计信息
|
|
140
|
+
*/
|
|
141
|
+
getStats() {
|
|
142
|
+
const stats = {};
|
|
143
|
+
for (const record of this.states.values()) {
|
|
144
|
+
stats[record.status] = (stats[record.status] || 0) + 1;
|
|
145
|
+
}
|
|
146
|
+
return stats;
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=task-state-machine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-state-machine.js","sourceRoot":"","sources":["../../src/task/task-state-machine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAY,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAG9D,eAAe;AACf,MAAM,iBAAiB,GAA8C;IACnE,OAAO,EAAE,CAAC,YAAY,EAAE,UAAU,EAAE,WAAW,CAAC;IAChD,UAAU,EAAE,CAAC,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC;IAC7D,QAAQ,EAAE,CAAC,SAAS,EAAE,WAAW,CAAC;IAClC,OAAO,EAAE,CAAC,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,CAAC;IACxD,MAAM,EAAE,EAAE,EAAK,KAAK;IACpB,SAAS,EAAE,EAAE,EAAE,KAAK;IACpB,QAAQ,EAAE,EAAE,EAAG,KAAK;IACpB,OAAO,EAAE,EAAE,EAAI,KAAK;IACpB,SAAS,EAAE,EAAE,EAAE,KAAK;CACrB,CAAC;AAEF,MAAM,OAAO,gBAAgB;IACnB,MAAM,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC;IAC1C,QAAQ,CAAW;IACnB,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEpD,YAAY,QAAkB;QAC5B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,IAAU;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAoB;YAC9B,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;YAChD,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,MAAc,EAAE,SAAqB,EAAE,MAAe;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;QACpC,MAAM,OAAO,GAAG,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAEjD,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CACb,WAAW,MAAM,MAAM,aAAa,MAAM,SAAS,EAAE;gBACrD,SAAS,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,GAAG,CACvC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC;QAC1B,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC;QACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;QAEnE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,MAAM,MAAM,aAAa,MAAM,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACrF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,kBAAkB,EAAE;YACrD,MAAM;YACN,IAAI,EAAE,aAAa;YACnB,EAAE,EAAE,SAAS;YACb,MAAM;SACP,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,MAAc,EAAE,SAAqB,EAAE,MAAe;QAClE,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAC3C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,MAAc;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAc;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,MAAc;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QACzB,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACtF,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,MAAc,EAAE,KAAyC;QAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,eAAe,GAAG,KAAK,CAAC;YAC/B,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,MAAc;QACpB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,eAAe;QACb,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,MAAM,KAAK,GAA2B,EAAE,CAAC;QACzC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,KAAmC,CAAC;IAC7C,CAAC;CACF"}
|