snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

persistence/sql/update/update.go

@@ -0,0 +1,76 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package update
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/pkg/errors"
+	"go.opentelemetry.io/otel/trace"
+
+	"github.com/ory/pop/v6"
+	"github.com/ory/pop/v6/columns"
+	"github.com/ory/x/otelx"
+	"github.com/ory/x/sqlcon"
+)
+
+func Generic(ctx context.Context, c *pop.Connection, tracer trace.Tracer, v any, columnNames ...string) (err error) {
+	ctx, span := tracer.Start(ctx, "persistence.sql.update")
+	defer otelx.End(span, &err)
+
+	quoter, ok := c.Dialect.(interface{ Quote(key string) string })
+	if !ok {
+		return errors.Errorf("store is not a quoter: %T", c.Store)
+	}
+
+	model := pop.NewModel(v, ctx)
+	tn := model.TableName()
+
+	cols := columns.Columns{}
+	if len(columnNames) > 0 && tn == model.TableName() {
+		cols = columns.NewColumnsWithAlias(tn, model.As, model.IDField())
+		cols.Add(columnNames...)
+	} else {
+		cols = columns.ForStructWithAlias(v, tn, model.As, model.IDField())
+	}
+
+	//#nosec G201 -- TableName is static
+	stmt := fmt.Sprintf("UPDATE %s AS %s SET %s WHERE (%s) AND %s.nid = :nid",
+		quoter.Quote(model.TableName()),
+		model.Alias(),
+		cols.Writeable().QuotedUpdateString(quoter),
+		model.WhereNamedID(),
+		model.Alias(),
+	)
+
+	if result, err := c.Store.NamedExecContext(ctx, stmt, v); err != nil {
+		return sqlcon.HandleError(err)
+	} else if affected, err := result.RowsAffected(); err != nil {
+		return sqlcon.HandleError(err)
+	} else if affected == 0 {
+		return errors.WithStack(sqlcon.ErrNoRows())
+	}
+
+	return nil
+}
+
+// GenericExcept updates v like Generic, but omits excludeColumns from the
+// UPDATE ... SET list (every other writeable column is still written).
+//
+// Leaving a column that backs a unique index out of the SET list — when its
+// value has not changed — lets the database skip the uniqueness check it would
+// otherwise plan for that index, which can be expensive.
+func GenericExcept(ctx context.Context, c *pop.Connection, tracer trace.Tracer, v any, excludeColumns ...string) error {
+	model := pop.NewModel(v, ctx)
+	cols := columns.ForStructWithAlias(v, model.TableName(), model.As, model.IDField())
+	cols.Remove(excludeColumns...)
+
+	writeable := cols.Writeable().Cols
+	names := make([]string, 0, len(writeable))
+	for name := range writeable {
+		names = append(names, name)
+	}
+	return Generic(ctx, c, tracer, v, names...)
+}