Class: Presto::Client::Query
- Inherits:
-
Object
- Object
- Presto::Client::Query
- Defined in:
- lib/presto/client/query.rb
Class Method Summary collapse
- .faraday_client(options) ⇒ Object
- .kill(query_id, options) ⇒ Object
- .resume(next_uri, options) ⇒ Object
- .start(query, options) ⇒ Object
Instance Method Summary collapse
- #advance ⇒ Object
- #cancel ⇒ Object
- #close ⇒ Object
- #columns ⇒ Object
- #current_results ⇒ Object
- #current_results_headers ⇒ Object
- #each_row(&block) ⇒ Object
- #each_row_chunk(&block) ⇒ Object
-
#initialize(api) ⇒ Query
constructor
A new instance of Query.
- #next_uri ⇒ Object
- #query_info ⇒ Object
- #raise_if_failed ⇒ Object
- #rows ⇒ Object
Constructor Details
#initialize(api) ⇒ Query
Returns a new instance of Query.
46 47 48 |
# File 'lib/presto/client/query.rb', line 46 def initialize(api) @api = api end |
Class Method Details
.faraday_client(options) ⇒ Object
42 43 44 |
# File 'lib/presto/client/query.rb', line 42 def self.faraday_client() Presto::Client.faraday_client() end |
.kill(query_id, options) ⇒ Object
34 35 36 37 38 39 40 |
# File 'lib/presto/client/query.rb', line 34 def self.kill(query_id, ) faraday = faraday_client() response = faraday.delete do |req| req.url "/v1/query/#{query_id}" end return response.status / 100 == 2 end |
.resume(next_uri, options) ⇒ Object
30 31 32 |
# File 'lib/presto/client/query.rb', line 30 def self.resume(next_uri, ) new StatementClient.new(faraday_client(), nil, , next_uri) end |
.start(query, options) ⇒ Object
26 27 28 |
# File 'lib/presto/client/query.rb', line 26 def self.start(query, ) new StatementClient.new(faraday_client(), query, ) end |
Instance Method Details
#advance ⇒ Object
58 59 60 |
# File 'lib/presto/client/query.rb', line 58 def advance @api.advance end |
#cancel ⇒ Object
124 125 126 |
# File 'lib/presto/client/query.rb', line 124 def cancel @api.cancel_leaf_stage end |
#close ⇒ Object
128 129 130 131 |
# File 'lib/presto/client/query.rb', line 128 def close @api.close nil end |
#columns ⇒ Object
82 83 84 85 86 |
# File 'lib/presto/client/query.rb', line 82 def columns wait_for_columns return @api.current_results.columns end |
#current_results ⇒ Object
50 51 52 |
# File 'lib/presto/client/query.rb', line 50 def current_results @api.current_results end |
#current_results_headers ⇒ Object
54 55 56 |
# File 'lib/presto/client/query.rb', line 54 def current_results_headers @api.current_results_headers end |
#each_row(&block) ⇒ Object
96 97 98 99 100 |
# File 'lib/presto/client/query.rb', line 96 def each_row(&block) each_row_chunk {|chunk| chunk.each(&block) } end |
#each_row_chunk(&block) ⇒ Object
102 103 104 105 106 107 108 109 110 111 112 113 114 |
# File 'lib/presto/client/query.rb', line 102 def each_row_chunk(&block) wait_for_data if self.columns == nil raise PrestoError, "Query #{@api.current_results.id} has no columns" end begin if data = @api.current_results.data block.call(data) end end while advance_and_raise end |
#next_uri ⇒ Object
120 121 122 |
# File 'lib/presto/client/query.rb', line 120 def next_uri @api.current_results.next_uri end |
#query_info ⇒ Object
116 117 118 |
# File 'lib/presto/client/query.rb', line 116 def query_info @api.query_info end |
#raise_if_failed ⇒ Object
133 134 135 136 137 138 139 140 141 |
# File 'lib/presto/client/query.rb', line 133 def raise_if_failed if @api.client_aborted? raise PrestoClientError, "Query aborted by user" elsif @api.query_failed? results = @api.current_results error = results.error raise PrestoQueryError.new("Query #{results.id} failed: #{error.}", results.id, error.error_code, error.error_name, error.error_type, error.error_location, error.failure_info) end end |
#rows ⇒ Object
88 89 90 91 92 93 94 |
# File 'lib/presto/client/query.rb', line 88 def rows rows = [] each_row_chunk {|chunk| rows.concat(chunk) } return rows end |