golang使用TiDB的SQL Parse对SQL指纹化

复制代码
//mkdir sqlaprse && cd sqlaprse
//export GOPROXY=https://proxy.golang.com.cn,direct
//go mod init sqlaprse && touch main.go
//go get -v github.com/pingcap/tidb/parser@40b72e7
//go get github.com/fatih/color
//go get github.com/pingcap/tidb/types/parser_driver@40b72e7
//go run main.go -sql="select * from t1 where id=1 and date>'2023-08-18'"
package main

import (
	"bytes"
	"flag"
	"fmt"
	"github.com/fatih/color"
	"github.com/pingcap/tidb/parser"
	"github.com/pingcap/tidb/parser/ast"
	"github.com/pingcap/tidb/parser/format"
	driver "github.com/pingcap/tidb/types/parser_driver"
	"sync" // 导入 sync 包用于 WaitGroup
)

type FingerprintVisitor struct{}

func (f *FingerprintVisitor) Enter(n ast.Node) (node ast.Node, skipChildren bool) {
	if v, ok := n.(*driver.ValueExpr); ok {
		v.SetValue([]byte("?"))
	}
	return n, false
}

func (f *FingerprintVisitor) Leave(n ast.Node) (node ast.Node, ok bool) {
	return n, true
}

var (
	Success *color.Color
	Err     *color.Color
	sql     string
)

func processSQL(sql string, wg *sync.WaitGroup) {
	defer wg.Done() // 通知 WaitGroup 此 goroutine 已完成
	p := parser.New()
	stmt, err := p.ParseOneStmt(sql, "", "")
	if err != nil {
		Err.Println("解析错误:" + err.Error())
		return
	}
	stmt.Accept(&FingerprintVisitor{})

	buf := new(bytes.Buffer)
	restoreCtx := format.NewRestoreCtx(format.RestoreKeyWordUppercase|format.RestoreNameBackQuotes, buf)
	err = stmt.Restore(restoreCtx)
	if nil != err {
		Err.Println("解析错误:" + err.Error())
		return
	}
	Success.Println(buf.String())
}

func main() {
	Success = color.New(color.FgHiGreen, color.Bold, color.Underline)
	Err = color.New(color.FgHiRed, color.Bold)
	flag.StringVar(&sql, "sql", "", "SQL语句")
	flag.Parse()

	// 分割 SQL 语句
	statements := splitSQLStatements(sql)

	// 创建 WaitGroup 以等待所有 goroutine 完成
	var wg sync.WaitGroup

	for _, stmt := range statements {
		wg.Add(1) // 为每个 goroutine 增加 WaitGroup 计数
		go processSQL(stmt, &wg)
	}

	// 等待所有 goroutine 完成
	wg.Wait()
}

// 用于分割 SQL 语句的辅助函数
func splitSQLStatements(sql string) []string {
	// 在此实现分割 SQL 语句的逻辑
	// 例如,如果每个语句位于单独的行上并以分号结尾,则可以按分号分割
	// 并根据需要修改此函数。这是为了演示目的的简化示例。
	// 您可能需要处理边缘情况并改进分割逻辑。
	statements := []string{sql}
	return statements
}
相关推荐
float_六七3 分钟前
SQL六大核心类别全解析
数据库·sql·oracle
Code季风2 小时前
将 gRPC 服务注册到 Consul:从配置到服务发现的完整实践(上)
数据库·微服务·go·json·服务发现·consul
Boilermaker19922 小时前
【Java EE】SpringIoC
前端·数据库·spring
霸王龙的小胳膊2 小时前
泛微虚拟视图-数据虚拟化集成
数据库
灵犀学长2 小时前
解锁Spring Boot多项目共享Redis:优雅Key命名结构指南
数据库·redis
轩情吖2 小时前
Qt的信号与槽(二)
数据库·c++·qt·信号·connect·信号槽·
ZeroNews内网穿透3 小时前
服装零售企业跨区域运营难题破解方案
java·大数据·运维·服务器·数据库·tcp/ip·零售
可观测性用观测云3 小时前
达梦数据库监控观测最佳实践
数据库
时序数据说3 小时前
IoTDB:专为物联网场景设计的高性能时序数据库
大数据·数据库·物联网·开源·时序数据库·iotdb
码农小站4 小时前
ClickHouse 时间范围查询:精准筛选「本月数据」
数据库