113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
|
package schduler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
|||
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type scheduler struct {
|
|||
|
|
// 是否延期
|
|||
|
|
IsDelay int
|
|||
|
|
// 总进度
|
|||
|
|
FinishedPtr int
|
|||
|
|
// 总量
|
|||
|
|
Total int
|
|||
|
|
// 实体
|
|||
|
|
WorkStatus *entity.WorkSchedule
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Schdule map[string]scheduler
|
|||
|
|
|
|||
|
|
// New 初始化计划
|
|||
|
|
func New(ScheduleData []entity.WorkSchedule) (Schdule, error) {
|
|||
|
|
ctx := context.Background()
|
|||
|
|
scheduleMap := make(Schdule)
|
|||
|
|
|
|||
|
|
for _, task := range ScheduleData {
|
|||
|
|
// 计划进度
|
|||
|
|
expectedProgress := task.PlanNum
|
|||
|
|
// 实际完成度
|
|||
|
|
actualProgress := task.FinishedNum
|
|||
|
|
|
|||
|
|
// 判断是否延期
|
|||
|
|
isDelayed := isDelay(task.EndAt.Format("Y-m-d"), expectedProgress, actualProgress)
|
|||
|
|
|
|||
|
|
// 如果 WorkID 不存在
|
|||
|
|
if _, exists := scheduleMap[task.WorkId]; !exists {
|
|||
|
|
newScheduler := scheduler{
|
|||
|
|
IsDelay: isDelayed,
|
|||
|
|
WorkStatus: &task,
|
|||
|
|
Total: expectedProgress,
|
|||
|
|
FinishedPtr: actualProgress,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
scheduleMap[task.WorkId] = newScheduler // 添加到 map 中
|
|||
|
|
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
existingScheduler := scheduleMap[task.WorkId] // 存在则取出 Value
|
|||
|
|
|
|||
|
|
// 判断延期
|
|||
|
|
if existingScheduler.IsDelay != 1 { // 为了防止在存在多个计划的情况下，未延期的计划覆盖了已经延期的状态。
|
|||
|
|
// 修改 Work_Status 中 is_delay 修改为延期状态 1
|
|||
|
|
_, err := dao.WorkStatus.Ctx(ctx).Where(dao.WorkStatus.Columns().WorkId, task.WorkId).
|
|||
|
|
Data(g.Map{dao.WorkStatus.Columns().IsDelay: 1}).Update()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
existingScheduler.IsDelay = isDelayed // 修改 map 中的状态
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 多个计划的情况下叠加数据
|
|||
|
|
existingScheduler.Total += expectedProgress // 计划总量叠加
|
|||
|
|
existingScheduler.FinishedPtr += actualProgress // 实际进度叠加
|
|||
|
|
scheduleMap[task.WorkId] = existingScheduler
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return scheduleMap, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取指定 WorkID 的计划和总量
|
|||
|
|
func (s Schdule) Get(workID string) (finishedPtr, total int) {
|
|||
|
|
if sch, ok := s[workID]; ok {
|
|||
|
|
return sch.FinishedPtr, sch.Total
|
|||
|
|
}
|
|||
|
|
return 0, 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetStatus 根据 WorkID 获取工作的状态，判断是否延期
|
|||
|
|
// s[workID] 中记录了该 WorkID 的计划总量和实际进度
|
|||
|
|
// total 是计划的总量
|
|||
|
|
// 如果实际进度（FinishedPtr）小于计划总量，则判断工作为延期（返回1）
|
|||
|
|
// 如果实际进度达到或超过计划总量，或者 WorkID 不存在于计划中，判断工作未延期（返回0）
|
|||
|
|
func (s Schdule) GetStatus(workID string, total int) int {
|
|||
|
|
if sch, ok := s[workID]; ok {
|
|||
|
|
if sch.FinishedPtr < total {
|
|||
|
|
return sch.IsDelay
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// isDelay 如果实际完成数量小于计划数量，且当前时间大于结束时间，则判断为延期
|
|||
|
|
// endAt 结束时间
|
|||
|
|
// planNum 计划数量
|
|||
|
|
// finishedNum 实际数量
|
|||
|
|
func isDelay(endAt string, planNum, finishedNum int) int {
|
|||
|
|
// 当天时间
|
|||
|
|
nowTime := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
|
|||
|
|
|
|||
|
|
// 如果实际完成数量小于计划数量，且当前时间大于结束时间，则判断为延期
|
|||
|
|
if finishedNum < planNum && nowTime > endAt {
|
|||
|
|
return 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return 0
|
|||
|
|
}
|