cubefs

1
// Copyright 2022 The CubeFS Authors.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
12
// implied. See the License for the specific language governing
13
// permissions and limitations under the License.
14

15
package common
16

17
import "time"
18

19
type Retry struct {
20
	retryTimes int
21
	delayTime  uint32
22
}
23

24
func (r Retry) On(caller func() error) error {
25
	var lastErr error
26
	for i := 0; i < r.retryTimes; i++ {
27
		err := caller()
28
		if err == nil {
29
			return nil
30
		}
31

32
		i++
33
		time.Sleep(time.Duration(r.delayTime) * time.Millisecond)
34

35
	}
36
	return lastErr
37
}
38

39
// return a Retry
40
// delayTime is millisecond
41
func Timed(retryTimes int, delayTime uint32) Retry {
42
	return Retry{
43
		retryTimes: retryTimes,
44
		delayTime:  delayTime,
45
	}
46
}
47