@sonamu-kit/tasks 0.1.3 → 0.3.0

package/src/core/step.ts

@@ -1,7 +1,7 @@
-import type { DurationString } from "./duration";
+import { type DurationString } from "./duration";
 import { parseDuration } from "./duration";
-import type { JsonValue } from "./json";
-import type { Result } from "./result";
+import { type JsonValue } from "./json";
+import { type Result } from "./result";
 import { err, ok } from "./result";
 
 /**
@@ -14,6 +14,7 @@ export type StepKind = "function" | "sleep";
  */
 export type StepAttemptStatus =
   | "running"
+  | "paused"
   | "succeeded" // deprecated in favor of 'completed'
   | "completed"
   | "failed";

package/src/core/workflow.test.ts

@@ -1,5 +1,6 @@
 import { describe, expect, test } from "vitest";
-import type { StandardSchemaV1 } from "./schema";
+
+import { type StandardSchemaV1 } from "./schema";
 import { DEFAULT_WORKFLOW_RESULT_CONFIG, isTerminalStatus, validateInput } from "./workflow";
 
 describe("validateInput", () => {

package/src/core/workflow.ts

@@ -1,6 +1,6 @@
-import type { SerializedError } from "./error";
-import type { JsonValue } from "./json";
-import type { StandardSchemaV1 } from "./schema";
+import { type SerializedError } from "./error";
+import { type JsonValue } from "./json";
+import { type StandardSchemaV1 } from "./schema";
 
 /**
  * Status of a workflow run through its lifecycle.
@@ -9,6 +9,7 @@ export type WorkflowRunStatus =
   | "pending"
   | "running"
   | "sleeping"
+  | "paused"
   | "succeeded" // deprecated in favor of 'completed'
   | "completed"
   | "failed"

package/src/database/backend.test.ts

@@ -1,4 +1,5 @@
 import { randomUUID } from "node:crypto";
+
 import { KNEX_GLOBAL_CONFIG } from "../testing/connection";
 import { BackendPostgres } from "./backend";
 import { testBackend } from "./backend.testsuite";

package/src/database/backend.testsuite.ts

@@ -1,8 +1,11 @@
 import { randomUUID } from "node:crypto";
+
 import { afterAll, beforeAll, describe, expect, test } from "vitest";
-import type { Backend } from "..//backend";
-import type { StepAttempt } from "../core/step";
-import type { WorkflowRun } from "../core/workflow";
+
+import { type Backend } from "..//backend";
+import { type SerializableRetryPolicy } from "../core/retry";
+import { type StepAttempt } from "../core/step";
+import { type WorkflowRun } from "../core/workflow";
 
 /**
  * Options for the Backend test suite.
@@ -137,7 +140,7 @@ export function testBackend(options: TestBackendOptions): void {
         // p2
         const page2 = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page1.pagination.next!,
         });
         expect(page2.data).toHaveLength(2);
@@ -149,7 +152,7 @@ export function testBackend(options: TestBackendOptions): void {
         // p3
         const page3 = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page2.pagination.next!,
         });
         expect(page3.data).toHaveLength(1);
@@ -160,7 +163,7 @@ export function testBackend(options: TestBackendOptions): void {
         // p2 again
         const page2Back = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           before: page3.pagination.prev!,
         });
         expect(page2Back.data).toHaveLength(2);
@@ -202,7 +205,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const page2 = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page1.pagination.next!,
         });
         expect(page2.data).toHaveLength(2);
@@ -212,7 +215,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const page3 = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page2.pagination.next!,
         });
         expect(page3.data).toHaveLength(1);
@@ -221,7 +224,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const page2Back = await backend.listWorkflowRuns({
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           before: page3.pagination.prev!,
         });
         expect(page2Back.data).toHaveLength(2);
@@ -532,6 +535,124 @@ export function testBackend(options: TestBackendOptions): void {
 
         await teardown(backend);
       });
+
+      test("marks workflow run as failed when maxAttempts is reached", async () => {
+        const backend = await setup();
+
+        // retryPolicy에 maxAttempts: 2를 지정하여 생성
+        const retryPolicy: SerializableRetryPolicy = {
+          maxAttempts: 2,
+          initialIntervalMs: 100,
+        };
+        await backend.createWorkflowRun({
+          workflowName: randomUUID(),
+          version: null,
+          idempotencyKey: null,
+          input: null,
+          config: {},
+          context: null,
+          availableAt: null,
+          deadlineAt: null,
+          retryPolicy,
+        });
+
+        // 첫 번째 시도 - 실패하면 pending으로 스케줄링
+        let workerId = randomUUID();
+        let claimed = await backend.claimWorkflowRun({
+          workerId,
+          leaseDurationMs: 100,
+        });
+        if (!claimed) throw new Error("Expected workflow run to be claimed");
+        expect(claimed.attempts).toBe(1);
+
+        const firstFailed = await backend.failWorkflowRun({
+          workflowRunId: claimed.id,
+          workerId,
+          error: { message: "first failure" },
+        });
+        expect(firstFailed.status).toBe("pending"); // 아직 maxAttempts(2) 미달
+
+        await sleep(150); // 100ms backoff 대기
+
+        // 두 번째 시도 - maxAttempts에 도달하면 failed로 종료
+        workerId = randomUUID();
+        claimed = await backend.claimWorkflowRun({
+          workerId,
+          leaseDurationMs: 100,
+        });
+        if (!claimed) throw new Error("Expected workflow run to be claimed");
+        expect(claimed.attempts).toBe(2);
+
+        const secondFailed = await backend.failWorkflowRun({
+          workflowRunId: claimed.id,
+          workerId,
+          error: { message: "second failure" },
+        });
+
+        // maxAttempts에 도달했으므로 failed로 종료
+        expect(secondFailed.status).toBe("failed");
+        expect(secondFailed.availableAt).toBeNull();
+        expect(secondFailed.finishedAt).not.toBeNull();
+
+        await teardown(backend);
+      });
+
+      test("marks workflow run as failed immediately when forceComplete is true", async () => {
+        const backend = await setup();
+
+        await createPendingWorkflowRun(backend);
+
+        const workerId = randomUUID();
+        const claimed = await backend.claimWorkflowRun({
+          workerId,
+          leaseDurationMs: 100,
+        });
+        if (!claimed) throw new Error("Expected workflow run to be claimed");
+
+        // forceComplete: true로 호출하면 재시도 없이 즉시 failed
+        const failed = await backend.failWorkflowRun({
+          workflowRunId: claimed.id,
+          workerId,
+          error: { message: "forced failure" },
+          forceComplete: true,
+        });
+
+        expect(failed.status).toBe("failed");
+        expect(failed.availableAt).toBeNull();
+        expect(failed.finishedAt).not.toBeNull();
+
+        await teardown(backend);
+      });
+
+      test("stores retryPolicy in config when creating workflow run", async () => {
+        const backend = await setup();
+
+        const retryPolicy: SerializableRetryPolicy = {
+          maxAttempts: 10,
+          initialIntervalMs: 500,
+          backoffCoefficient: 1.5,
+          maximumIntervalMs: 30000,
+        };
+
+        const created = await backend.createWorkflowRun({
+          workflowName: randomUUID(),
+          version: null,
+          idempotencyKey: null,
+          input: null,
+          config: { existingKey: "existingValue" },
+          context: null,
+          availableAt: null,
+          deadlineAt: null,
+          retryPolicy,
+        });
+
+        // config에 retryPolicy가 저장되어 있는지 확인
+        const config = created.config as Record<string, unknown>;
+        expect(config.existingKey).toBe("existingValue");
+        expect(config.retryPolicy).toEqual(retryPolicy);
+
+        await teardown(backend);
+      });
     });
 
     describe("createStepAttempt()", () => {
@@ -559,7 +680,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const created = await backend.createStepAttempt({
           workflowRunId: expected.workflowRunId,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: workflowRun.workerId!,
           stepName: expected.stepName,
           kind: expected.kind,
@@ -585,7 +706,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const created = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -614,7 +735,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const first = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -624,7 +745,7 @@ export function testBackend(options: TestBackendOptions): void {
         await backend.completeStepAttempt({
           workflowRunId: claimed.id,
           stepAttemptId: first.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           output: { ok: true },
         });
@@ -633,7 +754,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const second = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -654,7 +775,7 @@ export function testBackend(options: TestBackendOptions): void {
         for (let i = 0; i < 5; i++) {
           await backend.createStepAttempt({
             workflowRunId: claimed.id,
-            // biome-ignore lint/style/noNonNullAssertion: for test
+            // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
             workerId: claimed.workerId!,
             stepName: `step-${String(i)}`,
             kind: "function",
@@ -680,7 +801,7 @@ export function testBackend(options: TestBackendOptions): void {
         const page2 = await backend.listStepAttempts({
           workflowRunId: claimed.id,
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page1.pagination.next!,
         });
         expect(page2.data).toHaveLength(2);
@@ -693,7 +814,7 @@ export function testBackend(options: TestBackendOptions): void {
         const page3 = await backend.listStepAttempts({
           workflowRunId: claimed.id,
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           after: page2.pagination.next!,
         });
         expect(page3.data).toHaveLength(1);
@@ -705,7 +826,7 @@ export function testBackend(options: TestBackendOptions): void {
         const page2Back = await backend.listStepAttempts({
           workflowRunId: claimed.id,
           limit: 2,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           before: page3.pagination.prev!,
         });
         expect(page2Back.data).toHaveLength(2);
@@ -729,7 +850,7 @@ export function testBackend(options: TestBackendOptions): void {
         const claimed = await createClaimedWorkflowRun(backend);
         await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: "step-1",
           kind: "function",
@@ -761,7 +882,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const created = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -773,15 +894,16 @@ export function testBackend(options: TestBackendOptions): void {
         const completed = await backend.completeStepAttempt({
           workflowRunId: claimed.id,
           stepAttemptId: created.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           output,
         });
 
-        expect(completed.status).toBe("completed");
-        expect(completed.output).toEqual(output);
-        expect(completed.error).toBeNull();
-        expect(completed.finishedAt).not.toBeNull();
+        expect(completed).not.toBeNull();
+        expect(completed?.status).toBe("completed");
+        expect(completed?.output).toEqual(output);
+        expect(completed?.error).toBeNull();
+        expect(completed?.finishedAt).not.toBeNull();
 
         const fetched = await backend.getStepAttempt({
           stepAttemptId: created.id,
@@ -805,7 +927,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const stepAttempt = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -816,7 +938,7 @@ export function testBackend(options: TestBackendOptions): void {
         // complete the workflow so it's no longer running
         await backend.completeWorkflowRun({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           output: null,
         });
@@ -826,7 +948,7 @@ export function testBackend(options: TestBackendOptions): void {
           backend.completeStepAttempt({
             workflowRunId: claimed.id,
             stepAttemptId: stepAttempt.id,
-            // biome-ignore lint/style/noNonNullAssertion: for test
+            // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
             workerId: claimed.workerId!,
             output: { foo: "bar" },
           }),
@@ -843,7 +965,7 @@ export function testBackend(options: TestBackendOptions): void {
           backend.completeStepAttempt({
             workflowRunId: claimed.id,
             stepAttemptId: randomUUID(),
-            // biome-ignore lint/style/noNonNullAssertion: for test
+            // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
             workerId: claimed.workerId!,
             output: { foo: "bar" },
           }),
@@ -867,7 +989,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const created = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -879,15 +1001,16 @@ export function testBackend(options: TestBackendOptions): void {
         const failed = await backend.failStepAttempt({
           workflowRunId: claimed.id,
           stepAttemptId: created.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           error,
         });
 
-        expect(failed.status).toBe("failed");
-        expect(failed.error).toEqual(error);
-        expect(failed.output).toBeNull();
-        expect(failed.finishedAt).not.toBeNull();
+        expect(failed).not.toBeNull();
+        expect(failed?.status).toBe("failed");
+        expect(failed?.error).toEqual(error);
+        expect(failed?.output).toBeNull();
+        expect(failed?.finishedAt).not.toBeNull();
 
         const fetched = await backend.getStepAttempt({
           stepAttemptId: created.id,
@@ -911,7 +1034,7 @@ export function testBackend(options: TestBackendOptions): void {
 
         const stepAttempt = await backend.createStepAttempt({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           stepName: randomUUID(),
           kind: "function",
@@ -922,7 +1045,7 @@ export function testBackend(options: TestBackendOptions): void {
         // complete the workflow so it's no longer running
         await backend.completeWorkflowRun({
           workflowRunId: claimed.id,
-          // biome-ignore lint/style/noNonNullAssertion: for test
+          // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
           workerId: claimed.workerId!,
           output: null,
         });
@@ -932,7 +1055,7 @@ export function testBackend(options: TestBackendOptions): void {
           backend.failStepAttempt({
             workflowRunId: claimed.id,
             stepAttemptId: stepAttempt.id,
-            // biome-ignore lint/style/noNonNullAssertion: for test
+            // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
             workerId: claimed.workerId!,
             error: { message: "nope" },
           }),
@@ -949,7 +1072,7 @@ export function testBackend(options: TestBackendOptions): void {
           backend.failStepAttempt({
             workflowRunId: claimed.id,
             stepAttemptId: randomUUID(),
-            // biome-ignore lint/style/noNonNullAssertion: for test
+            // oxlint-disable-next-line @typescript-eslint/no-non-null-assertion -- for test
             workerId: claimed.workerId!,
             error: { message: "nope" },
           }),