Method formatExpression

cmd/gosqlx/cmd/sql_formatter.go:619–886

formatExpression formats SQL expressions

(expr ast.Expression)

// formatExpression formats SQL expressions
func (f *SQLFormatter) formatExpression(expr ast.Expression) error {
	switch e := expr.(type) {
	case *ast.Identifier:
		if e.Table != "" {
			f.builder.WriteString(e.Table + ".")
		}
		f.builder.WriteString(e.Name)
	case *ast.LiteralValue:
		// Handle string literals with proper quoting
		switch e.Type {
		case "string", "STRING":
			// Escape single quotes in the string value and wrap in quotes
			// Use type assertion for efficiency instead of fmt.Sprintf
			var strVal string
			if str, ok := e.Value.(string); ok {
				strVal = str
			} else {
				strVal = fmt.Sprintf("%v", e.Value)
			}
			escaped := strings.ReplaceAll(strVal, "'", "''")
			f.builder.WriteString("'")
			f.builder.WriteString(escaped)
			f.builder.WriteString("'")
		case "null", "NULL":
			f.writeKeyword("NULL")
		default:
			// For non-string types, use type assertions for common types
			switch v := e.Value.(type) {
			case string:
				f.builder.WriteString(v)
			case int:
				f.builder.WriteString(strconv.Itoa(v))
			case int64:
				f.builder.WriteString(strconv.FormatInt(v, 10))
			case float64:
				f.builder.WriteString(strconv.FormatFloat(v, 'f', -1, 64))
			case bool:
				if v {
					f.writeKeyword("TRUE")
				} else {
					f.writeKeyword("FALSE")
				}
			default:
				f.builder.WriteString(fmt.Sprintf("%v", e.Value))
			}
		}
	case *ast.BinaryExpression:
		// Handle IS NULL / IS NOT NULL specially
		if e.Operator == "IS NULL" {
			if err := f.formatExpression(e.Left); err != nil {
				return err
			}
			if e.Not {
				f.builder.WriteString(" IS NOT NULL")
			} else {
				f.builder.WriteString(" IS NULL")
			}
			return nil