other_mirrors/link
1
0
Fork 0
mirror of https://git.swurl.xyz/swirl/link.git synced 2023-11-08 10:44:52 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix(main.go): Add error case checking to retry.

Browse source
This commit is contained in:
fshee 2021-06-24 01:38:46 +00:00
parent 15d3d2837f
commit 5ee34dfc2f
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
main.go
View file

@ -44,6 +44,13 @@ type Retry struct {
retryAttemptCount int retryAttemptCount int
} }
func NewRetry(retryAttemptCount int) (Retry, error) {
if retryAttemptCount < 1 {
return Retry{}, errors.New("retry attempt count must be greater than zero")
}
return Retry{retryAttemptCount}, nil
}
func (r Retry) Do(f func() error) (err error) { func (r Retry) Do(f func() error) (err error) {
for i := 0; i < r.retryAttemptCount; i++ { for i := 0; i < r.retryAttemptCount; i++ {
err = f() err = f()
@ -275,7 +282,12 @@ func main() {
if *v { if *v {
applicationLogger = log.New(os.Stdout, logPrefix, 0) applicationLogger = log.New(os.Stdout, logPrefix, 0)
} }
db, err := NewDB(applicationLogger, *dbFilePath, *hashSeed, Retry{3}) retry, err := NewRetry(3)
if err != nil {
startupLogger.Fatal(err)
return
}
db, err := NewDB(applicationLogger, *dbFilePath, *hashSeed, retry)
if err != nil { if err != nil {
startupLogger.Fatal(err) startupLogger.Fatal(err)
return return