danvics/zerobyte
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

refactor: reconnect agent automatically

Browse source
This commit is contained in:
Nicolas Meienberger 2026-03-30 18:26:48 +02:00
parent 5684efed27
commit d291bb0382
1 changed files with 23 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
apps/agent/src/index.ts
View file

@ -3,12 +3,23 @@ import { createControllerSession, type ControllerSession } from "./controller-se
const controllerUrl = process.env.ZEROBYTE_CONTROLLER_URL; const controllerUrl = process.env.ZEROBYTE_CONTROLLER_URL;
const agentToken = process.env.ZEROBYTE_AGENT_TOKEN; const agentToken = process.env.ZEROBYTE_AGENT_TOKEN;
const reconnectDelayMs = 1_000;
class Agent { export class Agent {
private ws: WebSocket | null = null; private ws: WebSocket | null = null;
private controllerSession: ControllerSession | null = null; private controllerSession: ControllerSession | null = null;
private reconnectTimeout: ReturnType<typeof setTimeout> | null = null;
connect() { connect() {
if (this.reconnectTimeout) {
clearTimeout(this.reconnectTimeout);
this.reconnectTimeout = null;
}
if (this.ws) {
return;
}
if (!controllerUrl) { if (!controllerUrl) {
throw new Error("Env variable ZEROBYTE_CONTROLLER_URL is not set"); throw new Error("Env variable ZEROBYTE_CONTROLLER_URL is not set");
} }
@ -36,6 +47,13 @@ class Agent {
this.controllerSession = null; this.controllerSession = null;
this.ws = null; this.ws = null;
logger.info("Agent disconnected from controller"); logger.info("Agent disconnected from controller");
if (!this.reconnectTimeout) {
this.reconnectTimeout = setTimeout(() => {
this.reconnectTimeout = null;
this.connect();
}, reconnectDelayMs);
}
}; };
this.ws.onerror = (error) => { this.ws.onerror = (error) => {
logger.error("Agent encountered an error:", error); logger.error("Agent encountered an error:", error);
@ -43,5 +61,7 @@ class Agent {
} }
} }
const agent = new Agent(); if (import.meta.main) {
agent.connect(); const agent = new Agent();
agent.connect();
}