You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
639 B
36 lines
639 B
package common
|
|
|
|
import "time"
|
|
|
|
/* RepeatTimer */
|
|
type RepeatTimer struct {
|
|
Ch chan struct{}
|
|
quit chan struct{}
|
|
dur time.Duration
|
|
timer *time.Timer
|
|
}
|
|
|
|
func NewRepeatTimer(dur time.Duration) *RepeatTimer {
|
|
var ch = make(chan struct{})
|
|
var quit = make(chan struct{})
|
|
var t = &RepeatTimer{Ch: ch, dur: dur, quit: quit}
|
|
t.timer = time.AfterFunc(dur, t.fireHandler)
|
|
return t
|
|
}
|
|
|
|
func (t *RepeatTimer) fireHandler() {
|
|
select {
|
|
case t.Ch <- struct{}{}:
|
|
t.timer.Reset(t.dur)
|
|
case <-t.quit:
|
|
}
|
|
}
|
|
|
|
func (t *RepeatTimer) Reset() {
|
|
t.timer.Reset(t.dur)
|
|
}
|
|
|
|
func (t *RepeatTimer) Stop() bool {
|
|
close(t.quit)
|
|
return t.timer.Stop()
|
|
}
|