mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-12 11:37:32 +00:00
chore: move the go code into a subfolder
This commit is contained in:
144
apps/api/internal/eventstore/v3/sequence.go
Normal file
144
apps/api/internal/eventstore/v3/sequence.go
Normal file
@@ -0,0 +1,144 @@
|
||||
package eventstore
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/zitadel/logging"
|
||||
|
||||
"github.com/zitadel/zitadel/internal/api/authz"
|
||||
"github.com/zitadel/zitadel/internal/database"
|
||||
"github.com/zitadel/zitadel/internal/eventstore"
|
||||
"github.com/zitadel/zitadel/internal/zerrors"
|
||||
)
|
||||
|
||||
type latestSequence struct {
|
||||
aggregate *eventstore.Aggregate
|
||||
sequence uint64
|
||||
}
|
||||
|
||||
//go:embed sequences_query.sql
|
||||
var latestSequencesStmt string
|
||||
|
||||
func latestSequences(ctx context.Context, tx database.Tx, commands []eventstore.Command) ([]*latestSequence, error) {
|
||||
sequences := commandsToSequences(ctx, commands)
|
||||
|
||||
conditions, args := sequencesToSql(sequences)
|
||||
rows, err := tx.QueryContext(ctx, fmt.Sprintf(latestSequencesStmt, strings.Join(conditions, " UNION ALL ")), args...)
|
||||
if err != nil {
|
||||
return nil, zerrors.ThrowInternal(err, "V3-5jU5z", "Errors.Internal")
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
if err := scanToSequence(rows, sequences); err != nil {
|
||||
return nil, zerrors.ThrowInternal(err, "V3-Ydiwv", "Errors.Internal")
|
||||
}
|
||||
}
|
||||
|
||||
if rows.Err() != nil {
|
||||
return nil, zerrors.ThrowInternal(rows.Err(), "V3-XApDk", "Errors.Internal")
|
||||
}
|
||||
return sequences, nil
|
||||
}
|
||||
|
||||
func searchSequenceByCommand(sequences []*latestSequence, command eventstore.Command) *latestSequence {
|
||||
for _, sequence := range sequences {
|
||||
if sequence.aggregate.Type == command.Aggregate().Type &&
|
||||
sequence.aggregate.ID == command.Aggregate().ID &&
|
||||
sequence.aggregate.InstanceID == command.Aggregate().InstanceID {
|
||||
return sequence
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func searchSequence(sequences []*latestSequence, aggregateType eventstore.AggregateType, aggregateID, instanceID string) *latestSequence {
|
||||
for _, sequence := range sequences {
|
||||
if sequence.aggregate.Type == aggregateType &&
|
||||
sequence.aggregate.ID == aggregateID &&
|
||||
sequence.aggregate.InstanceID == instanceID {
|
||||
return sequence
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func commandsToSequences(ctx context.Context, commands []eventstore.Command) []*latestSequence {
|
||||
sequences := make([]*latestSequence, 0, len(commands))
|
||||
|
||||
for _, command := range commands {
|
||||
if searchSequenceByCommand(sequences, command) != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
if command.Aggregate().InstanceID == "" {
|
||||
command.Aggregate().InstanceID = authz.GetInstance(ctx).InstanceID()
|
||||
}
|
||||
sequences = append(sequences, &latestSequence{
|
||||
aggregate: command.Aggregate(),
|
||||
})
|
||||
}
|
||||
|
||||
return sequences
|
||||
}
|
||||
|
||||
const argsPerCondition = 3
|
||||
|
||||
func sequencesToSql(sequences []*latestSequence) (conditions []string, args []any) {
|
||||
args = make([]interface{}, 0, len(sequences)*argsPerCondition)
|
||||
conditions = make([]string, len(sequences))
|
||||
|
||||
for i, sequence := range sequences {
|
||||
conditions[i] = fmt.Sprintf(`(SELECT instance_id, aggregate_type, aggregate_id, "sequence" FROM eventstore.events2 WHERE instance_id = $%d AND aggregate_type = $%d AND aggregate_id = $%d ORDER BY "sequence" DESC LIMIT 1)`,
|
||||
i*argsPerCondition+1,
|
||||
i*argsPerCondition+2,
|
||||
i*argsPerCondition+3,
|
||||
)
|
||||
args = append(args, sequence.aggregate.InstanceID, sequence.aggregate.Type, sequence.aggregate.ID)
|
||||
}
|
||||
|
||||
return conditions, args
|
||||
}
|
||||
|
||||
func scanToSequence(rows *sql.Rows, sequences []*latestSequence) error {
|
||||
var aggregateType eventstore.AggregateType
|
||||
var aggregateID, instanceID string
|
||||
var currentSequence uint64
|
||||
var resourceOwner string
|
||||
|
||||
if err := rows.Scan(&instanceID, &resourceOwner, &aggregateType, &aggregateID, ¤tSequence); err != nil {
|
||||
return zerrors.ThrowInternal(err, "V3-OIWqj", "Errors.Internal")
|
||||
}
|
||||
|
||||
sequence := searchSequence(sequences, aggregateType, aggregateID, instanceID)
|
||||
if sequence == nil {
|
||||
logging.WithFields(
|
||||
"aggType", aggregateType,
|
||||
"aggID", aggregateID,
|
||||
"instance", instanceID,
|
||||
).Panic("no sequence found")
|
||||
// added return for linting
|
||||
return nil
|
||||
}
|
||||
sequence.sequence = currentSequence
|
||||
if resourceOwner != "" && sequence.aggregate.ResourceOwner != "" && sequence.aggregate.ResourceOwner != resourceOwner {
|
||||
logging.WithFields(
|
||||
"current_sequence", sequence.sequence,
|
||||
"instance_id", sequence.aggregate.InstanceID,
|
||||
"agg_type", sequence.aggregate.Type,
|
||||
"agg_id", sequence.aggregate.ID,
|
||||
"current_owner", resourceOwner,
|
||||
"provided_owner", sequence.aggregate.ResourceOwner,
|
||||
).Info("would have set wrong resource owner")
|
||||
}
|
||||
// set resource owner from previous events
|
||||
if resourceOwner != "" {
|
||||
sequence.aggregate.ResourceOwner = resourceOwner
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user