Retry only on Yt::ConnectionError

nullscreen · Jun 28, 2017 · 946e731 · 946e731
1 parent 2d5a8e5
commit 946e731
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/yt/http_request.rb b/lib/yt/http_request.rb
@@ -50,16 +50,18 @@ def run
           parse_response!
         end
       else
-        retry_run
+        raise Yt::HTTPError, error_message
       end
+    rescue Yt::ConnectionError
+      retry_run
     end
 
   private
 
     # retry the run method in case of a random 500 error from YouTube API
     def retry_run
       if @retried
-        raise Yt::HTTPError, error_message
+        raise Yt::ConnectionError, error_message
       else
         @retried = true
         @response = nil