feat: 显式注册函数插桩支持

IdeaLeap · Sep 2, 2023 · a22db59 · a22db59
1 parent 664163f
commit a22db59
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/package/utils.ts b/package/utils.ts
@@ -35,6 +35,12 @@ export class PipeRegistry {
       return result;
     },
   };
+
+  static customFn: Record<
+    string,
+    (result: any, context: PipelineContext) => any
+  > = {};
+
   static init() {
     const pipeRegistry = new PipeRegistry();
     //遍历commonPreProcess进行pipeRegistry.register注册
@@ -45,6 +51,10 @@ export class PipeRegistry {
     Object.entries(PipeRegistry.commonPostProcess).forEach(([type, fn]) => {
       pipeRegistry.register(type, fn);
     });
+    //遍历customFn进行pipeRegistry.register注册
+    Object.entries(PipeRegistry.customFn).forEach(([type, fn]) => {
+      pipeRegistry.register(type, fn);
+    });
     return pipeRegistry;
   }
 }