feat: 🎸 add support for arguments in mutex decorator

src/__tests__/mutex.spec.ts

@@ -38,3 +38,17 @@ test('works as a class method decorator', async () => {
   const res4 = await test.code();
   expect(res4).toStrictEqual(3);
 });
+
+test('passes through arguments in the decorated method', async () => {
+  let cnt = 0;
+  class Test {
+    @mutex async code(divisor: number) {
+      return cnt++ / divisor;
+    }
+  }
+  const test = new Test();
+  const res = await Promise.all([test.code(10), test.code(10), test.code(10)]);
+  expect(res).toStrictEqual([0, 0, 0]);
+  const res2 = await Promise.all([test.code(10), test.code(10), test.code(10), test.code(10)]);
+  expect(res2).toStrictEqual([0.1, 0.1, 0.1, 0.1]);
+});

src/mutex.ts

@@ -7,12 +7,12 @@ import {codeMutex} from './codeMutex';
  * {@link mutex} can be used as a class method decorator or a higher order
  * function.
  */
-export function mutex<This, Return>(
-  target: (this: This) => Promise<Return>,
-  context?: ClassMethodDecoratorContext<This, (this: This) => Promise<Return>>,
+export function mutex<This, Args extends any[], Return>(
+  target: (this: This, ...args: Args) => Promise<Return>,
+  context?: ClassMethodDecoratorContext<This, (this: This, ...args: Args) => Promise<Return>>,
 ) {
   const mut = codeMutex<Return>();
-  return async function (this: This): Promise<Return> {
-    return await mut(async () => await target.call(this));
+  return async function (this: This, ...args: Args): Promise<Return> {
+    return await mut(async () => await target.call(this, ...args));
   };
 }