bourdon
77087d36c6
- 扩展 Schema 支持更多数据类型（Duration、URL、JSON 等） - 优化 SSTable 编码解码性能 - 添加多个新示例程序： - all_types: 展示所有支持的数据类型 - new_types: 演示新增类型的使用 - struct_tags: 展示结构体标签功能 - time_duration: 时间和持续时间处理示例 - 完善测试用例和文档 - 优化代码结构和错误处理
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)
|
||
}
|