sqlinclause.go

package sqlutil

import (
	"fmt"
	"strconv"
	"strings"
)

const (
	sqlConditionIn    = "IN"
	sqlConditionNotIn = "NOT IN"
)

// BuildInClauseString prepares a SQL IN clause with the given list of string values.
func (c *SQLUtil) BuildInClauseString(field string, values []string) string {
	return c.composeInClause(sqlConditionIn, field, c.formatStrings(values))
}

// BuildNotInClauseString prepares a SQL NOT IN clause with the given list of string values.
func (c *SQLUtil) BuildNotInClauseString(field string, values []string) string {
	return c.composeInClause(sqlConditionNotIn, field, c.formatStrings(values))
}

// BuildInClauseInt prepares a SQL IN clause with the given list of integer values.
func (c *SQLUtil) BuildInClauseInt(field string, values []int) string {
	return c.composeInClause(sqlConditionIn, field, formatInts(values))
}

// BuildNotInClauseInt prepares a SQL NOT IN clause with the given list of integer values.
func (c *SQLUtil) BuildNotInClauseInt(field string, values []int) string {
	return c.composeInClause(sqlConditionNotIn, field, formatInts(values))
}

// BuildInClauseUint prepares a SQL IN clause with the given list of integer values.
func (c *SQLUtil) BuildInClauseUint(field string, values []uint64) string {
	return c.composeInClause(sqlConditionIn, field, formatUints(values))
}

// BuildNotInClauseUint prepares a SQL NOT IN clause with the given list of integer values.
func (c *SQLUtil) BuildNotInClauseUint(field string, values []uint64) string {
	return c.composeInClause(sqlConditionNotIn, field, formatUints(values))
}

// composeInClause constructs the final IN or NOT IN clause string.
func (c *SQLUtil) composeInClause(condition string, field string, values []string) string {
	if len(values) == 0 {
		return ""
	}

	return fmt.Sprintf("%s %s (%s)", c.QuoteID(field), condition, strings.Join(values, ","))
}

// formatStrings quotes each string value for safe SQL inclusion.
func (c *SQLUtil) formatStrings(values []string) []string {
	items := make([]string, len(values))

	for k, v := range values {
		items[k] = c.QuoteValue(v)
	}

	return items
}

// formatInts converts each integer value to its string representation.
func formatInts(values []int) []string {
	items := make([]string, len(values))

	for k, v := range values {
		items[k] = strconv.Itoa(v)
	}

	return items
}

// formatUints converts each unsigned integer value to its string representation.
func formatUints(values []uint64) []string {
	items := make([]string, len(values))

	for k, v := range values {
		items[k] = strconv.FormatUint(v, 10)
	}

	return items
}