frostfs-node/pkg/innerring/timers/timers.go
Stanislav Bogatyrev b7b5079934 Add Inner Ring code
2020-07-24 17:07:37 +03:00

86 lines
1.6 KiB
Go

package timers
import (
"context"
"time"
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
"github.com/pkg/errors"
"go.uber.org/zap"
)
type (
localTimer struct {
duration time.Duration
timer *time.Timer
handler event.Handler
}
// Timers is a component for local inner ring timers to produce local events.
Timers struct {
log *zap.Logger
epoch localTimer
}
// Params for timers instance constructor.
Params struct {
Log *zap.Logger
EpochDuration time.Duration
}
)
const (
// EpochTimer is a type for HandlerInfo structure.
EpochTimer = "EpochTimer"
)
// New creates instance of timers component.
func New(p *Params) *Timers {
return &Timers{
log: p.Log,
epoch: localTimer{duration: p.EpochDuration},
}
}
// Start runs all available local timers.
func (t *Timers) Start(ctx context.Context) {
t.epoch.timer = time.NewTimer(t.epoch.duration)
go t.serve(ctx)
}
func (t *Timers) serve(ctx context.Context) {
for {
select {
case <-ctx.Done():
t.log.Info("timers are getting stopped")
t.epoch.timer.Stop()
return
case <-t.epoch.timer.C:
// reset timer so it can tick once again
t.epoch.timer.Reset(t.epoch.duration)
// call handler if it is set
if t.epoch.handler != nil {
t.epoch.handler(NewEpochTick{})
}
}
}
}
// RegisterHandler of local timers events.
func (t *Timers) RegisterHandler(h event.HandlerInfo) error {
if h.Handler() == nil {
return errors.New("ir/timers: can't register nil handler")
}
switch h.GetType() {
case EpochTimer:
t.epoch.handler = h.Handler()
default:
return errors.New("ir/timers: unknown handler type")
}
return nil
}