| OLD | NEW |
| (Empty) | |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 package retry |
| 6 |
| 7 import ( |
| 8 "io" |
| 9 "time" |
| 10 ) |
| 11 |
| 12 // Default defines the default retry parameters that should be used throughout |
| 13 // the program. It is fine to update this variable on start up. |
| 14 var Default = &Config{ |
| 15 10, |
| 16 100 * time.Millisecond, |
| 17 500 * time.Millisecond, |
| 18 5 * time.Second, |
| 19 } |
| 20 |
| 21 // Config defines the retry properties. |
| 22 type Config struct { |
| 23 MaxTries int // Maximum number of retries. |
| 24 SleepMax time.Duration // Maximum duration of a single sleep. |
| 25 SleepBase time.Duration // Base sleep duration. |
| 26 SleepMultiplicative time.Duration // Incremental sleep duration for each
additional try. |
| 27 } |
| 28 |
| 29 // Do runs a Retriable, potentially retrying it multiple times. |
| 30 func (c *Config) Do(r Retriable) (err error) { |
| 31 defer func() { |
| 32 if err2 := r.Close(); err == nil { |
| 33 err = err2 |
| 34 } |
| 35 }() |
| 36 for i := 0; i < c.MaxTries; i++ { |
| 37 err = r.Do() |
| 38 if _, ok := err.(Error); !ok { |
| 39 return err |
| 40 } |
| 41 if i != c.MaxTries-1 { |
| 42 s := c.SleepBase + time.Duration(i)*c.SleepMultiplicativ
e |
| 43 if s > c.SleepMax { |
| 44 s = c.SleepMax |
| 45 } |
| 46 time.Sleep(s) |
| 47 } |
| 48 } |
| 49 return |
| 50 } |
| 51 |
| 52 // Error is an error that can be retried. |
| 53 type Error struct { |
| 54 Err error |
| 55 } |
| 56 |
| 57 func (e Error) Error() string { |
| 58 return e.Err.Error() |
| 59 } |
| 60 |
| 61 // Retriable is a task that can be retried. It is important that Do be |
| 62 // idempotent. |
| 63 type Retriable interface { |
| 64 io.Closer |
| 65 Do() error |
| 66 } |
| OLD | NEW |