Merge pull request #25 from closetool/master

feat: add updateFilteredPolicies

adapter.go

@@ -471,6 +471,32 @@ func (a *Adapter) UpdatePolicies(sec string, ptype string, oldRules, newRules []
 	return a.updatePolicies(oldLines, newLines)
 }
 
+func (a *Adapter) UpdateFilteredPolicies(sec string, ptype string, oldRules, newRules [][]string) error {
+	var oldLines []*CasbinRule
+	for _, rule := range oldRules {
+		line := savePolicyLine(ptype, rule)
+		oldLines = append(oldLines, line)
+	}
+
+	var newLines []*CasbinRule
+	for _, rule := range newRules {
+		line := savePolicyLine(ptype, rule)
+		newLines = append(newLines, line)
+	}
+
+	err := a.db.RunInTransaction(func(tx *pg.Tx) error {
+		_, err := tx.Model(&oldLines).Delete()
+		if err != nil {
+			return err
+		}
+		_, err = tx.Model(&newLines).
+			OnConflict("DO NOTHING").
+			Insert()
+		return err
+	})
+	return err
+}
+
 func (a *Adapter) updatePolicies(oldLines, newLines []*CasbinRule) error {
 	tx, err := a.db.Begin()
 	if err != nil {

adapter_test.go

@@ -324,6 +324,25 @@ func (s *AdapterTestSuite) TestUpdatePolicyWithLoadFilteredPolicy() {
 	s.assertPolicy(s.e.GetPolicy(), [][]string{{"alice", "data1", "read"}, {"bob", "data2", "write"}, {"bob", "data2", "read"}, {"alice", "data2", "write"}})
 }
 
+func (s *AdapterTestSuite) TestUpdateFilteredPolicies() {
+
+	var err error
+	s.e, err = casbin.NewEnforcer("examples/rbac_model.conf", "examples/rbac_policy.csv")
+	s.Require().NoError(err)
+
+	s.e.SetAdapter(s.a)
+
+	err = s.e.SavePolicy()
+	s.Require().NoError(err)
+
+	err = s.a.UpdateFilteredPolicies("p", "p", [][]string{{"alice", "data1", "read"}, {"bob", "data2", "write"}}, [][]string{{"alice", "data2", "write"}, {"bob", "data1", "read"}})
+	s.Require().NoError(err)
+
+	err = s.e.LoadPolicy()
+	s.Require().NoError(err)
+
+	s.assertPolicy(s.e.GetPolicy(), [][]string{{"data2_admin", "data2", "read"}, {"data2_admin", "data2", "write"}, {"alice", "data2", "write"}, {"bob", "data1", "read"}})
+}
 func TestAdapterTestSuite(t *testing.T) {
 	suite.Run(t, new(AdapterTestSuite))
 }