105 lines
1.9 KiB
Go
105 lines
1.9 KiB
Go
package db
|
|
|
|
import (
|
|
"bufio"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"be.ems/src/framework/config"
|
|
)
|
|
|
|
// ImportSQL 导入SQL
|
|
func ImportSQL() {
|
|
sqlPath := config.Get("sqlPath").(string)
|
|
if sqlPath == "" {
|
|
return
|
|
}
|
|
sqlSource := config.Get("sqlSource").(string)
|
|
if sqlSource == "" {
|
|
sqlSource = config.Get("database.defaultDataSourceName").(string)
|
|
}
|
|
|
|
// 数据源
|
|
db := DB(sqlSource)
|
|
if db == nil {
|
|
log.Fatalln("not database source")
|
|
return
|
|
}
|
|
|
|
// 获取路径信息
|
|
fileInfo, err := os.Stat(sqlPath)
|
|
if err != nil {
|
|
log.Fatalln(err.Error())
|
|
return
|
|
}
|
|
|
|
// 处理目录或文件
|
|
if fileInfo.IsDir() {
|
|
// 处理目录
|
|
files, err := os.ReadDir(sqlPath)
|
|
if err != nil {
|
|
log.Fatalln(err.Error())
|
|
return
|
|
}
|
|
|
|
for _, file := range files {
|
|
if file.IsDir() {
|
|
continue
|
|
}
|
|
if !strings.HasSuffix(file.Name(), ".sql") {
|
|
continue
|
|
}
|
|
processSQLFile(db, filepath.Join(sqlPath, file.Name()))
|
|
}
|
|
} else {
|
|
// 处理单个文件
|
|
processSQLFile(db, sqlPath)
|
|
}
|
|
|
|
log.Println("process success")
|
|
os.Exit(0)
|
|
}
|
|
|
|
// 处理单个SQL文件的通用函数
|
|
func processSQLFile(db *gorm.DB, filePath string) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
log.Fatalln(err.Error())
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
// 逐行读取 SQL 文件
|
|
scanner := bufio.NewScanner(file)
|
|
var sqlBuilder strings.Builder
|
|
for scanner.Scan() {
|
|
line := strings.TrimSpace(scanner.Text())
|
|
// 跳过注释和空行
|
|
if strings.HasPrefix(line, "--") || strings.TrimSpace(line) == "" {
|
|
continue
|
|
}
|
|
// 跳过配置语句
|
|
if strings.HasPrefix(line, "/*!") {
|
|
continue
|
|
}
|
|
|
|
sqlBuilder.WriteString(line + "\n")
|
|
|
|
// 当遇到分号时，执行 SQL 语句
|
|
if strings.HasSuffix(line, ";") {
|
|
// 执行 SQL 语句
|
|
if err := db.Exec(sqlBuilder.String()).Error; err != nil {
|
|
log.Fatalln(err.Error())
|
|
return
|
|
}
|
|
|
|
sqlBuilder.Reset()
|
|
continue
|
|
}
|
|
}
|
|
}
|