forked from free-lancers/electricity_bill_calc_service
refactor(context):改进超时上下文的生成。
This commit is contained in:
@@ -5,6 +5,12 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
func TimeoutContext(timeout time.Duration) (context.Context, context.CancelFunc) {
|
||||
// 生成一个超时时间为5秒的倍率的上下文,如果不传递任何值,默认生成6倍的上下文,即超时时间为30秒。
|
||||
func TimeoutContext(multiply ...int64) (context.Context, context.CancelFunc) {
|
||||
var ratio int64 = 6
|
||||
if len(multiply) > 0 {
|
||||
ratio = multiply[0]
|
||||
}
|
||||
timeout := time.Duration(ratio*5) * time.Second
|
||||
return context.WithTimeout(context.Background(), timeout)
|
||||
}
|
||||
|
Reference in New Issue
Block a user