86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
_ "github.com/lib/pq"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
// PostgreSQL连接信息
|
||
|
|
connStr := "host=localhost port=5432 user=postgres password=postgres dbname=trustlog sslmode=disable"
|
||
|
|
|
||
|
|
db, err := sql.Open("postgres", connStr)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("❌ Failed to connect to database: %v", err)
|
||
|
|
}
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
// 测试连接
|
||
|
|
if err := db.Ping(); err != nil {
|
||
|
|
log.Fatalf("❌ Failed to ping database: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("✅ Connected to PostgreSQL database: trustlog")
|
||
|
|
|
||
|
|
// 删除旧表
|
||
|
|
fmt.Println("\n📋 Dropping old tables...")
|
||
|
|
dropSQL := []string{
|
||
|
|
"DROP TABLE IF EXISTS operation CASCADE",
|
||
|
|
"DROP TABLE IF EXISTS trustlog_cursor CASCADE",
|
||
|
|
"DROP TABLE IF EXISTS trustlog_retry CASCADE",
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, sql := range dropSQL {
|
||
|
|
if _, err := db.Exec(sql); err != nil {
|
||
|
|
log.Printf("⚠️ Warning dropping table: %v", err)
|
||
|
|
} else {
|
||
|
|
fmt.Println("✅", sql)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 读取并执行新的DDL
|
||
|
|
fmt.Println("\n📋 Creating new tables with op_code...")
|
||
|
|
ddlFile := "api/persistence/sql/postgresql.sql"
|
||
|
|
ddlContent, err := os.ReadFile(ddlFile)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("❌ Failed to read DDL file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := db.Exec(string(ddlContent)); err != nil {
|
||
|
|
log.Fatalf("❌ Failed to execute DDL: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("✅ Tables created successfully")
|
||
|
|
|
||
|
|
// 验证表结构
|
||
|
|
fmt.Println("\n📋 Verifying table structure...")
|
||
|
|
|
||
|
|
var count int
|
||
|
|
query := `SELECT COUNT(*) FROM information_schema.tables WHERE table_schema = 'public' AND table_name IN ('operation', 'trustlog_cursor', 'trustlog_retry')`
|
||
|
|
if err := db.QueryRow(query).Scan(&count); err != nil {
|
||
|
|
log.Fatalf("❌ Failed to verify tables: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if count != 3 {
|
||
|
|
log.Fatalf("❌ Expected 3 tables, found %d", count)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("✅ All %d tables verified\n", count)
|
||
|
|
|
||
|
|
// 验证op_code列
|
||
|
|
var dataType string
|
||
|
|
colQuery := `SELECT data_type FROM information_schema.columns WHERE table_name = 'operation' AND column_name = 'op_code'`
|
||
|
|
if err := db.QueryRow(colQuery).Scan(&dataType); err != nil {
|
||
|
|
log.Fatalf("❌ Failed to verify op_code column: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("✅ op_code column type: %s\n", dataType)
|
||
|
|
|
||
|
|
fmt.Println("\n🎉 Database reset completed successfully!")
|
||
|
|
}
|
||
|
|
|