add forked docker process

launchSearch.ts

@@ -0,0 +1,78 @@
+/*!
+ * Copyright © 2023 United States Government as represented by the
+ * Administrator of the National Aeronautics and Space Administration.
+ * All Rights Reserved.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import Dockerode from 'dockerode'
+
+const dataDir = process.argv[2]
+const logsDir = process.argv[3]
+const engine = process.argv[4]
+const port = process.argv[5]
+const options = process.argv[6].split(',')
+
+type Message = {
+  action: string
+}
+
+let dockerConatiner: Dockerode.Container
+
+async function launchDocker() {
+  const Image =
+    engine === 'elasticsearch'
+      ? 'elastic/elasticsearch:8.6.2'
+      : 'opensearchproject/opensearch:2.11.0'
+  console.log('Launching Docker container', Image)
+  const docker = new Dockerode()
+
+  const container = await docker.createContainer({
+    Env: [...options, 'path.data=/var/lib/search', 'path.logs=/var/log/search'],
+    HostConfig: {
+      AutoRemove: true,
+      Mounts: [
+        { Source: dataDir, Target: '/var/lib/search', Type: 'bind' },
+        { Source: logsDir, Target: '/var/log/search', Type: 'bind' },
+      ],
+      PortBindings: {
+        [`${port}/tcp`]: [{ HostIP: '127.0.0.1', HostPort: `${port}` }],
+      },
+    },
+    Image,
+  })
+  const stream = await container.attach({ stream: true, stderr: true })
+  stream.pipe(process.stderr)
+  await container.start()
+  dockerConatiner = container
+}
+
+async function stopping() {
+  if (process.send) process.send('containerStopped')
+}
+
+async function waiting() {
+  await dockerConatiner.wait()
+  stopping()
+}
+
+launchDocker()
+
+process.once('message', async (message: Message) => {
+  if (message.action === 'wait') {
+    waiting()
+  } else {
+    await dockerConatiner.kill()
+  }
+})
+
+process.on('SIGTERM', async () => {
+  await dockerConatiner.kill()
+  process.exit(0)
+})
+
+process.on('SIGINT', async () => {
+  await dockerConatiner.kill()
+  process.exit(0)
+})

package.json

@@ -14,11 +14,13 @@
   ],
   "type": "module",
   "files": [
-    "index.js"
+    "index.js",
+    "launchSearch.js"
   ],
   "scripts": {
     "prepare:husky": "husky install",
     "prepare:esbuild": "esbuild index.ts --bundle --packages=external --outfile=index.js --platform=node --format=esm --tree-shaking=true",
+    "prepare:docker": "esbuild launchSearch.ts --bundle --packages=external --outfile=launchSearch.js --platform=node --format=esm --tree-shaking=true",
     "prepare": "run-p prepare:*"
   },
   "engines": {

run.ts

@@ -14,8 +14,8 @@ import { mkdirP, temp } from './paths.js'
 import rimraf from 'rimraf'
 import { spawn, untilTerminated } from './processes.js'
 import type { SandboxEngine } from './engines.js'
-import Dockerode from 'dockerode'
 import { UnexpectedResolveError, neverResolve } from './promises.js'
+import { ForkOptions, fork } from 'child_process'
 
 type SearchEngineLauncherFunction<T = object> = (
   props: T & {
@@ -63,37 +63,34 @@ const launchDocker: SearchEngineLauncherFunction = async ({
   port,
   options,
 }) => {
-  const Image =
-    engine === 'elasticsearch'
-      ? 'elastic/elasticsearch:8.6.2'
-      : 'opensearchproject/opensearch:2.11.0'
-  console.log('Launching Docker container', Image)
-  const docker = new Dockerode()
-  const container = await docker.createContainer({
-    Env: [...options, 'path.data=/var/lib/search', 'path.logs=/var/log/search'],
-    HostConfig: {
-      AutoRemove: true,
-      Mounts: [
-        { Source: dataDir, Target: '/var/lib/search', Type: 'bind' },
-        { Source: logsDir, Target: '/var/log/search', Type: 'bind' },
-      ],
-      PortBindings: {
-        [`${port}/tcp`]: [{ HostIP: '127.0.0.1', HostPort: `${port}` }],
-      },
-    },
-    Image,
+  const subprocess = fork(
+    './node_modules/@nasa-gcn/architect-plugin-search/launchSearch.js',
+    [dataDir, logsDir, engine, port, options] as ForkOptions
+  )
+
+  subprocess.stdout?.on('data', (data) => {
+    console.log(`subprocess stdout: ${data}`)
+  })
+
+  subprocess.stderr?.on('data', (data) => {
+    console.error(`subprocess stderr: ${data}`)
+  })
+
+  const waitUntilStopped = new Promise<void>((resolve) => {
+    subprocess.on('message', (message) => {
+      if (message === 'containerStopped') {
+        resolve()
+      }
+    })
   })
-  const stream = await container.attach({ stream: true, stderr: true })
-  stream.pipe(process.stderr)
-  await container.start()
 
   return {
     async kill() {
       console.log('Killing Docker container')
-      await container.kill()
+      subprocess.send({ action: 'kill' })
     },
     async waitUntilStopped() {
-      await container.wait()
+      await waitUntilStopped
     },
   }
 }