zitadel/internal/eventstore/aggregate.go
Silvan 444f682e25
refactor(notification): use new queue package (#9360)
# Which Problems Are Solved

The recently introduced notification queue have potential race conditions.

# How the Problems Are Solved

Current code is refactored to use the queue package, which is safe in
regards of concurrency.

# Additional Changes

- the queue is included in startup
- improved code quality of queue

# Additional Context

- closes https://github.com/zitadel/zitadel/issues/9278
2025-02-27 11:49:12 +01:00

112 lines
2.6 KiB
Go

package eventstore
import (
"context"
"github.com/zitadel/zitadel/internal/api/authz"
)
type aggregateOpt func(*Aggregate)
// NewAggregate is the default constructor of an aggregate
// opts overwrite values calculated by given parameters
func NewAggregate(
ctx context.Context,
id string,
typ AggregateType,
version Version,
opts ...aggregateOpt,
) *Aggregate {
a := &Aggregate{
ID: id,
Type: typ,
ResourceOwner: authz.GetCtxData(ctx).OrgID,
InstanceID: authz.GetInstance(ctx).InstanceID(),
Version: version,
}
for _, opt := range opts {
opt(a)
}
return a
}
// WithResourceOwner overwrites the resource owner of the aggregate
// by default the resource owner is set by the context
func WithResourceOwner(resourceOwner string) aggregateOpt {
return func(aggregate *Aggregate) {
aggregate.ResourceOwner = resourceOwner
}
}
// WithInstanceID overwrites the instance id of the aggregate
// by default the instance is set by the context
func WithInstanceID(id string) aggregateOpt {
return func(aggregate *Aggregate) {
aggregate.InstanceID = id
}
}
// AggregateFromWriteModel maps the given WriteModel to an Aggregate.
// Deprecated: Creates linter errors on missing context. Use [AggregateFromWriteModelCtx] instead.
func AggregateFromWriteModel(
wm *WriteModel,
typ AggregateType,
version Version,
) *Aggregate {
return AggregateFromWriteModelCtx(context.Background(), wm, typ, version)
}
// AggregateFromWriteModelCtx maps the given WriteModel to an Aggregate.
func AggregateFromWriteModelCtx(
ctx context.Context,
wm *WriteModel,
typ AggregateType,
version Version,
) *Aggregate {
return NewAggregate(
ctx,
wm.AggregateID,
typ,
version,
WithResourceOwner(wm.ResourceOwner),
WithInstanceID(wm.InstanceID),
)
}
// Aggregate is the basic implementation of Aggregater
type Aggregate struct {
// ID is the unique identitfier of this aggregate
ID string `json:"id"`
// Type is the name of the aggregate.
Type AggregateType `json:"type"`
// ResourceOwner is the org this aggregates belongs to
ResourceOwner string `json:"resourceOwner"`
// InstanceID is the instance this aggregate belongs to
InstanceID string `json:"instanceId"`
// Version is the semver this aggregate represents
Version Version `json:"version"`
}
// AggregateType is the object name
type AggregateType string
func isAggregateTypes(a *Aggregate, types ...AggregateType) bool {
for _, typ := range types {
if a.Type == typ {
return true
}
}
return false
}
func isAggregateIDs(a *Aggregate, ids ...string) bool {
for _, id := range ids {
if a.ID == id {
return true
}
}
return false
}