diff --git a/packages/extension-query/src/index.ts b/packages/extension-query/src/index.ts index c5694cd..eff2bfd 100644 --- a/packages/extension-query/src/index.ts +++ b/packages/extension-query/src/index.ts @@ -15,6 +15,24 @@ export interface UseQueryResult { } +const interruptRunningQuery = (fiberRef: Ref.Ref>>) => fiberRef.pipe( + Effect.flatMap(Option.match({ + onSome: Fiber.interrupt, + onNone: () => Effect.void, + })) +) + +const runQuery = ( + effect: Effect.Effect, + stateRef: Ref.Ref>, +) => effect.pipe( + Effect.matchCauseEffect({ + onSuccess: v => Ref.set(stateRef, AsyncData.success(v)), + onFailure: c => Ref.set(stateRef, AsyncData.failure(c)), + }) +) + + export const QueryExtension = ReffuseExtension.make(() => ({ useQuery( this: ReffuseHelpers.ReffuseHelpers, @@ -22,45 +40,31 @@ export const QueryExtension = ReffuseExtension.make(() => ({ ): UseQueryResult { const context = this.useContext() - const fiberRef = this.useRef(Option.none>()) + const fiberRef = this.useRef(Option.none>()) const stateRef = this.useRef(AsyncData.noData()) - const interruptRunningQuery = React.useMemo(() => fiberRef.pipe( - Effect.flatMap(Option.match({ - onSome: Fiber.interrupt, - onNone: () => Effect.void, - })) - ), [fiberRef]) - - const runQuery = React.useMemo(() => props.effect().pipe( - Effect.matchCauseEffect({ - onSuccess: v => Ref.set(stateRef, AsyncData.success(v)), - onFailure: c => Ref.set(stateRef, AsyncData.failure(c)), - }) - ), [...props.deps, stateRef]) - - const triggerRefresh = React.useMemo(() => interruptRunningQuery.pipe( + const triggerRefresh = React.useMemo(() => interruptRunningQuery(fiberRef).pipe( Effect.andThen(Ref.update(stateRef, prev => AsyncData.isSuccess(prev) || AsyncData.isFailure(prev) ? AsyncData.refreshing(prev) : AsyncData.loading() )), - Effect.andThen(runQuery), + Effect.andThen(runQuery(props.effect(), stateRef)), Effect.provide(context), Effect.scoped, Effect.forkDaemon, Effect.flatMap(fiber => Ref.set(fiberRef, Option.some(fiber))), - ), [interruptRunningQuery, stateRef, runQuery, context, fiberRef]) + ), [stateRef, context, fiberRef]) - this.useEffect(() => interruptRunningQuery.pipe( + this.useEffect(() => interruptRunningQuery(fiberRef).pipe( Effect.andThen(Ref.set(stateRef, AsyncData.loading())), - Effect.andThen(runQuery), + Effect.andThen(runQuery(props.effect(), stateRef)), Effect.scoped, Effect.forkDaemon, Effect.flatMap(fiber => Ref.set(fiberRef, Option.some(fiber))), - ), [stateRef, runQuery, fiberRef]) + ), [stateRef, fiberRef]) return React.useMemo(() => ({ state: stateRef, triggerRefresh }), [stateRef, triggerRefresh]) }