82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
"yuyue/database"
|
|||
|
|
"yuyue/models"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// GenerateTwoMonthsTimeSlots 生成最近两个月的预约时间槽
|
|||
|
|
func GenerateTwoMonthsTimeSlots() error {
|
|||
|
|
// 计算日期范围：明天开始的两个月
|
|||
|
|
now := time.Now()
|
|||
|
|
startDate := time.Date(now.Year(), now.Month(), now.Day()+1, 0, 0, 0, 0, now.Location()) // 明天
|
|||
|
|
endDate := startDate.AddDate(0, 2, 0) // 两个月后
|
|||
|
|
|
|||
|
|
fmt.Printf("开始生成时间槽：%s 到 %s\n", startDate.Format("2006-01-02"), endDate.Format("2006-01-02"))
|
|||
|
|
|
|||
|
|
generatedSlots := []models.TimeSlot{}
|
|||
|
|
conflictCount := 0
|
|||
|
|
|
|||
|
|
currentDate := startDate
|
|||
|
|
for !currentDate.After(endDate) {
|
|||
|
|
// 排除周末
|
|||
|
|
weekday := currentDate.Weekday()
|
|||
|
|
if weekday == time.Saturday || weekday == time.Sunday {
|
|||
|
|
currentDate = currentDate.AddDate(0, 0, 1)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 生成工作日的时间槽 (9:00-18:00，每小时一个时段)
|
|||
|
|
startTime := time.Date(currentDate.Year(), currentDate.Month(), currentDate.Day(),
|
|||
|
|
9, 0, 0, 0, currentDate.Location())
|
|||
|
|
endTime := time.Date(currentDate.Year(), currentDate.Month(), currentDate.Day(),
|
|||
|
|
18, 0, 0, 0, currentDate.Location())
|
|||
|
|
|
|||
|
|
currentSlotTime := startTime
|
|||
|
|
for currentSlotTime.Before(endTime) {
|
|||
|
|
slotEnd := currentSlotTime.Add(time.Hour)
|
|||
|
|
|
|||
|
|
// 检查是否已经存在相同的时间槽
|
|||
|
|
var existingSlot models.TimeSlot
|
|||
|
|
err := database.GetDB().Where("date = ? AND start_time = ? AND end_time = ?",
|
|||
|
|
currentDate, currentSlotTime, slotEnd).First(&existingSlot).Error
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
// 不存在，创建新的时间槽
|
|||
|
|
newSlot := models.TimeSlot{
|
|||
|
|
Date: currentDate,
|
|||
|
|
StartTime: currentSlotTime,
|
|||
|
|
EndTime: slotEnd,
|
|||
|
|
MaxPeople: 5, // 每个时段最多5人
|
|||
|
|
CurrentPeople: 0,
|
|||
|
|
IsActive: true,
|
|||
|
|
}
|
|||
|
|
generatedSlots = append(generatedSlots, newSlot)
|
|||
|
|
} else {
|
|||
|
|
conflictCount++
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
currentSlotTime = slotEnd
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
currentDate = currentDate.AddDate(0, 0, 1)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 批量插入到数据库
|
|||
|
|
if len(generatedSlots) > 0 {
|
|||
|
|
if err := database.GetDB().CreateInBatches(generatedSlots, 100).Error; err != nil {
|
|||
|
|
return fmt.Errorf("批量创建时间槽失败: %v", err)
|
|||
|
|
}
|
|||
|
|
fmt.Printf("成功生成 %d 个时间槽\n", len(generatedSlots))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if conflictCount > 0 {
|
|||
|
|
fmt.Printf("跳过 %d 个已存在的时间槽\n", conflictCount)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Printf("总共处理 %d 个时间槽\n", len(generatedSlots)+conflictCount)
|
|||
|
|
return nil
|
|||
|
|
}
|