feat: one transaction is notified only once (#100)

jcasbin · Jun 7, 2024 · aef3b87 · aef3b87
1 parent d1b6524
commit aef3b87
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/src/main/java/org/casbin/spring/boot/autoconfigure/TxWatcher.java b/src/main/java/org/casbin/spring/boot/autoconfigure/TxWatcher.java
@@ -10,9 +10,11 @@
 
 public class TxWatcher implements Watcher {
     private final Watcher watcher;
+    private final AfterCommitExecute afterCommitExecute;
 
     public TxWatcher(Watcher watcher) {
         this.watcher = watcher;
+        this.afterCommitExecute = new AfterCommitExecute(watcher);
     }
 
     @Override
@@ -28,14 +30,22 @@ public void setUpdateCallback(Consumer<String> consumer) {
     @Override
     public void update() {
         if (isActualTransactionActive()) {
-            registerSynchronization(new TransactionSynchronization() {
-                @Override
-                public void afterCommit() {
-                    watcher.update();
-                }
-            });
+            registerSynchronization(afterCommitExecute);
         } else {
             watcher.update();
         }
     }
+
+    public static class AfterCommitExecute implements TransactionSynchronization {
+        private final Watcher watcher;
+
+        public AfterCommitExecute(Watcher watcher) {
+            this.watcher = watcher;
+        }
+
+        @Override
+        public void afterCommit() {
+            watcher.update();
+        }
+    }
 }