boundary/internal/db/schema/edition.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

package schema

import (
	"embed"
	"fmt"
	"sync"

	"github.com/hashicorp/boundary/internal/db/schema/internal/edition"
)

// Dialect is same as edition.Dialect
type Dialect = edition.Dialect

// Supported dialects.
const (
	Postgres Dialect = "postgres"
)

var supportedDialects = map[Dialect]struct{}{
	Postgres: {},
}

// type dialects map[Dialect]edition.Editions

type dialects struct {
	sync.Mutex

	m map[Dialect]edition.Editions
}

var editions = dialects{
	m: make(map[Dialect]edition.Editions),
}

// RegisterEdition registers an edition for use by the Manager. It will panic if:
// - An unsupported dialect is provided.
// - The same (dialect, name) is registered.
// - The same (dialect, priority) is registered.
func RegisterEdition(name string, dialect Dialect, fs embed.FS, priority int, opt ...edition.Option) {
	editions.Lock()
	defer editions.Unlock()

	if _, ok := supportedDialects[dialect]; !ok {
		panic(fmt.Sprintf("unsupported dialect: %s", dialect))
	}
	var e edition.Editions
	var ok bool

	e, ok = editions.m[dialect]
	if !ok {
		e = make(edition.Editions, 0)
	}

	for _, ee := range e {
		if ee.Name == name {
			panic(fmt.Sprintf("edition %s with dialect %s already registered", name, dialect))
		}

		if ee.Priority == priority {
			panic(fmt.Sprintf("edition %s with dialect %s and priority %d has same priority as edition %s", name, dialect, priority, ee.Name))
		}
	}

	ee, err := edition.New(name, dialect, fs, priority, opt...)
	if err != nil {
		panic(err.Error())
	}
	e = append(e, ee)
	e.Sort()

	editions.m[dialect] = e
}