add rollbackBackend() method

Change-Id: I9a870d41c0675ad79cc6118591a0ae121401afff
apache · Jun 23, 2021 · 4d72a1e · 4d72a1e
1 parent 071aa26
commit 4d72a1e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/AbstractTransaction.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/AbstractTransaction.java
@@ -231,8 +231,7 @@ public void rollback() throws BackendException {
         LOG.debug("Transaction rollback()...");
         this.reset();
         if (this.committing2Backend) {
-            this.committing2Backend = false;
-            this.store.rollbackTx();
+            this.rollbackBackend();
         }
     }
 
@@ -301,6 +300,11 @@ protected void commitMutation2Backend(BackendMutation... mutations) {
         this.committing2Backend = false;
     }
 
+    protected void rollbackBackend() {
+        this.committing2Backend = false;
+        this.store.rollbackTx();
+    }
+
     protected BackendMutation prepareCommit() {
         // For sub-class preparing data, nothing to do here
         LOG.debug("Transaction prepareCommit()...");

diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/GraphTransaction.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/GraphTransaction.java
@@ -483,7 +483,7 @@ private void commitPartOfEdgeDeletions(Map<Id, HugeEdge> removedEdges) {
         try {
             this.commitMutation2Backend(mutation, idxMutation);
         } catch (Throwable e) {
-            this.store().rollbackTx();
+            this.rollbackBackend();
         } finally {
             mutation.clear();
             idxMutation.clear();