add docker heartbeat

.gitignore

@@ -1,3 +1,3 @@
 node_modules
-*.js
+index.js
 *.tgz

heartbeat.js

@@ -0,0 +1,41 @@
+import { exec } from 'child_process';
+
+const pidInput = process.argv[2];
+const containerName = process.argv[3]
+let healthy = true
+let currentIntervalId
+
+function checkProcessStatus(pid) {
+  return new Promise((resolve) => {
+    exec(`ps -p ${pid}`, (error, stdout) => {
+      if (error) {
+        resolve({ pid, running: false });
+      } else {
+        const processRunning = stdout.includes(pid.toString());
+        resolve({ pid, running: processRunning });
+      }
+    });
+  });
+}
+
+async function monitorProcess(pid, container) {
+  try {
+    const status = await checkProcessStatus(pid);
+    if (!status.running){
+      exec(`docker stop ${container}`)
+      healthy = false
+      process.exit(0)
+    }
+  } catch (error) {
+    console.error('Error checking process status:', error);
+  }
+}
+
+currentIntervalId = setInterval(() => {
+  monitorProcess(parseInt(pidInput), containerName);
+}, 10000); // ping every 10 seconds
+process.stdout.write('\u0007')
+
+if (!healthy){
+  clearInterval(currentIntervalId);
+}

run.ts

@@ -30,6 +30,29 @@ type SearchEngineLauncherFunction<T = object> = (
   waitUntilStopped: () => Promise<void>
 }>
 
+async function launchListener({
+  parentPID,
+  containerId,
+}: {
+  parentPID: number
+  containerId: string
+}) {
+  const subprocess = await spawn(
+    'node',
+    [
+      './node_modules/@nasa-gcn/architect-plugin-search/heartbeat.js',
+      parentPID.toString(),
+      containerId,
+    ],
+    {
+      detached: true,
+      stdio: 'ignore',
+    }
+  )
+
+  subprocess.unref()
+}
+
 const launchBinary: SearchEngineLauncherFunction<{ bin: string }> = async ({
   bin,
   dataDir,
@@ -86,6 +109,7 @@ const launchDocker: SearchEngineLauncherFunction = async ({
   const stream = await container.attach({ stream: true, stderr: true })
   stream.pipe(process.stderr)
   await container.start()
+  await launchListener({ parentPID: process.ppid, containerId: container.id })
 
   return {
     async kill() {