Add Inner Ring code

pkg/innerring/timers/timers.go

@@ -0,0 +1,86 @@
+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
+}