Fix update of newly created document using variables in a SQL batch s…

…cript Resolves: #9000
orientechnologies · Oct 1, 2019 · 7413a31 · 7413a31
1 parent 704572c
commit 7413a31
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/core/src/main/java/com/orientechnologies/orient/core/sql/executor/FetchFromVariableStep.java b/core/src/main/java/com/orientechnologies/orient/core/sql/executor/FetchFromVariableStep.java
@@ -92,7 +92,9 @@ private void init() {
     }
     inited = true;
     Object src = ctx.getVariable(variableName);
-    if (src instanceof OResultSet) {
+    if(src instanceof OInternalResultSet){
+      source = ((OInternalResultSet)src).copy();
+    } else if (src instanceof OResultSet) {
       source = (OResultSet) src;
       source.reset();
     } else if (src instanceof OElement) {

diff --git a/core/src/main/java/com/orientechnologies/orient/core/sql/executor/OInternalResultSet.java b/core/src/main/java/com/orientechnologies/orient/core/sql/executor/OInternalResultSet.java
@@ -56,4 +56,10 @@ public void reset() {
   public int size() {
     return content.size();
   }
+
+  public OInternalResultSet copy() {
+    OInternalResultSet result = new OInternalResultSet();
+    result.content = this.content;
+    return result;
+  }
 }