Support multiple statement queries

Author: formulahendry

CHANGELOG.md

@@ -1,3 +1,6 @@
+## 0.1.0 (2017-11-30)
+* Support multiple statement queries
+
 ## 0.0.3 (2017-11-26)
 * Activate extension when running MySQL query
 

package.json

@@ -2,14 +2,15 @@
     "name": "vscode-mysql",
     "displayName": "MySQL",
     "description": "MySQL management tool",
-    "version": "0.0.3",
+    "version": "0.1.0",
     "publisher": "formulahendry",
     "icon": "logo.png",
     "engines": {
         "vscode": "^1.18.0"
     },
     "categories": [
-        "Languages"
+        "Languages",
+        "Azure"
     ],
     "keywords": [
         "SQL",

src/common/utility.ts

@@ -44,25 +44,34 @@ export class Utility {
 
         sql = sql ? sql : vscode.window.activeTextEditor.document.getText();
         connectionOptions = connectionOptions ? connectionOptions : Global.activeConnection;
+        connectionOptions.multipleStatements = true;
         const connection = mysql.createConnection(connectionOptions);
 
         OutputChannel.appendLine("[Start] Executing MySQL query...");
-        Utility.queryPromise<any>(connection, sql)
-            .then((result) => {
-                if (Array.isArray(result)) {
-                    OutputChannel.appendLine(asciitable(result));
+        connection.query(sql, (err, rows) => {
+            if (Array.isArray(rows)) {
+                if (rows.some(((row) => Array.isArray(row)))) {
+                    rows.forEach((row) => {
+                        if (Array.isArray(row)) {
+                            OutputChannel.appendLine(asciitable(row));
+                        } else {
+                            OutputChannel.appendLine(JSON.stringify(row));
+                        }
+                    });
                 } else {
-                    OutputChannel.appendLine(JSON.stringify(result));
+                    OutputChannel.appendLine(asciitable(rows));
                 }
-                AppInsightsClient.sendEvent("runQuery.end", { Result: "Success" });
-            })
-            .catch((err) => {
+            } else {
+                OutputChannel.appendLine(JSON.stringify(rows));
+            }
+            if (err) {
                 OutputChannel.appendLine(err);
                 AppInsightsClient.sendEvent("runQuery.end", { Result: "Fail", ErrorMessage: err });
-            })
-            .then(() => {
-                OutputChannel.appendLine("[Done] Finished MySQL query.");
-            });
+            } else {
+                AppInsightsClient.sendEvent("runQuery.end", { Result: "Success" });
+            }
+            OutputChannel.appendLine("[Done] Finished MySQL query.");
+        });
     }
 
     public static async createSQLTextDocument(sql: string = "") {

src/model/connection.ts

@@ -4,4 +4,5 @@ export interface IConnection {
     readonly password?: string;
     readonly port: string;
     readonly database?: string;
+    multipleStatements?: boolean;
 }