quill-sql 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b3d503d9223878a246289559453c27d45e4393620f9b548433ea5c072fc4c396
-  data.tar.gz: 1820f9f817ab7a29c947555a564e6981b283486cb0cdd0f65396cd01aaaecffd
+  metadata.gz: 5b91688c1cc7cc762aabc68cf5a68306f0fca7b45fcdd8adfef93e845f2c1d1d
+  data.tar.gz: f69676f9e1f4f1046ddef72ae2a68c476f8dc2c28ad3cd856d05e68a4c72c768
 SHA512:
-  metadata.gz: b25af374393d2d9f6bc083f7cb426f631802df7582c0e4bc540c574e54123884e723030ca879dc9c937da21e40e6047be35c9ee59c001ee9bce78339db353ed9
-  data.tar.gz: 7a3251a382f48867d20e7600a4d00f69b95963a435328a4ae3348505ceeb1fe9671834b2715e0f3545f53cee507bad086d87f1acf75150dde218a9005d758d80
+  metadata.gz: 9541d51c08c50c7f03dc34358aba4b48635a2a6f2acdbcda78e6fbcdbdd85bb9bd02db4045398db13445266101576128fb7d121ffafdc5ddc718ab0d49f830df
+  data.tar.gz: 27006da9ca08d16c6ee34fb1c6fa1449b5e448ad140ed7b393c1b3f67c72a472711ec57e58e3eb227cc28ace871744b5f2d339a4cfdc1de4e6fd3f2ca37f6ddc

data/lib/db/databricks.rb

@@ -73,7 +73,6 @@ module DatabricksHelper
 
       json_like_types = ['ARRAY', 'MAP', 'STRUCT', 'JSON']
 
-
       fields = columns.map do |column|
         {
           name: column['name'],
@@ -81,30 +80,34 @@ module DatabricksHelper
         }
       end
 
-      # Transform the data array into rows of hashes
-      rows = result['result']['data_array'].map do |row_array|
-        row_hash = column_names.zip(row_array).to_h
-        transformed_row = {}
-      
-        column_names.each_with_index do |key, idx|
-          value = row_hash[key]
-          type = column_types[idx]
-          # parse JSON if the field is a JSON type
-          transformed_value = if value.is_a?(String) && json_like_types.include?(type.upcase)
-            begin
-              parsed = JSON.parse(value)
-              parsed.is_a?(Array) ? parsed.map(&:to_s) : parsed
-            rescue JSON::ParserError
+      # Handle empty results
+      rows = if result['result'].nil? || result['result']['data_array'].nil?
+        []
+      else
+        # Transform the data array into rows of hashes
+        result['result']['data_array'].map do |row_array|
+          row_hash = column_names.zip(row_array).to_h
+          transformed_row = {}
+        
+          column_names.each_with_index do |key, idx|
+            value = row_hash[key]
+            type = column_types[idx]
+            # parse JSON if the field is a JSON type
+            transformed_value = if value.is_a?(String) && json_like_types.include?(type.upcase)
+              begin
+                parsed = JSON.parse(value)
+                parsed.is_a?(Array) ? parsed.map(&:to_s) : parsed
+              rescue JSON::ParserError
+                value
+              end
+            else
               value
             end
-          else
-            value
+            transformed_row[key] = transformed_value
           end
-      
-          transformed_row[key] = transformed_value
+        
+          transformed_row
         end
-      
-        transformed_row
       end
 
       {

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: quill-sql
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Shawn Magee, Albert Yan