111 lines
2.8 KiB
TypeScript
111 lines
2.8 KiB
TypeScript
import { TOKEN_KEY, ROLES_KEY, USER_INFO_KEY, DB_DICT_DATA_KEY, TENANT_ID, LOGIN_INFO_KEY, SYS_CONFIG_INFO_KEY, PROJ_CFG_KEY } from '/@/enums/cacheEnum';
|
||
import { omit } from 'lodash-es';
|
||
|
||
export interface Cache<V = any> {
|
||
value?: V;
|
||
timeoutId?: ReturnType<typeof setTimeout>;
|
||
time?: number;
|
||
alive?: number;
|
||
}
|
||
|
||
const NOT_ALIVE = 0;
|
||
|
||
export class Memory<T = any, V = any> {
|
||
private cache: { [key in keyof T]?: Cache<V> } = {};
|
||
private alive: number;
|
||
|
||
constructor(alive = NOT_ALIVE) {
|
||
// Unit second
|
||
this.alive = alive * 1000;
|
||
}
|
||
|
||
get getCache() {
|
||
return this.cache;
|
||
}
|
||
|
||
setCache(cache) {
|
||
this.cache = cache;
|
||
}
|
||
|
||
// get<K extends keyof T>(key: K) {
|
||
// const item = this.getItem(key);
|
||
// const time = item?.time;
|
||
// if (!isNullOrUnDef(time) && time < new Date().getTime()) {
|
||
// this.remove(key);
|
||
// }
|
||
// return item?.value ?? undefined;
|
||
// }
|
||
|
||
get<K extends keyof T>(key: K) {
|
||
return this.cache[key];
|
||
}
|
||
|
||
set<K extends keyof T>(key: K, value: V, expires?: number) {
|
||
let item = this.get(key);
|
||
|
||
if (!expires || (expires as number) <= 0) {
|
||
expires = this.alive;
|
||
}
|
||
if (item) {
|
||
if (item.timeoutId) {
|
||
clearTimeout(item.timeoutId);
|
||
item.timeoutId = undefined;
|
||
}
|
||
item.value = value;
|
||
} else {
|
||
item = { value, alive: expires };
|
||
this.cache[key] = item;
|
||
}
|
||
|
||
if (!expires) {
|
||
return value;
|
||
}
|
||
const now = new Date().getTime();
|
||
item.time = now + this.alive;
|
||
item.timeoutId = setTimeout(
|
||
() => {
|
||
this.remove(key);
|
||
},
|
||
expires > now ? expires - now : expires
|
||
);
|
||
|
||
return value;
|
||
}
|
||
|
||
remove<K extends keyof T>(key: K) {
|
||
const item = this.get(key);
|
||
Reflect.deleteProperty(this.cache, key);
|
||
if (item) {
|
||
clearTimeout(item.timeoutId!);
|
||
return item.value;
|
||
}
|
||
}
|
||
|
||
resetCache(cache: { [K in keyof T]: Cache }) {
|
||
Object.keys(cache).forEach((key) => {
|
||
const k = key as any as keyof T;
|
||
const item = cache[k];
|
||
if (item && item.time) {
|
||
const now = new Date().getTime();
|
||
const expire = item.time;
|
||
if (expire > now) {
|
||
this.set(k, item.value, expire);
|
||
}
|
||
}
|
||
});
|
||
}
|
||
|
||
clear() {
|
||
console.log('------clear------进入clear方法');
|
||
Object.keys(this.cache).forEach((key) => {
|
||
const item = this.cache[key];
|
||
item.timeoutId && clearTimeout(item.timeoutId);
|
||
});
|
||
//update-begin---author:liusq Date:20220108 for：不删除登录用户的租户id，其他缓存信息都清除----
|
||
this.cache = {
|
||
...omit(this.cache, [TOKEN_KEY, USER_INFO_KEY, ROLES_KEY, DB_DICT_DATA_KEY, TENANT_ID, LOGIN_INFO_KEY, SYS_CONFIG_INFO_KEY, PROJ_CFG_KEY]),
|
||
};
|
||
//update-end---author:liusq Date:20220108 for：不删除登录用户的租户id，其他缓存信息都清除----
|
||
}
|
||
}
|