feat: add sync mode

Zxilly · Zxilly · commit 70e4e12610dc · 2021-03-18T22:04:55.000+08:00
Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;
diff --git a/src/syncedEnforcer.ts b/src/syncedEnforcer.ts
@@ -15,6 +15,7 @@
 import { Enforcer, newEnforcerWithClass } from './enforcer';
 import AwaitLock from 'await-lock';
 import { Watcher } from './persist';
+import { MatchingFunc } from './rbac';
 
 // SyncedEnforcer wraps Enforcer and provides synchronized access
 export class SyncedEnforcer extends Enforcer {
@@ -477,6 +478,28 @@ export class SyncedEnforcer extends Enforcer {
     await this.lock.acquireAsync();
     return super.removeFilteredNamedGroupingPolicy(ptype, fieldIndex, ...fieldValues).finally(() => this.lock.release());
   }
+
+  /**
+   * add matching function to RoleManager by ptype
+   * @param ptype g
+   * @param fn the function will be added
+   */
+  public async addNamedMatchingFunc(ptype: string, fn: MatchingFunc): Promise<void> {
+    await this.lock.acquireAsync();
+    return super.addNamedMatchingFunc(ptype, fn).finally(() => this.lock.release());
+  }
+
+  /**
+   * add domain matching function to RoleManager by ptype
+   * @param ptype g
+   * @param fn the function will be added
+   */
+  public async addNamedDomainMatchingFunc(ptype: string, fn: MatchingFunc): Promise<void> {
+    await this.lock.acquireAsync();
+    return super.addNamedDomainMatchingFunc(ptype, fn).finally(() => {
+      this.lock.release();
+    });
+  }
 }
 
 // newSyncedEnforcer creates a synchronized enforcer via file or DB.
