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
- #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.
45 46 47 |
# File 'lib/presto/client/query.rb', line 45 def initialize(api) @api = api end |
Class Method Details
.faraday_client(options) ⇒ Object
41 42 43 |
# File 'lib/presto/client/query.rb', line 41 def self.faraday_client() Presto::Client.faraday_client() end |
.kill(query_id, options) ⇒ Object
33 34 35 36 37 38 39 |
# File 'lib/presto/client/query.rb', line 33 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
29 30 31 |
# File 'lib/presto/client/query.rb', line 29 def self.resume(next_uri, ) new StatementClient.new(faraday_client(), nil, , next_uri) end |
.start(query, options) ⇒ Object
25 26 27 |
# File 'lib/presto/client/query.rb', line 25 def self.start(query, ) new StatementClient.new(faraday_client(), query, ) end |
Instance Method Details
#advance ⇒ Object
53 54 55 |
# File 'lib/presto/client/query.rb', line 53 def advance @api.advance end |
#cancel ⇒ Object
119 120 121 |
# File 'lib/presto/client/query.rb', line 119 def cancel @api.cancel_leaf_stage end |
#close ⇒ Object
123 124 125 126 |
# File 'lib/presto/client/query.rb', line 123 def close @api.cancel_leaf_stage nil end |
#columns ⇒ Object
77 78 79 80 81 |
# File 'lib/presto/client/query.rb', line 77 def columns wait_for_columns return @api.current_results.columns end |
#current_results ⇒ Object
49 50 51 |
# File 'lib/presto/client/query.rb', line 49 def current_results @api.current_results end |
#each_row(&block) ⇒ Object
91 92 93 94 95 |
# File 'lib/presto/client/query.rb', line 91 def each_row(&block) each_row_chunk {|chunk| chunk.each(&block) } end |
#each_row_chunk(&block) ⇒ Object
97 98 99 100 101 102 103 104 105 106 107 108 109 |
# File 'lib/presto/client/query.rb', line 97 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
115 116 117 |
# File 'lib/presto/client/query.rb', line 115 def next_uri @api.current_results.next_uri end |
#query_info ⇒ Object
111 112 113 |
# File 'lib/presto/client/query.rb', line 111 def query_info @api.query_info end |
#raise_if_failed ⇒ Object
128 129 130 131 132 133 134 135 136 137 138 139 |
# File 'lib/presto/client/query.rb', line 128 def raise_if_failed if @api.closed? raise PrestoClientError, "Query aborted by user" elsif @api.exception? # query is gone raise @api.exception 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.failure_info) end end |
#rows ⇒ Object
83 84 85 86 87 88 89 |
# File 'lib/presto/client/query.rb', line 83 def rows rows = [] each_row_chunk {|chunk| rows.concat(chunk) } return rows end |