feat: buildRegFn add reset func

msgbyte · Oct 11, 2023 · af16ebe · af16ebe
1 parent 18bc2ed
commit af16ebe
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/client/shared/manager/buildReg.ts b/client/shared/manager/buildReg.ts
@@ -9,7 +9,7 @@ export function buildRegFn<F extends (...args: any[]) => any>(
   name: string,
   defaultFunc?: F
 ) {
-  let func: F;
+  let func: F | undefined;
 
   const get = (...args: Parameters<F>): ReturnType<F> => {
     if (!func) {
@@ -26,7 +26,11 @@ export function buildRegFn<F extends (...args: any[]) => any>(
     func = fn;
   };
 
-  return [get, set] as const;
+  const reset = (): void => {
+    func = defaultFunc;
+  };
+
+  return [get, set, reset] as const;
 }
 
 /**