Wrap the query with double quotes (#2565) (#2570)

(cherry picked from commit 405068b) Signed-off-by: Louis Chu <[email protected]> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
opensearch-project · Mar 20, 2024 · c205b6d · c205b6d
1 parent b4c5415
commit c205b6d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/spark/src/main/java/org/opensearch/sql/spark/asyncquery/model/SparkSubmitParameters.java b/spark/src/main/java/org/opensearch/sql/spark/asyncquery/model/SparkSubmitParameters.java
@@ -86,7 +86,8 @@ public Builder clusterName(String clusterName) {
     }
 
     public Builder query(String query) {
-      config.put(FLINT_JOB_QUERY, query);
+      String wrappedQuery = "\"" + query + "\""; // Wrap the query with double quotes
+      config.put(FLINT_JOB_QUERY, wrappedQuery);
       return this;
     }
 

diff --git a/spark/src/test/java/org/opensearch/sql/spark/dispatcher/SparkQueryDispatcherTest.java b/spark/src/test/java/org/opensearch/sql/spark/dispatcher/SparkQueryDispatcherTest.java
@@ -947,6 +947,7 @@ private String constructExpectedSparkSubmitParameterString(
       authParamConfigBuilder.append(authParams.get(key));
       authParamConfigBuilder.append(" ");
     }
+    query = "\"" + query + "\"";
     return " --class org.apache.spark.sql.FlintJob  --conf"
                + " spark.hadoop.fs.s3.customAWSCredentialsProvider=com.amazonaws.emr.AssumeRoleAWSCredentialsProvider"
                + "  --conf"