Class: Presto::Client::Query
- Inherits:
-
Object
- Object
- Presto::Client::Query
- Defined in:
- lib/presto/client/query.rb
Class Method Summary collapse
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.
- #query_info ⇒ Object
- #raise_if_failed ⇒ Object
- #rows ⇒ Object
Constructor Details
#initialize(api) ⇒ Query
Returns a new instance of Query.
39 40 41 |
# File 'lib/presto/client/query.rb', line 39 def initialize(api) @api = api end |
Class Method Details
.start(query, options) ⇒ Object
24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# File 'lib/presto/client/query.rb', line 24 def self.start(query, ) server = [:server] unless server raise ArgumentError, ":server option is required" end faraday = Faraday.new(url: "http://#{server}") do |faraday| #faraday.request :url_encoded faraday.response :logger if [:http_debug] faraday.adapter Faraday.default_adapter end new StatementClient.new(faraday, query, ) end |
Instance Method Details
#advance ⇒ Object
47 48 49 |
# File 'lib/presto/client/query.rb', line 47 def advance @api.advance end |
#cancel ⇒ Object
106 107 108 |
# File 'lib/presto/client/query.rb', line 106 def cancel @api.cancel_leaf_stage end |
#close ⇒ Object
110 111 112 113 |
# File 'lib/presto/client/query.rb', line 110 def close @api.cancel_leaf_stage nil end |
#columns ⇒ Object
64 65 66 67 68 69 70 |
# File 'lib/presto/client/query.rb', line 64 def columns wait_for_columns raise_if_failed return @api.current_results.columns end |
#current_results ⇒ Object
43 44 45 |
# File 'lib/presto/client/query.rb', line 43 def current_results @api.current_results end |
#each_row(&block) ⇒ Object
80 81 82 83 84 |
# File 'lib/presto/client/query.rb', line 80 def each_row(&block) each_row_chunk {|chunk| chunk.each(&block) } end |
#each_row_chunk(&block) ⇒ Object
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
# File 'lib/presto/client/query.rb', line 86 def each_row_chunk(&block) wait_for_data raise_if_failed 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 @api.advance end |
#query_info ⇒ Object
102 103 104 |
# File 'lib/presto/client/query.rb', line 102 def query_info @api.query_info end |
#raise_if_failed ⇒ Object
115 116 117 118 119 120 121 122 123 124 125 126 |
# File 'lib/presto/client/query.rb', line 115 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.failure_info) end end |
#rows ⇒ Object
72 73 74 75 76 77 78 |
# File 'lib/presto/client/query.rb', line 72 def rows rows = [] each_row_chunk {|chunk| rows.concat(chunk) } return rows end |