mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-12 04:07:31 +00:00
chore: move the go code into a subfolder
This commit is contained in:
159
apps/api/internal/execution/handlers.go
Normal file
159
apps/api/internal/execution/handlers.go
Normal file
@@ -0,0 +1,159 @@
|
||||
package execution
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
"github.com/riverqueue/river"
|
||||
|
||||
"github.com/zitadel/zitadel/internal/api/authz"
|
||||
"github.com/zitadel/zitadel/internal/domain"
|
||||
"github.com/zitadel/zitadel/internal/eventstore"
|
||||
"github.com/zitadel/zitadel/internal/eventstore/handler/v2"
|
||||
"github.com/zitadel/zitadel/internal/query"
|
||||
"github.com/zitadel/zitadel/internal/queue"
|
||||
exec_repo "github.com/zitadel/zitadel/internal/repository/execution"
|
||||
)
|
||||
|
||||
const (
|
||||
HandlerTable = "projections.execution_handler"
|
||||
)
|
||||
|
||||
type Queue interface {
|
||||
Insert(ctx context.Context, args river.JobArgs, opts ...queue.InsertOpt) error
|
||||
}
|
||||
|
||||
type Queries interface {
|
||||
TargetsByExecutionID(ctx context.Context, ids []string) (execution []*query.ExecutionTarget, err error)
|
||||
InstanceByID(ctx context.Context, id string) (instance authz.Instance, err error)
|
||||
}
|
||||
|
||||
type eventHandler struct {
|
||||
eventTypes []string
|
||||
aggregateTypeFromEventType func(typ eventstore.EventType) eventstore.AggregateType
|
||||
query Queries
|
||||
queue Queue
|
||||
}
|
||||
|
||||
func NewEventHandler(
|
||||
ctx context.Context,
|
||||
config handler.Config,
|
||||
eventTypes []string,
|
||||
aggregateTypeFromEventType func(typ eventstore.EventType) eventstore.AggregateType,
|
||||
query Queries,
|
||||
queue Queue,
|
||||
) *handler.Handler {
|
||||
return handler.NewHandler(ctx, &config, &eventHandler{
|
||||
eventTypes: eventTypes,
|
||||
aggregateTypeFromEventType: aggregateTypeFromEventType,
|
||||
query: query,
|
||||
queue: queue,
|
||||
})
|
||||
}
|
||||
|
||||
func (u *eventHandler) Name() string {
|
||||
return HandlerTable
|
||||
}
|
||||
|
||||
func (u *eventHandler) Reducers() []handler.AggregateReducer {
|
||||
aggList := make(map[eventstore.AggregateType][]eventstore.EventType)
|
||||
for _, eventType := range u.eventTypes {
|
||||
aggType := u.aggregateTypeFromEventType(eventstore.EventType(eventType))
|
||||
aggEventTypes := aggList[aggType]
|
||||
if !slices.Contains(aggEventTypes, eventstore.EventType(eventType)) {
|
||||
aggList[aggType] = append(aggList[aggType], eventstore.EventType(eventType))
|
||||
}
|
||||
}
|
||||
|
||||
aggReducers := make([]handler.AggregateReducer, 0, len(aggList))
|
||||
for aggType, aggEventTypes := range aggList {
|
||||
eventReducers := make([]handler.EventReducer, len(aggEventTypes))
|
||||
for j, eventType := range aggEventTypes {
|
||||
eventReducers[j] = handler.EventReducer{
|
||||
Event: eventType,
|
||||
Reduce: u.reduce,
|
||||
}
|
||||
}
|
||||
aggReducers = append(aggReducers, handler.AggregateReducer{
|
||||
Aggregate: aggType,
|
||||
EventReducers: eventReducers,
|
||||
})
|
||||
}
|
||||
return aggReducers
|
||||
}
|
||||
|
||||
// FilterGlobalEvents implements [handler.GlobalProjection]
|
||||
func (u *eventHandler) FilterGlobalEvents() {}
|
||||
|
||||
func groupsFromEventType(s string) []string {
|
||||
parts := strings.Split(s, ".")
|
||||
groups := make([]string, len(parts))
|
||||
for i := range parts {
|
||||
groups[i] = strings.Join(parts[:i+1], ".")
|
||||
if i < len(parts)-1 {
|
||||
groups[i] += ".*"
|
||||
}
|
||||
}
|
||||
slices.Reverse(groups)
|
||||
return groups
|
||||
}
|
||||
|
||||
func idsForEventType(eventType string) []string {
|
||||
ids := make([]string, 0)
|
||||
for _, group := range groupsFromEventType(eventType) {
|
||||
ids = append(ids,
|
||||
exec_repo.ID(domain.ExecutionTypeEvent, group),
|
||||
)
|
||||
}
|
||||
return append(ids,
|
||||
exec_repo.IDAll(domain.ExecutionTypeEvent),
|
||||
)
|
||||
}
|
||||
|
||||
func (u *eventHandler) reduce(e eventstore.Event) (*handler.Statement, error) {
|
||||
ctx := HandlerContext(context.Background(), e.Aggregate())
|
||||
|
||||
targets, err := u.query.TargetsByExecutionID(ctx, idsForEventType(string(e.Type())))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// no execution from worker necessary
|
||||
if len(targets) == 0 {
|
||||
return handler.NewNoOpStatement(e), nil
|
||||
}
|
||||
|
||||
return handler.NewStatement(e, func(ctx context.Context, ex handler.Executer, projectionName string) error {
|
||||
ctx = HandlerContext(ctx, e.Aggregate())
|
||||
req, err := NewRequest(e, targets)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return u.queue.Insert(ctx,
|
||||
req,
|
||||
queue.WithQueueName(exec_repo.QueueName),
|
||||
)
|
||||
}), nil
|
||||
}
|
||||
|
||||
func NewRequest(e eventstore.Event, targets []*query.ExecutionTarget) (*exec_repo.Request, error) {
|
||||
targetsData, err := json.Marshal(targets)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
eventData, err := json.Marshal(e)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &exec_repo.Request{
|
||||
Aggregate: e.Aggregate(),
|
||||
Sequence: e.Sequence(),
|
||||
EventType: e.Type(),
|
||||
CreatedAt: e.CreatedAt(),
|
||||
UserID: e.Creator(),
|
||||
EventData: eventData,
|
||||
TargetsData: targetsData,
|
||||
}, nil
|
||||
}
|
Reference in New Issue
Block a user