Files
go-trustlog/scripts/reset_db.go
ryan fb182adef4 feat: OpType重构为OpCode (int32) - 完整实现
🎯 核心变更:
- OpType (string) → OpCode (int32)
- 20+ OpCode枚举常量 (基于DOIP/IRP标准)
- 类型安全 + 性能优化

📊 影响范围:
- 核心模型: Operation结构体、CBOR序列化
- 数据库: schema.go + SQL DDL (PostgreSQL/MySQL/SQLite)
- 持久化: repository.go查询、cursor_worker.go
- API接口: Protobuf定义 + gRPC客户端
- 测试代码: 60+ 测试文件更新

 测试结果:
- 通过率: 100% (所有87个测试用例)
- 总体覆盖率: 53.7%
- 核心包覆盖率: logger(100%), highclient(95.3%), model(79.1%)

📝 文档:
- 精简README (1056行→489行,减少54%)
- 完整的OpCode枚举说明
- 三种持久化策略示例
- 数据库表结构和架构图

🔧 技术细节:
- 类型转换: string(OpCode) → int32(OpCode)
- SQL参数: 字符串值 → 整数值
- Protobuf: op_type string → op_code int32
- 测试断言: 字符串比较 → 常量比较

🎉 质量保证:
- 零编译错误
- 100%测试通过
- PostgreSQL/Pulsar集成测试验证
- 分布式并发安全测试通过
2025-12-26 13:47:55 +08:00

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!")
}