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;
+ }
 }