presto-client 0.5.13 → 0.5.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 25d825cdb3befbfa69b368f05d2267ed4bc3affd
-  data.tar.gz: abbc72ac27018abef03fbe2c5a122cc96aff68e1
+  metadata.gz: 2569367597808d3c63e5a35556fbe14009992dda
+  data.tar.gz: e0a85cb6e8ca545005fba0d46e6c7a87f1a70b4e
 SHA512:
-  metadata.gz: f5988faa2e414bdb31a1368eb3a829e1a30872195a8bf6aa1a0ece60edb832ca64af70f6e3cfa8b2e85e7b20786b2ab943fdfa8b340158fab33d5d8279b8a440
-  data.tar.gz: 00eb3921405da85361bda22d588982e3e8d706ac118bd694cfab94849812bdb25e27ce8e5c0ba43d437d21a1fdbf03d9a09075722e4569082680ce5fa376c25e
+  metadata.gz: '0598af44256badcf033967ed6835944a580f6f326428f776ac19d153f4d462caf9770934e2a4aa35f1ba633860d333cd81f67c758e594f0e8d58aeb87da70649'
+  data.tar.gz: ba6f95f3d41d50ebaafe7047cbfbe4d63e33add4e9d805d50e134cee53ce1527c445b2a80e9243fb8389dcdc1b6fbacd1bcd93350a089b0fec6ea02f37e11d2d

data/ChangeLog

@@ -1,3 +1,7 @@
+2019-01-30 version 0.5.14
+
+* Added `Query#current_results_headers` that returns HTTP response headers
+
 2018-12-10 version 0.5.13
 
 * Added `query_timeout` and `plan_timeout` options with default disabled

data/lib/presto/client/query.rb

@@ -50,6 +50,10 @@ module Presto::Client
       @api.current_results
     end
 
+    def current_results_headers
+      @api.current_results_headers
+    end
+
     def advance
       @api.advance
     end

data/lib/presto/client/statement_client.rb

@@ -51,6 +51,7 @@ module Presto::Client
 
       if next_uri
         response = faraday_get_with_retry(next_uri)
+        @results_headers = response.headers
         @results = @models::QueryResults.decode(parse_body(response))
       else
         post_query_request!
@@ -78,6 +79,7 @@ module Presto::Client
         raise PrestoHttpError.new(response.status, "Failed to start query: #{response.body} (#{response.status})")
       end
 
+      @results_headers = response.headers
       @results = decode_model(uri, parse_body(response), @models::QueryResults)
     end
 
@@ -111,6 +113,10 @@ module Presto::Client
       @results
     end
 
+    def current_results_headers
+      @results_headers
+    end
+
     def has_next?
       !!@results.next_uri
     end
@@ -122,6 +128,7 @@ module Presto::Client
 
       uri = @results.next_uri
       response = faraday_get_with_retry(uri)
+      @results_headers = response.headers
       @results = decode_model(uri, parse_body(response), @models::QueryResults)
 
       raise_if_timeout!

data/lib/presto/client/version.rb

@@ -15,6 +15,6 @@
 #
 module Presto
   module Client
-    VERSION = "0.5.13"
+    VERSION = "0.5.14"
   end
 end

data/spec/statement_client_spec.rb

@@ -132,6 +132,22 @@ describe Presto::Client::StatementClient do
     end.should raise_error(TypeError, /String to Hash/)
   end
 
+  it "receives headers of POST" do
+    stub_request(:post, "localhost/v1/statement").
+      with(body: query).to_return(body: response_json2.to_json, headers: {"X-Test-Header" => "123"})
+
+    sc = StatementClient.new(faraday, query, options.merge(http_open_timeout: 1))
+    sc.current_results_headers["X-Test-Header"].should == "123"
+  end
+
+  it "receives headers of POST through Query" do
+    stub_request(:post, "localhost/v1/statement").
+      with(body: query).to_return(body: response_json2.to_json, headers: {"X-Test-Header" => "123"})
+
+    q = Presto::Client.new(options).query(query)
+    q.current_results_headers["X-Test-Header"].should == "123"
+  end
+
   describe '#query_info' do
     let :headers do
       {

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: presto-client
 version: !ruby/object:Gem::Version
-  version: 0.5.13
+  version: 0.5.14
 platform: ruby
 authors:
 - Sadayuki Furuhashi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-12-10 00:00:00.000000000 Z
+date: 2019-01-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday