Ben Lin
2024-06-23 200eb764e83c7a77defeaf98130801d300dbee5d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
type Timer = ReturnType<typeof setTimeout>;
type CachedKey = string | number;
 
export interface CachedData<TData = any, TParams = any> {
  data: TData;
  params: TParams;
  time: number;
}
 
interface RecordData extends CachedData {
  timer: Timer | undefined;
}
 
const cache = new Map<CachedKey, RecordData>();
 
export const setCache = (key: CachedKey, cacheTime: number, cachedData: CachedData) => {
  const currentCache = cache.get(key);
  if (currentCache?.timer) {
    clearTimeout(currentCache.timer);
  }
 
  let timer: Timer | undefined = undefined;
 
  if (cacheTime > -1) {
    // if cache out, clear it
    timer = setTimeout(() => {
      cache.delete(key);
    }, cacheTime);
  }
 
  cache.set(key, {
    ...cachedData,
    timer,
  });
};
 
export const getCache = (key: CachedKey) => {
  return cache.get(key);
};
 
export const clearCache = (key?: string | string[]) => {
  if (key) {
    const cacheKeys = Array.isArray(key) ? key : [key];
    cacheKeys.forEach((cacheKey) => cache.delete(cacheKey));
  } else {
    cache.clear();
  }
};