zitadel/internal/database/dialect/config.go

package dialect

import (
	"database/sql"
	"sync"

	"github.com/jackc/pgx/v5/pgxpool"
)

type Dialect struct {
	Matcher   Matcher
	Config    Connector
	IsDefault bool
}

var (
	dialects       []*Dialect
	defaultDialect *Dialect
	dialectsMu     sync.Mutex
)

type Matcher interface {
	MatchName(string) bool
	Decode([]any) (Connector, error)
	Type() DatabaseType
}

type DatabaseType uint8

const (
	DatabaseTypePostgres DatabaseType = iota
	DatabaseTypeCockroach
)

const (
	DefaultAppName = "zitadel"
)

type Connector interface {
	Connect(useAdmin bool) (*sql.DB, *pgxpool.Pool, error)
	Password() string
	Database
}

type Database interface {
	DatabaseName() string
	Username() string
	Type() DatabaseType
}

func Register(matcher Matcher, config Connector, isDefault bool) {
	dialectsMu.Lock()
	defer dialectsMu.Unlock()

	d := &Dialect{Matcher: matcher, Config: config}

	if isDefault {
		defaultDialect = d
		return
	}

	dialects = append(dialects, d)
}

func SelectByConfig(config map[string]interface{}) *Dialect {
	for key := range config {
		for _, d := range dialects {
			if d.Matcher.MatchName(key) {
				return d
			}
		}
	}

	return defaultDialect
}