feat: simplify staleTime for queryOptions

packages/react-query-observable/src/__tests__/observableQueryOptions.test.ts

@@ -1,5 +1,5 @@
-import { type Query } from '@tanstack/react-query'
-import { of } from 'rxjs'
+import { hashKey, type Query } from '@tanstack/react-query'
+import { of, type Subscription } from 'rxjs'
 import { beforeEach, describe, expect, it } from 'vitest'
 import { mock } from 'vitest-mock-extended'
 
@@ -31,12 +31,17 @@ describe('observableQueryOptions', () => {
     expect(staleTimeValue).toBe(0)
   })
 
-  it('should set staleTime to Infinity when the query has updated data', () => {
+  it('should set staleTime to Infinity when the query has an active subscription', () => {
     const options = observableQueryOptions({
       queryKey,
       observableFn: () => of('value'),
     })
 
+    // Add a subscription to simulate an active query subscription
+    const mockSubscription = mock<Subscription>()
+    const queryKeyHash = hashKey(queryKey)
+    queriesSubscriptions.set(queryKeyHash, mockSubscription)
+
     const staleTimeValue =
       typeof options.staleTime === 'function'
         ? options.staleTime(

packages/react-query-observable/src/observableQueryOptions.ts

@@ -1,6 +1,13 @@
-import { type DefaultError, type QueryKey, queryOptions, type UnusedSkipTokenOptions } from '@tanstack/react-query'
+import {
+  type DefaultError,
+  hashKey,
+  type QueryKey,
+  queryOptions,
+  type UnusedSkipTokenOptions,
+} from '@tanstack/react-query'
 
 import { queryFnFromObservableFn } from './queryFn/observableQueryFn'
+import { queriesSubscriptions } from './queryFn/queriesWithObservable'
 import { type ObservableQueryFunction } from './types'
 
 export interface ObservableQueryOptions<
@@ -31,7 +38,7 @@ export const observableQueryOptions = <
 ) =>
   queryOptions({
     queryFn: queryFnFromObservableFn(options.observableFn),
-    staleTime: ({ state }) => (state.dataUpdateCount === 0 || state.status === 'error' ? 0 : Infinity),
+    staleTime: ({ queryKey }) => (queriesSubscriptions.has(hashKey(queryKey)) ? Infinity : 0),
     gcTime: 10_000,
     ...options,
   })