1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/common/signal/timer.go
2017-01-31 14:15:34 +01:00

50 lines
752 B
Go

package signal
import (
"context"
"time"
)
type ActivityTimer struct {
updated chan bool
timeout time.Duration
ctx context.Context
cancel context.CancelFunc
}
func (t *ActivityTimer) UpdateActivity() {
select {
case t.updated <- true:
default:
}
}
func (t *ActivityTimer) run() {
for {
select {
case <-time.After(t.timeout):
case <-t.ctx.Done():
return
}
select {
case <-t.updated:
// Updated keep waiting.
default:
t.cancel()
return
}
}
}
func CancelAfterInactivity(ctx context.Context, cancel context.CancelFunc, timeout time.Duration) *ActivityTimer {
timer := &ActivityTimer{
ctx: ctx,
cancel: cancel,
timeout: timeout,
updated: make(chan bool, 1),
}
go timer.run()
return timer
}