Merge pull request #60 from saving-satoshi/kill-unused-container

Cron Job to klll and remove unused containers every 10 minutes

Author: 0tuedon

.env.example

@@ -3,4 +3,5 @@ PORT=8000
 SECRET=12345678
 WHITELIST=http://localhost:3000
 MAX_SCRIPT_EXECUTION_TIME=30000
-DATABASE_URL="postgresql://postgres:postgres@localhost:5432/postgres?schema=public"
+DATABASE_URL="postgresql://postgres:postgres@localhost:5432/postgres?schema=public"
+CONTAINER_TERMINATION_SCHEDULE= 10 # Scheduled Time to clear your container in minutes 1<=59

.gitignore

@@ -1,3 +1,4 @@
 node_modules
 .env
 dist
+logs

package.json

@@ -32,6 +32,7 @@
   "dependencies": {
     "@prisma/client": "5.2.0",
     "@types/dockerode": "^3.3.31",
+    "@types/node-cron": "^3.0.11",
     "@types/uuid": "^10.0.0",
     "ansi-to-html": "^0.7.2",
     "body-parser": "^1.20.1",
@@ -44,6 +45,7 @@
     "jsonwebtoken": "^9.0.0",
     "lodash": "^4.17.21",
     "module-alias": "^2.2.2",
+    "node-cron": "^3.0.3",
     "pg": "^8.9.0",
     "uuid": "^9.0.0",
     "winston": "^3.16.0",

src/config.ts

@@ -1,4 +1,6 @@
 import path from 'path'
 
 export const SUPPORTED_LANGUAGES = ['javascript', 'python', 'go', 'rust', 'cpp']
+export const CONTAINERS_TO_KEEP_ON = ['/saving-satoshi']
+export const CONTAINERS_SCHEDULE =  parseInt(process.env.CONTAINER_TERMINATION_SCHEDULE || "10")
 export const LANG_PATH = path.join(__dirname, 'languages')

src/index.ts

@@ -7,16 +7,16 @@ import * as http from 'http'
 import * as WebSocket from 'ws'
 import * as repl from 'lib/repl'
 import logger from 'lib/logger'
+import cron from 'node-cron'
 
 import { v1 } from './routes'
 
 import { JobManager } from './lib/jobManager'
-
-const jobManager = new JobManager()
+import { killContainers } from 'lib/docker'
+import { CONTAINERS_SCHEDULE } from 'config'
 
 const port = process.env.PORT
 
-const JOBS = {}
 const WHITELIST = process.env.WHITELIST.split(',').map((a) => a.trim())
 
 function isAllowedOrigin(origin: string): boolean {
@@ -38,6 +38,24 @@ async function run() {
   const server = http.createServer(app)
   const wss = new WebSocket.Server({ server })
 
+// Cron Job Wrapper
+  function scheduleRemovalOfContainers(
+    expression: string,
+    fn: () => Promise<void>,
+    options = {}
+  ) {
+    cron.schedule(
+      expression,
+      () => fn().catch((e) => logger.error(e)),
+      options
+    )
+  }
+
+  // Cron job to run every specified minutes
+  scheduleRemovalOfContainers(`*/${CONTAINERS_SCHEDULE} * * * *`, async () => {
+    await killContainers()
+  })
+
   // WebSocket connection handler
   wss.on(
     'connection',

src/lib/docker.ts

@@ -3,6 +3,7 @@ import Docker from 'dockerode'
 import type { Container } from 'dockerode'
 import logger from './logger'
 import { JobManager } from './jobManager'
+import { CONTAINERS_SCHEDULE, CONTAINERS_TO_KEEP_ON } from 'config'
 
 export const docker = new Docker()
 
@@ -67,6 +68,31 @@ async function buildImage(
   }
 }
 
+export async function killContainers() {
+  const allContainers = await docker.listContainers()
+  let containersKilled = 0
+  for (let container of allContainers) {
+    if (container.Status) {
+      const stringTime = parseInt(container.Status.split(' ')[1])
+      const minutes = container.Status.split(' ')[2]
+      if (minutes === 'minutes' && stringTime >= CONTAINERS_SCHEDULE) {
+        const isSafeContainer = CONTAINERS_TO_KEEP_ON.includes(
+          container.Names[0]
+        )
+        // So our database migration doesn't get killed
+        if (!isSafeContainer) {
+          await docker.getContainer(container.Id).kill()
+          await docker.getContainer(container.Id).remove()
+          containersKilled++
+        }
+      }
+    }
+  }
+  if (containersKilled) {
+    logger.info(`${containersKilled} containers were killed`)
+  }
+}
+
 function sanitizeContainerId(id: string): string {
   return id.slice(0, 8)
 }