zitadel/backend/v3/storage/database/statement.go

package database

import (
	"strconv"
	"strings"
)

type Instruction string

const (
	DefaultInstruction       Instruction = "DEFAULT"
	NowInstruction           Instruction = "NOW()"
	NullInstruction          Instruction = "NULL"
	GenRandomUUIDInstruction Instruction = "gen_random_uuid()"
)

// StatementBuilder is a helper to build SQL statement.
type StatementBuilder struct {
	strings.Builder
	args []any
}

type argWriter interface {
	WriteArg(builder *StatementBuilder)
}

func NewStatementBuilder(baseQuery string, baseArgs ...any) *StatementBuilder {
	b := &StatementBuilder{}
	b.WriteString(baseQuery)
	b.AppendArgs(baseArgs...)
	return b
}

// WriteArgs adds the argument to the statement and writes the placeholder to the query.
func (b *StatementBuilder) WriteArg(arg any) {
	if writer, ok := arg.(argWriter); ok {
		writer.WriteArg(b)
		return
	}
	b.WriteString(b.AppendArg(arg))
}

// WriteArgs adds the arguments to the statement and writes the placeholders to the query.
// The placeholders are comma separated.
func (b *StatementBuilder) WriteArgs(args ...any) {
	for i, arg := range args {
		if i > 0 {
			b.WriteString(", ")
		}
		b.WriteArg(arg)
	}
}

// AppendArg adds the argument to the statement and returns the placeholder.
func (b *StatementBuilder) AppendArg(arg any) (placeholder string) {
	if instruction, ok := arg.(Instruction); ok {
		return string(instruction)
	}

	b.args = append(b.args, arg)
	placeholder = "$" + strconv.Itoa(len(b.args))
	return placeholder
}

// AppendArgs adds the arguments to the statement and doesn't return the placeholders.
func (b *StatementBuilder) AppendArgs(args ...any) {
	for _, arg := range args {
		b.AppendArg(arg)
	}
}

// Args returns the arguments added to the statement.
func (b *StatementBuilder) Args() []any {
	return b.args
}