mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-12-15 20:26:36 +08:00
feat: sql解析器替换、工单统一由‘我的流程’发起、流程定义支持自定义条件触发审批、资源隐藏编号、model支持物理删除等
This commit is contained in:
128
server/internal/db/dbm/sqlparser/base/base.go
Normal file
128
server/internal/db/dbm/sqlparser/base/base.go
Normal file
@@ -0,0 +1,128 @@
|
||||
package base
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/antlr4-go/antlr/v4"
|
||||
)
|
||||
|
||||
// SingleSQL is a separate SQL split from multi-SQL.
|
||||
type SingleSQL struct {
|
||||
Text string
|
||||
// BaseLine is the line number of the first line of the SQL in the original SQL.
|
||||
// HINT: ZERO based.
|
||||
BaseLine int
|
||||
// FirstStatementLine is the line number of the first non-comment and non-blank line of the SQL in the original SQL.
|
||||
// HINT: ZERO based.
|
||||
FirstStatementLine int
|
||||
// FirstStatementColumn is the column number of the first non-comment and non-blank line of the SQL in the original SQL.
|
||||
// HINT: ZERO based.
|
||||
FirstStatementColumn int
|
||||
// LastLine is the line number of the last line of the SQL in the original SQL.
|
||||
// HINT: ZERO based.
|
||||
LastLine int
|
||||
// LastColumn is the column number of the last line of the SQL in the original SQL.
|
||||
// HINT: ZERO based.
|
||||
LastColumn int
|
||||
// The sql is empty, such as `/* comments */;` or just `;`.
|
||||
Empty bool
|
||||
|
||||
// ByteOffsetStart is the start position of the sql.
|
||||
// This field may not be present for every engine.
|
||||
// ByteOffsetStart is intended for sql execution log display. It may not represent the actual sql that is sent to the database.
|
||||
ByteOffsetStart int
|
||||
// ByteOffsetEnd is the end position of the sql.
|
||||
// This field may not be present for every engine.
|
||||
// ByteOffsetEnd is intended for sql execution log display. It may not represent the actual sql that is sent to the database.
|
||||
ByteOffsetEnd int
|
||||
}
|
||||
|
||||
// SyntaxError is a syntax error.
|
||||
type SyntaxError struct {
|
||||
Line int
|
||||
Column int
|
||||
Message string
|
||||
}
|
||||
|
||||
// Error returns the error message.
|
||||
func (e *SyntaxError) Error() string {
|
||||
return e.Message
|
||||
}
|
||||
|
||||
// ParseErrorListener is a custom error listener for PLSQL parser.
|
||||
type ParseErrorListener struct {
|
||||
BaseLine int
|
||||
Err *SyntaxError
|
||||
}
|
||||
|
||||
// SyntaxError returns the errors.
|
||||
func (l *ParseErrorListener) SyntaxError(_ antlr.Recognizer, token any, line, column int, msg string, _ antlr.RecognitionException) {
|
||||
if l.Err == nil {
|
||||
errMessage := ""
|
||||
if token, ok := token.(*antlr.CommonToken); ok {
|
||||
stream := token.GetInputStream()
|
||||
start := token.GetStart() - 40
|
||||
if start < 0 {
|
||||
start = 0
|
||||
}
|
||||
stop := token.GetStop()
|
||||
if stop >= stream.Size() {
|
||||
stop = stream.Size() - 1
|
||||
}
|
||||
errMessage = fmt.Sprintf("related text: %s", stream.GetTextFromInterval(antlr.NewInterval(start, stop)))
|
||||
}
|
||||
l.Err = &SyntaxError{
|
||||
Line: line + l.BaseLine,
|
||||
Column: column,
|
||||
Message: fmt.Sprintf("Syntax error at line %d:%d \n%s", line+l.BaseLine, column, errMessage),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ReportAmbiguity reports an ambiguity.
|
||||
func (*ParseErrorListener) ReportAmbiguity(recognizer antlr.Parser, dfa *antlr.DFA, startIndex, stopIndex int, exact bool, ambigAlts *antlr.BitSet, configs *antlr.ATNConfigSet) {
|
||||
antlr.ConsoleErrorListenerINSTANCE.ReportAmbiguity(recognizer, dfa, startIndex, stopIndex, exact, ambigAlts, configs)
|
||||
}
|
||||
|
||||
// ReportAttemptingFullContext reports an attempting full context.
|
||||
func (*ParseErrorListener) ReportAttemptingFullContext(recognizer antlr.Parser, dfa *antlr.DFA, startIndex, stopIndex int, conflictingAlts *antlr.BitSet, configs *antlr.ATNConfigSet) {
|
||||
antlr.ConsoleErrorListenerINSTANCE.ReportAttemptingFullContext(recognizer, dfa, startIndex, stopIndex, conflictingAlts, configs)
|
||||
}
|
||||
|
||||
// ReportContextSensitivity reports a context sensitivity.
|
||||
func (*ParseErrorListener) ReportContextSensitivity(recognizer antlr.Parser, dfa *antlr.DFA, startIndex, stopIndex, prediction int, configs *antlr.ATNConfigSet) {
|
||||
antlr.ConsoleErrorListenerINSTANCE.ReportContextSensitivity(recognizer, dfa, startIndex, stopIndex, prediction, configs)
|
||||
}
|
||||
|
||||
func FilterEmptySQL(list []SingleSQL) []SingleSQL {
|
||||
var result []SingleSQL
|
||||
for _, sql := range list {
|
||||
if !sql.Empty {
|
||||
result = append(result, sql)
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func FilterEmptySQLWithIndexes(list []SingleSQL) ([]SingleSQL, []int32) {
|
||||
var result []SingleSQL
|
||||
var originalIndex []int32
|
||||
for i, sql := range list {
|
||||
if !sql.Empty {
|
||||
result = append(result, sql)
|
||||
originalIndex = append(originalIndex, int32(i))
|
||||
}
|
||||
}
|
||||
return result, originalIndex
|
||||
}
|
||||
|
||||
func GetOffsetLength(total int) int {
|
||||
length := 1
|
||||
for {
|
||||
if total < 10 {
|
||||
return length
|
||||
}
|
||||
total /= 10
|
||||
length++
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user