rb_snowflake_client 1.0.6 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: af3e56d97ebae43ae931c9c50784d7a3af0fe0e4a70f7e2a28c932d30371eff5
-  data.tar.gz: cdd2e9f55805e8d27e344925b8917b6d292aa1b07ed87ac8443060d31e224ccc
+  metadata.gz: '089c46288d7dc44f8717542fce8732ce32c65db796e5ac5d49fa6fff6cb1f29d'
+  data.tar.gz: '035589f8bcc072830f037c333379a2059eb6a205ad8ae6daecb2bb2ea9c4761a'
 SHA512:
-  metadata.gz: 3c3aae7bf833cab432ea952419e951f51584100766eccd07d3a978c449fd1a2fd532a5756a0789242ff1baa7319dc360a162ba608d502d12436837f456033b97
-  data.tar.gz: cf1828f9a00b3be7c6030c5bc3e6f1be834206e364be477aa944faffaaf3638dc65a61bbc215c6f7428fd103ee62a1077e9658fa08055479e632b29d15f008f7
+  metadata.gz: e390973b02ce165e3bb5722c84cc79cb8c1fcc8457b57ceaf9317f2675fdfe2e5b6276d7c8a496ec5605ab66c15fcdfa2247182d1f54bed39df3c8ce4240fed7
+  data.tar.gz: '092ba2997da8f64f6e9e7ff912d7bdb43882681f5d564028f9a9c6ad51854515e237de27b51de35329acb3fd021de3d4212551cbc31a58942592627805f2e298'

data/.github/dependabot.yml

@@ -0,0 +1,13 @@
+version: 2
+updates:
+  #
+  # BUNDLER
+  #
+  - package-ecosystem: bundler
+    directory: '/'
+    schedule:
+      interval: weekly
+    allow:
+      - dependency-type: direct
+      - dependency-type: indirect
+    open-pull-requests-limit: 10

data/Gemfile

@@ -6,13 +6,7 @@ source "https://rubygems.org"
 gemspec
 
 gem "bundler"
-gem "concurrent-ruby"
-gem "connection_pool"
-gem "dotenv"
-gem "jwt"
-gem "oj"
 gem "rake"
-gem "retryable"
 
 group :development do
   gem "parallel"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rb_snowflake_client (1.0.6)
+    rb_snowflake_client (1.1.1)
       concurrent-ruby (>= 1.2)
       connection_pool (>= 2.4)
       dotenv (>= 2.8)
@@ -12,52 +12,49 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
+    base64 (0.2.0)
+    bigdecimal (3.1.8)
     coderay (1.1.3)
-    concurrent-ruby (1.2.2)
+    concurrent-ruby (1.3.1)
     connection_pool (2.4.1)
-    diff-lcs (1.5.0)
-    dotenv (2.8.1)
-    jwt (2.7.1)
-    method_source (1.0.0)
-    oj (3.16.1)
-    parallel (1.23.0)
+    diff-lcs (1.5.1)
+    dotenv (3.1.2)
+    jwt (2.8.1)
+      base64
+    method_source (1.1.0)
+    oj (3.16.3)
+      bigdecimal (>= 3.0)
+    parallel (1.24.0)
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
-    rake (13.1.0)
+    rake (13.2.1)
     retryable (3.0.5)
-    rspec (3.12.0)
-      rspec-core (~> 3.12.0)
-      rspec-expectations (~> 3.12.0)
-      rspec-mocks (~> 3.12.0)
-    rspec-core (3.12.2)
-      rspec-support (~> 3.12.0)
-    rspec-expectations (3.12.3)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.0)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.12.0)
-    rspec-mocks (3.12.6)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.12.0)
-    rspec-support (3.12.1)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.1)
 
 PLATFORMS
-  arm64-darwin-21
   arm64-darwin-22
-  x86_64-linux
+  ruby
 
 DEPENDENCIES
   bundler
-  concurrent-ruby
-  connection_pool
-  dotenv
-  jwt
-  oj
   parallel
   pry
   rake
   rb_snowflake_client!
-  retryable
   rspec
 
 BUNDLED WITH
-   2.4.19
+   2.5.4

data/README.md

@@ -105,6 +105,30 @@ Clients are not warehouse specific, you can override the default warehouse per q
 client.query("SELECT * FROM BIGTABLE", warehouse: "FAST_WH")
 ```
 
+## Specifying a schema
+
+```ruby
+client.query("SELECT * FROM BIGTABLE", schema: "MY_SCHEMA")
+```
+
+## Binding parameters
+
+Say we have `BIGTABLE` with a `data` column of a type `VARIANT`.
+
+```ruby
+json_string = '{"valid": "json"}'
+query = "insert into BIGTABLE(data) select parse_json(?)"
+bindings = {
+  "1": {
+    "type": "TEXT",
+    "value": json_string
+  }
+}
+client.query(query, bindings: bindings)
+```
+
+For additional information about binding parameters refer to snowflake documentation: https://docs.snowflake.com/en/developer-guide/sql-api/submitting-requests#using-bind-variables-in-a-statement
+
 # Configuration Options
 
 The client supports the following configuration options, each with their own getter/setter except connection pool options which must be set at construction. Additionally, all except logger can be configured with environment variables (see above, but the pattern is like: "SNOWFLAKE_HTTP_RETRIES". Configuration options can only be set on initialization through `new` or `from_env`.

data/lib/ruby_snowflake/client.rb

@@ -128,7 +128,7 @@ module RubySnowflake
       @_enable_polling_queries = false
     end
 
-    def query(query, warehouse: nil, streaming: false, database: nil, schema: nil)
+    def query(query, warehouse: nil, streaming: false, database: nil, schema: nil, bindings: nil)
       warehouse ||= @default_warehouse
       database ||= @default_database
 
@@ -136,11 +136,11 @@ module RubySnowflake
       response = nil
       connection_pool.with do |connection|
         request_body = {
-          "database" =>  database&.upcase,
+          "warehouse" => warehouse&.upcase,
           "schema" => schema&.upcase,
+          "database" =>  database&.upcase,  
           "statement" => query,
-          "timeout" => @query_timeout,
-          "warehouse" => warehouse&.upcase,
+          "bindings" => bindings
         }
 
         response = request_with_auth_and_headers(
@@ -150,7 +150,7 @@ module RubySnowflake
           Oj.dump(request_body)
         )
       end
-      retreive_result_set(query_start_time, query, response, streaming)
+      retrieve_result_set(query_start_time, query, response, streaming)
     end
 
     alias fetch query
@@ -223,7 +223,7 @@ module RubySnowflake
       def retryable_http_response_code?(code)
         # retry (in order): bad request, forbidden (token expired in flight), method not allowed,
         #   request timeout, too many requests, anything in the 500 range (504 is fairly common)
-        [400, 403, 405, 408, 429, 504].include?(code.to_i) || (500..599).include?(code)
+        [400, 403, 405, 408, 429, 502, 504].include?(code.to_i) || (500..599).include?(code.to_i)
       end
 
       def retryable_log_method
@@ -267,7 +267,7 @@ module RubySnowflake
         false
       end
 
-      def retreive_result_set(query_start_time, query, response, streaming)
+      def retrieve_result_set(query_start_time, query, response, streaming)
         json_body = Oj.load(response.body, OJ_OPTIONS)
         statement_handle = json_body["statementHandle"]
 
@@ -277,18 +277,18 @@ module RubySnowflake
         end
 
         num_threads = number_of_threads_to_use(json_body["resultSetMetaData"]["partitionInfo"].size)
-        retreive_proc = ->(index) { retreive_partition_data(statement_handle, index) }
+        retrieve_proc = ->(index) { retrieve_partition_data(statement_handle, index) }
 
         if streaming
-          StreamingResultStrategy.result(json_body, retreive_proc)
+          StreamingResultStrategy.result(json_body, retrieve_proc)
         elsif num_threads == 1
-          SingleThreadInMemoryStrategy.result(json_body, retreive_proc)
+          SingleThreadInMemoryStrategy.result(json_body, retrieve_proc)
         else
-          ThreadedInMemoryStrategy.result(json_body, retreive_proc, num_threads)
+          ThreadedInMemoryStrategy.result(json_body, retrieve_proc, num_threads)
         end
       end
 
-      def retreive_partition_data(statement_handle, partition_index)
+      def retrieve_partition_data(statement_handle, partition_index)
         partition_response = nil
         connection_pool.with do |connection|
           partition_response = request_with_auth_and_headers(

data/lib/ruby_snowflake/version.rb

@@ -1,3 +1,3 @@
 module RubySnowflake
-  VERSION = "1.0.6"
+  VERSION = "1.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rb_snowflake_client
 version: !ruby/object:Gem::Version
-  version: 1.0.6
+  version: 1.1.1
 platform: ruby
 authors:
 - Rinsed
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-06-05 00:00:00.000000000 Z
+date: 2024-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -105,6 +105,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".github/dependabot.yml"
 - ".github/workflows/ci.yml"
 - ".github/workflows/release-gh-packages.yml"
 - ".github/workflows/release-rubygems.yml"