Function replacerCacheKeyFrom

rivershared/sqlctemplate/sqlc_template.go:288–324

Builds a cache key for the given SQL and context container. A key is only built if the given SQL/templates are cacheable, which means all template values must be stable. The second return value is a boolean indicating whether a cache key was built or not. If false, the input is not eligible for cac

(sql string, container *contextContainer)

// indicating whether a cache key was built or not. If false, the input is not
// eligible for caching, and no check against the cache should be made.
func replacerCacheKeyFrom(sql string, container *contextContainer) (replacerCacheKey, bool) {
	// Only eligible for caching if all replacements are stable.
	for _, replacement := range container.Replacements {
		if !replacement.Stable {
			return replacerCacheKey{}, false
		}
	}

	var (
		namedArgsBuilder strings.Builder

		// Named args must be sorted for key stability because Go maps don't
		// provide any ordering guarantees.
		sortedNamedArgs = maputil.Keys(container.NamedArgs)
	)
	slices.Sort(sortedNamedArgs)
	for _, namedArg := range sortedNamedArgs {
		namedArgsBuilder.WriteRune('@') // useful as separator because not valid in the name of a named arg
		namedArgsBuilder.WriteString(namedArg)
	}

	var (
		replacementValuesBuilder strings.Builder
		sortedReplacements       = maputil.Keys(container.Replacements)
	)
	slices.Sort(sortedReplacements)
	for _, template := range sortedReplacements {
		replacementValuesBuilder.WriteRune('•') // use a separator that SQL would reject under most circumstances (this may be imperfect)
		replacementValuesBuilder.WriteString(container.Replacements[template].Value)
	}

	return replacerCacheKey{
		namedArgs:         namedArgsBuilder.String(),
		replacementValues: replacementValuesBuilder.String(),
		sql:               sql,
	}, true
}