zitadel/internal/database/dialect/config.go

package dialect

import (
	"database/sql"
	"sync"
	"time"

	"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([]interface{}) (Connector, error)
}

const (
	// QueryAppName             = "zitadel_queries"
	// EventstorePusherAppName  = "zitadel_es_pusher"
	// ProjectionSpoolerAppName = "zitadel_projection_spooler"
	defaultAppName = "zitadel"
)

// DBPurpose is what the resulting connection pool is used for.
// type DBPurpose int

// const (
// 	DBPurposeQuery DBPurpose = iota
// 	DBPurposeEventPusher
// 	DBPurposeProjectionSpooler
// )

// func (p DBPurpose) AppName() string {
// 	switch p {
// 	case DBPurposeQuery:
// 		return QueryAppName
// 	case DBPurposeEventPusher:
// 		return EventstorePusherAppName
// 	case DBPurposeProjectionSpooler:
// 		return ProjectionSpoolerAppName
// 	default:
// 		return defaultAppName
// 	}
// }

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

type Database interface {
	DatabaseName() string
	Username() string
	Type() string
	Timetravel(time.Duration) string
}

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
}