presto-client 0.5.13 → 0.5.14
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/ChangeLog +4 -0
- data/lib/presto/client/query.rb +4 -0
- data/lib/presto/client/statement_client.rb +7 -0
- data/lib/presto/client/version.rb +1 -1
- data/spec/statement_client_spec.rb +16 -0
- metadata +2 -2
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA1:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 2569367597808d3c63e5a35556fbe14009992dda
|
4
|
+
data.tar.gz: e0a85cb6e8ca545005fba0d46e6c7a87f1a70b4e
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: '0598af44256badcf033967ed6835944a580f6f326428f776ac19d153f4d462caf9770934e2a4aa35f1ba633860d333cd81f67c758e594f0e8d58aeb87da70649'
|
7
|
+
data.tar.gz: ba6f95f3d41d50ebaafe7047cbfbe4d63e33add4e9d805d50e134cee53ce1527c445b2a80e9243fb8389dcdc1b6fbacd1bcd93350a089b0fec6ea02f37e11d2d
|
data/ChangeLog
CHANGED
data/lib/presto/client/query.rb
CHANGED
@@ -51,6 +51,7 @@ module Presto::Client
|
|
51
51
|
|
52
52
|
if next_uri
|
53
53
|
response = faraday_get_with_retry(next_uri)
|
54
|
+
@results_headers = response.headers
|
54
55
|
@results = @models::QueryResults.decode(parse_body(response))
|
55
56
|
else
|
56
57
|
post_query_request!
|
@@ -78,6 +79,7 @@ module Presto::Client
|
|
78
79
|
raise PrestoHttpError.new(response.status, "Failed to start query: #{response.body} (#{response.status})")
|
79
80
|
end
|
80
81
|
|
82
|
+
@results_headers = response.headers
|
81
83
|
@results = decode_model(uri, parse_body(response), @models::QueryResults)
|
82
84
|
end
|
83
85
|
|
@@ -111,6 +113,10 @@ module Presto::Client
|
|
111
113
|
@results
|
112
114
|
end
|
113
115
|
|
116
|
+
def current_results_headers
|
117
|
+
@results_headers
|
118
|
+
end
|
119
|
+
|
114
120
|
def has_next?
|
115
121
|
!!@results.next_uri
|
116
122
|
end
|
@@ -122,6 +128,7 @@ module Presto::Client
|
|
122
128
|
|
123
129
|
uri = @results.next_uri
|
124
130
|
response = faraday_get_with_retry(uri)
|
131
|
+
@results_headers = response.headers
|
125
132
|
@results = decode_model(uri, parse_body(response), @models::QueryResults)
|
126
133
|
|
127
134
|
raise_if_timeout!
|
@@ -132,6 +132,22 @@ describe Presto::Client::StatementClient do
|
|
132
132
|
end.should raise_error(TypeError, /String to Hash/)
|
133
133
|
end
|
134
134
|
|
135
|
+
it "receives headers of POST" do
|
136
|
+
stub_request(:post, "localhost/v1/statement").
|
137
|
+
with(body: query).to_return(body: response_json2.to_json, headers: {"X-Test-Header" => "123"})
|
138
|
+
|
139
|
+
sc = StatementClient.new(faraday, query, options.merge(http_open_timeout: 1))
|
140
|
+
sc.current_results_headers["X-Test-Header"].should == "123"
|
141
|
+
end
|
142
|
+
|
143
|
+
it "receives headers of POST through Query" do
|
144
|
+
stub_request(:post, "localhost/v1/statement").
|
145
|
+
with(body: query).to_return(body: response_json2.to_json, headers: {"X-Test-Header" => "123"})
|
146
|
+
|
147
|
+
q = Presto::Client.new(options).query(query)
|
148
|
+
q.current_results_headers["X-Test-Header"].should == "123"
|
149
|
+
end
|
150
|
+
|
135
151
|
describe '#query_info' do
|
136
152
|
let :headers do
|
137
153
|
{
|
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: presto-client
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.5.
|
4
|
+
version: 0.5.14
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Sadayuki Furuhashi
|
8
8
|
autorequire:
|
9
9
|
bindir: bin
|
10
10
|
cert_chain: []
|
11
|
-
date:
|
11
|
+
date: 2019-01-30 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: faraday
|