Merge pull request #132 from tangyang9464/batch

fix: change AddPolicies to batch insert

Author: hsluoyz

adapter.go

@@ -553,15 +553,12 @@ func (a *Adapter) RemovePolicy(sec string, ptype string, rule []string) error {
 
 // AddPolicies adds multiple policy rules to the storage.
 func (a *Adapter) AddPolicies(sec string, ptype string, rules [][]string) error {
-	return a.db.Transaction(func(tx *gorm.DB) error {
-		for _, rule := range rules {
-			line := a.savePolicyLine(ptype, rule)
-			if err := tx.Create(&line).Error; err != nil {
-				return err
-			}
-		}
-		return nil
-	})
+	var lines []CasbinRule
+	for _, rule := range rules {
+		line := a.savePolicyLine(ptype, rule)
+		lines = append(lines, line)
+	}
+	return a.db.Create(&lines).Error
 }
 
 // RemovePolicies removes multiple policy rules from the storage.

adapter_test.go

@@ -516,3 +516,12 @@ func TestAdapters(t *testing.T) {
 	//testUpdatePolicy(t, a)
 	//testUpdatePolicies(t, a)
 }
+
+func TestAddPolicies(t *testing.T) {
+	a := initAdapter(t, "mysql", "root:@tcp(127.0.0.1:3306)/", "casbin", "casbin_rule")
+	e, _ := casbin.NewEnforcer("examples/rbac_model.conf", a)
+	e.AddPolicies([][]string{{"jack", "data1", "read"}, {"jack2", "data1", "read"}})
+	e.LoadPolicy()
+
+	testGetPolicy(t, e, [][]string{{"alice", "data1", "read"}, {"bob", "data2", "write"}, {"data2_admin", "data2", "read"}, {"data2_admin", "data2", "write"}, {"jack", "data1", "read"}, {"jack2", "data1", "read"}})
+}