Updated QueryStore type

src/createCacheGroup.tsx

@@ -5,7 +5,7 @@ import { CacheGroup } from './types/CacheGroup';
 import { QueryCacheBag } from './types/QueryCacheBag';
 import { QueryStore } from './types/QueryStore';
 
-export const createCacheGroup = (customStore?: QueryStore): CacheGroup => {
+export const createCacheGroup = (customStore?: QueryStore<unknown>): CacheGroup => {
     const CacheContext = createSafeContext<QueryCacheBag>();
 
     return {

src/hooks/useQueryCache.ts

@@ -8,8 +8,8 @@ import { QueryStore } from '../types/QueryStore';
 import { convertKeyToArguments } from '../utils/convertKeyToArguments';
 import { QueryKeyMap } from '../utils/QueryKeyMap';
 
-export const useQueryCache = (customCache?: QueryStore): QueryCacheBag => {
-    const cache = useRef<QueryKeyMap>(new QueryKeyMap(customCache));
+export const useQueryCache = (customCache?: QueryStore<unknown>): QueryCacheBag => {
+    const cache = useRef<QueryKeyMap<unknown>>(new QueryKeyMap(customCache));
 
     const getValue = useCallback(
         <TData, TKey extends Key>(key: TKey, fetcher: Fetcher<TData, TKey>): FetchResult<TData> => {

src/index.ts

@@ -4,4 +4,5 @@ export * from './hooks/useQuery';
 
 // Types
 export * from './types/CacheGroup';
+export * from './types/FetchResult';
 export * from './types/QueryStore';

src/types/QueryStore.ts

@@ -1,5 +1,7 @@
-export type QueryStore = {
-    get: <T>(key: string) => T | undefined;
-    set: <T>(key: string, value: T) => void;
+import { FetchResult } from './FetchResult';
+
+export type QueryStore<TData> = {
+    get: (key: string) => FetchResult<TData> | undefined;
+    set: (key: string, value: FetchResult<TData>) => void;
     has: (key: string) => boolean;
 };

src/utils/QueryKeyMap.ts

@@ -3,16 +3,16 @@ import { Key } from '../types/Key';
 import { QueryStore } from '../types/QueryStore';
 import { stringifyKey } from './stringifyKey';
 
-export class QueryKeyMap {
-    constructor(private storage: QueryStore = new Map()) {}
+export class QueryKeyMap<TData> {
+    constructor(private storage: QueryStore<TData> = new Map<string, FetchResult<TData>>()) {}
 
-    public get = <TData>(key: Key): FetchResult<TData> => {
+    public get = (key: Key): FetchResult<TData> => {
         const stringifiedKey = stringifyKey(key);
 
         return this.storage.get(stringifiedKey) as FetchResult<TData>;
     };
 
-    public set = <TData>(key: Key, value: FetchResult<TData>): void => {
+    public set = (key: Key, value: FetchResult<TData>): void => {
         const stringifiedKey = stringifyKey(key);
 
         this.storage.set(stringifiedKey, value);

test/customQueryStorage.test.tsx

@@ -1,13 +1,14 @@
 import { render, waitFor } from '@testing-library/react';
 import React, { Suspense } from 'react';
 import { createCacheGroup, QueryStore, useQuery } from '../src';
+import { FetchResult } from '../src/types/FetchResult';
 import '@testing-library/jest-dom';
 
-const object: Record<string, unknown> = {};
+const object: Record<string, FetchResult<unknown>> = {};
 
-const globalQueryStore: QueryStore = {
-    get: <T,>(key: string) => {
-        return object[key] as T;
+const globalQueryStore: QueryStore<unknown> = {
+    get: (key: string) => {
+        return object[key];
     },
     set: (key, value) => {
         object[key] = value;