KennethCheng
f4c42bbb9e
完成 Phase 1.2,把 GORM 真正接进来。Phase 1.1 留下的 /health 和骨架保持不变, 本 commit 在此之上铺好数据访问层,Phase 1.3 直接在 service / handler 上挂业务即可。 新增: - internal/model/vehicle.go 车辆实体(严格按 plan §7.1) - internal/model/refuel_record.go 加油记录实体(严格按 plan §7.2) - internal/db/db.go SQLite 连接 + AutoMigrate 入口 * glebarez/sqlite 纯 Go,免 CGO,Windows / Docker 通吃 * DSN 自动启用 WAL + 外键约束 + busy_timeout * 表目录若不存在会自动创建 - internal/repository/vehicle.go 车辆数据访问接口 + GORM 实现 * Create / Get / List / Update / Delete * ClearDefault / SetDefault(切换默认车辆用,在事务里跑避免脏状态) - internal/repository/refuel_record.go 加油记录数据访问接口 + GORM 实现 * Create / Get / Update / Delete * ListByVehicle 支持可选时间窗口 * PrevFullRefuel 找上一次加满,给油耗计算用 main.go 改动: - 启动期 db.Open + db.Migrate,失败直接退出 - application 持有 *gorm.DB,Phase 1.3 起注入 repo / service 设计要点: - repository 全是接口,Service 持有接口便于单测用 fake 替换 - 所有 repo 错误统一包成 apperror.NotFound / apperror.Internal, 不让 GORM 原始错误泄漏到 handler - model.TableName() 显式指定单数,避免 GORM 复数 - refuel_record 用 (vehicle_id, refuel_time) 复合场景常见,已加索引 未做(留给后续 commit): - service 层与 DTO - handler / router - 显式 migration 文件(AGENTS.md 要求 schema 变更必须写 migration; 此次为首次建表,AutoMigrate 即可,后续 schema 改动会补 goose / 自管文件) 验证: go mod tidy all modules verified go vet ./... 0 issues go build 25.9 MB binary 冷启动一次,DB 文件生成在 fuellog-api/data/fuellog.db(WAL/SHM 齐备) listtables 脚本 dump 出来的两表 schema 与 plan §7 完全一致
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
// Package db 负责 GORM 数据库连接的初始化与迁移注册。
|
|
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
|
|
"github.com/kennethcheng/fuellog-api/internal/model"
|
|
)
|
|
|
|
// Open 打开 SQLite 数据库连接,并确保目标目录存在。
|
|
//
|
|
// 设计要点:
|
|
// - 使用 pure-Go glebarez/sqlite,避免 CGO;Windows / Docker 都友好
|
|
// - 日志走 slog,关闭 GORM 默认的 stdout 输出
|
|
// - 启用 WAL + 外键约束,提升并发与一致性
|
|
func Open(path string) (*gorm.DB, error) {
|
|
if dir := filepath.Dir(path); dir != "" {
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
return nil, fmt.Errorf("ensure db dir: %w", err)
|
|
}
|
|
}
|
|
gormLogger := logger.New(
|
|
slog.NewLogLogger(slog.Default().Handler(), slog.LevelWarn),
|
|
logger.Config{
|
|
LogLevel: logger.Warn,
|
|
Colorful: false,
|
|
},
|
|
)
|
|
dsn := fmt.Sprintf("file:%s?_pragma=journal_mode(WAL)&_pragma=foreign_keys(ON)&_pragma=busy_timeout(5000)", path)
|
|
g, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{Logger: gormLogger})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("open sqlite: %w", err)
|
|
}
|
|
return g, nil
|
|
}
|
|
|
|
// Migrate 在数据库上注册所有模型。Phase 1.2 之前,业务表只读;Phase 1.2
|
|
// 起每次 schema 变更必须追加迁移文件,不得依赖此处注册。
|
|
func Migrate(g *gorm.DB) error {
|
|
return g.AutoMigrate(
|
|
&model.Vehicle{},
|
|
&model.RefuelRecord{},
|
|
)
|
|
}
|