Handle drill errors correctly

JohannWilfridCalixte · Feb 8, 2017 · 810efce · 810efce
1 parent e5cfa0f
commit 810efce
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/blazer/adapters/drill_adapter.rb b/lib/blazer/adapters/drill_adapter.rb
@@ -17,8 +17,12 @@ def run_statement(statement, comment)
 
         begin
           response = JSON.parse(http.post(uri.request_uri, data.to_json, header).body)
-          columns = response["columns"]
-          rows = response["rows"].map { |r| r.values }
+          if response["errorMessage"]
+            error = response["errorMessage"]
+          else
+            columns = response["columns"]
+            rows = response["rows"].map { |r| r.values }
+          end
         rescue => e
           error = e.message
         end