107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"code.tczkiot.com/wlw/srdb"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
fmt.Println("=== Testing Time and Duration Types (Simple) ===\n")
|
|||
|
|
|
|||
|
|
// 1. 创建 Schema
|
|||
|
|
schema, err := srdb.NewSchema("events", []srdb.Field{
|
|||
|
|
{Name: "name", Type: srdb.String, Comment: "事件名称"},
|
|||
|
|
{Name: "created_at", Type: srdb.Time, Comment: "创建时间"},
|
|||
|
|
{Name: "duration", Type: srdb.Duration, Comment: "持续时间"},
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("创建 Schema 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("✓ Schema 创建成功")
|
|||
|
|
for _, field := range schema.Fields {
|
|||
|
|
fmt.Printf(" - %s: %s\n", field.Name, field.Type.String())
|
|||
|
|
}
|
|||
|
|
fmt.Println()
|
|||
|
|
|
|||
|
|
// 2. 创建数据库和表
|
|||
|
|
os.RemoveAll("./test_data")
|
|||
|
|
db, err := srdb.Open("./test_data")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("打开数据库失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
table, err := db.CreateTable("events", schema)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("创建表失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("✓ 表创建成功\n")
|
|||
|
|
|
|||
|
|
// 3. 插入数据
|
|||
|
|
now := time.Now()
|
|||
|
|
duration := 2 * time.Hour
|
|||
|
|
|
|||
|
|
err = table.Insert(map[string]any{
|
|||
|
|
"name": "event1",
|
|||
|
|
"created_at": now,
|
|||
|
|
"duration": duration,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("插入数据失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("✓ 插入数据成功")
|
|||
|
|
fmt.Printf(" 时间: %v\n", now.Format(time.RFC3339))
|
|||
|
|
fmt.Printf(" 持续时间: %v\n\n", duration)
|
|||
|
|
|
|||
|
|
// 4. 立即查询（从 MemTable）
|
|||
|
|
fmt.Println("4. 查询数据（从 MemTable）")
|
|||
|
|
rows, err := table.Query().Rows()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("查询失败: %v", err)
|
|||
|
|
}
|
|||
|
|
defer rows.Close()
|
|||
|
|
|
|||
|
|
success := true
|
|||
|
|
for rows.Next() {
|
|||
|
|
row := rows.Row()
|
|||
|
|
data := row.Data()
|
|||
|
|
|
|||
|
|
name := data["name"]
|
|||
|
|
createdAt := data["created_at"]
|
|||
|
|
dur := data["duration"]
|
|||
|
|
|
|||
|
|
fmt.Printf(" 名称: %v\n", name)
|
|||
|
|
|
|||
|
|
// 验证类型
|
|||
|
|
if t, ok := createdAt.(time.Time); ok {
|
|||
|
|
fmt.Printf(" 时间: %v (类型: time.Time) ✓\n", t.Format(time.RFC3339))
|
|||
|
|
} else {
|
|||
|
|
fmt.Printf(" 时间: %v (类型: %T) ✗ FAILED\n", createdAt, createdAt)
|
|||
|
|
success = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if d, ok := dur.(time.Duration); ok {
|
|||
|
|
fmt.Printf(" 持续时间: %v (类型: time.Duration) ✓\n", d)
|
|||
|
|
} else {
|
|||
|
|
fmt.Printf(" 持续时间: %v (类型: %T) ✗ FAILED\n", dur, dur)
|
|||
|
|
success = false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if success {
|
|||
|
|
fmt.Println("\n✅ 测试通过! Time 和 Duration 类型正确保留")
|
|||
|
|
} else {
|
|||
|
|
fmt.Println("\n❌ 测试失败! 类型未正确保留")
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 快速退出，不等待清理
|
|||
|
|
os.Exit(0)
|
|||
|
|
}
|