feat: add initRmMap

Zxilly · Zxilly · commit 87f801109e9c · 2021-03-18T23:21:45.000+08:00
Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;
diff --git a/src/coreEnforcer.ts b/src/coreEnforcer.ts
@@ -143,15 +143,15 @@ export class CoreEnforcer {
     this.model.clearPolicy();
   }
 
-  // public initRmMap(): void {
-  //   this.rmMap = new Map<string, RoleManager>();
-  //   const rm = this.model.model.get('g');
-  //   if (rm) {
-  //     for (const ptype of rm.keys()) {
-  //       this.rmMap.set(ptype, new DefaultRoleManager(10));
-  //     }
-  //   }
-  // }
+  public initRmMap(): void {
+    this.rmMap = new Map<string, RoleManager>();
+    const rm = this.model.model.get('g');
+    if (rm) {
+      for (const ptype of rm.keys()) {
+        this.rmMap.set(ptype, new DefaultRoleManager(10));
+      }
+    }
+  }
 
   /**
    * loadPolicy reloads the policy from file/database.
@@ -160,6 +160,8 @@ export class CoreEnforcer {
     this.model.clearPolicy();
     await this.adapter.loadPolicy(this.model);
 
+    this.initRmMap();
+
     if (this.autoBuildRoleLinks) {
       await this.buildRoleLinksInternal();
     }
@@ -180,6 +182,8 @@ export class CoreEnforcer {
       throw new Error('filtered policies are not supported by this adapter');
     }
 
+    this.initRmMap();
+
     if (this.autoBuildRoleLinks) {
       await this.buildRoleLinksInternal();
     }
