zitadel/internal/eventstore/v1/spooler/config.go

package spooler

import (
	"math/rand"

	"github.com/zitadel/logging"

	v1 "github.com/zitadel/zitadel/internal/eventstore/v1"
	"github.com/zitadel/zitadel/internal/eventstore/v1/query"
	"github.com/zitadel/zitadel/internal/id"
)

type Config struct {
	Eventstore        v1.Eventstore
	Locker            Locker
	ViewHandlers      []query.Handler
	ConcurrentWorkers int
}

func (c *Config) New() *Spooler {
	lockID, err := id.SonyFlakeGenerator.Next()
	logging.OnError(err).Panic("unable to generate lockID")

	//shuffle the handlers for better balance when running multiple pods
	rand.Shuffle(len(c.ViewHandlers), func(i, j int) {
		c.ViewHandlers[i], c.ViewHandlers[j] = c.ViewHandlers[j], c.ViewHandlers[i]
	})

	return &Spooler{
		handlers:   c.ViewHandlers,
		lockID:     lockID,
		eventstore: c.Eventstore,
		locker:     c.Locker,
		queue:      make(chan *spooledHandler, len(c.ViewHandlers)),
		workers:    c.ConcurrentWorkers,
	}
}