replace all run engine find uniques with find first

Author: nicktrn

internal-packages/run-engine/src/engine/db/worker.ts

@@ -264,13 +264,10 @@ export async function getManagedWorkerFromCurrentlyPromotedDeployment(
   prisma: PrismaClientOrTransaction,
   environmentId: string
 ): Promise<WorkerDeploymentWithWorkerTasks | null> {
-  // TODO: fixme
-  const promotion = await prisma.workerDeploymentPromotion.findUnique({
+  const promotion = await prisma.workerDeploymentPromotion.findFirst({
     where: {
-      environmentId_label: {
-        environmentId,
-        label: CURRENT_DEPLOYMENT_LABEL,
-      },
+      environmentId,
+      label: CURRENT_DEPLOYMENT_LABEL,
     },
     include: {
       deployment: {

internal-packages/run-engine/src/engine/systems/batchSystem.ts

@@ -34,7 +34,7 @@ export class BatchSystem {
    */
   async #tryCompleteBatch({ batchId }: { batchId: string }) {
     return startSpan(this.$.tracer, "#tryCompleteBatch", async (span) => {
-      const batch = await this.$.prisma.batchTaskRun.findUnique({
+      const batch = await this.$.prisma.batchTaskRun.findFirst({
         select: {
           status: true,
           runtimeEnvironmentId: true,

internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts

@@ -139,12 +139,10 @@ export class RunAttemptSystem {
             throw new ServiceValidationError("Task run is not locked", 400);
           }
 
-          const queue = await prisma.taskQueue.findUnique({
+          const queue = await prisma.taskQueue.findFirst({
             where: {
-              runtimeEnvironmentId_name: {
-                runtimeEnvironmentId: environment.id,
-                name: taskRun.queue,
-              },
+              runtimeEnvironmentId: environment.id,
+              name: taskRun.queue,
             },
           });
 
@@ -1199,7 +1197,7 @@ export class RunAttemptSystem {
 
   async #getAuthenticatedEnvironmentFromRun(runId: string, tx?: PrismaClientOrTransaction) {
     const prisma = tx ?? this.$.prisma;
-    const taskRun = await prisma.taskRun.findUnique({
+    const taskRun = await prisma.taskRun.findFirst({
       where: {
         id: runId,
       },

internal-packages/run-engine/src/engine/systems/ttlSystem.ts

@@ -33,7 +33,7 @@ export class TtlSystem {
       }
 
       //only expire "PENDING" runs
-      const run = await prisma.taskRun.findUnique({ where: { id: runId } });
+      const run = await prisma.taskRun.findFirst({ where: { id: runId } });
 
       if (!run) {
         this.$.logger.debug("Could not find enqueued run to expire", {

internal-packages/run-engine/src/engine/systems/waitpointSystem.ts

@@ -159,12 +159,10 @@ export class WaitpointSystem {
     const prisma = tx ?? this.$.prisma;
 
     const existingWaitpoint = idempotencyKey
-      ? await prisma.waitpoint.findUnique({
+      ? await prisma.waitpoint.findFirst({
           where: {
-            environmentId_idempotencyKey: {
-              environmentId,
-              idempotencyKey,
-            },
+            environmentId,
+            idempotencyKey,
           },
         })
       : undefined;
@@ -241,12 +239,10 @@ export class WaitpointSystem {
     tags?: string[];
   }): Promise<{ waitpoint: Waitpoint; isCached: boolean }> {
     const existingWaitpoint = idempotencyKey
-      ? await this.$.prisma.waitpoint.findUnique({
+      ? await this.$.prisma.waitpoint.findFirst({
           where: {
-            environmentId_idempotencyKey: {
-              environmentId,
-              idempotencyKey,
-            },
+            environmentId,
+            idempotencyKey,
           },
         })
       : undefined;