Working query
All checks were successful
Lint / lint (push) Successful in 13s

This commit is contained in:
Julien Valverdé
2025-03-04 01:19:42 +01:00
parent 40e8bf6a1f
commit b2f1626268
2 changed files with 63 additions and 51 deletions

View File

@@ -5,11 +5,11 @@ import { Effect, Fiber, identity, Option, Ref, SubscriptionRef } from "effect"
export interface QueryRunner<A, E, R> {
readonly queryRef: SubscriptionRef.SubscriptionRef<Effect.Effect<A, E, R>>
readonly stateRef: SubscriptionRef.SubscriptionRef<AsyncData.AsyncData<A, E>>
readonly fiberRef: SubscriptionRef.SubscriptionRef<Option.Option<Fiber.Fiber<void>>>
readonly fiberRef: SubscriptionRef.SubscriptionRef<Option.Option<Fiber.RuntimeFiber<void>>>
readonly interrupt: Effect.Effect<void>
readonly forkFetch: Effect.Effect<void>
readonly forkRefetch: Effect.Effect<void>
readonly forkInterrupt: Effect.Effect<Fiber.RuntimeFiber<void>>
readonly forkFetch: Effect.Effect<Fiber.RuntimeFiber<void>>
readonly forkRefresh: Effect.Effect<Fiber.RuntimeFiber<void>>
}
@@ -20,66 +20,80 @@ export const make = <A, E, R>(
const queryRef = yield* SubscriptionRef.make(query)
const stateRef = yield* SubscriptionRef.make(AsyncData.noData<A, E>())
const fiberRef = yield* SubscriptionRef.make(Option.none<Fiber.Fiber<void>>())
const fiberRef = yield* SubscriptionRef.make(Option.none<Fiber.RuntimeFiber<void>>())
const interrupt = fiberRef.pipe(
Effect.flatMap(Option.match({
onSome: fiber => Ref.set(fiberRef, Option.none()).pipe(
Effect.andThen(Fiber.interrupt(fiber))
),
onSome: Fiber.interrupt,
onNone: () => Effect.void,
}))
)
const forkInterrupt = fiberRef.pipe(
Effect.flatMap(Option.match({
onSome: fiber => Ref.set(fiberRef, Option.none()).pipe(
Effect.andThen(Fiber.interruptFork(fiber))
),
onNone: () => Effect.void,
}))
)
const forkInterrupt = Effect.forkDaemon(interrupt)
const forkFetch = interrupt.pipe(
Effect.andThen(Ref.set(stateRef, AsyncData.loading())),
Effect.andThen(queryRef.pipe(Effect.flatMap(identity))),
Effect.matchCauseEffect({
onSuccess: v => Ref.set(stateRef, AsyncData.success(v)),
onFailure: c => Ref.set(stateRef, AsyncData.failure(c)),
}),
Effect.andThen(Ref.set(fiberRef, Option.none())),
Effect.provide(context),
Effect.forkDaemon,
Effect.andThen(
Effect.addFinalizer(() => Ref.set(fiberRef, Option.none())).pipe(
Effect.andThen(Ref.set(stateRef, AsyncData.loading())),
Effect.andThen(queryRef.pipe(Effect.flatMap(identity))),
Effect.matchCauseEffect({
onSuccess: v => Ref.set(stateRef, AsyncData.success(v)),
onFailure: c => Ref.set(stateRef, AsyncData.failure(c)),
}),
Effect.flatMap(fiber => Ref.set(fiberRef, Option.some(fiber))),
Effect.provide(context),
Effect.scoped,
Effect.fork,
)
),
Effect.flatMap(fiber =>
Ref.set(fiberRef, Option.some(fiber)).pipe(
Effect.andThen(Fiber.join(fiber))
)
),
Effect.forkDaemon,
)
const forkRefetch = interrupt.pipe(
Effect.andThen(Ref.update(stateRef, previous => {
if (AsyncData.isSuccess(previous) || AsyncData.isFailure(previous))
return AsyncData.refreshing(previous)
if (AsyncData.isRefreshing(previous))
return AsyncData.refreshing(previous.previous)
return AsyncData.loading()
})),
Effect.andThen(queryRef.pipe(Effect.flatMap(identity))),
Effect.matchCauseEffect({
onSuccess: v => Ref.set(stateRef, AsyncData.success(v)),
onFailure: c => Ref.set(stateRef, AsyncData.failure(c)),
}),
Effect.andThen(Ref.set(fiberRef, Option.none())),
Effect.provide(context),
Effect.forkDaemon,
const forkRefresh = interrupt.pipe(
Effect.andThen(
Effect.addFinalizer(() => Ref.set(fiberRef, Option.none())).pipe(
Effect.andThen(Ref.update(stateRef, previous => {
if (AsyncData.isSuccess(previous) || AsyncData.isFailure(previous))
return AsyncData.refreshing(previous)
if (AsyncData.isRefreshing(previous))
return AsyncData.refreshing(previous.previous)
return AsyncData.loading()
})),
Effect.andThen(queryRef.pipe(Effect.flatMap(identity))),
Effect.matchCauseEffect({
onSuccess: v => Ref.set(stateRef, AsyncData.success(v)),
onFailure: c => Ref.set(stateRef, AsyncData.failure(c)),
}),
Effect.flatMap(fiber => Ref.set(fiberRef, Option.some(fiber))),
Effect.provide(context),
Effect.scoped,
Effect.fork,
)
),
Effect.flatMap(fiber =>
Ref.set(fiberRef, Option.some(fiber)).pipe(
Effect.andThen(Fiber.join(fiber))
)
),
Effect.forkDaemon,
)
return {
queryRef,
stateRef,
fiberRef,
interrupt,
forkInterrupt,
forkFetch,
forkRefetch,
forkRefresh,
}
})