47 lines
810 B
Go
47 lines
810 B
Go
package callticker
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
type CallTicker struct {
|
|
manualTrigger chan interface{}
|
|
function func(...interface{}) interface{}
|
|
ticker *time.Ticker
|
|
duration time.Duration
|
|
}
|
|
|
|
func NewCallTicker(d time.Duration, function func(...interface{}) interface{}) *CallTicker {
|
|
ticker := &CallTicker{
|
|
manualTrigger: make(chan interface{}),
|
|
function: function,
|
|
ticker: time.NewTicker(d),
|
|
duration: d,
|
|
}
|
|
return ticker
|
|
}
|
|
|
|
func (c *CallTicker) Stop() {
|
|
c.ticker.Stop()
|
|
}
|
|
|
|
func (c *CallTicker) Reset(d time.Duration) {
|
|
c.ticker.Reset(d)
|
|
}
|
|
|
|
func (c *CallTicker) Trigger() {
|
|
c.manualTrigger <- true
|
|
c.Reset(c.duration)
|
|
}
|
|
|
|
func (c *CallTicker) Run() {
|
|
for {
|
|
select {
|
|
case _ = <-c.ticker.C:
|
|
c.function()
|
|
case _ = <-c.manualTrigger:
|
|
c.function()
|
|
}
|
|
}
|
|
}
|