Function WithReplacements

rivershared/sqlctemplate/sqlc_template.go:258–273

WithReplacements adds sqlctemplate templates to the given context (they go in context because it's the only way to get them down into the innards of sqlc). namedArgs can also be passed in to replace arguments found in If sqlctemplate params are already present in context, the two sets are merged, w

(ctx context.Context, replacements map[string]Replacement, namedArgs map[string]any)

// If sqlctemplate params are already present in context, the two sets are
// merged, with the new params taking precedent.
func WithReplacements(ctx context.Context, replacements map[string]Replacement, namedArgs map[string]any) context.Context {
	if container, ok := ctx.Value(contextKey{}).(*contextContainer); ok {
		maps.Copy(container.NamedArgs, namedArgs)
		maps.Copy(container.Replacements, replacements)
		return ctx
	}

	if namedArgs == nil {
		namedArgs = make(map[string]any)
	}

	return context.WithValue(ctx, contextKey{}, &contextContainer{
		NamedArgs:    namedArgs,
		Replacements: replacements,
	})
}

// Comparable struct that's used as a key for template caching.
type replacerCacheKey struct {