22 lines
520 B
TypeScript
22 lines
520 B
TypeScript
import type { Ref, WatchSource } from 'vue';
|
|
import { ref, watch } from 'vue';
|
|
|
|
export default function useMemo<T>(
|
|
getValue: () => T,
|
|
condition: (WatchSource<unknown> | object)[],
|
|
shouldUpdate?: (prev: any[], next: any[]) => boolean,
|
|
) {
|
|
const cacheRef: Ref<T> = ref(getValue() as any);
|
|
watch(condition, (next, pre) => {
|
|
if (shouldUpdate) {
|
|
if (shouldUpdate(next, pre)) {
|
|
cacheRef.value = getValue();
|
|
}
|
|
} else {
|
|
cacheRef.value = getValue();
|
|
}
|
|
});
|
|
|
|
return cacheRef;
|
|
}
|