SQL格式化：使用Go语言实现SQL语句的语法感知缩进与换行

碧海醫心 2026-01-13 00:00:00 次阅读

本文介绍在go中对sql字符串进行结构化格式化的专业方案，强调仅靠正则或简单关键词分割的局限性，并推荐基于真实sql解析器的可靠实现路径。

在Go语言中，将扁平SQL字符串（如 select col1, col2 from foo where col1 > 1000）转换为可读性强、缩进规范的格式化输出（如多行+大写关键字+嵌套缩进），看似可通过字符串切分+关键词匹配（如 "SELECT"、"FROM"、"AND"）实现，但实际极易出错。例如：

// ❌ 危险的“伪格式化”示例（不推荐生产使用）
strings.ReplaceAll(sql, "select ", "SELECT\n")
strings.ReplaceAll(sql, " from ", "\nFROM\n")
// ……很快会崩溃于：SELECT * FROM users WHERE name LIKE 'a% and b%'

这类基于子串替换或空格分割的“启发式”方法，无法识别SQL语法结构：

关键字可能出现在字符串字面量中（如 'SELECT * FROM t'）；
括号嵌套（WHERE (a > 1 AND b
控制流块（如 BEGIN ... END, IF ... THEN ... END）需上下文感知缩进层级；
大小写敏感性、注释（--、/* */）、引号包裹的标识符等进一步增加复杂度。

✅ 正确路径是：使用成熟的SQL解析器生成抽象语法树（AST），再遍历AST进行语义化格式化。

Go生态中较成熟的选择是 vitess/sqlparser（原 code.google.com/p/vitess/go/vt/sqlparser 已迁移）。它支持MySQL语法（兼容大部分标准SQL），能精准解析SELECT/INSERT/BEGIN/IF等语句，并保留位置、类型和嵌套关系：

import (
    "github.com/vitessio/vitess/go/vt/sqlparser"
)

func FormatSQL(sql string) (string, error) {
    stmt, err := sqlparser.Parse(sql)
    if err != nil {
        return "", err
    }
    // vitess 默认不提供格式化器，需自定义遍历逻辑
    // 示例：对 *sqlparser.SelectStmt 实现缩进打印
    return formatSelect(stmt), nil
}

func formatSelect(stmt sqlparser.Statement) string {
    switch s := stmt.(type) {
    case *sqlparser.SelectStmt:
        var b strings.Builder
        b.WriteString("SELECT ")
        b.WriteString(sqlparser.String(s.SelectExprs))
        b.WriteString("\nFROM ")
        b.WriteString(sqlparser.String(s.From))
        if s.Where != nil {
            b.WriteString("\nWHERE ")
            b.WriteString(indent(sqlparser.String(s.Where.Expr), 4)) // 缩进条件
        }
        return b.String()
    default:
        return sqlparser.String(stmt)
    }
}