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 完全一致
103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/kennethcheng/fuellog-api/internal/model"
|
|
"github.com/kennethcheng/fuellog-api/pkg/apperror"
|
|
)
|
|
|
|
// RefuelRecordRepository 加油记录表的数据访问接口。
|
|
type RefuelRecordRepository interface {
|
|
Create(ctx context.Context, r *model.RefuelRecord) error
|
|
Get(ctx context.Context, id uint64) (*model.RefuelRecord, error)
|
|
Update(ctx context.Context, r *model.RefuelRecord) error
|
|
Delete(ctx context.Context, id uint64) error
|
|
// List 返回指定车辆的加油记录,按 refuel_time DESC 排序。
|
|
// 支持可选时间窗口。
|
|
ListByVehicle(ctx context.Context, vehicleID uint64, start, end *time.Time) ([]model.RefuelRecord, error)
|
|
// PrevFullRefuel 返回指定车辆在 time 之前最近一次 is_full=true 的记录。
|
|
// 用于计算百公里油耗。
|
|
PrevFullRefuel(ctx context.Context, vehicleID uint64, before time.Time) (*model.RefuelRecord, error)
|
|
}
|
|
|
|
func NewRefuelRecordRepository(g *gorm.DB) RefuelRecordRepository {
|
|
return &refuelRepo{g: g}
|
|
}
|
|
|
|
type refuelRepo struct{ g *gorm.DB }
|
|
|
|
func (r *refuelRepo) Create(ctx context.Context, rec *model.RefuelRecord) error {
|
|
if err := r.g.WithContext(ctx).Create(rec).Error; err != nil {
|
|
return apperror.Wrap(apperror.CodeInternal, "创建加油记录失败", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *refuelRepo) Get(ctx context.Context, id uint64) (*model.RefuelRecord, error) {
|
|
var rec model.RefuelRecord
|
|
if err := r.g.WithContext(ctx).First(&rec, id).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, apperror.New(apperror.CodeNotFound, "加油记录不存在")
|
|
}
|
|
return nil, apperror.Wrap(apperror.CodeInternal, "查询加油记录失败", err)
|
|
}
|
|
return &rec, nil
|
|
}
|
|
|
|
func (r *refuelRepo) Update(ctx context.Context, rec *model.RefuelRecord) error {
|
|
res := r.g.WithContext(ctx).Save(rec)
|
|
if res.Error != nil {
|
|
return apperror.Wrap(apperror.CodeInternal, "更新加油记录失败", res.Error)
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
return apperror.New(apperror.CodeNotFound, "加油记录不存在")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *refuelRepo) Delete(ctx context.Context, id uint64) error {
|
|
res := r.g.WithContext(ctx).Delete(&model.RefuelRecord{}, id)
|
|
if res.Error != nil {
|
|
return apperror.Wrap(apperror.CodeInternal, "删除加油记录失败", res.Error)
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
return apperror.New(apperror.CodeNotFound, "加油记录不存在")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *refuelRepo) ListByVehicle(ctx context.Context, vehicleID uint64, start, end *time.Time) ([]model.RefuelRecord, error) {
|
|
q := r.g.WithContext(ctx).Where("vehicle_id = ?", vehicleID)
|
|
if start != nil {
|
|
q = q.Where("refuel_time >= ?", *start)
|
|
}
|
|
if end != nil {
|
|
q = q.Where("refuel_time <= ?", *end)
|
|
}
|
|
var out []model.RefuelRecord
|
|
if err := q.Order("refuel_time DESC, id DESC").Find(&out).Error; err != nil {
|
|
return nil, apperror.Wrap(apperror.CodeInternal, "查询加油记录列表失败", err)
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (r *refuelRepo) PrevFullRefuel(ctx context.Context, vehicleID uint64, before time.Time) (*model.RefuelRecord, error) {
|
|
var rec model.RefuelRecord
|
|
err := r.g.WithContext(ctx).
|
|
Where("vehicle_id = ? AND is_full = ? AND refuel_time < ?", vehicleID, true, before).
|
|
Order("refuel_time DESC, id DESC").
|
|
First(&rec).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, apperror.New(apperror.CodeNotFound, "此前没有加满油记录")
|
|
}
|
|
return nil, apperror.Wrap(apperror.CodeInternal, "查询前次加满记录失败", err)
|
|
}
|
|
return &rec, nil
|
|
}
|