presto-client 0.4.13 → 0.4.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0bee1b8fb22804f12e8a2fe9ee8f9fca6262cf4f
-  data.tar.gz: 2bd49e06c846141c5d909c507716970454565270
+  metadata.gz: 1f05ad7fb71f97a4804e24451a201d8043d3518a
+  data.tar.gz: 369f6523a275ead5350cf42a95f0c22d56b52c22
 SHA512:
-  metadata.gz: 0128a6bdf303a34f156916fb460c4e133277d41de5ff9b839fb19c969e569e71bdef2174a40be8e5bb5b43868ab41d192afb7a9320dbb750ca83990929fd734d
-  data.tar.gz: 30486a370eb854050e71dd0cdec598e17fe0aa3bec43feb6f1b20f7292c8b8d8362e29ce35d06a5757726d547c3f507af2751726092fbbf8580325d17cc66823
+  metadata.gz: 938b3b5ae20a354a659713f3ad29952619403d45f01dbf41a1ca8907659250a7b5597787fdcfc3b028e049cc27e819c39b6721b38a0ab5695d86dcbc6b55e911
+  data.tar.gz: 81353b8f0f53784533c282e2471f1c79d4f28de2cdd8506ceb7e291fb8d7fb3c7d979342d952dc7e47f09b1c4b3a7833abcefed85b425bf9a7ed25c59ee88ea5

data/ChangeLog

@@ -1,3 +1,8 @@
+2016-08-02 version 0.4.14:
+
+* Added support for resuming fetching query results by using new `Query.resume(next_uri, options)` method (@tetrakai++)
+
+
 2016-08-02 version 0.4.13:
 
 * Added support for :http_proxy option to use a HTTP proxy server

data/lib/presto/client/client.rb

@@ -36,6 +36,10 @@ module Presto::Client
       end
     end
 
+    def resume_query(next_uri)
+      return Query.resume(next_uri, @options)
+    end
+
     def run(query)
       q = Query.start(query, @options)
       begin

data/lib/presto/client/query.rb

@@ -22,6 +22,14 @@ module Presto::Client
 
   class Query
     def self.start(query, options)
+      new StatementClient.new(faraday_client(options), query, options)
+    end
+
+    def self.resume(next_uri, options)
+      new StatementClient.new(faraday_client(options), nil, options, next_uri)
+    end
+
+    def self.faraday_client(options)
       server = options[:server]
       unless server
         raise ArgumentError, ":server option is required"
@@ -34,9 +42,11 @@ module Presto::Client
         faraday.adapter Faraday.default_adapter
       end
 
-      new StatementClient.new(faraday, query, options)
+      return faraday
     end
 
+    private_class_method :faraday_client
+
     def initialize(api)
       @api = api
     end
@@ -104,6 +114,10 @@ module Presto::Client
       @api.query_info
     end
 
+    def next_uri
+      @api.current_results.next_uri
+    end
+
     def cancel
       @api.cancel_leaf_stage
     end

data/lib/presto/client/statement_client.rb

@@ -39,7 +39,7 @@ module Presto::Client
       "User-Agent" => "presto-ruby/#{VERSION}",
     }
 
-    def initialize(faraday, query, options)
+    def initialize(faraday, query, options, next_uri=nil)
       @faraday = faraday
       @faraday.headers.merge!(HEADERS)
 
@@ -49,7 +49,13 @@ module Presto::Client
       @exception = nil
 
       @faraday.headers.merge!(optional_headers)
-      post_query_request!
+
+      if next_uri
+        body = faraday_get_with_retry(next_uri)
+        @results = Models::QueryResults.decode(MultiJson.load(body))
+      else
+        post_query_request!
+      end
     end
 
     def optional_headers

data/lib/presto/client/version.rb

@@ -15,6 +15,6 @@
 #
 module Presto
   module Client
-    VERSION = "0.4.13"
+    VERSION = "0.4.14"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: presto-client
 version: !ruby/object:Gem::Version
-  version: 0.4.13
+  version: 0.4.14
 platform: ruby
 authors:
 - Sadayuki Furuhashi