Golang中解析SQL语句为JSON格式常用的库介绍

在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": {...}// }}

选择建议：

1. 通用SQL解析：推荐使用sqlparser或vitess-sqlparser

   • 支持 MySQL 语法
   • 活跃维护
   • 文档相对完善

2. PostgreSQL专用：使用pg_query_go

   • 基于 PostgreSQL 官方解析器
   • 支持最新 PostgreSQL 语法

3. 需要精细控制：可以基于这些库构建自己的JSON格式

这些库都能将SQL解析为AST（抽象语法树），然后你可以选择将其序列化为JSON格式。注意原始AST对象直接Marshal可能包含循环引用，需要适当处理。