github/oneuptime
1
1
Fork 0
mirror of https://github.com/OneUptime/oneuptime.git synced 2026-04-06 00:32:12 +02:00
Code Issues 193 Packages Projects Releases Wiki Activity

Refactor InfrastructureAgent/Index.ts and update daemon management logic

Browse Source
This commit is contained in:
Simon Larsen
2024-04-08 20:09:20 +01:00
parent bcefa8ae3a
commit 39d7b76e23
1 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
InfrastructureAgent/Index.ts
View File

@@ -1,7 +1,7 @@
#!/usr/bin/env tsx
import yargs from 'yargs';
import { ChildProcess, spawn } from 'node:child_process';
import { spawn } from 'node:child_process';
import Logger from './Utils/Logger';
import PackageJson from './package.json';
import MonitorInfrastructure from './Jobs/MonitorInfrastructure';
@@ -15,8 +15,6 @@ export interface ArgumentType {
$0: string;
}
let daemon: ChildProcess | null = null;
const usage: string =
'\nUsage: oneuptime-infrastructure-agent start --secret-key <secret-key>.';
@@ -101,11 +99,14 @@ const returnValue:
fs.writeFileSync('./out.log', '');
fs.writeFileSync('./err.log', '');
daemon = spawn('tsx', startArguments, {
const daemon = spawn('tsx', startArguments, {
detached: true,
stdio: ['ignore', out, err],
});
// save the process id to a file
fs.writeFileSync('./daemon.pid', daemon?.pid?.toString() || '');
daemon.unref();
Logger.info('OneUptime Infrastructure Agent started as daemon');
} catch (err) {
@@ -119,8 +120,21 @@ const returnValue:
'Stop the daemon',
() => {},
() => {
if (daemon && daemon.pid) {
process.kill(daemon.pid);
// read the pid from file
let pid: string | number = fs
.readFileSync('./daemon.pid', 'utf-8')
.trim();
if (pid) {
if (typeof pid === 'string') {
pid = parseInt(pid);
}
process.kill(pid);
// remove the pid file
fs.unlinkSync('./daemon.pid');
Logger.info('OneUptime Infrastructure Agent stopped');
} else {
Logger.info('OneUptime Infrastructure Agent not running');
@@ -132,7 +146,11 @@ const returnValue:
'Show status of daemon',
() => {},
() => {
if (daemon && daemon.pid) {
const pid: string | number = fs
.readFileSync('./daemon.pid', 'utf-8')
.trim();
if (pid) {
Logger.info('OneUptime Infrastructure Agent is running');
} else {
Logger.info('OneUptime Infrastructure Agent is not running');
@@ -146,7 +164,11 @@ const returnValue:
() => {
// show logs from daemon
if (daemon && daemon.pid) {
const pid: string | number = fs
.readFileSync('./daemon.pid', 'utf-8')
.trim();
if (pid) {
const logs: string = fs.readFileSync('./out.log', 'utf-8');
Logger.info(logs);
} else {