Version: 0.3.9.dev.260226

fix: 🐛 修复智能排程接口任务块信息缺失与数据插入错误

* 修复智能排程接口返回的任务块信息缺失问题,确保任务数据完整返回
* 修复 `UserInsertTaskItemRequestToModel` DTO 函数未填入起始时间字段的问题,解决多个接口插入数据时出现 500 错误

  * 错误源自上次更新“获取最近完成动态任务接口”时,未同步更改数据库字段对应逻辑
* 将智能排程接口的 `ctx` 超时恢复为 1 秒,优化接口响应性能

feat: 🎯 新增正式应用日程接口

* 新增正式应用日程接口,并完成功能测试,确保业务流程无异常
This commit is contained in:
LoveLosita
2026-02-26 20:07:55 +08:00
parent 1399b38f16
commit fca4004024
12 changed files with 397 additions and 10 deletions

View File

@@ -155,9 +155,8 @@ func (s *ScheduleAPI) SmartPlanning(c *gin.Context) {
return
}
//3.调用服务层方法进行智能规划
/*ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)*/
ctx := context.Background()
/*defer cancel() // 记得释放资源*/
ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)
defer cancel() // 记得释放资源
res, err := s.scheduleService.SmartPlanning(ctx, userID, intTaskClassID)
if err != nil {
respond.DealWithError(c, err)

View File

@@ -171,3 +171,22 @@ func (api *TaskClassHandler) DeleteTaskClass(c *gin.Context) {
}
c.JSON(http.StatusOK, respond.Ok)
}
func (api *TaskClassHandler) UserInsertBatchTaskClassItemsIntoSchedule(c *gin.Context) {
var req model.UserInsertTaskClassItemToScheduleRequestBatch
err := c.ShouldBindJSON(&req)
if err != nil {
c.JSON(http.StatusBadRequest, respond.WrongParamType)
return
}
userID := c.GetInt("user_id")
// 创建一个带 1 秒超时的上下文
ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)
defer cancel() // 记得释放资源
err = api.svc.BatchApplyPlans(ctx, req.TaskClassID, userID, &req)
if err != nil {
respond.DealWithError(c, err)
return
}
c.JSON(http.StatusOK, respond.Ok)
}

View File

@@ -474,6 +474,7 @@ func buildBrief(slot slotInfo, day, start, end, span, order int) model.WeeklyEve
brief.Name = *slot.plan.Content
brief.Type = "task"
brief.Status = "suggested" // 标记为建议状态
brief.ID = slot.plan.ID // 虚日程的 ID 直接使用 TaskClassItem 的 ID方便前端追踪和操作
}
return brief

View File

@@ -138,7 +138,7 @@ func ProcessUserGetCompleteTaskClassRequest(taskClass *model.TaskClass) (*model.
}
// UserInsertTaskItemRequestToModel 用于将填入空闲时段日程的请求转换为 Schedule 模型
func UserInsertTaskItemRequestToModel(req *model.UserInsertTaskClassItemToScheduleRequest, item *model.TaskClassItem, taskID *int, userID, startSection, endSection int) ([]model.Schedule, *model.ScheduleEvent) {
func UserInsertTaskItemRequestToModel(req *model.UserInsertTaskClassItemToScheduleRequest, item *model.TaskClassItem, taskID *int, userID, startSection, endSection int) ([]model.Schedule, *model.ScheduleEvent, error) {
var schedules []model.Schedule
for section := startSection; section <= endSection; section++ {
req1 := &model.Schedule{
@@ -151,14 +151,20 @@ func UserInsertTaskItemRequestToModel(req *model.UserInsertTaskClassItemToSchedu
}
schedules = append(schedules, *req1)
}
startTime, endTime, err := RelativeTimeToRealTime(req.Week, req.DayOfWeek, startSection, endSection)
if err != nil {
return nil, nil, err
}
req2 := &model.ScheduleEvent{
UserID: userID, // 由调用方填充
Name: safeStr(item.Content), // 任务内容作为事件名称
Type: "task",
RelID: &item.ID, // 关联到 TaskClassItem 的 ID
CanBeEmbedded: false, // 任务事件允许嵌入其他任务(如果需要的话)
StartTime: startTime,
EndTime: endTime,
}
return schedules, req2
return schedules, req2, nil
}
// --- 🛡️ 辅助工具函数:保持代码清爽并防止 Panic ---

View File

@@ -62,7 +62,7 @@ type SectionTime struct {
End string // 第一个结束
}
var sectionTimeMap2 = map[int]SectionTime{
var SectionTimeMap2 = map[int]SectionTime{
1: {Start: "08:00", End: "08:45"},
2: {Start: "08:55", End: "09:40"},
3: {Start: "10:15", End: "11:00"},
@@ -83,8 +83,8 @@ func RelativeTimeToRealTime(week, dayOfWeek, startSection, endSection int) (time
return time.Time{}, time.Time{}, respond.InvalidSectionRange
}
startTimeInfo, okStart := sectionTimeMap2[startSection]
endTimeInfo, okEnd := sectionTimeMap2[endSection]
startTimeInfo, okStart := SectionTimeMap2[startSection]
endTimeInfo, okEnd := SectionTimeMap2[endSection]
if !okStart || !okEnd {
return time.Time{}, time.Time{}, respond.InvalidSectionNumber
}

View File

@@ -573,3 +573,55 @@ func (d *ScheduleDAO) GetUserSchedulesByTimeRange(ctx context.Context, userID in
}
return schedules, nil
}
func (d *ScheduleDAO) BatchEmbedTaskIntoSchedule(ctx context.Context, eventIDs, taskItemIDs []int) error {
if len(eventIDs) == 0 {
return nil
}
if len(eventIDs) != len(taskItemIDs) {
return fmt.Errorf("eventIDs length != taskItemIDs length")
}
db := d.db.WithContext(ctx)
for i, eventID := range eventIDs {
taskItemID := taskItemIDs[i]
// 1) 校验该 event 是否为 course
var typ string
if err := db.
Table("schedule_events").
Select("type").
Where("id = ?", eventID).
Scan(&typ).Error; err != nil {
return err
}
if typ != "course" {
continue
}
// 2) 一 event 对多 schedules批量写入 embedded_task_id
if err := db.
Table("schedules").
Where("event_id = ?", eventID).
Update("embedded_task_id", taskItemID).Error; err != nil {
return err
}
}
return nil
}
func (d *ScheduleDAO) InsertScheduleEvents(ctx context.Context, events []model.ScheduleEvent) ([]int, error) {
if len(events) == 0 {
return nil, nil
}
if err := d.db.WithContext(ctx).Create(&events).Error; err != nil {
return nil, err
}
ids := make([]int, len(events))
for i, e := range events {
ids[i] = e.ID
}
return ids, nil
}

View File

@@ -212,6 +212,21 @@ func (dao *TaskClassDAO) IfTaskClassItemArranged(ctx context.Context, taskID int
return item.EmbeddedTime != nil, nil
}
func (dao *TaskClassDAO) BatchCheckIfTaskClassItemsArranged(ctx context.Context, itemIDs []int) (bool, error) {
if len(itemIDs) == 0 {
return false, nil
}
var count int64
err := dao.db.WithContext(ctx).
Model(&model.TaskClassItem{}).
Where("id IN ? AND embedded_time IS NOT NULL", itemIDs).
Count(&count).Error
if err != nil {
return false, err
}
return count > 0, nil
}
func (dao *TaskClassDAO) DeleteTaskClassItemByID(ctx context.Context, id int) error {
err := dao.db.WithContext(ctx).
Where("id = ?", id).
@@ -231,3 +246,55 @@ func (dao *TaskClassDAO) DeleteTaskClassByID(ctx context.Context, id int) error
}
return nil
}
func (dao *TaskClassDAO) BatchUpdateTaskClassItemEmbeddedTime(ctx context.Context, itemIDs []int, updates []*model.TargetTime) error {
if len(itemIDs) == 0 {
return nil
}
if len(itemIDs) != len(updates) {
return errors.New("itemIDs length mismatch updates length")
}
// 单条 SQL 批量更新UPDATE ... SET embedded_time = CASE id WHEN ? THEN ? ... END WHERE id IN (?)
caseSQL := "CASE id"
args := make([]any, 0, len(itemIDs)*2)
for i, id := range itemIDs {
caseSQL += " WHEN ? THEN ?"
args = append(args, id, updates[i])
}
caseSQL += " END"
res := dao.db.WithContext(ctx).
Model(&model.TaskClassItem{}).
Where("id IN ?", itemIDs).
Update("embedded_time", gorm.Expr(caseSQL, args...))
return res.Error
}
func (dao *TaskClassDAO) ValidateTaskItemIDsBelongToTaskClass(ctx context.Context, taskClassID int, itemIDs []int) (bool, error) {
if len(itemIDs) == 0 {
return true, nil
}
var count int64
err := dao.db.WithContext(ctx).
Model(&model.TaskClassItem{}).
Where("id IN ? AND category_id = ?", itemIDs, taskClassID).
Count(&count).Error
if err != nil {
return false, err
}
return count == int64(len(itemIDs)), nil
}
func (dao *TaskClassDAO) GetTaskClassItemsByIDs(ctx context.Context, itemIDs []int) ([]model.TaskClassItem, error) {
var items []model.TaskClassItem
err := dao.db.WithContext(ctx).
Where("id IN ?", itemIDs).
Find(&items).Error
if err != nil {
return nil, err
}
return items, nil
}

View File

@@ -129,6 +129,20 @@ type UserInsertTaskClassItemToScheduleRequest struct {
EmbedCourseEventID int `json:"embed_course_event_id"` // 可选,嵌入的课程日程事件 ID
}
type UserInsertTaskClassItemToScheduleRequestBatch struct {
TaskClassID int `json:"task_class_id" binding:"required"`
Items []SingleTaskClassItem `json:"items" binding:"required,dive,required"`
}
type SingleTaskClassItem struct {
TaskItemID int `json:"task_item_id" binding:"required"`
Week int `json:"week" binding:"required,min=1"`
DayOfWeek int `json:"day_of_week" binding:"required,min=1,max=7"`
StartSection int `json:"start_section" binding:"required,min=1"`
EndSection int `json:"end_section" binding:"required,min=1,gtefield=StartSection"`
EmbedCourseEventID int `json:"embed_course_event_id"` // 可选,嵌入的课程日程事件 ID
}
// Value 实现 driver.Valuer 接口,负责将 TargetTime 转换为数据库存储的格式
func (t *TargetTime) Value() (driver.Value, error) {
if t == nil {

View File

@@ -304,4 +304,13 @@ var ( //请求相关的响应
Status: "40047",
Info: "time not enough for auto scheduling",
}
TaskClassItemNotBelongToTaskClass = Response{ //任务类项目不属于任务类
Status: "40048",
Info: "task class item does not belong to task class",
}
TaskClassItemTryingToInsertOutOfTimeRange = Response{ //任务类项目试图插入超出时间范围
Status: "40049",
Info: "task class item trying to insert out of time range",
}
)

View File

@@ -71,6 +71,7 @@ func RegisterRouters(handlers *api.ApiHandlers, cache *dao.CacheDAO, limiter *pk
taskClassGroup.POST("/insert-into-schedule", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserAddTaskClassItemIntoSchedule)
taskClassGroup.DELETE("/delete-item", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.DeleteTaskClassItem)
taskClassGroup.DELETE("/delete-class", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.DeleteTaskClass)
taskClassGroup.PUT("/apply-batch-into-schedule", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserInsertBatchTaskClassItemsIntoSchedule)
}
scheduleGroup := apiGroup.Group("/schedule")
{

View File

@@ -163,11 +163,14 @@ func (ss *ScheduleService) DeleteScheduleEvent(ctx context.Context, requests []m
//直接构造Schedule模型
sections := make([]int, 0, taskClassItem.EmbeddedTime.SectionTo-taskClassItem.EmbeddedTime.SectionFrom+1)
// 这里的 req 主要是为了传递 Week 和 DayOfWeek其他字段不需要了
schedules, scheduleEvent := conv.UserInsertTaskItemRequestToModel(
schedules, scheduleEvent, err := conv.UserInsertTaskItemRequestToModel(
&model.UserInsertTaskClassItemToScheduleRequest{
Week: taskClassItem.EmbeddedTime.Week,
DayOfWeek: taskClassItem.EmbeddedTime.DayOfWeek},
taskClassItem, nil, userID, taskClassItem.EmbeddedTime.SectionFrom, taskClassItem.EmbeddedTime.SectionTo)
if err != nil {
return err
}
//将节次区间转换为节次切片,方便后续检查冲突
for section := taskClassItem.EmbeddedTime.SectionFrom; section <= taskClassItem.EmbeddedTime.SectionTo; section++ {
sections = append(sections, section)

View File

@@ -3,7 +3,9 @@ package service
import (
"context"
"errors"
"fmt"
"log"
"sort"
"time"
"github.com/LoveLosita/smartflow/backend/conv"
@@ -192,7 +194,10 @@ func (sv *TaskClassService) AddTaskClassItemIntoSchedule(ctx context.Context, re
}
//4.否则构造Schedule模型
sections := make([]int, 0, req.EndSection-req.StartSection+1)
schedules, scheduleEvent := conv.UserInsertTaskItemRequestToModel(req, taskItem, nil, userID, req.StartSection, req.EndSection)
schedules, scheduleEvent, err := conv.UserInsertTaskItemRequestToModel(req, taskItem, nil, userID, req.StartSection, req.EndSection)
if err != nil {
return err
}
//将节次区间转换为节次切片,方便后续检查冲突
for section := req.StartSection; section <= req.EndSection; section++ {
sections = append(sections, section)
@@ -309,3 +314,214 @@ func (sv *TaskClassService) DeleteTaskClass(ctx context.Context, userID int, tas
}
return nil
}
func (sv *TaskClassService) BatchApplyPlans(ctx context.Context, taskClassID int, userID int, plans *model.UserInsertTaskClassItemToScheduleRequestBatch) error {
//1.通过任务类id获取任务类详情
taskClass, err := sv.taskClassRepo.GetCompleteTaskClassByID(ctx, taskClassID, userID)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return respond.WrongTaskClassID
}
return err
}
//2.校验任务类的参数是否合法
if taskClass == nil {
return respond.WrongTaskClassID
}
if *taskClass.Mode != "auto" {
return respond.TaskClassModeNotAuto
}
//3.获取任务类安排的时间范围内的全部周数信息(左右边界不足一周的情况也要算作一周),用于下方冲突检查
startWeekTime := conv.CalculateFirstDayOfWeek(*taskClass.StartDate)
endWeekTime := conv.CalculateLastDayOfWeek(*taskClass.EndDate)
schedules, err := sv.scheduleRepo.GetUserSchedulesByTimeRange(ctx, userID, startWeekTime, endWeekTime)
if err != nil {
return err
}
startWeek, _, err := conv.RealDateToRelativeDate(startWeekTime.Format("2006-01-02"))
if err != nil {
return err
}
endWeek, _, err := conv.RealDateToRelativeDate(endWeekTime.Format("2006-01-02"))
if err != nil {
return err
}
//4.统一检查冲突(避免逐条查库)
//先将日程放入一个map中key是"周-星期-节次"value是课程信息方便后续检查冲突
courseMap := make(map[string]model.Schedule)
for _, schedule := range schedules {
key := fmt.Sprintf("%d-%d-%d", schedule.Week, schedule.DayOfWeek, schedule.Section)
courseMap[key] = schedule
}
//再遍历每个任务块的安排时间,检查是否和课程冲突(目前逻辑是只要有一个时段冲突就算冲突,后续可以优化为统计冲突的时段数量,或者提供具体的冲突时段信息)
for _, plan := range plans.Items {
if plan.Week < startWeek || plan.Week > endWeek {
return respond.TaskClassItemTryingToInsertOutOfTimeRange
}
for section := plan.StartSection; section <= plan.EndSection; section++ {
key := fmt.Sprintf("%d-%d-%d", plan.Week, plan.DayOfWeek, section)
// 如果课程存在,并且满足以下任一条件则认为冲突:
// 1. 课程时段已经被其他任务块嵌入了(不允许多个任务块嵌入同一课程)
// 2. 当前时段的课的EventID与用户计划中指定的EmbedCourseEventID不匹配说明用户计划要嵌入的课程和当前时段的课不是同一节
// 3. 用户计划中没有指定EmbedCourseEventID即EmbedCourseEventID为0但当前时段有课不允许在有课的时段安排任务块
// 4. 当前时段的课不允许被嵌入即使用户计划中指定了EmbedCourseEventID但如果课程本身不允许被嵌入了也算冲突
if course, exists := courseMap[key]; exists && ((plan.EmbedCourseEventID != 0 && course.EmbeddedTask != nil) ||
(plan.EmbedCourseEventID != course.EventID) || plan.EmbedCourseEventID == 0 || !course.Event.CanBeEmbedded) {
return respond.ScheduleConflict
}
}
}
//5.分流批量写入数据库(通过 RepoManager 统一管理事务)
//先分流
toEmbed := make([]model.SingleTaskClassItem, 0) //需要嵌入课程的任务块
toNormal := make([]model.SingleTaskClassItem, 0) //需要新建日程的任务块
for _, item := range plans.Items {
if item.EmbedCourseEventID != 0 {
toEmbed = append(toEmbed, item)
} else {
toNormal = append(toNormal, item)
}
}
//再开事务批量写库
if err := sv.repoManager.Transaction(ctx, func(txM *dao.RepoManager) error {
//5.1 先处理需要嵌入课程的任务块
//先提取出需要嵌入的课程ID和TaskItemID列表
courseIDs := make([]int, 0, len(toEmbed))
for _, item := range toEmbed {
courseIDs = append(courseIDs, item.EmbedCourseEventID)
}
itemIDs := make([]int, 0, len(toEmbed))
for _, item := range toEmbed {
itemIDs = append(itemIDs, item.TaskItemID)
}
//检查任务块本身是否已经被安排
result, err := sv.taskClassRepo.BatchCheckIfTaskClassItemsArranged(ctx, itemIDs)
if err != nil {
return err
}
if result {
return respond.TaskClassItemAlreadyArranged
}
//验证一下plans中的taskItemID确实都属于这个用户和这个任务类避免用户恶意构造请求把别的用户的任务块或者不属于任何任务类的任务块也安排了
//同时也能检查是否重复
result, err = sv.taskClassRepo.ValidateTaskItemIDsBelongToTaskClass(ctx, taskClassID, itemIDs)
if err != nil {
return err
}
if !result {
return respond.TaskClassItemNotBelongToTaskClass
}
//批量更新日程表中对应课程的embedded_task_id字段目前业务限制一个课程只能被一个任务块嵌入了所以直接批量更新不用担心覆盖问题
err = txM.Schedule.BatchEmbedTaskIntoSchedule(ctx, courseIDs, itemIDs)
if err != nil {
return err
}
//批量更新任务块的embedded_time字段
targetTimes := make([]*model.TargetTime, 0, len(toEmbed))
for _, item := range toEmbed {
targetTimes = append(targetTimes, &model.TargetTime{
DayOfWeek: item.DayOfWeek,
Week: item.Week,
SectionFrom: item.StartSection,
SectionTo: item.EndSection,
})
}
err = txM.TaskClass.BatchUpdateTaskClassItemEmbeddedTime(ctx, itemIDs, targetTimes)
if err != nil {
return err
}
//5.2 再处理需要新建日程的任务块
//先提取出需要新建日程的任务块ID列表
normalItemIDs := make([]int, 0, len(toNormal))
for _, item := range toNormal {
normalItemIDs = append(normalItemIDs, item.TaskItemID)
}
//验证一下plans中的taskItemID确实都属于这个任务类避免用户恶意构造请求把别的用户的任务块或者不属于任何任务类的任务块也安排了
result, err = sv.taskClassRepo.ValidateTaskItemIDsBelongToTaskClass(ctx, taskClassID, normalItemIDs)
if err != nil {
return err
}
if !result {
return respond.TaskClassItemNotBelongToTaskClass
}
//批量提取TaskItems
taskItems, err := txM.TaskClass.GetTaskClassItemsByIDs(ctx, normalItemIDs)
if err != nil {
return err
}
if len(taskItems) != len(normalItemIDs) {
log.Printf("警告批量提取任务块时返回的任务块数量与请求中的任务块ID数量不匹配可能存在数据问题。请求ID数量%d返回任务块数量%d", len(normalItemIDs), len(taskItems))
return respond.InternalError(errors.New("返回的任务块数量与请求中的任务块ID数量不匹配可能存在数据问题"))
}
//将toNormal按照TaskItemID升序排序将taskItems也按照ID升序排序保证一一对应关系上面已经检查过重复
//如果请求中的任务块ID有重复这里就无法保证一一对应关系了后续可以考虑在请求层面加一个校验拒绝包含重复任务块ID的请求
sort.SliceStable(toNormal, func(i, j int) bool {
return toNormal[i].TaskItemID < toNormal[j].TaskItemID
})
sort.SliceStable(taskItems, func(i, j int) bool {
return taskItems[i].ID < taskItems[j].ID
})
//开始构建event和schedules
finalSchedules := make([]model.Schedule, 0) //最终要插入数据库的Schedule切片
finalScheduleEvents := make([]model.ScheduleEvent, 0) //最终要插入数据库的ScheduleEvent切片
pos := make([]int, 0) //记录每个任务块对应的Schedule在finalSchedules中的位置方便后续批量插入数据库后回填EventID
for i := 0; i < len(toNormal); i++ {
item := toNormal[i]
taskItem := taskItems[i]
if item.StartSection < 1 || item.EndSection > 12 || item.StartSection > item.EndSection {
return respond.InvalidSectionRange
}
schedules, scheduleEvent, err := conv.UserInsertTaskItemRequestToModel(&model.UserInsertTaskClassItemToScheduleRequest{
Week: item.Week,
DayOfWeek: item.DayOfWeek,
StartSection: item.StartSection,
EndSection: item.EndSection,
EmbedCourseEventID: 0, //不嵌入课程
}, &taskItem, nil, userID, item.StartSection, item.EndSection)
if err != nil {
return err
}
finalScheduleEvents = append(finalScheduleEvents, *scheduleEvent)
for range schedules {
pos = append(pos, len(finalScheduleEvents)-1)
}
finalSchedules = append(finalSchedules, schedules...)
}
//最后批量插入数据库
//先插入ScheduleEvent表获取生成的EventID再批量插入Schedule表最后批量更新TaskClassItem的embedded_time字段
ids, err := txM.Schedule.InsertScheduleEvents(ctx, finalScheduleEvents)
if err != nil {
return err
}
// 将生成的 ScheduleEvent ID 赋值给对应的 Schedule 的 EventID 字段
for i := range finalSchedules {
finalSchedules[i].EventID = ids[pos[i]]
}
if _, err = txM.Schedule.AddSchedules(finalSchedules); err != nil {
return err
}
//批量更新任务块的embedded_time字段
targetTimes = make([]*model.TargetTime, 0, len(toEmbed))
for _, item := range toNormal {
targetTimes = append(targetTimes, &model.TargetTime{
DayOfWeek: item.DayOfWeek,
Week: item.Week,
SectionFrom: item.StartSection,
SectionTo: item.EndSection,
})
}
//提取出所有需要更新的任务块ID
itemIDs = make([]int, 0, len(toNormal))
for _, item := range toNormal {
itemIDs = append(itemIDs, item.TaskItemID)
}
err = txM.TaskClass.BatchUpdateTaskClassItemEmbeddedTime(ctx, itemIDs, targetTimes)
if err != nil {
return err
}
return nil
}); err != nil {
return err
}
return nil
}