Cleanup from PR comments

This commit is contained in:
Ethan Frey
2017-12-08 10:07:04 +01:00
parent cc7a87e27c
commit ec4adf21e0
3 changed files with 12 additions and 15 deletions

View File

@ -20,7 +20,7 @@ type RepeatTimer struct {
stopped bool
}
type repeatCommand int32
type repeatCommand int8
const (
Reset repeatCommand = iota
@ -67,21 +67,21 @@ func (t *RepeatTimer) run() {
// stop goroutine if the input says so
// don't close channels, as closed channels mess up select reads
done = t.processInput(cmd)
case <-t.ticker.C:
t.trySend()
case tick := <-t.ticker.C:
t.trySend(tick)
}
}
}
// trySend performs non-blocking send on t.Ch
func (t *RepeatTimer) trySend() {
func (t *RepeatTimer) trySend(tick time.Time) {
// NOTE: this was blocking in previous version (t.Ch <- t_)
// should I use that behavior unstead of unblocking as per throttle?
// probably not: https://golang.org/src/time/sleep.go#L132
select {
case t.output <- time.Now():
default:
}
// probably better not: https://golang.org/src/time/sleep.go#L132
t.output <- tick
// select {
// case t.output <- tick:
// default:
// }
}
// all modifications of the internal state of ThrottleTimer

View File

@ -22,7 +22,7 @@ type ThrottleTimer struct {
stopped bool
}
type throttleCommand int32
type throttleCommand int8
const (
Set throttleCommand = iota
@ -83,7 +83,6 @@ func (t *ThrottleTimer) processInput(cmd throttleCommand) (shutdown bool) {
t.timer.Reset(t.dur)
}
case TQuit:
t.stopped = true
shutdown = true
fallthrough
case Unset:
@ -125,5 +124,6 @@ func (t *ThrottleTimer) Stop() bool {
return false
}
t.input <- TQuit
t.stopped = true
return true
}

View File

@ -95,9 +95,6 @@ func TestThrottle(test *testing.T) {
stopped := t.Stop()
assert.True(stopped)
time.Sleep(longwait)
assert.Equal(5, c.Count())
// extra calls to stop don't block
t.Stop()
}