improve

Change-Id: I0f473063f6df1e43170f01a655a87a1c790f028a
apache · Jan 13, 2021 · 4047231 · 4047231
1 parent 921efba
commit 4047231
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/task/TaskCallable.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/task/TaskCallable.java
@@ -22,6 +22,7 @@
 import java.util.Date;
 import java.util.concurrent.Callable;
 
+import org.apache.tinkerpop.gremlin.structure.Transaction;
 import org.slf4j.Logger;
 
 import com.baidu.hugegraph.HugeException;
@@ -56,8 +57,9 @@ protected void cancelled() {
     }
 
     protected void closeTx() {
-        if (this.graph().tx().isOpen()) {
-            this.graph().tx().close();
+        Transaction tx = this.graph().tx();
+        if (tx.isOpen()) {
+            tx.close();
         }
     }