mattermost-community-enterprise/vendor/github.com/splitio/go-split-commons/v7/tasks/worker.go

package tasks

import (
	"errors"

	"github.com/splitio/go-toolkit/v5/logging"
)

// SegmentWorker struct contains resources and functions for fetching segments and storing them
type SegmentWorker struct {
	name        string
	failureTime int64
	toExecute   func(name string, till *int64) error
	logger      logging.LoggerInterface
}

// NewSegmentWorker some
func NewSegmentWorker(name string, failureTime int64, logger logging.LoggerInterface, toExecute func(name string, till *int64) error) *SegmentWorker {
	return &SegmentWorker{
		name:        name,
		failureTime: failureTime,
		toExecute:   toExecute,
		logger:      logger,
	}
}

// Name Returns the name of the worker
func (w *SegmentWorker) Name() string {
	return w.name
}

// FailureTime Returns how much time should be waited after an error, before the worker resumes execution
func (w *SegmentWorker) FailureTime() int64 {
	return w.failureTime
}

// DoWork performs the actual work and returns an error if something goes wrong
func (w *SegmentWorker) DoWork(msg interface{}) error {
	segmentName, ok := msg.(string)
	if !ok {
		return errors.New("segment name popped from queue is not a string")
	}

	return w.toExecute(segmentName, nil)
}

// OnError callback does nothing
func (w *SegmentWorker) OnError(e error) { w.logger.Error("error in segmentWorker: ", e) }

// Cleanup callback does nothing
func (w *SegmentWorker) Cleanup() error { return nil }