pinot-client 1.16.0 → 1.18.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cfa501b22e0b1a69addd7cb33728772fb9717f60fd3986c3fd98c31ba0f3d2a9
-  data.tar.gz: 4e20a3f3c908b5da5afd357dcff60139334929ee086fc58ac044732c5c73205b
+  metadata.gz: 22de970391b9faa4d6f889e1891e84e69c2d045b4f22132d5d892a619b85659d
+  data.tar.gz: 64d0af1ca75e2a1ac9b074d8958d1a8b9df4599fdaf1f0570d95a1395f2faeb0
 SHA512:
-  metadata.gz: bb715615d2af17557a6ea45b97c14b8a00079841c28259708b66431f837290cfaa31ae3b863d13bfaf416d78c568f2f8bfe1d2758eded1e6e6081e84203abdd8
-  data.tar.gz: 738c06c6e6c5629a675fe793ccec5ea919dadae5b39c4da39cc6eefcdac88ec9962a8929f6fd016e16367003aa309f8c595450c736e721e55a27dbf3f792d91e
+  metadata.gz: 4654c1ca25d89bea2e4b1723153e079f016b15af06b736f7c3577475c34eb728ba2e19e8889273a6bf2f39742e9fb243460bb64fae7a9fe701cb7c9b8e4ecbd5
+  data.tar.gz: 52c5de1088080bba980bca3914b2e626c57e8d0e965b0127043ded49e3a165fbc6273e1f485e57111260d191086b0e35937ca5d40a950a46d47f9701a3913545

data/lib/pinot/connection.rb

@@ -32,8 +32,6 @@ module Pinot
         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)

data/lib/pinot/grpc_transport.rb

@@ -18,7 +18,7 @@ module Pinot
     def execute(broker_address, request)
       stub = build_stub(broker_address)
       grpc_request = build_request(request)
-      call_opts = build_call_opts
+      call_opts = build_call_opts(request)
 
       grpc_response = stub.submit(grpc_request, **call_opts)
       BrokerResponse.from_json(grpc_response.payload)
@@ -55,12 +55,19 @@ module Pinot
     def build_query_options(request)
       parts = ["groupByMode=sql", "responseFormat=sql"]
       parts << "useMultistageEngine=true" if request.use_multistage_engine
+      parts << "timeoutMs=#{request.query_timeout_ms}" if request.query_timeout_ms
       parts.join(";")
     end
 
-    def build_call_opts
+    def build_call_opts(request)
       opts = {}
-      opts[:deadline] = Time.now + @config.timeout if @config.timeout
+      # Per-request timeout takes precedence over config-level timeout
+      timeout_s = if request.query_timeout_ms
+                    request.query_timeout_ms / 1000.0
+                  elsif @config.timeout
+                    @config.timeout
+                  end
+      opts[:deadline] = Time.now + timeout_s if timeout_s
       opts[:metadata] = @config.extra_metadata unless @config.extra_metadata.empty?
       opts
     end

data/lib/pinot/simple_broker_selector.rb

@@ -4,6 +4,8 @@ module Pinot
 
     def initialize(broker_list)
       @broker_list = broker_list.dup.freeze
+      @mutex = Mutex.new
+      @index = 0
     end
 
     def init
@@ -12,7 +14,11 @@ module Pinot
 
     def select_broker(_table)
       raise BrokerNotFoundError, "no pre-configured broker lists" if @broker_list.empty?
-      @broker_list.sample
+      @mutex.synchronize do
+        broker = @broker_list[@index % @broker_list.size]
+        @index += 1
+        broker
+      end
     end
   end
 end

data/lib/pinot/transport.rb

@@ -38,6 +38,16 @@ module Pinot
       end
     end
 
+    def close
+      @reaper.kill rescue nil
+      @pool_mutex.synchronize do
+        @pool.each_value do |entries|
+          entries.each { |entry| entry.http.finish rescue nil }
+        end
+        @pool.clear
+      end
+    end
+
     private
 
     def with_connection(url)

data/lib/pinot/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pinot-client
 version: !ruby/object:Gem::Version
-  version: 1.16.0
+  version: 1.18.0
 platform: ruby
 authors:
 - Xiang Fu
@@ -115,7 +115,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.6'
+      version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="