Merge pull request #6 from basakest/BatchAdapter

try to implement BatchAdapter

Author: leeqvip

Diff for: src/Adapter.php

@@ -9,14 +9,16 @@
 use Casbin\Persist\FilteredAdapter;
 use Casbin\Persist\Adapters\Filter;
 use Casbin\Exceptions\InvalidFilterTypeException;
+use Casbin\Persist\BatchAdapter;
 use Closure;
+use Throwable;
 
 /**
  * DatabaseAdapter.
  *
  * @author [email protected]
  */
-class Adapter implements AdapterContract, FilteredAdapter
+class Adapter implements AdapterContract, FilteredAdapter, BatchAdapter
 {
     use AdapterHelper;
 
@@ -136,6 +138,38 @@ public function addPolicy(string $sec, string $ptype, array $rule): void
         $this->savePolicyLine($ptype, $rule);
     }
 
+    public function addPolicies(string $sec, string $ptype, array $rules): void
+    {
+        $table = $this->casbinRuleTableName;
+        $columns = ['p_type', 'v0', 'v1', 'v2', 'v3', 'v4', 'v5'];
+        $values = [];
+        $sets = [];
+        $columnsCount = count($columns);
+        foreach ($rules as $rule) {
+            $values = array_merge($values, array_pad($rule, $columnsCount, null));
+            $sets[] = array_pad([], $columnsCount, '?');
+        }
+        $valuesStr = implode(', ', array_map(function ($set) {
+            return '(' . implode(', ', $set) . ')';
+        }, $sets));
+        $sql = 'INSERT INTO ' . $table . ' (' . implode(', ', $columns) . ')' .
+            ' VALUES' . $valuesStr;
+    }
+
+    public function removePolicies(string $sec, string $ptype, array $rules): void
+    {
+        $this->connection->getPdo()->beginTransaction();
+        try {
+            foreach($rules as $rule) {
+                $this->removePolicy($sec, $ptype, $rule);
+            }
+            $this->connection->getPdo()->commit();
+        } catch (Throwable $e){
+            $this->connection->getPdo()->rollback();
+            throw $e;
+        }
+    }
+
     /**
      * This is part of the Auto-Save feature.
      *

Diff for: tests/AdapterTest.php

@@ -125,6 +125,21 @@ public function testAddPolicy()
         $this->assertTrue($e->enforce('eve', 'data3', 'read'));
     }
 
+    public function testAddPolicies()
+    {
+        $policies = [
+            ['u1', 'd1', 'read'],
+            ['u2', 'd2', 'read'],
+            ['u3', 'd3', 'read'],
+        ];
+        $e = $this->getEnforcer();
+        $e->clearPolicy();
+        $this->initDb(DatabaseAdapter::newAdapter($this->config));
+        $this->assertEquals([], $e->getPolicy());
+        $e->addPolicies($policies);
+        $this->assertEquals($policies, $e->getPolicy());
+    }
+
     public function testSavePolicy()
     {
         $e = $this->getEnforcer();
@@ -149,6 +164,27 @@ public function testRemovePolicy()
         $this->assertFalse($e->enforce('alice', 'data5', 'read'));
     }
 
+    public function testRemovePolicies()
+    {
+        $e = $this->getEnforcer();
+        $this->assertEquals([
+            ['alice', 'data1', 'read'],
+            ['bob', 'data2', 'write'],
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+        ], $e->getPolicy());
+
+        $e->removePolicies([
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+        ]);
+
+        $this->assertEquals([
+            ['alice', 'data1', 'read'],
+            ['bob', 'data2', 'write']
+        ], $e->getPolicy());
+    }
+
     public function testRemoveFilteredPolicy()
     {
         $e = $this->getEnforcer();