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-05-08 17:09:21 +02:00

58 lines
886 B
Go

package signal
import (
"context"
"time"
)
type ActivityTimer interface {
Update()
}
type realActivityTimer struct {
updated chan bool
timeout time.Duration
ctx context.Context
cancel context.CancelFunc
}
func (t *realActivityTimer) Update() {
select {
case t.updated <- true:
default:
}
}
func (t *realActivityTimer) run() {
ticker := time.NewTicker(t.timeout)
defer ticker.Stop()
for {
select {
case <-ticker.C:
case <-t.ctx.Done():
return
}
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 := &realActivityTimer{
ctx: ctx,
cancel: cancel,
timeout: timeout,
updated: make(chan bool, 1),
}
go timer.run()
return ctx, timer
}