rb_snowflake_client 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: df4fb0a8172d346a1cf606a8f5ed5333725c1459dcaefc22ae149387c052e3aa
-  data.tar.gz: b8d9ebe83e5553aca0be2983b8b47857b447afb304147bda40da2661bca791e4
+  metadata.gz: 4aa627645364f92fc0e0d9ef949519a7649088e43de1e9ca7480ba284a9efba5
+  data.tar.gz: e18d9a6a1f297cbbb5a10aafcac9dce440c43e0ca430f0c6310c31cea28b4507
 SHA512:
-  metadata.gz: 6f57ff1d40be759b042767cc8cb35c705f2228b638790659f0d9b238cdad9afcf8ebab389705e26bab87a89f64aba25d22feb5096f67312fd90e4712dd45142c
-  data.tar.gz: 190ee97111afb77599e7d0e2bacdf4ab4c2d1b374bc7c0fb976e155809ab0f2f8a8e3e4d2f72df71619a468e6616de96a8a8c16db682610265ae405b05f43241
+  metadata.gz: 80ac86515ff08d723c5eff134ac090b60c079695da9c68cf39b06f8db5d1661f700b78930f5da2c19b3f4686563d4dbcda47df70f9ad35200af9e7582b09ea8b
+  data.tar.gz: d4c10ff8bba434ccd0ee380bbaaf9d14bfc7aed22543c34397fd6e1f83611daab65efeacf8caf241a98fd5e2a0f3591592b47f7c76de4997cfe2ea2fb951a5b8

data/Gemfile.lock

@@ -1,7 +1,13 @@
 PATH
   remote: .
   specs:
-    rb_snowflake_client (0.1.1)
+    rb_snowflake_client (0.1.2)
+      concurrent-ruby (>= 1.2)
+      connection_pool (>= 2.4)
+      dotenv (>= 2.8)
+      jwt (>= 2.7)
+      oj (>= 3.16)
+      retryable (>= 3.0)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -32,6 +32,7 @@ client = RubySnowflake::Client.new(
   "snowflake-account",                                  # typically your subdomain
   "snowflake-user",                                     # Your snowflake user
   "some_warehouse",                                     # The name of your warehouse to use by default
+  "some_database",                                      # The name of the database in the context of which the queries will run
   max_connections: 12,                                  # Config options can be passed in
   connection_timeout: 45,                               # See below for the full set of options
 )
@@ -48,6 +49,7 @@ Available ENV variables (see below in the config section for details)
 `SNOWFLAKE_ACCOUNT`
 `SNOWFLAKE_USER`
 `SNOWFLAKE_DEFAULT_WAREHOUSE`
+`SNOWFLAKE_DEFAULT_DATABASE`
 `SNOWFLAKE_JWT_TOKEN_TTL`
 `SNOWFLAKE_CONNECTION_TIMEOUT`
 `SNOWFLAKE_MAX_CONNECTIONS`
@@ -75,6 +77,14 @@ result = client.query("SELECT * FROM HUGETABLE", streaming: true)
 result.each do |row|
   puts row
 end
+
+
+# You can also overwrite the database specified in the initializer, and run
+# your query with a different context.
+result = client.query("SELECT * FROM SECRET_TABLE", database: "OTHER_DB")
+result.each do |row|
+  puts row
+end
 ```
 
 # Configuration Options
@@ -97,6 +107,7 @@ Example configuration:
   client.max_connections = 24
   client.http_retries = 1
 end
+```
 
 # Gotchas
 
@@ -153,6 +164,7 @@ client = RubySnowflake::Client.new(
   "snowflake-account",                                  # typically your subdomain
   "snowflake-user",                                     # Your snowflake user
   "some_warehouse",                                     # The name of your warehouse to use by default
+  "some_database",                                      # The name of the database in the context of which the queries will run
 )
 ```
 

data/lib/ruby_snowflake/client.rb

@@ -69,6 +69,7 @@ module RubySnowflake
         ENV["SNOWFLAKE_ACCOUNT"],
         ENV["SNOWFLAKE_USER"],
         ENV["SNOWFLAKE_DEFAULT_WAREHOUSE"],
+        ENV["SNOWFLAKE_DEFAULT_DATABASE"],
         jwt_token_ttl: env_option("SNOWFLAKE_JWT_TOKEN_TTL", DEFAULT_JWT_TOKEN_TTL),
         connection_timeout: env_option("SNOWFLAKE_CONNECTION_TIMEOUT", DEFAULT_CONNECTION_TIMEOUT ),
         max_connections: env_option("SNOWFLAKE_MAX_CONNECTIONS", DEFAULT_MAX_CONNECTIONS ),
@@ -78,15 +79,17 @@ module RubySnowflake
       )
     end
 
-    def initialize(uri, private_key, organization, account, user, default_warehouse,
-                   logger: DEFAULT_LOGGER,
-                   log_level: DEFAULT_LOG_LEVEL,
-                   jwt_token_ttl: DEFAULT_JWT_TOKEN_TTL,
-                   connection_timeout: DEFAULT_CONNECTION_TIMEOUT,
-                   max_connections: DEFAULT_MAX_CONNECTIONS,
-                   max_threads_per_query: DEFAULT_MAX_THREADS_PER_QUERY,
-                   thread_scale_factor: DEFAULT_THREAD_SCALE_FACTOR,
-                   http_retries: DEFAULT_HTTP_RETRIES)
+    def initialize(
+      uri, private_key, organization, account, user, default_warehouse, default_database,
+      logger: DEFAULT_LOGGER,
+      log_level: DEFAULT_LOG_LEVEL,
+      jwt_token_ttl: DEFAULT_JWT_TOKEN_TTL,
+      connection_timeout: DEFAULT_CONNECTION_TIMEOUT,
+      max_connections: DEFAULT_MAX_CONNECTIONS,
+      max_threads_per_query: DEFAULT_MAX_THREADS_PER_QUERY,
+      thread_scale_factor: DEFAULT_THREAD_SCALE_FACTOR,
+      http_retries: DEFAULT_HTTP_RETRIES
+    )
       @base_uri = uri
       @private_key_pem = private_key
       @organization = organization
@@ -94,6 +97,7 @@ module RubySnowflake
       @user = user
       @default_warehouse = default_warehouse
       @public_key_fingerprint = public_key_fingerprint(@private_key_pem)
+      @default_database = default_database
 
       # set defaults for config settings
       @logger = logger
@@ -110,12 +114,15 @@ module RubySnowflake
       @token_semaphore = Concurrent::Semaphore.new(1)
     end
 
-    def query(query, warehouse: nil, streaming: false)
+    def query(query, warehouse: nil, streaming: false, database: nil)
       warehouse ||= @default_warehouse
+      database ||= @default_database
 
       response = nil
       connection_pool.with do |connection|
-        request_body = { "statement" => query, "warehouse" => warehouse }
+        request_body = {
+          "statement" => query, "warehouse" => warehouse, "database" =>  database
+        }
 
         response = request_with_auth_and_headers(
           connection,

data/lib/ruby_snowflake/version.rb

@@ -1,3 +1,3 @@
 module RubySnowflake
-  VERSION = '0.1.1'
+  VERSION = '0.1.2'
 end

data/rb_snowflake_client.gemspec

@@ -18,4 +18,10 @@ Gem::Specification.new do |s|
   end
 
   s.require_paths = ["lib"]
+  s.add_dependency "concurrent-ruby", ">= 1.2"
+  s.add_dependency "connection_pool", ">= 2.4"
+  s.add_dependency "dotenv", ">= 2.8"
+  s.add_dependency "jwt", ">= 2.7"
+  s.add_dependency "oj", ">= 3.16"
+  s.add_dependency "retryable", ">= 3.0"
 end

metadata

@@ -1,15 +1,99 @@
 --- !ruby/object:Gem::Specification
 name: rb_snowflake_client
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Rinsed
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-12-01 00:00:00.000000000 Z
-dependencies: []
+date: 2023-12-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: connection_pool
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.4'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: jwt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.7'
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.16'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.16'
+- !ruby/object:Gem::Dependency
+  name: retryable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.0'
 description: 'Using the HTTP V2 Api for Snowflake runs queries & creates native Ruby
   objects.