Add graph commits to OrientIOInterpreter

ItsMeijers · Oct 2, 2016 · 07be009 · 07be009
1 parent 7056b19
commit 07be009
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/main/scala/oriented/syntax/syntax.scala b/src/main/scala/oriented/syntax/syntax.scala
@@ -56,6 +56,8 @@ package object syntax {
 
       val result = orientIO.foldMap(unsafeInterpreter)
 
+      if(enableTransactions) graph.commit()
+
       graph.shutdown()
 
       result
@@ -82,7 +84,9 @@ package object syntax {
 
       val result = orientIO.foldMap(tryInterpreter)
 
-      if(result.isFailure && enableTransactions) graph.rollback()
+      if(result.isFailure && enableTransactions)
+        graph.rollback()
+      else if(enableTransactions) graph.commit()
 
       graph.shutdown()
 
@@ -135,6 +139,7 @@ package object syntax {
       })
 
       result.onSuccess(PartialFunction { _ =>
+        if(enableTransactions) graph.commit()
         graph.shutdown()
       })
 
@@ -158,10 +163,10 @@ package object syntax {
       else orientClient.graphNoTransaction
 
       val interpreter: OrientProgram ~> EitherT[Future, Throwable, ?] =
-        SafeAsyncEdgeInterpreter() or
-       (SafeAsyncVertexInterpreter() or
+        SafeAsyncEdgeInterpreter()    or
+       (SafeAsyncVertexInterpreter()  or
        (SafeAsyncElementInterpreter() or
-       (SafeAsyncClientInterpreter() or
+       (SafeAsyncClientInterpreter()  or
        SafeAsyncSqlInterpreter())))
 
 
@@ -176,6 +181,7 @@ package object syntax {
 
       isLeft.onSuccess(PartialFunction { failed =>
         if(failed && enableTransactions) graph.rollback()
+        else if(enableTransactions) graph.commit()
         graph.shutdown()
       })