优化逻辑
This commit is contained in:
34
internal/routes/standby_ctrl/interval.go
Normal file
34
internal/routes/standby_ctrl/interval.go
Normal file
@@ -0,0 +1,34 @@
|
||||
package standby_ctrl
|
||||
|
||||
import (
|
||||
"context"
|
||||
"go.uber.org/zap"
|
||||
"time"
|
||||
)
|
||||
|
||||
func Interval(interval int64, play func(c context.Context) error) func(c context.Context) error {
|
||||
return func(c context.Context) error {
|
||||
zap.S().Infoln("待机间隔控制器: ", interval)
|
||||
defer zap.S().Infoln("待机间隔控制器结束: ", interval)
|
||||
|
||||
for {
|
||||
err := play(c)
|
||||
if err != nil {
|
||||
zap.S().Errorln("执行后续操作异常: ", err)
|
||||
}
|
||||
if interval > 0 {
|
||||
select {
|
||||
case <-c.Done():
|
||||
return nil
|
||||
case <-time.After(time.Duration(interval) * time.Second):
|
||||
}
|
||||
} else {
|
||||
select {
|
||||
case <-c.Done():
|
||||
return nil
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user