72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"database/sql"
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/niangaodev/art-code/config"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
// 初始化数据库连接
|
|||
|
|
config.InitDB()
|
|||
|
|
db := config.DB
|
|||
|
|
defer config.CloseDB()
|
|||
|
|
|
|||
|
|
// 读取SQL文件
|
|||
|
|
sqlFile := "scripts/create_inquiries_tables.sql"
|
|||
|
|
content, err := ioutil.ReadFile(sqlFile)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("Error reading SQL file: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 分割SQL语句
|
|||
|
|
statements := strings.Split(string(content), ";")
|
|||
|
|
|
|||
|
|
// 执行每一条SQL语句
|
|||
|
|
for _, stmt := range statements {
|
|||
|
|
stmt = strings.TrimSpace(stmt)
|
|||
|
|
if stmt == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_, err := db.Exec(stmt)
|
|||
|
|
if err != nil {
|
|||
|
|
// 忽略"表已存在"的错误（虽然我们用了IF NOT EXISTS，但为了保险）
|
|||
|
|
if !strings.Contains(err.Error(), "already exists") {
|
|||
|
|
log.Printf("Error executing statement: %s\nError: %v", stmt, err)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 只打印前50个字符
|
|||
|
|
displayStmt := stmt
|
|||
|
|
if len(displayStmt) > 50 {
|
|||
|
|
displayStmt = displayStmt[:50] + "..."
|
|||
|
|
}
|
|||
|
|
fmt.Printf("Successfully executed: %s\n", displayStmt)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查表是否创建成功
|
|||
|
|
checkTable(db, "inquiries")
|
|||
|
|
checkTable(db, "email_suffixes")
|
|||
|
|
|
|||
|
|
fmt.Println("Inquiries tables migration completed successfully!")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func checkTable(db *sql.DB, tableName string) {
|
|||
|
|
var name string
|
|||
|
|
err := db.QueryRow("SHOW TABLES LIKE ?", tableName).Scan(&name)
|
|||
|
|
if err != nil {
|
|||
|
|
if err == sql.ErrNoRows {
|
|||
|
|
fmt.Printf("Table %s does NOT exist!\n", tableName)
|
|||
|
|
} else {
|
|||
|
|
fmt.Printf("Error checking table %s: %v\n", tableName, err)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
fmt.Printf("Table %s exists.\n", tableName)
|
|||
|
|
}
|
|||
|
|
}
|