coder/coderd/database/sqlxqueries/sqlxqueries.go

package sqlxqueries

import (
	"bytes"
	"embed"
	"sync"
	"text/template"

	"golang.org/x/xerrors"
)

//go:embed *.gosql
var sqlxQueries embed.FS

var (
	// Only parse the queries once.
	once   sync.Once
	cached *template.Template
	//nolint:errname
	cachedError error
)

// LoadQueries parses the embedded queries and returns the template.
// Results are cached.
func LoadQueries() (*template.Template, error) {
	once.Do(func() {
		tpls, err := template.New("").
			Funcs(template.FuncMap{
				"int32": func(i int) int32 { return int32(i) },
			}).ParseFS(sqlxQueries, "*.gosql")
		if err != nil {
			cachedError = xerrors.Errorf("developer error parse sqlx queries: %w", err)
			return
		}
		cached = tpls
	})

	return cached, cachedError
}

// query executes the named template with the given data and returns the result.
// The returned query string is SQL.
func query(name string, data interface{}) (string, error) {
	tpls, err := LoadQueries()
	if err != nil {
		return "", err
	}

	var out bytes.Buffer
	err = tpls.ExecuteTemplate(&out, name, data)
	if err != nil {
		return "", xerrors.Errorf("execute template %s: %w", name, err)
	}
	return out.String(), nil
}