Merge pull request #8 from tangyang9464/select-db

feat: add select db support
jcasbin · Sep 29, 2021 · dbfa188 · dbfa188
2 parents 714af60 + 91433d7
commit dbfa188
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/.github/workflows/maven-ci.yml b/.github/workflows/maven-ci.yml
@@ -43,7 +43,7 @@ jobs:
       - name: Set up Node.js
         uses: actions/setup-node@v1
         with:
-          node-version: 12
+          node-version: 14
 
       - name: Sematic Release
         run: |

diff --git a/src/main/java/org/casbin/adapter/RedisAdapter.java b/src/main/java/org/casbin/adapter/RedisAdapter.java
@@ -130,6 +130,12 @@ public void close() {
         jedis.close();
     }
 
+    public void selectDb(int dbIndex) {
+        if(jedis != null) {
+            jedis.select(dbIndex);
+        }
+    }
+
     private void newRedisAdapter(String host, int port, String key, String password) {
         this.key = key;
 

diff --git a/src/test/java/org/casbin/adapter/RedisAdapterTest.java b/src/test/java/org/casbin/adapter/RedisAdapterTest.java
@@ -99,6 +99,21 @@ public void testAdapter() {
         redisAdapter.close();
     }
 
+    @Test
+    public void testSelectDb() {
+        Enforcer enforcer = new Enforcer("examples/rbac_model.conf", "examples/rbac_policy.csv");
+        redisAdapter = new RedisAdapter(host, port);
+        // save policy to db 0
+        redisAdapter.savePolicy(enforcer.getModel());
+        // select db to 1
+        redisAdapter.selectDb(1);
+        // add policy to db 1
+        redisAdapter.addPolicy("p", "p", Arrays.asList("paul", "data2", "read"));
+        enforcer.clearPolicy();
+        redisAdapter.loadPolicy(enforcer.getModel());
+        testGetPolicy(enforcer, Arrays.asList(Arrays.asList("paul", "data2", "read")));
+    }
+
     private void testGetPolicy(Enforcer e, List<List<String>> res) {
         List<List<String>> policies = e.getPolicy();
         Assert.assertEquals(res, policies);