mirror of
https://gitee.com/dromara/mayfly-go
synced 2026-05-19 01:15:40 +08:00
41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
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)
|
|
}
|