|
| 1 | +import { describe, it } from "@effect/vitest" |
| 2 | +import { assertTrue } from "@effect/vitest/utils" |
| 3 | +import * as Effect from "effect/Effect" |
| 4 | +import * as Exit from "effect/Exit" |
| 5 | +import * as Fiber from "effect/Fiber" |
| 6 | +import * as Request from "effect/Request" |
| 7 | +import * as RequestResolver from "effect/RequestResolver" |
| 8 | + |
| 9 | +class GetValue extends Request.TaggedClass("GetValue")<string, never, { readonly id: number }> {} |
| 10 | + |
| 11 | +describe("batched resolver defect", () => { |
| 12 | + // When a batched resolver dies with a defect, the request Deferreds are |
| 13 | + // never completed and consumers hang forever. The cleanup that completes |
| 14 | + // uncompleted entries only runs on success (flatMap/OP_ON_SUCCESS) but |
| 15 | + // should run on all exits. |
| 16 | + it.live("resolver defect should not hang consumers", () => |
| 17 | + Effect.gen(function*() { |
| 18 | + const resolver = RequestResolver.makeBatched((_requests: Array<GetValue>) => Effect.die("boom")) |
| 19 | + |
| 20 | + const fiber = yield* Effect.request(new GetValue({ id: 1 }), resolver).pipe( |
| 21 | + Effect.fork |
| 22 | + ) |
| 23 | + |
| 24 | + // Wait briefly then check if the fiber completed. |
| 25 | + // If the bug is present, the fiber hangs on deferredAwait forever. |
| 26 | + yield* Effect.sleep("500 millis") |
| 27 | + const poll = yield* Fiber.poll(fiber) |
| 28 | + |
| 29 | + assertTrue( |
| 30 | + poll._tag === "Some", |
| 31 | + "Fiber should have completed — resolver defect must not leave consumers hanging" |
| 32 | + ) |
| 33 | + |
| 34 | + if (poll._tag === "Some") { |
| 35 | + assertTrue(Exit.isFailure(poll.value)) |
| 36 | + } |
| 37 | + })) |
| 38 | + |
| 39 | + it.live("resolver defect should not hang multiple consumers", () => |
| 40 | + Effect.gen(function*() { |
| 41 | + const resolver = RequestResolver.makeBatched((_requests: Array<GetValue>) => Effect.die("boom")) |
| 42 | + |
| 43 | + const fiber = yield* Effect.forEach( |
| 44 | + [1, 2, 3], |
| 45 | + (id) => Effect.request(new GetValue({ id }), resolver), |
| 46 | + { batching: true, concurrency: "unbounded" } |
| 47 | + ).pipe(Effect.fork) |
| 48 | + |
| 49 | + yield* Effect.sleep("500 millis") |
| 50 | + const poll = yield* Fiber.poll(fiber) |
| 51 | + |
| 52 | + assertTrue( |
| 53 | + poll._tag === "Some", |
| 54 | + "Fiber should have completed — resolver defect must not leave consumers hanging" |
| 55 | + ) |
| 56 | + |
| 57 | + if (poll._tag === "Some") { |
| 58 | + assertTrue(Exit.isFailure(poll.value)) |
| 59 | + } |
| 60 | + })) |
| 61 | +}) |
0 commit comments