clickhouse-native 0.2.0-aarch64-linux-gnu → 0.3.0-aarch64-linux-gnu

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dd8c58eb13a3296eadce8aaee22141dccdcfffb76ea0f478b80c6929f3606d46
-  data.tar.gz: e5c2eab3b02b4421781a5bb0007ad45a90e85fee3500e3070fbe72ee69f253fd
+  metadata.gz: 39aa2d8f1b276afbfbeb825e1736803e9f5f0b6cfb75b7bc1d26991bddee1831
+  data.tar.gz: f54a40e3820d497ea7d61be1a632cace76241c130a67482ee96d54de0b94b77c
 SHA512:
-  metadata.gz: 98fbe2d751559f0658f40f524246cc6fb6ea02fdb34073828652d9a2bf96cda3d752e86d8a0485c1c6de5e272161926f8ec716694451ab236b67416a7b6b38ac
-  data.tar.gz: b4c8ccac5d8e2f225c17dce7836ebcba9fedff8a2e3bfba02cd24eb5bb4c0385e6b4d8adef17fbf03454299e30c21b8be4aecb5ad58685a97bc5c94b79117e52
+  metadata.gz: 1fbb467425c7861c8284c29bb49094eaea5c6f2964b7b9fcfebc06b6c44eeb0ff7c8ee4f2a555c51a3514ffb05a310310d5e6be114f292b0e02ac34dd9c2a1ac
+  data.tar.gz: cb79e291335acbbe15fcad4c2f20cc61e1efebf1fe95e65f74ba29513bcbdf6dcd4ca21f59c24292fe3b6c3069eb3667e33b0ecb43e37d0ce09041bfa69c5522

data/lib/clickhouse_native/pool.rb

@@ -4,11 +4,20 @@ require "connection_pool"
 
 module ClickhouseNative
   class Pool
+    attr_reader :host, :port, :database
+
     def initialize(host:, port:, database: "default", user: "default", password: "",
-                   compression: :none, logger: nil, pool_size: 5, pool_timeout: 5)
+                   compression: :none, logger: nil, settings: {},
+                   pool_size: 5, pool_timeout: 5)
+      @host = host
+      @port = port
+      @database = database
       client_kwargs = { host:, port:, database:, user:, password:, compression:, logger: }
+      set_sql = settings_sql(settings)
       @pool = ConnectionPool.new(size: pool_size, timeout: pool_timeout) do
-        Client.new(**client_kwargs)
+        client = Client.new(**client_kwargs)
+        client.execute(set_sql) if set_sql
+        client
       end
     end
 
@@ -47,5 +56,21 @@ module ClickhouseNative
     def describe_table(table, db_name: nil)
       @pool.with { |c| c.describe_table(table, db_name:) }
     end
+
+    private
+
+    # Render a `SET key1 = val1, key2 = val2` statement once at pool setup
+    # so every checked-out connection starts with the same session
+    # settings. Matches how the HTTP driver injected global_params per
+    # request. Values: Integer / Float render bare; anything else is
+    # quoted as a SQL string literal.
+    def settings_sql(settings)
+      return nil if settings.nil? || settings.empty?
+      parts = settings.map do |k, v|
+        literal = v.is_a?(Numeric) ? v.to_s : "'#{v.to_s.gsub("'", "''")}'"
+        "#{k} = #{literal}"
+      end
+      "SET #{parts.join(', ')}"
+    end
   end
 end

data/lib/clickhouse_native/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ClickhouseNative
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: clickhouse-native
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: aarch64-linux-gnu
 authors:
 - Yuri Smirnov