Merge pull request casbin#366 from nodece/master

feat: controls whether to automatically notify Watcher

enforcer.go

@@ -44,6 +44,7 @@ type Enforcer struct {
 	enabled            bool
 	autoSave           bool
 	autoBuildRoleLinks bool
+	autoNotifyWatcher  bool
 }
 
 // NewEnforcer creates an enforcer via file or DB.
@@ -171,6 +172,7 @@ func (e *Enforcer) initialize() {
 	e.enabled = true
 	e.autoSave = true
 	e.autoBuildRoleLinks = true
+	e.autoNotifyWatcher = true
 }
 
 // LoadModel reloads the model from the model CONF file.
@@ -316,6 +318,11 @@ func (e *Enforcer) EnableLog(enable bool) {
 	log.GetLogger().EnableLog(enable)
 }
 
+// EnableAutoNotifyWatcher controls whether to save a policy rule automatically notify the Watcher when it is added or removed.
+func (e *Enforcer) EnableAutoNotifyWatcher(enable bool)  {
+	e.autoNotifyWatcher = enable
+}
+
 // EnableAutoSave controls whether to save a policy rule automatically to the adapter when it is added or removed.
 func (e *Enforcer) EnableAutoSave(autoSave bool) {
 	e.autoSave = autoSave

internal_api.go

@@ -31,12 +31,12 @@ func (e *Enforcer) addPolicy(sec string, ptype string, rule []string) (bool, err
 				return ruleAdded, err
 			}
 		}
+	}
 
-		if e.watcher != nil {
-			err := e.watcher.Update()
-			if err != nil {
-				return ruleAdded, err
-			}
+	if e.watcher !=nil && e.autoNotifyWatcher {
+		err := e.watcher.Update()
+		if err != nil {
+			return ruleAdded, err
 		}
 	}
 
@@ -56,11 +56,12 @@ func (e *Enforcer) removePolicy(sec string, ptype string, rule []string) (bool,
 				return ruleRemoved, err
 			}
 		}
-		if e.watcher != nil {
-			err := e.watcher.Update()
-			if err != nil {
-				return ruleRemoved, err
-			}
+	}
+
+	if e.watcher !=nil && e.autoNotifyWatcher {
+		err := e.watcher.Update()
+		if err != nil {
+			return ruleRemoved, err
 		}
 	}
 
@@ -80,11 +81,12 @@ func (e *Enforcer) removeFilteredPolicy(sec string, ptype string, fieldIndex int
 				return ruleRemoved, err
 			}
 		}
-		if e.watcher != nil {
-			err := e.watcher.Update()
-			if err != nil {
-				return ruleRemoved, err
-			}
+	}
+
+	if e.watcher !=nil && e.autoNotifyWatcher {
+		err := e.watcher.Update()
+		if err != nil {
+			return ruleRemoved, err
 		}
 	}