Control remote fleet workers over SSH with user, host, and key path
Some checks are pending
offline / test (push) Waiting to run
Some checks are pending
offline / test (push) Waiting to run
NS1 is marchon@70.88.205.138 using ~/.ssh/id_ed25519. Private key bytes stay off git. Spawn, health, and kill go through ssh; remote workers bind loopback only.
This commit is contained in:
parent
b325f6f697
commit
e89e006542
12 changed files with 425 additions and 34 deletions
|
|
@ -23,7 +23,12 @@ operator --HTTP 127.0.0.1:3850--> fleet control
|
|||
Define hosts in `packages/verae-fleet/machines.json` (or **Add machine** on the monitor). Each machine has `capacity` and `roles` it may run. The supervisor places new replicas on the **least-loaded** eligible host.
|
||||
|
||||
- `kind: local` — spawn on the control plane.
|
||||
- `kind: agent` — HTTP to `http://host:agentPort` (`node src/agent.js` on that box). Do not publish NATS.
|
||||
- `kind: ssh` — `user@host` with `identityFile` (path only, e.g. `~/.ssh/id_ed25519`). Control plane `scp`s the worker and `ssh`es spawn/kill/health. Workers stay on `127.0.0.1` on the remote host; NATS is not published.
|
||||
- `kind: agent` — HTTP to `http://host:agentPort` (`node src/agent.js` on that box).
|
||||
|
||||
Override credentials in gitignored `machines.secrets.json` (see `machines.secrets.json.example`). Never put private-key bytes in git.
|
||||
|
||||
`node src/cli.js ssh-check ns1` verifies login as marchon@70.88.205.138.
|
||||
|
||||
Raising `tree-node.max` and adding machines increases bulk-summary lookup capacity.
|
||||
|
||||
|
|
|
|||
2
packages/verae-fleet/.gitignore
vendored
2
packages/verae-fleet/.gitignore
vendored
|
|
@ -1 +1,3 @@
|
|||
data/
|
||||
machines.secrets.json
|
||||
|
||||
|
|
|
|||
|
|
@ -27,7 +27,13 @@ node src/cli.js stop webhook-deliver # disable that service
|
|||
node src/cli.js start webhook-deliver
|
||||
```
|
||||
|
||||
Add capacity by editing `machines.json` or the monitor **Add machine** form. Remote boxes run `node src/agent.js` (`FLEET_AGENT_PORT=3851`). New replicas land on the least-loaded host that allows the role.
|
||||
Add capacity in `machines.json` or the monitor **Add machine** form (`kind=ssh`, user, host, identity file path). New replicas land on the least-loaded eligible host.
|
||||
|
||||
```bash
|
||||
node src/cli.js ssh-check ns1 # marchon@70.88.205.138 with ~/.ssh/id_ed25519
|
||||
```
|
||||
|
||||
Private keys stay on disk (`~/.ssh/id_ed25519`); git stores only the path. Optional overrides: `machines.secrets.json` (gitignored).
|
||||
|
||||
The UI shows message-processing **min / avg / p50 / p90** RTT per service, instance, and machine.
|
||||
|
||||
|
|
|
|||
|
|
@ -12,10 +12,13 @@
|
|||
{
|
||||
"id": "ns1",
|
||||
"title": "NS1.GEORGELAMBERT.ORG",
|
||||
"kind": "agent",
|
||||
"kind": "ssh",
|
||||
"host": "70.88.205.138",
|
||||
"agentPort": 3851,
|
||||
"enabled": false,
|
||||
"user": "marchon",
|
||||
"sshPort": 22,
|
||||
"identityFile": "~/.ssh/id_ed25519",
|
||||
"remoteDir": "~/verae-fleet-runtime",
|
||||
"enabled": true,
|
||||
"capacity": 24,
|
||||
"roles": [
|
||||
"tree-node",
|
||||
|
|
@ -24,7 +27,21 @@
|
|||
"job-poller",
|
||||
"webhook-deliver"
|
||||
],
|
||||
"notes": "Run `node src/agent.js` on NS1, then enable this machine to spread tree-node / WORM capacity. Agent port is loopback-plus-SSH-tunnel or private LAN — do not publish NATS."
|
||||
"notes": "SSH as marchon with ~/.ssh/id_ed25519. Workers bind 127.0.0.1 on NS1; control plane probes via ssh+curl. Private key is never stored in git."
|
||||
},
|
||||
{
|
||||
"id": "lan-134",
|
||||
"title": "70.88.205.134",
|
||||
"kind": "ssh",
|
||||
"host": "70.88.205.134",
|
||||
"user": "marchon",
|
||||
"sshPort": 22,
|
||||
"identityFile": "~/.ssh/id_ed25519",
|
||||
"remoteDir": "~/verae-fleet-runtime",
|
||||
"enabled": false,
|
||||
"capacity": 12,
|
||||
"roles": ["tree-node", "archive-worm"],
|
||||
"notes": "Same user/key as NS1. Enable after confirming node is on PATH."
|
||||
}
|
||||
]
|
||||
}
|
||||
|
|
|
|||
13
packages/verae-fleet/machines.secrets.json.example
Normal file
13
packages/verae-fleet/machines.secrets.json.example
Normal file
|
|
@ -0,0 +1,13 @@
|
|||
{
|
||||
"ns1": {
|
||||
"user": "marchon",
|
||||
"host": "70.88.205.138",
|
||||
"sshPort": 22,
|
||||
"identityFile": "~/.ssh/id_ed25519"
|
||||
},
|
||||
"lan-134": {
|
||||
"user": "marchon",
|
||||
"host": "70.88.205.134",
|
||||
"identityFile": "~/.ssh/id_ed25519"
|
||||
}
|
||||
}
|
||||
|
|
@ -44,14 +44,17 @@
|
|||
<label>id <input name="id" required placeholder="ns1-b"/></label>
|
||||
<label>host <input name="host" required placeholder="10.0.0.12"/></label>
|
||||
<label>kind
|
||||
<select name="kind"><option value="local">local</option><option value="agent">agent</option></select>
|
||||
<select name="kind"><option value="local">local</option><option value="ssh">ssh</option><option value="agent">agent</option></select>
|
||||
</label>
|
||||
<label>user <input name="user" value="marchon"/></label>
|
||||
<label>ssh port <input name="sshPort" type="number" value="22"/></label>
|
||||
<label>identity file <input name="identityFile" placeholder="~/.ssh/id_ed25519" value="~/.ssh/id_ed25519"/></label>
|
||||
<label>capacity <input name="capacity" type="number" min="1" value="8"/></label>
|
||||
<label>roles <input name="roles" value="*" placeholder="tree-node,archive-worm"/></label>
|
||||
<button type="submit">Add machine</button>
|
||||
</form>
|
||||
<table>
|
||||
<thead><tr><th>Machine</th><th>Host</th><th>Kind</th><th>Capacity</th><th>Running</th><th>RTT ms min / avg / p50 / p90</th><th></th></tr></thead>
|
||||
<thead><tr><th>Machine</th><th>SSH / host</th><th>Kind</th><th>Capacity</th><th>Running</th><th>RTT ms min / avg / p50 / p90</th><th></th></tr></thead>
|
||||
<tbody id="machines"></tbody>
|
||||
</table>
|
||||
<h2>Services</h2>
|
||||
|
|
@ -74,13 +77,14 @@ async function draw() {
|
|||
const on = m.enabled;
|
||||
return `<tr class="${on ? (m.running > 0 ? 'state-operational' : '') : 'state-down'}">
|
||||
<td><strong>${m.id}</strong><br><span style="color:#5b6d78">${m.title || ''}</span></td>
|
||||
<td><code>${m.host}${m.kind === 'agent' ? ':' + m.agentPort : ''}</code></td>
|
||||
<td><code>${m.kind === 'ssh' ? (m.user || 'marchon') + '@' : ''}${m.host}${m.kind === 'agent' ? ':' + m.agentPort : ''}${m.kind === 'ssh' && m.identityFile ? ' key=' + m.identityFile : ''}</code></td>
|
||||
<td>${m.kind} ${on ? '<span class="pill good">on</span>' : '<span class="pill bad">off</span>'}</td>
|
||||
<td>${m.running} / ${m.capacity}</td>
|
||||
<td>${m.available} avail</td>
|
||||
<td>${rttCell(m.rtt)}</td>
|
||||
<td>
|
||||
<button onclick="act('/api/machines/${m.id}/${on ? 'disable' : 'enable'}')">${on ? 'disable' : 'enable'}</button>
|
||||
${m.kind === 'ssh' ? `<button onclick="act('/api/machines/${m.id}/check')">ssh-check</button>` : ''}
|
||||
</td>
|
||||
</tr>`;
|
||||
}).join('');
|
||||
|
|
@ -126,6 +130,9 @@ document.getElementById('addMachine').addEventListener('submit', async (e) => {
|
|||
id: f.id.value,
|
||||
host: f.host.value,
|
||||
kind: f.kind.value,
|
||||
user: f.user.value,
|
||||
sshPort: Number(f.sshPort.value || 22),
|
||||
identityFile: f.identityFile.value,
|
||||
capacity: Number(f.capacity.value || 8),
|
||||
roles,
|
||||
enabled: true,
|
||||
|
|
|
|||
|
|
@ -37,7 +37,8 @@ function help() {
|
|||
Central spec: fleet.json
|
||||
Machines: machines.json (add hosts to spread replicas)
|
||||
Per service: services/<id>.json
|
||||
Remote host: FLEET_AGENT_PORT=3851 node src/agent.js
|
||||
Remote SSH: machines.json user + host + identityFile (path only)
|
||||
ssh-check [id] test SSH login (default ns1)
|
||||
`);
|
||||
}
|
||||
|
||||
|
|
@ -74,6 +75,19 @@ async function main() {
|
|||
}
|
||||
print(`\ncentral replica spec: ${loaded.fleetPath}`);
|
||||
print(`tree-node floor: min=${loaded.services['tree-node'].min} keepFloor=${loaded.services['tree-node'].keepFloor}`);
|
||||
print('\nmachines:');
|
||||
for (const m of loaded.machines) {
|
||||
const ssh = m.kind === 'ssh' ? `${m.user}@${m.host} key=${m.identityFile}` : m.host;
|
||||
print(` ${m.id.padEnd(12)} ${m.kind.padEnd(6)} ${m.enabled ? 'on ' : 'off'} cap=${m.capacity} ${ssh}`);
|
||||
}
|
||||
return;
|
||||
}
|
||||
if (cmd === 'ssh-check') {
|
||||
const { sshCheck } = await import('./ssh.js');
|
||||
const id = target || 'ns1';
|
||||
const m = loaded.machines.find((x) => x.id === id);
|
||||
if (!m) throw new Error(`unknown machine ${id}`);
|
||||
print(await sshCheck(m));
|
||||
return;
|
||||
}
|
||||
if (cmd === 'serve' || cmd === 'monitor') {
|
||||
|
|
|
|||
|
|
@ -32,15 +32,29 @@ export function loadMachines(root, overlayPath) {
|
|||
const incoming = Array.isArray(extra) ? extra : extra.machines || [];
|
||||
for (const m of incoming) upsertMachine(list, m);
|
||||
}
|
||||
const secretsPath = path.join(root, 'machines.secrets.json');
|
||||
if (fs.existsSync(secretsPath)) {
|
||||
const secrets = JSON.parse(fs.readFileSync(secretsPath, 'utf8'));
|
||||
const byId = secrets.machines || secrets;
|
||||
for (const m of list) {
|
||||
const extra = byId[m.id];
|
||||
if (extra && typeof extra === 'object') Object.assign(m, extra);
|
||||
}
|
||||
}
|
||||
return list.map(normalizeMachine);
|
||||
}
|
||||
|
||||
export function normalizeMachine(m) {
|
||||
const kind = m.kind === 'agent' || m.kind === 'ssh' ? m.kind : 'local';
|
||||
return {
|
||||
id: String(m.id || '').trim(),
|
||||
title: m.title || m.id,
|
||||
kind: m.kind === 'agent' ? 'agent' : 'local',
|
||||
kind,
|
||||
host: m.host || '127.0.0.1',
|
||||
user: m.user || (kind === 'ssh' ? 'marchon' : ''),
|
||||
sshPort: Number(m.sshPort || 22),
|
||||
identityFile: m.identityFile || (kind === 'ssh' ? '~/.ssh/id_ed25519' : ''),
|
||||
remoteDir: m.remoteDir || '~/verae-fleet-runtime',
|
||||
agentPort: Number(m.agentPort || 3851),
|
||||
enabled: m.enabled !== false,
|
||||
capacity: Number(m.capacity || 8),
|
||||
|
|
@ -72,8 +86,9 @@ export function canHost(machine, serviceId, role) {
|
|||
* @param {string} serviceId
|
||||
* @param {string} [role]
|
||||
*/
|
||||
export function pickMachine(machines, instances, serviceId, role) {
|
||||
const eligible = machines.filter((m) => canHost(m, serviceId, role));
|
||||
export function pickMachine(machines, instances, serviceId, role, exclude = []) {
|
||||
const skip = new Set(exclude);
|
||||
const eligible = machines.filter((m) => canHost(m, serviceId, role) && !skip.has(m.id));
|
||||
if (!eligible.length) return null;
|
||||
const scored = eligible.map((m) => {
|
||||
const running = instances.filter((i) => i.machine === m.id && i.pid).length;
|
||||
|
|
|
|||
|
|
@ -47,9 +47,13 @@ export function startControlServer(sup, mon) {
|
|||
const m = sup.addMachine(spec);
|
||||
return json(res, 200, { machine: m, machines: sup.status().machines });
|
||||
}
|
||||
const mach = url.pathname.match(/^\/api\/machines\/([^/]+)\/(enable|disable)$/);
|
||||
const mach = url.pathname.match(/^\/api\/machines\/([^/]+)\/(enable|disable|check)$/);
|
||||
if (mach && req.method === 'POST') {
|
||||
const [, id, op] = mach;
|
||||
if (op === 'check') {
|
||||
const check = await sup.checkMachine(id);
|
||||
return json(res, check.ok ? 200 : 400, { check, machines: sup.status().machines });
|
||||
}
|
||||
const m = sup.setMachineEnabled(id, op === 'enable');
|
||||
return json(res, 200, { machine: m, machines: sup.status().machines });
|
||||
}
|
||||
|
|
|
|||
168
packages/verae-fleet/src/ssh.js
Normal file
168
packages/verae-fleet/src/ssh.js
Normal file
|
|
@ -0,0 +1,168 @@
|
|||
/**
|
||||
* SSH control of remote fleet workers. Uses username + host + identity file path.
|
||||
* Never logs or stores private-key material.
|
||||
* @module ssh
|
||||
*/
|
||||
|
||||
import { spawn } from 'node:child_process';
|
||||
import fs from 'node:fs';
|
||||
import os from 'node:os';
|
||||
import path from 'node:path';
|
||||
import { fileURLToPath } from 'node:url';
|
||||
|
||||
const HERE = path.dirname(fileURLToPath(import.meta.url));
|
||||
|
||||
export function expandHome(p) {
|
||||
if (!p) return p;
|
||||
if (p === '~') return os.homedir();
|
||||
if (p.startsWith('~/')) return path.join(os.homedir(), p.slice(2));
|
||||
return p;
|
||||
}
|
||||
|
||||
export function shellQuote(s) {
|
||||
return `'${String(s).replace(/'/g, `'\"'\"'`)}'`;
|
||||
}
|
||||
|
||||
export function sshTarget(machine) {
|
||||
return `${machine.user || 'marchon'}@${machine.host}`;
|
||||
}
|
||||
|
||||
export function sshBaseArgs(machine) {
|
||||
const ident = expandHome(machine.identityFile || '');
|
||||
const args = [
|
||||
'-o', 'BatchMode=yes',
|
||||
'-o', 'StrictHostKeyChecking=accept-new',
|
||||
'-o', 'IdentitiesOnly=yes',
|
||||
'-o', 'ConnectTimeout=8',
|
||||
'-p', String(machine.sshPort || 22),
|
||||
];
|
||||
if (ident) {
|
||||
if (!fs.existsSync(ident)) throw new Error(`identityFile not found: ${ident}`);
|
||||
args.push('-i', ident);
|
||||
}
|
||||
return args;
|
||||
}
|
||||
|
||||
export function scpBaseArgs(machine) {
|
||||
const ident = expandHome(machine.identityFile || '');
|
||||
const args = [
|
||||
'-o', 'BatchMode=yes',
|
||||
'-o', 'StrictHostKeyChecking=accept-new',
|
||||
'-o', 'IdentitiesOnly=yes',
|
||||
'-o', 'ConnectTimeout=8',
|
||||
'-P', String(machine.sshPort || 22),
|
||||
];
|
||||
if (ident) args.push('-i', ident);
|
||||
return args;
|
||||
}
|
||||
|
||||
export function sshExec(machine, remoteCommand, { timeoutMs = 20000 } = {}) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const args = [...sshBaseArgs(machine), sshTarget(machine), remoteCommand];
|
||||
const child = spawn('ssh', args, { stdio: ['ignore', 'pipe', 'pipe'] });
|
||||
const out = [];
|
||||
const err = [];
|
||||
child.stdout.on('data', (c) => out.push(c));
|
||||
child.stderr.on('data', (c) => err.push(c));
|
||||
const t = setTimeout(() => {
|
||||
child.kill('SIGKILL');
|
||||
reject(new Error(`ssh timeout: ${machine.id || machine.host}`));
|
||||
}, timeoutMs);
|
||||
child.on('error', (e) => {
|
||||
clearTimeout(t);
|
||||
reject(e);
|
||||
});
|
||||
child.on('close', (code) => {
|
||||
clearTimeout(t);
|
||||
const stdout = Buffer.concat(out).toString('utf8').trim();
|
||||
const stderr = Buffer.concat(err).toString('utf8').trim();
|
||||
if (code !== 0) {
|
||||
reject(new Error(stderr || stdout || `ssh exit ${code}`));
|
||||
return;
|
||||
}
|
||||
resolve({ stdout, stderr, code });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
export function scpTo(machine, localFiles, remoteDir) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const dest = `${sshTarget(machine)}:${remoteDir}/`;
|
||||
const args = [...scpBaseArgs(machine), ...localFiles, dest];
|
||||
const child = spawn('scp', args, { stdio: ['ignore', 'pipe', 'pipe'] });
|
||||
const err = [];
|
||||
child.stderr.on('data', (c) => err.push(c));
|
||||
const t = setTimeout(() => {
|
||||
child.kill('SIGKILL');
|
||||
reject(new Error('scp timeout'));
|
||||
}, 30000);
|
||||
child.on('close', (code) => {
|
||||
clearTimeout(t);
|
||||
if (code !== 0) reject(new Error(Buffer.concat(err).toString('utf8') || `scp exit ${code}`));
|
||||
else resolve();
|
||||
});
|
||||
child.on('error', (e) => {
|
||||
clearTimeout(t);
|
||||
reject(e);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
const synced = new Set();
|
||||
|
||||
export async function ensureRemoteRuntime(machine) {
|
||||
const remoteDir = machine.remoteDir || '~/verae-fleet-runtime';
|
||||
const key = `${machine.id}:${remoteDir}`;
|
||||
if (synced.has(key)) return remoteDir;
|
||||
await sshExec(machine, `mkdir -p ${remoteDir}/src ${remoteDir}/data`);
|
||||
await scpTo(machine, [path.join(HERE, 'worker.js'), path.join(HERE, 'rtt.js')], `${remoteDir}/src`);
|
||||
synced.add(key);
|
||||
return remoteDir;
|
||||
}
|
||||
|
||||
export async function sshSpawnWorker(machine, { instance, env }) {
|
||||
const remoteDir = await ensureRemoteRuntime(machine);
|
||||
const fleetEnv = Object.entries(env)
|
||||
.filter(([k]) => k.startsWith('FLEET_'))
|
||||
.map(([k, v]) => `${k}=${shellQuote(v)}`)
|
||||
.join(' ');
|
||||
const cmd = `mkdir -p ${remoteDir}/data/${instance} && cd ${remoteDir} && ${fleetEnv} nohup node src/worker.js >data/${instance}/worker.log 2>&1 & echo $!`;
|
||||
const { stdout } = await sshExec(machine, cmd);
|
||||
const pid = Number(String(stdout).split('\n').pop());
|
||||
if (!Number.isFinite(pid)) throw new Error(`remote spawn produced no pid: ${stdout}`);
|
||||
return { pid, remoteDir };
|
||||
}
|
||||
|
||||
export async function sshKillWorker(machine, pid, instance) {
|
||||
const cmd = pid
|
||||
? `kill -TERM ${Number(pid)} 2>/dev/null || true`
|
||||
: `pkill -f ${shellQuote(`FLEET_INSTANCE=${instance}`)} 2>/dev/null || true`;
|
||||
await sshExec(machine, cmd);
|
||||
}
|
||||
|
||||
export async function sshHttp(machine, method, urlPath, { port, body } = {}) {
|
||||
const url = `http://127.0.0.1:${port}${urlPath}`;
|
||||
let cmd = `curl -sS -m 3 -X ${method}`;
|
||||
if (body) {
|
||||
cmd += ` -H 'content-type: application/json' -d ${shellQuote(JSON.stringify(body))}`;
|
||||
}
|
||||
cmd += ` ${shellQuote(url)}`;
|
||||
const { stdout } = await sshExec(machine, cmd, { timeoutMs: 8000 });
|
||||
try {
|
||||
return JSON.parse(stdout || '{}');
|
||||
} catch {
|
||||
return { raw: stdout };
|
||||
}
|
||||
}
|
||||
|
||||
export async function sshCheck(machine) {
|
||||
const { stdout } = await sshExec(machine, 'echo OK; whoami; hostname; command -v node; node -v');
|
||||
const lines = stdout.split('\n');
|
||||
return {
|
||||
ok: lines[0] === 'OK',
|
||||
user: lines[1],
|
||||
hostname: lines[2],
|
||||
node: lines[3] ? `${lines[3]} ${lines[4] || ''}`.trim() : null,
|
||||
identityFile: expandHome(machine.identityFile || ''),
|
||||
};
|
||||
}
|
||||
|
|
@ -12,6 +12,7 @@ import { httpProbe, postControl } from './health.js';
|
|||
import { classifyService } from './classify.js';
|
||||
import { pickMachine, upsertMachine, saveOverlay } from './machines.js';
|
||||
import { summarizeRtt } from './rtt.js';
|
||||
import { sshSpawnWorker, sshKillWorker, sshHttp, sshCheck } from './ssh.js';
|
||||
|
||||
const WORKER = path.join(path.dirname(fileURLToPath(import.meta.url)), 'worker.js');
|
||||
|
||||
|
|
@ -55,6 +56,35 @@ export class Supervisor {
|
|||
return m;
|
||||
}
|
||||
|
||||
async checkMachine(id) {
|
||||
const m = this.machines().find((x) => x.id === id);
|
||||
if (!m) throw new Error(`unknown machine ${id}`);
|
||||
if (m.kind === 'ssh') return sshCheck(m);
|
||||
if (m.kind === 'agent') {
|
||||
const r = await httpProbe({ host: m.host, port: m.agentPort, path: '/health', timeoutMs: 2000 });
|
||||
return { ok: r.ok, kind: 'agent', body: r.body };
|
||||
}
|
||||
return { ok: true, kind: 'local', host: m.host };
|
||||
}
|
||||
|
||||
machineOf(rec) {
|
||||
return this.machines().find((x) => x.id === rec.machine);
|
||||
}
|
||||
|
||||
async controlInstance(rec, pathname, body) {
|
||||
if (rec.kind === 'ssh') {
|
||||
const m = this.machineOf(rec);
|
||||
if (!m) throw new Error(`machine ${rec.machine} missing`);
|
||||
return sshHttp(m, 'POST', pathname, { port: rec.healthPort, body });
|
||||
}
|
||||
return postControl({
|
||||
host: rec.host || '127.0.0.1',
|
||||
port: rec.healthPort,
|
||||
path: pathname,
|
||||
body,
|
||||
});
|
||||
}
|
||||
|
||||
log(type, detail) {
|
||||
const ev = { t: new Date().toISOString(), type, ...detail };
|
||||
this.events.push(ev);
|
||||
|
|
@ -100,7 +130,13 @@ export class Supervisor {
|
|||
if (this.instances.has(instance)) return this.instances.get(instance);
|
||||
const machine =
|
||||
opts.machine ||
|
||||
pickMachine(this.machines(), [...this.instances.values()], serviceId, spec.role || serviceId);
|
||||
pickMachine(
|
||||
this.machines(),
|
||||
[...this.instances.values()],
|
||||
serviceId,
|
||||
spec.role || serviceId,
|
||||
opts.excludeMachines || [],
|
||||
);
|
||||
if (!machine) {
|
||||
this.log('skip', { service: serviceId, reason: 'no-machine-capacity' });
|
||||
return null;
|
||||
|
|
@ -115,13 +151,25 @@ export class Supervisor {
|
|||
FLEET_INSTANCE: instance,
|
||||
FLEET_MACHINE: machine.id,
|
||||
FLEET_HEALTH_PORT: String(healthPort),
|
||||
FLEET_HEALTH_BIND: machine.kind === 'local' ? '127.0.0.1' : '0.0.0.0',
|
||||
FLEET_STATE_DIR: stateDir,
|
||||
FLEET_HEALTH_BIND: machine.kind === 'local' ? '127.0.0.1' : '127.0.0.1',
|
||||
FLEET_STATE_DIR: machine.kind === 'ssh' ? `${machine.remoteDir || '~/verae-fleet-runtime'}/data/${instance}` : stateDir,
|
||||
FLEET_MESSAGE_HZ: process.env.FLEET_MESSAGE_HZ || '8',
|
||||
};
|
||||
let child = null;
|
||||
let pid = null;
|
||||
if (machine.kind === 'agent') {
|
||||
if (machine.kind === 'ssh') {
|
||||
try {
|
||||
const spawned = await sshSpawnWorker(machine, { instance, env });
|
||||
pid = spawned.pid;
|
||||
} catch (err) {
|
||||
this.log('skip', { service: serviceId, reason: 'ssh-spawn-failed', machine: machine.id, error: err.message });
|
||||
const exclude = [...(opts.excludeMachines || []), machine.id];
|
||||
if (exclude.length < this.machines().length) {
|
||||
return this.startOne(serviceId, { ...opts, excludeMachines: exclude });
|
||||
}
|
||||
return null;
|
||||
}
|
||||
} else if (machine.kind === 'agent') {
|
||||
try {
|
||||
const spawned = await postControl({
|
||||
host: machine.host,
|
||||
|
|
@ -150,6 +198,7 @@ export class Supervisor {
|
|||
stateDir,
|
||||
child,
|
||||
kind: machine.kind,
|
||||
_machine: machine,
|
||||
paused: false,
|
||||
startedAt: Date.now(),
|
||||
restarts: 0,
|
||||
|
|
@ -167,7 +216,7 @@ export class Supervisor {
|
|||
}
|
||||
this.instances.set(instance, rec);
|
||||
this.log('start', { instance, service: serviceId, machine: machine.id, pid: rec.pid, healthPort });
|
||||
const h = await waitForHealth(rec.host, healthPort, 4000);
|
||||
const h = await waitForHealth(rec, 8000);
|
||||
rec.healthy = h.ok;
|
||||
rec.lastProbe = h;
|
||||
return rec;
|
||||
|
|
@ -177,7 +226,16 @@ export class Supervisor {
|
|||
const rec = this.instances.get(instanceId);
|
||||
if (!rec) return null;
|
||||
rec.stopping = true;
|
||||
if (rec.kind === 'agent') {
|
||||
if (rec.kind === 'ssh') {
|
||||
const m = this.machines().find((x) => x.id === rec.machine);
|
||||
if (m) {
|
||||
try {
|
||||
await sshKillWorker(m, rec.pid, instanceId);
|
||||
} catch {
|
||||
/* ignore */
|
||||
}
|
||||
}
|
||||
} else if (rec.kind === 'agent') {
|
||||
const m = this.machines().find((x) => x.id === rec.machine);
|
||||
if (m) {
|
||||
try {
|
||||
|
|
@ -199,7 +257,7 @@ export class Supervisor {
|
|||
async pauseInstance(instanceId, { replace = true } = {}) {
|
||||
const rec = this.instances.get(instanceId);
|
||||
if (!rec) throw new Error(`no instance ${instanceId}`);
|
||||
await postControl({ host: rec.host || '127.0.0.1', port: rec.healthPort, path: '/pause' });
|
||||
await this.controlInstance(rec, '/pause');
|
||||
rec.paused = true;
|
||||
this.log('pause', { instance: instanceId, service: rec.service });
|
||||
if (replace) await this.reconcile(rec.service);
|
||||
|
|
@ -209,7 +267,7 @@ export class Supervisor {
|
|||
async resumeInstance(instanceId) {
|
||||
const rec = this.instances.get(instanceId);
|
||||
if (!rec) throw new Error(`no instance ${instanceId}`);
|
||||
await postControl({ host: rec.host || '127.0.0.1', port: rec.healthPort, path: '/resume' });
|
||||
await this.controlInstance(rec, '/resume');
|
||||
rec.paused = false;
|
||||
this.log('resume', { instance: instanceId, service: rec.service });
|
||||
return rec;
|
||||
|
|
@ -229,25 +287,36 @@ export class Supervisor {
|
|||
async markUnhealthy(instanceId) {
|
||||
const rec = this.instances.get(instanceId);
|
||||
if (!rec) throw new Error(`no instance ${instanceId}`);
|
||||
await postControl({ host: rec.host || '127.0.0.1', port: rec.healthPort, path: '/unhealthy' });
|
||||
await this.controlInstance(rec, '/unhealthy');
|
||||
return rec;
|
||||
}
|
||||
|
||||
async probe(rec) {
|
||||
const spec = this.spec(rec.service);
|
||||
const host = rec.host || '127.0.0.1';
|
||||
const r = await httpProbe({
|
||||
host,
|
||||
port: rec.healthPort,
|
||||
path: spec.health?.path || '/health',
|
||||
timeoutMs: spec.health?.timeoutMs || 800,
|
||||
});
|
||||
const pathName = spec.health?.path || '/health';
|
||||
let r;
|
||||
if (rec.kind === 'ssh') {
|
||||
const m = this.machineOf(rec);
|
||||
try {
|
||||
const body = await sshHttp(m, 'GET', pathName, { port: rec.healthPort });
|
||||
r = { ok: body.ok !== false, paused: Boolean(body.paused), body, statusCode: body.ok === false ? 503 : 200 };
|
||||
} catch (err) {
|
||||
r = { ok: false, paused: false, error: err.message };
|
||||
}
|
||||
} else {
|
||||
r = await httpProbe({
|
||||
host: rec.host || '127.0.0.1',
|
||||
port: rec.healthPort,
|
||||
path: pathName,
|
||||
timeoutMs: spec.health?.timeoutMs || 800,
|
||||
});
|
||||
}
|
||||
rec.paused = Boolean(r.paused);
|
||||
rec.lastProbe = r;
|
||||
rec.healthy = r.ok;
|
||||
if (r.ok && !rec.paused) {
|
||||
try {
|
||||
const msg = await postControl({ host, port: rec.healthPort, path: '/message' });
|
||||
const msg = await this.controlInstance(rec, '/message');
|
||||
if (Number.isFinite(msg.rttMs)) {
|
||||
rec.rttSamples = rec.rttSamples || [];
|
||||
rec.rttSamples.push(msg.rttMs);
|
||||
|
|
@ -415,12 +484,27 @@ function waitExit(child, ms) {
|
|||
});
|
||||
}
|
||||
|
||||
async function waitForHealth(host, port, ms) {
|
||||
async function waitForHealth(rec, ms) {
|
||||
const deadline = Date.now() + ms;
|
||||
while (Date.now() < deadline) {
|
||||
const r = await httpProbe({ host: host || '127.0.0.1', port, path: '/health', timeoutMs: 300 });
|
||||
let r;
|
||||
if (rec.kind === 'ssh' && rec._machine) {
|
||||
try {
|
||||
const body = await sshHttp(rec._machine, 'GET', '/health', { port: rec.healthPort });
|
||||
r = { ok: body.ok !== false, statusCode: 200, paused: Boolean(body.paused), body };
|
||||
} catch {
|
||||
r = { ok: false };
|
||||
}
|
||||
} else {
|
||||
r = await httpProbe({
|
||||
host: rec.host || '127.0.0.1',
|
||||
port: rec.healthPort,
|
||||
path: '/health',
|
||||
timeoutMs: 400,
|
||||
});
|
||||
}
|
||||
if (r.ok || r.statusCode === 503) return r;
|
||||
await new Promise((r) => setTimeout(r, 50));
|
||||
await new Promise((res) => setTimeout(res, 80));
|
||||
}
|
||||
return { ok: false, error: 'start-timeout' };
|
||||
}
|
||||
|
|
|
|||
56
packages/verae-fleet/test/ssh.test.js
Normal file
56
packages/verae-fleet/test/ssh.test.js
Normal file
|
|
@ -0,0 +1,56 @@
|
|||
import { describe, it } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import fs from 'node:fs';
|
||||
import os from 'node:os';
|
||||
import path from 'node:path';
|
||||
import { expandHome, shellQuote, sshBaseArgs, sshTarget } from '../src/ssh.js';
|
||||
import { normalizeMachine, pickMachine } from '../src/machines.js';
|
||||
|
||||
describe('SSH remote machine config', () => {
|
||||
it('expands ~ in identityFile and quotes remote commands', () => {
|
||||
assert.equal(expandHome('~/foo'), `${os.homedir()}/foo`);
|
||||
assert.equal(shellQuote("a'b"), `'a'"'"'b'`);
|
||||
});
|
||||
|
||||
it('builds ssh argv with user, host, port, and key path (not key bytes)', () => {
|
||||
const key = path.join(os.tmpdir(), 'verae-fleet-test-id');
|
||||
fs.writeFileSync(key, 'not-a-real-key\n', { mode: 0o600 });
|
||||
const m = normalizeMachine({
|
||||
id: 'ns1',
|
||||
kind: 'ssh',
|
||||
host: '70.88.205.138',
|
||||
user: 'marchon',
|
||||
sshPort: 22,
|
||||
identityFile: key,
|
||||
});
|
||||
assert.equal(m.kind, 'ssh');
|
||||
assert.equal(sshTarget(m), 'marchon@70.88.205.138');
|
||||
const args = sshBaseArgs(m);
|
||||
assert.ok(args.includes('-i'));
|
||||
assert.ok(args.includes(key));
|
||||
assert.ok(!args.some((a) => a.includes('BEGIN')));
|
||||
assert.equal(m.user, 'marchon');
|
||||
fs.unlinkSync(key);
|
||||
});
|
||||
|
||||
it('places load on ssh hosts when they have free capacity', () => {
|
||||
const machines = [
|
||||
normalizeMachine({ id: 'local', kind: 'local', host: '127.0.0.1', capacity: 2, roles: ['*'] }),
|
||||
normalizeMachine({
|
||||
id: 'ns1',
|
||||
kind: 'ssh',
|
||||
host: '70.88.205.138',
|
||||
user: 'marchon',
|
||||
identityFile: '~/.ssh/id_ed25519',
|
||||
capacity: 8,
|
||||
roles: ['tree-node'],
|
||||
}),
|
||||
];
|
||||
const instances = [
|
||||
{ machine: 'local', pid: 1 },
|
||||
{ machine: 'local', pid: 2 },
|
||||
];
|
||||
const pick = pickMachine(machines, instances, 'tree-node', 'tree-node');
|
||||
assert.equal(pick.id, 'ns1');
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue