mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-08 10:20:26 +08:00
feat: sql解析器替换、工单统一由‘我的流程’发起、流程定义支持自定义条件触发审批、资源隐藏编号、model支持物理删除等
This commit is contained in:
59
server/internal/db/dbm/sqlparser/mysql/mysql.go
Normal file
59
server/internal/db/dbm/sqlparser/mysql/mysql.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package mysql
|
||||
|
||||
import (
|
||||
"mayfly-go/internal/db/dbm/sqlparser/base"
|
||||
mysqlparser "mayfly-go/internal/db/dbm/sqlparser/mysql/antlr4"
|
||||
"mayfly-go/internal/db/dbm/sqlparser/sqlstmt"
|
||||
"mayfly-go/pkg/logx"
|
||||
|
||||
"github.com/antlr4-go/antlr/v4"
|
||||
)
|
||||
|
||||
func GetMysqlParserTree(baseLine int, statement string) (antlr.ParseTree, *antlr.CommonTokenStream, error) {
|
||||
lexer := mysqlparser.NewMySqlLexer(antlr.NewInputStream(statement))
|
||||
stream := antlr.NewCommonTokenStream(lexer, antlr.TokenDefaultChannel)
|
||||
parser := mysqlparser.NewMySqlParser(stream)
|
||||
|
||||
lexerErrorListener := &base.ParseErrorListener{
|
||||
BaseLine: baseLine,
|
||||
}
|
||||
lexer.RemoveErrorListeners()
|
||||
lexer.AddErrorListener(lexerErrorListener)
|
||||
|
||||
parserErrorListener := &base.ParseErrorListener{
|
||||
BaseLine: baseLine,
|
||||
}
|
||||
parser.RemoveErrorListeners()
|
||||
parser.AddErrorListener(parserErrorListener)
|
||||
parser.BuildParseTrees = true
|
||||
|
||||
tree := parser.Root()
|
||||
|
||||
if lexerErrorListener.Err != nil {
|
||||
return nil, nil, lexerErrorListener.Err
|
||||
}
|
||||
|
||||
if parserErrorListener.Err != nil {
|
||||
return nil, nil, parserErrorListener.Err
|
||||
}
|
||||
|
||||
return tree, stream, nil
|
||||
}
|
||||
|
||||
type MysqlParser struct {
|
||||
}
|
||||
|
||||
func (*MysqlParser) Parse(stmt string) (stmts []sqlstmt.Stmt, err error) {
|
||||
defer func() {
|
||||
if e := recover(); e != nil {
|
||||
logx.ErrorTrace("mysql sql parser err: ", e)
|
||||
err = e.(error)
|
||||
}
|
||||
}()
|
||||
tree, _, err := GetMysqlParserTree(1, stmt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return tree.Accept(new(MysqlVisitor)).([]sqlstmt.Stmt), nil
|
||||
}
|
||||
Reference in New Issue
Block a user