feat: add support for multiple json contents in insert queries

orientechnologies · Nov 27, 2023 · 34aa90e · 34aa90e
1 parent 783da28
commit 34aa90e
Show file tree

Hide file tree

Showing 5 changed files with 3,745 additions and 3,615 deletions.
diff --git a/core/src/main/grammar/OrientSQL.jjt b/core/src/main/grammar/OrientSQL.jjt
@@ -1918,9 +1918,17 @@ OInsertBody InsertBody():
             ( <CONTENT>
                 (
                     content = Json() { jjtThis.addContent(content);}
-                    |
+                    | 
                     inputParamater = InputParameter() { jjtThis.addContentInputParam(inputParamater); }
                 )
+                (
+                	<COMMA>
+                	(
+                	    content = Json() { jjtThis.addContent(content);}
+                        | 
+                        inputParamater = InputParameter() { jjtThis.addContentInputParam(inputParamater); }
+                	)                	
+                )*
             )
          )
     )

diff --git a/...src/main/java/com/orientechnologies/orient/core/sql/executor/OInsertExecutionPlanner.java b/...src/main/java/com/orientechnologies/orient/core/sql/executor/OInsertExecutionPlanner.java
@@ -149,6 +149,18 @@ private void handleCreateRecord(
         && body.getValueExpressions().size() > 0) {
       tot = body.getValueExpressions().size();
     }
+    if (body != null
+        && body.getContentInputParam() != null
+        && body.getContentInputParam().size() > 0) {
+      tot = body.getContentInputParam().size();
+      if (body != null && body.getContent() != null && body.getContent().size() > 0) {
+        tot += body.getContent().size();
+      }
+    } else {
+      if (body != null && body.getContent() != null && body.getContent().size() > 0) {
+        tot = body.getContent().size();
+      }
+    }
     result.chain(new CreateRecordStep(ctx, tot, profilingEnabled));
   }