pinot-client 1.22.0 → 1.23.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8fa5d08fb08bee353252e2da56932948941faa5fdbdd4f3410e5ff8887755e04
-  data.tar.gz: c8bb9cbe06c5d209d8f9f63031776fc604a153730b4be207b7fd6d8a3b69e5f0
+  metadata.gz: 0752e62700790e556388c588d9a305049a5935fc4da0afdefeecab830a9dde0c
+  data.tar.gz: 4f5b1e86e84817febf245ec13498ec300ac68688c691eda58a2f4aafa79d219c
 SHA512:
-  metadata.gz: 6d5c85b83effd2ce352b09c03d350eedd817415a18994ed23ef7be972f80dfb8d72cbbbace9c83bfddedb095b6695af0b8b4538f2f0023ee40844e6b554849a9
-  data.tar.gz: c450ffb393a8798ae3ea3e608cf2cb976bdf5aeb6f7145121764c41bed33069807fdb3087d02c3963552f79568fb6f2020088794cccf47f97218d10fe7041c99
+  metadata.gz: 684b1e6b39e4e1f5f84bed527de2b50f056d101b81ba7621549454d81cc08f49dd044ce2d3dfa5cbb8084075339c0b4c45a89f32c7cf185f1aa73e6c9e40ec95
+  data.tar.gz: 4ea1edf2abac766c358f0e99ed073a05e500d19fb77101475c96f6b82fa4d308a0025565bc1d8d55103da7c7af470654343f9b2b5b57366838e257853e493223

data/lib/pinot/connection.rb

@@ -27,13 +27,14 @@ module Pinot
       @trace = false
     end
 
-    def execute_sql(table, query, query_timeout_ms: nil)
+    def execute_sql(table, query, query_timeout_ms: nil, headers: {})
       Pinot::Instrumentation.instrument(table: table, query: query) do
         logger.debug "Executing SQL on table=#{table}: #{query}"
         broker = @broker_selector.select_broker(table)
         effective_timeout = query_timeout_ms || @query_timeout_ms
         run_with_circuit_breaker(broker) do
-          @transport.execute(broker, build_request(query, timeout_ms: effective_timeout))
+          @transport.execute(broker, build_request(query, timeout_ms: effective_timeout),
+                             extra_request_headers: headers)
         end
       end
     end
@@ -42,11 +43,10 @@ module Pinot
       execute_sql(table, query, query_timeout_ms: timeout_ms)
     end
 
-    def execute_sql_with_params(table, query_pattern, params, query_timeout_ms: nil)
+    def execute_sql_with_params(table, query_pattern, params, query_timeout_ms: nil, headers: {})
       query = format_query(query_pattern, params)
-      execute_sql(table, query, query_timeout_ms: query_timeout_ms)
+      execute_sql(table, query, query_timeout_ms: query_timeout_ms, headers: headers)
     rescue => e
-      # Re-raise format errors directly (they already have the right message)
       raise e
     end
 

data/lib/pinot/prepared_statement.rb

@@ -60,7 +60,7 @@ module Pinot
       nil
     end
 
-    def execute
+    def execute(headers: {})
       @mutex.synchronize do
         raise PreparedStatementClosedError, "prepared statement is closed" if @closed
         @parameters.each_with_index do |p, i|
@@ -72,10 +72,10 @@ module Pinot
       rescue => e
         raise "failed to build query: #{e.message}"
       end
-      @connection.execute_sql(@table, query)
+      @connection.execute_sql(@table, query, headers: headers)
     end
 
-    def execute_with_params(*params)
+    def execute_with_params(*params, headers: {})
       @mutex.synchronize { raise PreparedStatementClosedError, "prepared statement is closed" if @closed }
       if params.length != @param_count
         raise "expected #{@param_count} parameters, got #{params.length}"
@@ -85,7 +85,7 @@ module Pinot
       rescue => e
         raise "failed to build query: #{e.message}"
       end
-      @connection.execute_sql(@table, query)
+      @connection.execute_sql(@table, query, headers: headers)
     end
 
     def clear_parameters

data/lib/pinot/transport.rb

@@ -188,7 +188,7 @@ module Pinot
       @retry_interval_ms = retry_interval_ms
     end
 
-    def execute(broker_address, request)
+    def execute(broker_address, request, extra_request_headers: {})
       logger.debug "Pinot query to #{broker_address}: #{request.query}"
 
       attempts = 0
@@ -202,6 +202,7 @@ module Pinot
         headers = DEFAULT_HEADERS
           .merge(@extra_headers)
           .merge("X-Correlation-Id" => SecureRandom.uuid)
+          .merge(extra_request_headers)
 
         resp = @http_client.post(url, body: body, headers: headers)
 

data/lib/pinot/version.rb

@@ -1,3 +1,3 @@
 module Pinot
-  VERSION = "1.22.0"
+  VERSION = "1.23.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pinot-client
 version: !ruby/object:Gem::Version
-  version: 1.22.0
+  version: 1.23.0
 platform: ruby
 authors:
 - Xiang Fu