chore: improve needpolling type (#741)

Author: DorianMaliszewski

packages/use-dataloader/src/DataLoaderProvider.tsx

@@ -43,10 +43,10 @@ type GetReloadsFn = {
 
 export interface IDataLoaderContext {
   addRequest: (key: string, args: UseDataLoaderInitializerArgs) => DataLoader
-  getOrAddRequest: (
+  getOrAddRequest: <T>(
     key: string,
-    args: UseDataLoaderInitializerArgs,
-  ) => DataLoader
+    args: UseDataLoaderInitializerArgs<T>,
+  ) => DataLoader<T>
   cacheKeyPrefix?: string
   onError?: (error: Error) => void | Promise<void>
   clearAllCachedData: () => void

packages/use-dataloader/src/types.ts

@@ -30,7 +30,7 @@ export interface UseDataLoaderConfig<T = unknown> {
    * Max time before data from previous success is considered as outdated (in millisecond)
    */
   maxDataLifetime?: number
-  needPolling?: NeedPollingType
+  needPolling?: NeedPollingType<T>
 }
 
 /**
@@ -76,7 +76,7 @@ export type UsePaginatedDataLoaderConfig<T = unknown> = {
    * Max time before data from previous success is considered as outdated (in millisecond)
    */
   maxDataLifetime?: number
-  needPolling?: NeedPollingType
+  needPolling?: NeedPollingType<T>
   initialPage?: number
   perPage?: number
 }

packages/use-dataloader/src/useDataLoader.ts

@@ -8,7 +8,6 @@ import {
 } from 'react'
 import { useDataLoaderContext } from './DataLoaderProvider'
 import { StatusEnum } from './constants'
-import DataLoader from './dataloader'
 import { PromiseType, UseDataLoaderConfig, UseDataLoaderResult } from './types'
 
 /**
@@ -52,7 +51,7 @@ const useDataLoader = <T>(
       method,
       needPolling,
       pollingInterval,
-    }) as DataLoader<T>
+    })
 
     unsubscribeRequestRef.current = () => newRequest.removeObserver(subscribeFn)
     newRequest.addObserver(subscribeFn)
@@ -82,7 +81,9 @@ const useDataLoader = <T>(
     }
   }, [onSuccess, onError, onErrorProvider, method, request])
 
-  const cancelMethodRef = useRef<(() => Promise<void>) | undefined>(request?.cancel)
+  const cancelMethodRef = useRef<(() => Promise<void>) | undefined>(
+    request?.cancel,
+  )
 
   const isLoading = useMemo(
     () =>