run-registry.js
4.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
/**
* RunRegistry - 进程运行注册表
*
* 参考 openclaw/src/process/supervisor/registry.ts
*
* 核心功能：
* - 状态机追踪: starting → running → exiting → exited
* - 按条件查询: byBackend, byState, active
* - 自动淘汰: exited 记录超过 maxExited 时 FIFO 清除
* - 统计输出: 给 /api/runs 端点提供数据
*/
/** @typedef {'starting'|'running'|'exiting'|'exited'} RunState */
/** @typedef {'manual-cancel'|'overall-timeout'|'no-output-timeout'|'spawn-error'|'signal'|'exit'|'unknown'} TerminationReason */
/**
* @typedef {object} RunRecord
* @property {string} runId
* @property {string} agentName
* @property {string} backendId - launcher / cliBackend 标识
* @property {string} [scopeKey] - 串行化 key
* @property {number} [pid]
* @property {RunState} state
* @property {TerminationReason} [terminationReason]
* @property {number} [exitCode]
* @property {number} createdAtMs
* @property {number} startedAtMs
* @property {number} lastOutputAtMs
* @property {number} [updatedAtMs]
*/
const DEFAULT_MAX_EXITED = 200;
class RunRegistry {
/**
* @param {object} [opts]
* @param {number} [opts.maxExitedRecords=200]
*/
constructor(opts = {}) {
/** @type {Map<string, RunRecord>} */
this._records = new Map();
this._maxExited = opts.maxExitedRecords || DEFAULT_MAX_EXITED;
}
/**
* 注册新的 run
* @param {object} params
* @param {string} params.runId
* @param {string} params.agentName
* @param {string} params.backendId
* @param {string} [params.scopeKey]
* @param {number} [params.pid]
* @returns {string} runId
*/
add(params) {
const now = Date.now();
/** @type {RunRecord} */
const record = {
runId: params.runId,
agentName: params.agentName,
backendId: params.backendId,
scopeKey: params.scopeKey,
pid: params.pid,
state: 'starting',
createdAtMs: now,
startedAtMs: 0,
lastOutputAtMs: 0,
};
this._records.set(record.runId, record);
this._pruneExited();
return record.runId;
}
/**
* 获取记录
* @param {string} runId
* @returns {RunRecord|undefined}
*/
get(runId) {
return this._records.get(runId);
}
/**
* 变更状态
* @param {string} runId
* @param {RunState} state
* @param {object} [extra] - 额外字段 (exitCode, terminationReason, pid 等)
*/
markState(runId, state, extra = {}) {
const record = this._records.get(runId);
if (!record) return;
record.state = state;
record.updatedAtMs = Date.now();
if (state === 'running' && !record.startedAtMs) {
record.startedAtMs = Date.now();
}
Object.assign(record, extra);
}
/** 记录输出活动 (给 ProcessGuard 联动) */
touchOutput(runId) {
const record = this._records.get(runId);
if (record) {
record.lastOutputAtMs = Date.now();
}
}
/** 获取所有活跃 runs (starting + running) */
listActive() {
return [...this._records.values()].filter(
r => r.state === 'starting' || r.state === 'running'
);
}
/** 按 backendId 查询 */
listByBackend(backendId) {
return [...this._records.values()].filter(r => r.backendId === backendId);
}
/** 按 agentName 查询 */
listByAgent(agentName) {
return [...this._records.values()].filter(r => r.agentName === agentName);
}
/** 获取全部记录 */
list() {
return [...this._records.values()];
}
/** 删除指定 run */
remove(runId) {
this._records.delete(runId);
}
/** 统计信息 */
stats() {
const all = [...this._records.values()];
return {
total: all.length,
starting: all.filter(r => r.state === 'starting').length,
running: all.filter(r => r.state === 'running').length,
exiting: all.filter(r => r.state === 'exiting').length,
exited: all.filter(r => r.state === 'exited').length,
};
}
/** 清空全部 */
clear() {
this._records.clear();
}
/** 淘汰过期 exited 记录 (FIFO) */
_pruneExited() {
const exited = [...this._records.entries()]
.filter(([, r]) => r.state === 'exited')
.sort((a, b) => (a[1].createdAtMs || 0) - (b[1].createdAtMs || 0));
while (exited.length > this._maxExited) {
const [id] = exited.shift();
this._records.delete(id);
}
}
}
module.exports = { RunRegistry, DEFAULT_MAX_EXITED };