zitadel/internal/authz/repository/eventsourcing/handler/handler.go

package handler

import (
	"github.com/caos/zitadel/internal/iam/repository/eventsourcing"
	"time"

	"github.com/caos/zitadel/internal/authz/repository/eventsourcing/view"
	sd "github.com/caos/zitadel/internal/config/systemdefaults"
	"github.com/caos/zitadel/internal/config/types"
	"github.com/caos/zitadel/internal/eventstore"
	"github.com/caos/zitadel/internal/eventstore/query"
	org_events "github.com/caos/zitadel/internal/org/repository/eventsourcing"
	project_events "github.com/caos/zitadel/internal/project/repository/eventsourcing"
)

type Configs map[string]*Config

type Config struct {
	MinimumCycleDuration types.Duration
}

type handler struct {
	view                *view.View
	bulkLimit           uint64
	cycleDuration       time.Duration
	errorCountUntilSkip uint64

	es eventstore.Eventstore
}

func (h *handler) Eventstore() eventstore.Eventstore {
	return h.es
}

type EventstoreRepos struct {
	IAMEvents     *eventsourcing.IAMEventstore
	OrgEvents     *org_events.OrgEventstore
	ProjectEvents *project_events.ProjectEventstore
}

func Register(configs Configs, bulkLimit, errorCount uint64, view *view.View, es eventstore.Eventstore, repos EventstoreRepos, systemDefaults sd.SystemDefaults) []query.Handler {
	return []query.Handler{
		newUserGrant(
			handler{view, bulkLimit, configs.cycleDuration("UserGrants"), errorCount, es},
			repos.IAMEvents,
			systemDefaults.IamID),
		newUserMembership(
			handler{view, bulkLimit, configs.cycleDuration("UserMemberships"), errorCount, es},
			repos.OrgEvents,
			repos.ProjectEvents),
		newApplication(
			handler{view, bulkLimit, configs.cycleDuration("Application"), errorCount, es}),
		newOrg(
			handler{view, bulkLimit, configs.cycleDuration("Org"), errorCount, es}),
	}
}

func (configs Configs) cycleDuration(viewModel string) time.Duration {
	c, ok := configs[viewModel]
	if !ok {
		return 3 * time.Minute
	}
	return c.MinimumCycleDuration.Duration
}

func (h *handler) MinimumCycleDuration() time.Duration {
	return h.cycleDuration
}

func (h *handler) LockDuration() time.Duration {
	return h.cycleDuration / 3
}

func (h *handler) QueryLimit() uint64 {
	return h.bulkLimit
}