mirror of
https://gitee.com/dromara/mayfly-go
synced 2026-05-20 18:05:33 +08:00
refactor: 移除antlr4减小包体积&ai助手优化
This commit is contained in:
40
server/internal/db/dbm/sqlparser/splitter.go
Normal file
40
server/internal/db/dbm/sqlparser/splitter.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package sqlparser
|
||||
|
||||
import (
|
||||
"io"
|
||||
|
||||
"mayfly-go/internal/pkg/utils"
|
||||
)
|
||||
|
||||
// SQLSplitter SQL 切割器接口
|
||||
// 不同方言有不同的切割规则,例如:
|
||||
// - WITH CTE 语句不能被分号切割
|
||||
// - PostgreSQL 的 DO $$ ... $$ 块
|
||||
// - MySQL 的 DELIMITER 命令
|
||||
// - 存储过程/函数中的分号
|
||||
type SQLSplitter interface {
|
||||
// SplitSQL 切割 SQL 语句
|
||||
// - r: 读取器
|
||||
// - callback: 回调函数,每条完整的 SQL 语句调用一次
|
||||
SplitSQL(r io.Reader, callback utils.StmtCallback) error
|
||||
}
|
||||
|
||||
// DefaultSplitter 默认 SQL 切割器
|
||||
// 适用于大多数标准 SQL 场景
|
||||
type DefaultSplitter struct {
|
||||
delimiter rune
|
||||
}
|
||||
|
||||
// NewDefaultSplitter 创建默认切割器
|
||||
func NewDefaultSplitter(delimiter ...rune) *DefaultSplitter {
|
||||
delim := rune(';')
|
||||
if len(delimiter) > 0 {
|
||||
delim = delimiter[0]
|
||||
}
|
||||
return &DefaultSplitter{delimiter: delim}
|
||||
}
|
||||
|
||||
// SplitSQL 实现 SQLSplitter 接口
|
||||
func (s *DefaultSplitter) SplitSQL(r io.Reader, callback utils.StmtCallback) error {
|
||||
return utils.SplitStmts(r, s.delimiter, callback)
|
||||
}
|
||||
Reference in New Issue
Block a user