From 684ae1d2d0160a36c704dbc87686d1524647aeff Mon Sep 17 00:00:00 2001
From: Zack Hobson
Date: Fri, 22 Nov 2013 10:09:27 -0800
Subject: app: retry on throttle

---
 lib/hcl/app.rb                | 4 ++++
 lib/hcl/timesheet_resource.rb | 9 +++++++++
 2 files changed, 13 insertions(+)

(limited to 'lib')

diff --git a/lib/hcl/app.rb b/lib/hcl/app.rb
index ac1b992..4185a67 100644
--- a/lib/hcl/app.rb
+++ b/lib/hcl/app.rb
@@ -61,6 +61,10 @@ module HCl
       rescue SocketError => e
         STDERR.puts "Connection failed. (#{e.message})"
         exit 1
+      rescue TimesheetResource::ThrottleFailure => e
+        STDERR.puts "Too many requests, retrying in #{e.retry_after+5} seconds..."
+        sleep e.retry_after+5
+        run
       rescue TimesheetResource::AuthFailure => e
         STDERR.puts "Unable to authenticate: #{e}"
         request_config
diff --git a/lib/hcl/timesheet_resource.rb b/lib/hcl/timesheet_resource.rb
index 69eeb75..d4624ad 100644
--- a/lib/hcl/timesheet_resource.rb
+++ b/lib/hcl/timesheet_resource.rb
@@ -17,6 +17,13 @@ module HCl
   class TimesheetResource
     class Failure < StandardError; end
     class AuthFailure < StandardError; end
+    class ThrottleFailure < StandardError
+      attr_reader :retry_after
+      def initialize response
+        @retry_after = response.headers['Retry-After'].to_i
+        super "Too many requests! Try again in #{@retry_after} seconds."
+      end
+    end
 
     def self.configure opts = nil
       if opts
@@ -68,6 +75,8 @@ module HCl
         response.body
       when Net::HTTPFound
         raise Failure, "Redirected! Perhaps your ssl configuration variable is set incorrectly?"
+      when Net::HTTPServiceUnavailable
+        raise ThrottleFailure, response
       when Net::HTTPUnauthorized
         raise AuthFailure, "Login failed."
       else
-- 
cgit v1.2.3