Adding simulate graphile job to easily debug graceful shutdown retries in docker image

Author: ericallam

apps/webapp/app/routes/admin.api.v1.simulate.ts

@@ -0,0 +1,34 @@
+import { ActionArgs, json, redirect } from "@remix-run/server-runtime";
+import { prisma } from "~/db.server";
+import { authenticateApiRequest } from "~/services/apiAuth.server";
+import { workerQueue } from "~/services/worker.server";
+
+export async function action({ request }: ActionArgs) {
+  // Next authenticate the request
+  const authenticationResult = await authenticateApiRequest(request);
+
+  if (!authenticationResult) {
+    return json({ error: "Invalid or Missing API key" }, { status: 401 });
+  }
+
+  const adminOrgMembers = await prisma.orgMember.findMany({
+    where: {
+      organizationId: authenticationResult.environment.organizationId,
+      user: {
+        admin: true,
+      },
+    },
+  });
+
+  if (!adminOrgMembers.length) {
+    return json({ error: "You must be an admin to perform this action" }, { status: 403 });
+  }
+
+  const body: any = await request.json();
+
+  await workerQueue.enqueue("simulate", {
+    seconds: body.seconds,
+  });
+
+  return json({ success: true });
+}

apps/webapp/app/services/worker.server.ts

@@ -76,6 +76,9 @@ const workerCatalog = {
   connectionCreated: z.object({
     id: z.string(),
   }),
+  simulate: z.object({
+    seconds: z.number(),
+  }),
 };
 
 const executionWorkerCatalog = {
@@ -313,6 +316,12 @@ function getWorkerQueue() {
           });
         },
       },
+      simulate: {
+        maxAttempts: 5,
+        handler: async (payload, job) => {
+          await new Promise((resolve) => setTimeout(resolve, payload.seconds * 1000));
+        },
+      },
     },
   });
 }

docker/dev-compose.yml

@@ -9,7 +9,7 @@ networks:
 
 services:
   db:
-    container_name: db
+    container_name: devdb
     image: postgres:14
     restart: always
     volumes:
@@ -30,6 +30,8 @@ services:
       - 3030:3030
     depends_on:
       - db
+    env_file:
+      - ../.env
     environment:
       DATABASE_URL: postgres://postgres:postgres@db:5432/postgres?schema=public
       DIRECT_URL: postgres://postgres:postgres@db:5432/postgres?schema=public