初始
This commit is contained in:
234
internal/app/system/logic/workSchedule/work_schedule.go
Normal file
234
internal/app/system/logic/workSchedule/work_schedule.go
Normal file
@ -0,0 +1,234 @@
|
||||
// ==========================================================================
|
||||
// GFast自动生成logic操作代码。
|
||||
// 生成日期：2024-03-20 10:02:26
|
||||
// 生成路径: internal/app/system/logic/work_schedule.go
|
||||
// 生成人：gfast
|
||||
// desc:工作计划
|
||||
// company:云南奇讯科技有限公司
|
||||
// ==========================================================================
|
||||
|
||||
package logic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"github.com/tiger1103/gfast/v3/api/v1/system"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/consts"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/model"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
|
||||
"github.com/tiger1103/gfast/v3/internal/app/system/service"
|
||||
"github.com/tiger1103/gfast/v3/library/liberr"
|
||||
)
|
||||
|
||||
func init() {
|
||||
service.RegisterWorkSchedule(New())
|
||||
}
|
||||
|
||||
func New() *sWorkSchedule {
|
||||
return &sWorkSchedule{}
|
||||
}
|
||||
|
||||
type sWorkSchedule struct{}
|
||||
|
||||
func (s *sWorkSchedule) List(ctx context.Context, req *system.WorkScheduleSearchReq) (listRes *system.WorkScheduleSearchRes, err error) {
|
||||
listRes = new(system.WorkScheduleSearchRes)
|
||||
err = g.Try(ctx, func(ctx context.Context) {
|
||||
m := dao.WorkSchedule.Ctx(ctx).WithAll()
|
||||
if req.Id != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().Id+" = ?", req.Id)
|
||||
}
|
||||
if req.WorkId != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().WorkId+" = ?", req.WorkId)
|
||||
}
|
||||
if req.StartAt != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().StartAt+" = ?", gconv.Time(req.StartAt))
|
||||
}
|
||||
if req.Detail != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().Detail+" = ?", req.Detail)
|
||||
}
|
||||
if req.Status != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().Status+" = ?", gconv.Int(req.Status))
|
||||
}
|
||||
if req.PlanNum != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().PlanNum+" = ?", gconv.Int(req.PlanNum))
|
||||
}
|
||||
if req.FinishedNum != "" {
|
||||
m = m.Where(dao.WorkSchedule.Columns().FinishedNum+" = ?", gconv.Int(req.FinishedNum))
|
||||
}
|
||||
listRes.Total, err = m.Count()
|
||||
liberr.ErrIsNil(ctx, err, "获取总行数失败")
|
||||
if req.PageNum == 0 {
|
||||
req.PageNum = 1
|
||||
}
|
||||
listRes.CurrentPage = req.PageNum
|
||||
if req.PageSize == 0 {
|
||||
req.PageSize = consts.PageSize
|
||||
}
|
||||
order := "id asc"
|
||||
if req.OrderBy != "" {
|
||||
order = req.OrderBy
|
||||
}
|
||||
var res []*model.WorkScheduleInfoRes
|
||||
err = m.Fields(system.WorkScheduleSearchRes{}).Page(req.PageNum, req.PageSize).Order(order).Scan(&res)
|
||||
liberr.ErrIsNil(ctx, err, "获取数据失败")
|
||||
listRes.List = make([]*model.WorkScheduleInfo2Res, len(res))
|
||||
|
||||
for k, v := range res {
|
||||
listRes.List[k] = &model.WorkScheduleInfo2Res{
|
||||
Id: v.Id,
|
||||
WorkId: v.WorkId,
|
||||
StartAt: v.StartAt,
|
||||
EndAt: v.EndAt,
|
||||
Status: v.Status,
|
||||
PlanNum: v.PlanNum,
|
||||
FinishedNum: v.FinishedNum,
|
||||
}
|
||||
|
||||
// detail 反序列化
|
||||
var scheduleDetails []model.WorkScheduleDetail
|
||||
if err := json.Unmarshal([]byte(v.Detail), &scheduleDetails); err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
listRes.List[k].Detail = scheduleDetails
|
||||
}
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func (s *sWorkSchedule) GetById(ctx context.Context, id uint) (res *model.WorkScheduleInfoRes, err error) {
|
||||
err = g.Try(ctx, func(ctx context.Context) {
|
||||
err = dao.WorkSchedule.Ctx(ctx).WithAll().Where(dao.WorkSchedule.Columns().Id, id).Scan(&res)
|
||||
liberr.ErrIsNil(ctx, err, "获取信息失败")
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func (s *sWorkSchedule) Add(ctx context.Context, req *system.WorkScheduleAddReq) (err error) {
|
||||
err = g.Try(ctx, func(ctx context.Context) {
|
||||
_, err = dao.WorkSchedule.Ctx(ctx).Insert(do.WorkSchedule{
|
||||
WorkId: req.WorkId,
|
||||
StartAt: req.StartAt,
|
||||
Detail: req.Detail,
|
||||
Status: req.Status,
|
||||
PlanNum: req.PlanNum,
|
||||
FinishedNum: req.FinishedNum,
|
||||
})
|
||||
liberr.ErrIsNil(ctx, err, "添加失败")
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func (s *sWorkSchedule) Edit(ctx context.Context, req *system.WorkScheduleEditReq) (err error) {
|
||||
err = g.Try(ctx, func(ctx context.Context) {
|
||||
_, err = dao.WorkSchedule.Ctx(ctx).WherePri(req.Id).Update(do.WorkSchedule{
|
||||
WorkId: req.WorkId,
|
||||
StartAt: req.StartAt,
|
||||
Detail: req.Detail,
|
||||
Status: req.Status,
|
||||
PlanNum: req.PlanNum,
|
||||
FinishedNum: req.FinishedNum,
|
||||
})
|
||||
liberr.ErrIsNil(ctx, err, "修改失败")
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func (s *sWorkSchedule) Delete(ctx context.Context, req *system.WorkScheduleDeleteReq) (err error) {
|
||||
err = g.Try(ctx, func(ctx context.Context) {
|
||||
// 获取该计划的详细信息
|
||||
var workSchedules entity.WorkSchedule
|
||||
err := dao.WorkSchedule.Ctx(ctx).Where(dao.WorkSchedule.Columns().Id, req.Ids).Scan(&workSchedules)
|
||||
liberr.ErrIsNil(ctx, err, "获取该计划失败")
|
||||
|
||||
_, err = dao.WorkSchedule.Ctx(ctx).Where(dao.WorkSchedule.Columns().Id, req.Ids).Delete()
|
||||
liberr.ErrIsNil(ctx, err, "删除计划失败")
|
||||
|
||||
// 反序列化 workSchedules.Detail
|
||||
var scheduleDetails []model.WorkScheduleDetail
|
||||
if err := json.Unmarshal([]byte(workSchedules.Detail), &scheduleDetails); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 遍历 scheduleDetails，将 pv_module 表中的状态修改为 0
|
||||
for _, scheduleDetail := range scheduleDetails {
|
||||
_, err = dao.PvModule.Ctx(ctx).Where(dao.PvModule.Columns().WorkId, workSchedules.WorkId).
|
||||
Where(dao.PvModule.Columns().DoneTime, scheduleDetail.Date).
|
||||
Data(g.Map{
|
||||
dao.PvModule.Columns().Status: 0,
|
||||
dao.PvModule.Columns().DoneTime: "",
|
||||
}).Update()
|
||||
liberr.ErrIsNil(ctx, err, "修改状态失败")
|
||||
}
|
||||
|
||||
// 重新统计在 work_schedule 表中的完成数量
|
||||
finishedNum, err := dao.WorkSchedule.Ctx(ctx).
|
||||
Where(dao.WorkSchedule.Columns().WorkId, workSchedules.WorkId).
|
||||
Sum(dao.WorkSchedule.Columns().FinishedNum)
|
||||
liberr.ErrIsNil(ctx, err, "重新获取完成数量失败")
|
||||
|
||||
isExist, err := dao.WorkSchedule.Ctx(ctx).
|
||||
Where(dao.WorkSchedule.Columns().WorkId, workSchedules.WorkId).Count()
|
||||
liberr.ErrIsNil(ctx, err, "获取计划数量失败")
|
||||
|
||||
// 如果不存在任何计划则将状态修改为 `未开始` 并将 `完成数量` 修改为 0
|
||||
if isExist == 0 {
|
||||
_, err = dao.WorkStatus.Ctx(ctx).
|
||||
Where(dao.WorkStatus.Columns().WorkId, workSchedules.WorkId).
|
||||
Data(g.Map{
|
||||
dao.WorkStatus.Columns().Finished: 0,
|
||||
dao.WorkStatus.Columns().Status: 0,
|
||||
}).Update()
|
||||
liberr.ErrIsNil(ctx, err, "更新计划完成数量失败")
|
||||
return
|
||||
}
|
||||
|
||||
_, err = dao.WorkStatus.Ctx(ctx).
|
||||
Where(dao.WorkStatus.Columns().WorkId, workSchedules.WorkId).
|
||||
Data(g.Map{
|
||||
dao.WorkStatus.Columns().Finished: finishedNum,
|
||||
dao.WorkStatus.Columns().Status: 1,
|
||||
}).Update()
|
||||
liberr.ErrIsNil(ctx, err, "更新计划完成数量失败")
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 判断在指定的时间周期内方阵或子项目是否有计划
|
||||
func (s *sWorkSchedule) IsExistWorkSchedule(ctx context.Context, subProjectID, fangzhenID string, startAt, endAt string) bool {
|
||||
var (
|
||||
id string // 子项目ID 或者 方阵ID
|
||||
column string
|
||||
)
|
||||
|
||||
// 获取子项目ID 或者 方阵ID
|
||||
if subProjectID != "" {
|
||||
id = subProjectID
|
||||
column = dao.WorkSchedule.Columns().SubProjectid
|
||||
} else if fangzhenID != "" {
|
||||
id = fangzhenID
|
||||
column = dao.WorkSchedule.Columns().FangzhenId
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
|
||||
// 获取下的所有工作计划
|
||||
workSchedule := []entity.WorkSchedule{}
|
||||
if err := dao.WorkSchedule.Ctx(ctx).Where(column, id).Scan(&workSchedule); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
// 遍历计划，判断是否在指定时间周期内有计划
|
||||
for _, v := range workSchedule {
|
||||
if v.StartAt.Format("Y-m-d") <= endAt {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
Reference in New Issue
Block a user