在Go中解析SQL语句为JSON格式,常用的库有以下几种:
1.sqlparser(最常用)
GitHub: https://github.com/xwb1989/sqlparser
import("github.com/xwb1989/sqlparser""encoding/json")funcparseSQLToJSON(sqlstring)(string,error){stmt,err:=sqlparser.Parse(sql)iferr!=nil{return"",err}// 转换为JSONjsonBytes,err:=json.Marshal(stmt)iferr!=nil{return"",err}returnstring(jsonBytes),nil}2.vitess-sqlparser(功能更全面)
GitHub: https://github.com/vitessio/vitess/tree/main/go/vt/sqlparser
import("vitess.io/vitess/go/vt/sqlparser""encoding/json")funcparseMySQLToJSON(sqlstring)(string,error){stmt,err:=sqlparser.Parse(sql)iferr!=nil{return"",err}// 使用自定义结构体格式化输出result:=map[string]interface{}{"type":sqlparser.ASTString(stmt),"parsed":stmt,}jsonBytes,_:=json.MarshalIndent(result,""," ")returnstring(jsonBytes),nil}3.pg_query_go(专门解析PostgreSQL)
GitHub: https://github.com/lfittl/pg_query_go
import("github.com/lfittl/pg_query_go""encoding/json")funcparsePostgreSQLToJSON(sqlstring)(string,error){result,err:=pg_query_go.ParseToJSON(sql)iferr!=nil{return"",err}returnresult,nil}4.自定义实现示例
packagemainimport("encoding/json""fmt""github.com/xwb1989/sqlparser")typeSQLNodestruct{Typestring`json:"type"`Valueinterface{}`json:"value,omitempty"`Children[]*SQLNode`json:"children,omitempty"`}funcparseToCustomJSON(sqlstring)(string,error){stmt,err:=sqlparser.Parse(sql)iferr!=nil{return"",err}root:=&SQLNode{Type:fmt.Sprintf("%T",stmt)}// 递归遍历ASTswitchnode:=stmt.(type){case*sqlparser.Select:root.Value=map[string]interface{}{"columns":parseSelectColumns(node.SelectExprs),"from":parseTable(node.From),"where":parseWhere(node.Where),"limit":parseLimit(node.Limit),}case*sqlparser.Insert:// 处理INSERT语句// ... 其他语句类型}jsonBytes,_:=json.MarshalIndent(root,""," ")returnstring(jsonBytes),nil}5.使用示例
funcmain(){sql:="SELECT id, name FROM users WHERE age > 18 LIMIT 10"// 使用sqlparserjsonOutput,err:=parseSQLToJSON(sql)iferr!=nil{panic(err)}fmt.Println("JSON输出:")fmt.Println(jsonOutput)// 输出示例结构// {// "type": "*sqlparser.Select",// "select_exprs": [...],// "from": [...],// "where": {...},// "limit": {...}// }}选择建议:
通用SQL解析:推荐使用sqlparser或vitess-sqlparser
- 支持 MySQL 语法
- 活跃维护
- 文档相对完善
PostgreSQL专用:使用pg_query_go
- 基于 PostgreSQL 官方解析器
- 支持最新 PostgreSQL 语法
需要精细控制:可以基于这些库构建自己的JSON格式
这些库都能将SQL解析为AST(抽象语法树),然后你可以选择将其序列化为JSON格式。注意原始AST对象直接Marshal可能包含循环引用,需要适当处理。