Fix empty queries (#378)

* fix empty queries * update changelog
ultram4rine · Apr 13, 2023 · 0a01c59 · 0a01c59
1 parent 241e527
commit 0a01c59
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Fixed
+
+- Fix empty output from queries that should return output.
+
 ## [0.3.0] - 2023-04-09
 
 ### Added

diff --git a/src/ls/driver.ts b/src/ls/driver.ts
@@ -56,7 +56,11 @@ export default class ClickHouseDriver
     return this.open().then((ch) => {
       return new Promise<NSDatabase.IResult[]>(async (resolve) => {
         const { requestId } = opt;
-        const method = query.toString().startsWith("SELECT") ? "query" : "exec";
+        const queryStart = query.toString().trimStart().toUpperCase();
+        const method =
+          queryStart.startsWith("SELECT") || queryStart.startsWith("WITH")
+            ? "query"
+            : "exec";
 
         try {
           if (method === "query") {