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-10-06 20:59:04 +08:00

68 lines
1.1 KiB
Go

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