feat: options for useWorkerFns()

mys1024 · Mar 9, 2024 · a841a0e · a841a0e
1 parent 762d90f
commit a841a0e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/use.ts b/src/use.ts
@@ -42,10 +42,14 @@ export function useWorkerFn<FN extends AnyFn>(
  * Invoke this function in the main thread to create proxy functions of all worker functions.
  *
  * @param worker - A Worker instance.
+ * @param options - An object containing options.
  * @returns Proxy functions.
  */
 export function useWorkerFns<FNS extends Record<string, AnyFn>>(
   worker: MsgPort,
+  options: {
+    [NAME in keyof FNS]?: UseWorkerFnOpts<FNS[NAME]>;
+  } = {},
 ): ProxyFns<FNS> {
   const memo = new Map<string, ProxyFn<AnyFn>>();
   const fns = new Proxy({}, {
@@ -56,7 +60,7 @@ export function useWorkerFns<FNS extends Record<string, AnyFn>>(
       if (memo.has(name)) {
         return memo.get(name);
       }
-      const fn = useWorkerFn(name, worker);
+      const fn = useWorkerFn(name, worker, options[name]);
       memo.set(name, fn);
       return fn;
     },

diff --git a/test/fns.test.ts b/test/fns.test.ts
@@ -26,7 +26,11 @@ Deno.test("fns", async (t) => {
         type: "module",
       },
     );
-    const { add, fib } = useWorkerFns<Fns>(worker);
+    const { add, fib } = useWorkerFns<Fns>(worker, {
+      fib: {
+        transfer: () => [],
+      },
+    });
     assertEquals(await add(1, 2), 3);
     assertEquals(await add(5, 5), 10);
     assertEquals(await add(10, 20), 30);