전체적인 리팩토링
This commit is contained in:
96
scripts/autotrade-worker.mjs
Normal file
96
scripts/autotrade-worker.mjs
Normal file
@@ -0,0 +1,96 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/**
|
||||
* [목적]
|
||||
* heartbeat가 끊긴 자동매매 세션을 주기적으로 정리하는 백그라운드 워커입니다.
|
||||
*
|
||||
* [데이터 흐름]
|
||||
* worker loop -> /api/autotrade/worker/tick 호출 -> 만료 세션 sweep -> 로그 출력
|
||||
*/
|
||||
const args = new Set(process.argv.slice(2));
|
||||
const once = args.has("--once");
|
||||
|
||||
const appUrl = process.env.AUTOTRADE_APP_URL || "http://127.0.0.1:3001";
|
||||
const workerToken = process.env.AUTOTRADE_WORKER_TOKEN || "autotrade-worker-local";
|
||||
const pollMsRaw = Number.parseInt(process.env.AUTOTRADE_WORKER_POLL_MS || "5000", 10);
|
||||
const pollMs = Number.isFinite(pollMsRaw) ? Math.max(1000, pollMsRaw) : 5000;
|
||||
|
||||
let isShuttingDown = false;
|
||||
|
||||
function log(message, level = "INFO") {
|
||||
const time = new Date().toISOString();
|
||||
const line = `[autotrade-worker][${level}][${time}] ${message}`;
|
||||
console.log(line);
|
||||
}
|
||||
|
||||
async function runTick() {
|
||||
// Next API로 tick 요청을 보내 세션 만료 정리를 수행합니다.
|
||||
const response = await fetch(`${appUrl}/api/autotrade/worker/tick`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"content-type": "application/json",
|
||||
"x-autotrade-worker-token": workerToken,
|
||||
},
|
||||
body: JSON.stringify({}),
|
||||
});
|
||||
|
||||
const payload = await response.json().catch(() => null);
|
||||
|
||||
if (!response.ok) {
|
||||
const message = payload?.message || `HTTP ${response.status}`;
|
||||
throw new Error(message);
|
||||
}
|
||||
|
||||
return payload;
|
||||
}
|
||||
|
||||
async function loop() {
|
||||
if (once) {
|
||||
const payload = await runTick();
|
||||
log(
|
||||
`single tick done: expired=${payload?.sweep?.expiredCount ?? 0}, running=${payload?.runningSessions ?? 0}`,
|
||||
);
|
||||
return;
|
||||
}
|
||||
|
||||
log(`started (poll=${pollMs}ms, app=${appUrl})`);
|
||||
|
||||
// 지정 주기마다 tick을 반복 호출합니다.
|
||||
while (!isShuttingDown) {
|
||||
try {
|
||||
const payload = await runTick();
|
||||
log(
|
||||
`tick ok: expired=${payload?.sweep?.expiredCount ?? 0}, running=${payload?.runningSessions ?? 0}, stopped=${payload?.stoppedSessions ?? 0}`,
|
||||
);
|
||||
} catch (error) {
|
||||
const message = error instanceof Error ? error.message : "tick failed";
|
||||
log(message, "ERROR");
|
||||
}
|
||||
|
||||
await wait(pollMs);
|
||||
}
|
||||
|
||||
log("graceful shutdown complete");
|
||||
}
|
||||
|
||||
function wait(ms) {
|
||||
return new Promise((resolve) => {
|
||||
setTimeout(resolve, ms);
|
||||
});
|
||||
}
|
||||
|
||||
process.on("SIGINT", () => {
|
||||
isShuttingDown = true;
|
||||
log("SIGINT received, stopping...");
|
||||
});
|
||||
|
||||
process.on("SIGTERM", () => {
|
||||
isShuttingDown = true;
|
||||
log("SIGTERM received, stopping...");
|
||||
});
|
||||
|
||||
loop().catch((error) => {
|
||||
const message = error instanceof Error ? error.message : "worker exited with error";
|
||||
log(message, "ERROR");
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user