mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 12:27:59 +00:00
1ee7a1ab7c
# Which Problems Are Solved Push is not capable of external transactions. # How the Problems Are Solved A new function `PushWithClient` is added to the eventstore framework which allows to pass a client which can either be a `*sql.Client` or `*sql.Tx` and is used during push. # Additional Changes Added interfaces to database package. # Additional Context - part of https://github.com/zitadel/zitadel/issues/8931 --------- Co-authored-by: Livio Spring <livio.a@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
97 lines
3.6 KiB
Go
97 lines
3.6 KiB
Go
package eventstore
|
|
|
|
import (
|
|
"context"
|
|
_ "embed"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
"github.com/zitadel/logging"
|
|
|
|
"github.com/zitadel/zitadel/internal/database"
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
"github.com/zitadel/zitadel/internal/zerrors"
|
|
)
|
|
|
|
var (
|
|
//go:embed unique_constraints_delete.sql
|
|
deleteConstraintStmt string
|
|
//go:embed unique_constraints_delete_placeholders.sql
|
|
deleteConstraintPlaceholdersStmt string
|
|
//go:embed unique_constraints_add.sql
|
|
addConstraintStmt string
|
|
)
|
|
|
|
func handleUniqueConstraints(ctx context.Context, tx database.Tx, commands []eventstore.Command) error {
|
|
deletePlaceholders := make([]string, 0)
|
|
deleteArgs := make([]any, 0)
|
|
|
|
addPlaceholders := make([]string, 0)
|
|
addArgs := make([]any, 0)
|
|
addConstraints := map[string]*eventstore.UniqueConstraint{}
|
|
deleteConstraints := map[string]*eventstore.UniqueConstraint{}
|
|
|
|
for _, command := range commands {
|
|
for _, constraint := range command.UniqueConstraints() {
|
|
instanceID := command.Aggregate().InstanceID
|
|
if constraint.IsGlobal {
|
|
instanceID = ""
|
|
}
|
|
switch constraint.Action {
|
|
case eventstore.UniqueConstraintAdd:
|
|
constraint.UniqueField = strings.ToLower(constraint.UniqueField)
|
|
addPlaceholders = append(addPlaceholders, fmt.Sprintf("($%d, $%d, $%d)", len(addArgs)+1, len(addArgs)+2, len(addArgs)+3))
|
|
addArgs = append(addArgs, instanceID, constraint.UniqueType, constraint.UniqueField)
|
|
addConstraints[fmt.Sprintf(uniqueConstraintPlaceholderFmt, instanceID, constraint.UniqueType, constraint.UniqueField)] = constraint
|
|
case eventstore.UniqueConstraintRemove:
|
|
deletePlaceholders = append(deletePlaceholders, fmt.Sprintf(deleteConstraintPlaceholdersStmt, len(deleteArgs)+1, len(deleteArgs)+2, len(deleteArgs)+3))
|
|
deleteArgs = append(deleteArgs, instanceID, constraint.UniqueType, constraint.UniqueField)
|
|
deleteConstraints[fmt.Sprintf(uniqueConstraintPlaceholderFmt, instanceID, constraint.UniqueType, constraint.UniqueField)] = constraint
|
|
case eventstore.UniqueConstraintInstanceRemove:
|
|
deletePlaceholders = append(deletePlaceholders, fmt.Sprintf("(instance_id = $%d)", len(deleteArgs)+1))
|
|
deleteArgs = append(deleteArgs, instanceID)
|
|
deleteConstraints[fmt.Sprintf(uniqueConstraintPlaceholderFmt, instanceID, constraint.UniqueType, constraint.UniqueField)] = constraint
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(deletePlaceholders) > 0 {
|
|
_, err := tx.ExecContext(ctx, fmt.Sprintf(deleteConstraintStmt, strings.Join(deletePlaceholders, " OR ")), deleteArgs...)
|
|
if err != nil {
|
|
logging.WithError(err).Warn("delete unique constraint failed")
|
|
errMessage := "Errors.Internal"
|
|
if constraint := constraintFromErr(err, deleteConstraints); constraint != nil {
|
|
errMessage = constraint.ErrorMessage
|
|
}
|
|
return zerrors.ThrowInternal(err, "V3-C8l3V", errMessage)
|
|
}
|
|
}
|
|
if len(addPlaceholders) > 0 {
|
|
_, err := tx.ExecContext(ctx, fmt.Sprintf(addConstraintStmt, strings.Join(addPlaceholders, ", ")), addArgs...)
|
|
if err != nil {
|
|
logging.WithError(err).Warn("add unique constraint failed")
|
|
errMessage := "Errors.Internal"
|
|
if constraint := constraintFromErr(err, addConstraints); constraint != nil {
|
|
errMessage = constraint.ErrorMessage
|
|
}
|
|
return zerrors.ThrowAlreadyExists(err, "V3-DKcYh", errMessage)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func constraintFromErr(err error, constraints map[string]*eventstore.UniqueConstraint) *eventstore.UniqueConstraint {
|
|
pgErr := new(pgconn.PgError)
|
|
if !errors.As(err, &pgErr) {
|
|
return nil
|
|
}
|
|
for key, constraint := range constraints {
|
|
if strings.Contains(pgErr.Detail, key) {
|
|
return constraint
|
|
}
|
|
}
|
|
return nil
|
|
}
|