diff --git a/packages/extension-query/src/MutationRunner.ts b/packages/extension-query/src/MutationRunner.ts index 881966c..d8c294e 100644 --- a/packages/extension-query/src/MutationRunner.ts +++ b/packages/extension-query/src/MutationRunner.ts @@ -30,7 +30,8 @@ export const make = ( const mutate = (...key: K) => QueryClient.pipe( Effect.flatMap(client => client.ErrorHandler), - Effect.flatMap(errorHandler => mutation(...key).pipe( + Effect.flatMap(errorHandler => Ref.set(stateRef, AsyncData.loading()).pipe( + Effect.andThen(mutation(...key)), errorHandler.handle, Effect.tapErrorCause(c => Ref.set(stateRef, AsyncData.failure(c))), Effect.tap(v => Ref.set(stateRef, AsyncData.success(v))),