pinot-client 1.24.0 → 1.25.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b893ea9b2cefc1aca8cf86940399d36be6e071e88adc8023affb660d94ad8ce7
-  data.tar.gz: 539a393551125f7169975b0e641f4d1b60fd0cc6390fc1c9c10b4f86c4036db0
+  metadata.gz: 1f76e9f295ba992d3a6c2ec7a4dc738230c8676bda0f64bafb3aa85af1530c34
+  data.tar.gz: 4ea8dfd7e0c530f38c860928c18c1facc76d5fabb7667cf7b802864629365dd1
 SHA512:
-  metadata.gz: 13d9d34e4a86ac6d62b15db000f4220e28cd5498e78cfdb42b2c9edcfac5645c68332a43048d96bc16089381fc4ea9c3d0ce429a95480c481766d1ebd9dfd6ca
-  data.tar.gz: 679bde8940819c309cbc351712cf2c4d4f3cfa1601e81ae023e5bf03a944404bd907b1f0a6fa44c2d5925909f3b8ce6533adf2a91cfea5aa284be7ab86afc9cf
+  metadata.gz: 2a07d7e620bf3adbbcd2f809c37854a1c96da7e1bbb416cbe994d6ce7ea5d17620839aad66b2059e5f1f0de683cd5a405868d1729b5da7dbba19ff820c1ac704
+  data.tar.gz: 2da45bfbb16a63ed13c72fdec7c2684511d27caac9a211155135a3697c88cbf36686a7ded2d569859cb751e8dac59a04b7d91aaf06b3d06136461a217166924e

data/lib/pinot/connection.rb

@@ -50,6 +50,35 @@ module Pinot
       raise e
     end
 
+    def execute_many(queries, max_concurrency: nil)
+      return [] if queries.empty?
+
+      results  = Array.new(queries.size)
+      # Queue acts as a counting semaphore: pre-filled with N tokens.
+      sem = max_concurrency ? build_semaphore(max_concurrency) : nil
+
+      threads = queries.each_with_index.map do |item, idx|
+        table      = item[:table]           || item["table"]           || ""
+        query      = item[:query]           || item["query"]           || ""
+        timeout_ms = item[:query_timeout_ms] || item["query_timeout_ms"]
+
+        Thread.new do
+          sem&.pop   # acquire
+          begin
+            resp = execute_sql(table, query, query_timeout_ms: timeout_ms)
+            results[idx] = QueryResult.new(table: table, query: query, response: resp, error: nil)
+          rescue => e
+            results[idx] = QueryResult.new(table: table, query: query, response: nil, error: e)
+          ensure
+            sem&.push(:token)  # release
+          end
+        end
+      end
+
+      threads.each(&:join)
+      results
+    end
+
     def prepare(table, query_template)
       raise ArgumentError, "table name cannot be empty" if table.nil? || table.strip.empty?
       raise ArgumentError, "query template cannot be empty" if query_template.nil? || query_template.strip.empty?
@@ -101,6 +130,12 @@ module Pinot
 
     private
 
+    def build_semaphore(n)
+      q = SizedQueue.new(n)
+      n.times { q.push(:token) }
+      q
+    end
+
     def run_with_circuit_breaker(broker, &block)
       return yield unless @circuit_breaker_registry
       @circuit_breaker_registry.for(broker).call(broker, &block)

data/lib/pinot/query_result.rb

@@ -0,0 +1,11 @@
+module Pinot
+  QueryResult = Struct.new(:table, :query, :response, :error, keyword_init: true) do
+    def success?
+      error.nil?
+    end
+
+    def error?
+      !error.nil?
+    end
+  end
+end

data/lib/pinot/version.rb

@@ -1,3 +1,3 @@
 module Pinot
-  VERSION = "1.24.0"
+  VERSION = "1.25.0"
 end

data/lib/pinot.rb

@@ -19,6 +19,7 @@ require_relative "pinot/controller_response"
 require_relative "pinot/controller_based_broker_selector"
 require_relative "pinot/transport"
 require_relative "pinot/circuit_breaker"
+require_relative "pinot/query_result"
 require_relative "pinot/connection"
 require_relative "pinot/prepared_statement"
 require_relative "pinot/connection_factory"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pinot-client
 version: !ruby/object:Gem::Version
-  version: 1.24.0
+  version: 1.25.0
 platform: ruby
 authors:
 - Xiang Fu
@@ -119,6 +119,7 @@ files:
 - lib/pinot/proto/broker_service.proto
 - lib/pinot/proto/broker_service_pb.rb
 - lib/pinot/proto/broker_service_services_pb.rb
+- lib/pinot/query_result.rb
 - lib/pinot/request.rb
 - lib/pinot/response.rb
 - lib/pinot/simple_broker_selector.rb