mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-27 18:45:37 +08:00
32 lines
566 B
Go
32 lines
566 B
Go
|
package taskmonitor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
F "github.com/sagernet/sing/common/format"
|
||
|
|
"github.com/sagernet/sing/common/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Monitor struct {
|
||
|
|
logger logger.Logger
|
||
|
|
timeout time.Duration
|
||
|
|
timer *time.Timer
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(logger logger.Logger, timeout time.Duration) *Monitor {
|
||
|
|
return &Monitor{
|
||
|
|
logger: logger,
|
||
|
|
timeout: timeout,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Monitor) Start(taskName ...any) {
|
||
|
|
m.timer = time.AfterFunc(m.timeout, func() {
|
||
|
|
m.logger.Warn(F.ToString(taskName...), " take too much time to finish!")
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Monitor) Finish() {
|
||
|
|
m.timer.Stop()
|
||
|
|
}
|