pinot-client 1.15.0 → 1.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9f8ec1d9b9dd17c63f38ddf2d30db5e331d0d1decd66a43335dc9488690ebef2
-  data.tar.gz: 518d0d4c6210660a9270166fdd0131c6bc363862e51449c531512bcaf2b37e8f
+  metadata.gz: cfa501b22e0b1a69addd7cb33728772fb9717f60fd3986c3fd98c31ba0f3d2a9
+  data.tar.gz: 4e20a3f3c908b5da5afd357dcff60139334929ee086fc58ac044732c5c73205b
 SHA512:
-  metadata.gz: 550bef73ba7ad87e2712058e97836b5ad10fd3c8489e1649985fbf6ae487afccee0040bb2169e842161aa8dc788257b80aa62143e4fa028b68197f2a0616ea82
-  data.tar.gz: 957ca4aa0a52098aa7bb7158a8a931764bfd1d4d6bb2630c5eae2356d6719dc0909e67801f1d1db50aa14036c323d1bce9070ae8af240f8660b348190d779049
+  metadata.gz: bb715615d2af17557a6ea45b97c14b8a00079841c28259708b66431f837290cfaa31ae3b863d13bfaf416d78c568f2f8bfe1d2758eded1e6e6081e84203abdd8
+  data.tar.gz: 738c06c6e6c5629a675fe793ccec5ea919dadae5b39c4da39cc6eefcdac88ec9962a8929f6fd016e16367003aa309f8c595450c736e721e55a27dbf3f792d91e

data/lib/pinot/connection.rb

@@ -25,27 +25,24 @@ module Pinot
       @trace = false
     end
 
-    def execute_sql(table, query)
+    def execute_sql(table, query, query_timeout_ms: nil)
       Pinot::Instrumentation.instrument(table: table, query: query) do
         logger.debug "Executing SQL on table=#{table}: #{query}"
         broker = @broker_selector.select_broker(table)
-        @transport.execute(broker, build_request(query))
+        effective_timeout = query_timeout_ms || @query_timeout_ms
+        @transport.execute(broker, build_request(query, timeout_ms: effective_timeout))
       end
     rescue => e
       raise "unable to execute SQL on table #{table}: #{e.message}"
     end
 
     def execute_sql_with_timeout(table, query, timeout_ms)
-      logger.debug "Executing SQL with timeout=#{timeout_ms}ms on table=#{table}: #{query}"
-      broker = @broker_selector.select_broker(table)
-      @transport.execute(broker, build_request(query, timeout_ms: timeout_ms))
-    rescue => e
-      raise "unable to execute SQL on table #{table}: #{e.message}"
+      execute_sql(table, query, query_timeout_ms: timeout_ms)
     end
 
-    def execute_sql_with_params(table, query_pattern, params)
+    def execute_sql_with_params(table, query_pattern, params, query_timeout_ms: nil)
       query = format_query(query_pattern, params)
-      execute_sql(table, query)
+      execute_sql(table, query, query_timeout_ms: query_timeout_ms)
     rescue => e
       # Re-raise format errors directly (they already have the right message)
       raise e

data/lib/pinot/connection_factory.rb

@@ -22,7 +22,8 @@ module Pinot
         transport: transport,
         broker_selector: selector,
         use_multistage_engine: config.use_multistage_engine || false,
-        logger: config.logger
+        logger: config.logger,
+        query_timeout_ms: config.query_timeout_ms
       )
 
       selector.init

data/lib/pinot/version.rb

@@ -1,3 +1,3 @@
 module Pinot
-  VERSION = "1.15.0"
+  VERSION = "1.16.0"
 end

metadata

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