package main
import (
	"bufio"
	"io"
	"os"
	"strings"
)
// readSQLFile 讀取SQL文件，并將文件中的SQL語(yǔ)句分割成字符串切片返回。
func readSQLFile(filename string) ([]string, error) {
	var statements []string // 用于存儲(chǔ)分割后的SQL語(yǔ)句
	file, err := os.Open(filename) // 打開文件
	if err != nil {
		return nil, err // 如果打開文件出錯(cuò)，返回錯(cuò)誤
	}
	defer file.Close() // 最后關(guān)閉文件
	reader := bufio.NewReader(file) // 創(chuàng)建一個(gè)緩沖讀取器
	var stmt strings.Builder // 用于構(gòu)建單個(gè)SQL語(yǔ)句
	var inMultiLineComment bool // 標(biāo)記是否處于多行注釋中
	for {
		line, err := reader.ReadString('\n') // 逐行讀取
		if err != nil {
			if err == io.EOF { // 如果讀到文件末尾
				if stmt.Len() > 0 { // 如果stmt中有內(nèi)容，說(shuō)明還有未結(jié)束的SQL語(yǔ)句
					stmt.WriteString(strings.TrimSpace(line)) // 去除末尾的空白字符
					statements = append(statements, stmt.String()) // 將構(gòu)建好的SQL語(yǔ)句添加到切片中
				}
				break // 結(jié)束循環(huán)
			}
			return nil, err // 如果讀取過(guò)程中發(fā)生錯(cuò)誤，返回錯(cuò)誤
		}
		line = strings.TrimSpace(line) // 去除行首尾的空白字符
		// 處理多行注釋
		if inMultiLineComment {
			if strings.Contains(line, "*/") { // 檢查注釋是否結(jié)束
				line = strings.Split(line, "*/")[1] // 獲取注釋結(jié)束后的內(nèi)容
				inMultiLineComment = false // 更新狀態(tài)
			} else {
				continue // 如果注釋未結(jié)束，跳過(guò)當(dāng)前行
			}
		}
		// 處理單行注釋
		if strings.HasPrefix(line, "/*") {
			commentEndIndex := strings.Index(line, "*/") // 查找注釋結(jié)束的位置
			if commentEndIndex == -1 { // 如果注釋未結(jié)束
				inMultiLineComment = true // 更新狀態(tài)
				continue
			} else {
				line = line[commentEndIndex+2:] // 獲取注釋結(jié)束后的內(nèi)容
			}
		}
		// 跳過(guò)單行注釋
		if strings.HasPrefix(line, "--") {
			continue
		}
		stmt.WriteString(line) // 將當(dāng)前行添加到stmt中
		// 檢查是否是SQL語(yǔ)句的結(jié)束
		if strings.HasSuffix(line, ";") {
			statements = append(statements, stmt.String()) // 將構(gòu)建好的SQL語(yǔ)句添加到切片中
			stmt.Reset() // 重置stmt，準(zhǔn)備構(gòu)建下一個(gè)SQL語(yǔ)句
		}
	}
	return statements, nil // 返回分割后的SQL語(yǔ)句切片
}
func main() {
	// 示例用法
	statements, err := readSQLFile("example.sql") // 讀取SQL文件
	if err != nil {
		panic(err) // 如果發(fā)生錯(cuò)誤，中斷程序
	}
	for _, stmt := range statements { // 遍歷并打印每個(gè)SQL語(yǔ)句
		println(stmt)
	}
}