support stacked lock

src/def.ts

@@ -3,7 +3,7 @@ import { Settings } from '@nanahira/redlock';
 
 export type RedisDriverOptions = RedisOptions & {
   uri?: string;
-  lock?: Settings & { duration?: number; prefix?: string };
+  lock?: Settings & { duration?: number; prefix?: string; stacked?: boolean };
   queueKey?: string;
 };
 

src/drivers/redis.ts

@@ -3,6 +3,7 @@ import Redis from 'ioredis';
 import Redlock from '@nanahira/redlock';
 import { Awaitable, RedisDriverOptions } from '../def';
 import { createPool } from 'generic-pool';
+import BetterLock from 'better-lock';
 
 export class RedisDriver extends BaseDriver {
   async createRedisClient() {
@@ -94,8 +95,11 @@ export class RedisDriver extends BaseDriver {
     await this.pool.use((r) => r.redis.del(keys));
   }
 
+  private betterLock = new BetterLock();
+
   override async lock<R>(keys: string[], cb: () => Promise<R>): Promise<R> {
-    return this.useTempRedisClient(async (redis) => {
+    const run = () =>
+      this.useTempRedisClient(async (redis) => {
         const redlock = new Redlock([redis], this.options.lock);
         return redlock.using(
           keys.map((key) => `${this.options.lock?.prefix || '_lock'}:${key}`),
@@ -103,6 +107,11 @@ export class RedisDriver extends BaseDriver {
           cb,
         );
       });
+    if (this.options?.lock?.stacked) {
+      return this.betterLock.acquire(keys, run);
+    } else {
+      return run();
+    }
   }
 
   override async isFree(keys: string[]): Promise<boolean> {