79b6be5f40
refactor: ♻️ 重命名部分接口接收器以避免与 dao 层包名冲突 - 调整部分接口接收器命名，避免与 dao 层包名重名 🧩 feat: 📅 新增获取用户今日日程接口并完成实现 fix: 🐛 修复现实日期与相对日期转换逻辑中的初始化时序问题 - 修复 conv/time.go 中日期转换函数的一个 bug ⏱️ - 解决 viper 在包级变量初始化时机过早的问题 - 避免因过早初始化导致无法读取配置中的学期开学/结束时间 📆
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type CourseDAO struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewCourseDAO 创建ScheduleDAO实例
|
||
func NewCourseDAO(db *gorm.DB) *CourseDAO {
|
||
return &CourseDAO{
|
||
db: db,
|
||
}
|
||
}
|
||
|
||
func (r *CourseDAO) WithTx(tx *gorm.DB) *CourseDAO {
|
||
return &CourseDAO{db: tx}
|
||
}
|
||
|
||
func (r *CourseDAO) AddUserCoursesIntoSchedule(ctx context.Context, courses []model.Schedule) error {
|
||
if err := r.db.WithContext(ctx).Create(&courses).Error; err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *CourseDAO) AddUserCoursesIntoScheduleEvents(ctx context.Context, events []model.ScheduleEvent) ([]int, error) {
|
||
if err := r.db.WithContext(ctx).Create(&events).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
ids := make([]int, 0, len(events))
|
||
for i := range events {
|
||
ids = append(ids, int(events[i].ID))
|
||
}
|
||
return ids, nil
|
||
}
|
||
|
||
// Transaction 在同一个数据库事务中执行传入的函数，供 service 层复用（自动提交/回滚）
|
||
// 规则：fn 返回 nil \-\> 提交；fn 返回 error 或发生 panic \-\> 回滚
|
||
// 说明：gorm\.\(\\\*DB\)\.Transaction 会在 fn 返回 error 时回滚，并在发生 panic 时自动回滚后继续向上抛出 panic
|
||
func (r *CourseDAO) Transaction(fn func(txDAO *CourseDAO) error) error {
|
||
return r.db.Transaction(func(tx *gorm.DB) error {
|
||
return fn(NewCourseDAO(tx))
|
||
})
|
||
}
|