26 lines
739 B
Ruby
26 lines
739 B
Ruby
module Vagrant
|
|
module Util
|
|
module Retryable
|
|
# Retries a given block a specified number of times in the
|
|
# event the specified exception is raised. If the retries
|
|
# run out, the final exception is raised.
|
|
#
|
|
# This code is adapted slightly from the following blog post:
|
|
# http://blog.codefront.net/2008/01/14/retrying-code-blocks-in-ruby-on-exceptions-whatever/
|
|
def retryable(opts=nil)
|
|
opts = { :tries => 1, :on => Exception }.merge(opts || {})
|
|
|
|
begin
|
|
return yield
|
|
rescue *opts[:on]
|
|
if (opts[:tries] -= 1) > 0
|
|
sleep opts[:sleep].to_f if opts[:sleep]
|
|
retry
|
|
end
|
|
raise
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|