zmkgC/internal/app/system/controller/project_schedule.go

// ==========================================================================
// GFast自动生成controller操作代码。
// 生成日期：2024-03-13 18:17:32
// 生成路径: internal/app/system/controller/project_schedule.go
// 生成人：gfast
// desc:项目排期
// company:云南奇讯科技有限公司
// ==========================================================================

package controller

import (
	"context"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/tiger1103/gfast/v3/api/v1/system"
	"github.com/tiger1103/gfast/v3/internal/app/system/dao"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
	"github.com/tiger1103/gfast/v3/internal/app/system/service"
	"github.com/tiger1103/gfast/v3/library/liberr"
)

type projectScheduleController struct {
	BaseController
}

var ProjectSchedule = new(projectScheduleController)

// List 列表
func (c *projectScheduleController) List(ctx context.Context, req *system.ProjectScheduleSearchReq) (res *system.ProjectScheduleSearchRes, err error) {
	res, err = service.ProjectSchedule().List(ctx, req)
	return
}

// Get 获取项目排期
func (c *projectScheduleController) Get(ctx context.Context, req *system.ProjectScheduleGetReq) (res *system.ProjectScheduleGetRes, err error) {
	res = new(system.ProjectScheduleGetRes)
	res.ProjectScheduleInfoRes, err = service.ProjectSchedule().GetById(ctx, req.Id)
	return
}

// Add 添加项目排期
func (c *projectScheduleController) Add(ctx context.Context, req *system.ProjectScheduleAddReq) (res *system.ProjectScheduleAddRes, err error) {
	err = service.ProjectSchedule().Add(ctx, req)
	return
}

// Edit 修改项目排期
func (c *projectScheduleController) Edit(ctx context.Context, req *system.ProjectScheduleEditReq) (res *system.ProjectScheduleEditRes, err error) {
	err = service.ProjectSchedule().Edit(ctx, req)
	return
}

// Delete 删除项目排期
func (c *projectScheduleController) Delete(ctx context.Context, req *system.ProjectScheduleDeleteReq) (res *system.ProjectScheduleDeleteRes, err error) {
	err = service.ProjectSchedule().Delete(ctx, req.Ids)
	return
}

// 添加一个计划
func (s *projectScheduleController) AddSchedule(ctx context.Context, req *system.ProjectScheduleAddPlanReq) (res *system.ProjectScheduleAddPlanRes, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		if req.Types == 1 {
			// 如果添加计划时，发现父级的 type ==0 的数据不存在，则将其父项目的开始时间修改为，第一个子任务的开始时间
			// 查询父级的 type == 0 的数据
			parentData, err := dao.ProjectSchedule.Ctx(ctx).Where(dao.ProjectSchedule.Columns().ParentId, req.ConstructionId).Where(dao.ProjectSchedule.Columns().Types, 0).One()
			liberr.ErrIsNil(ctx, err, "查询失败")
			if parentData == nil {
				// 如果父级的 type == 0 的数据不存在，则添加一条 parent_id = req.ConstructionId, types = 0 的数据
				_, err = dao.ProjectSchedule.Ctx(ctx).Insert(do.ProjectSchedule{
					ParentId:  req.ConstructionId,
					StartDate: req.StartDate,
					EndDate:   req.EndDate,
					Name:      req.Name,
					PlaneNum:  req.PlaneNum,
					Types:     0,
				})
				liberr.ErrIsNil(ctx, err, "更新失败")
			}

			// 插入一条 type = 1 的数据
			_, err = dao.ProjectSchedule.Ctx(ctx).Insert(do.ProjectSchedule{
				ParentId:  req.ConstructionId,
				StartDate: req.StartDate,
				EndDate:   req.EndDate,
				Name:      req.Name,
				PlaneNum:  req.PlaneNum,
				Types:     1,
			})
			liberr.ErrIsNil(ctx, err, "更新失败")
			return
		}
		// 如果数据不存在，则添加
		count, err := dao.ProjectSchedule.Ctx(ctx).Where(dao.ProjectSchedule.Columns().ParentId, req.ConstructionId).Count()
		liberr.ErrIsNil(ctx, err, "查询失败")
		if count == 0 {
			_, err = dao.ProjectSchedule.Ctx(ctx).Insert(do.ProjectSchedule{
				ParentId:  req.ConstructionId,
				StartDate: req.StartDate,
				EndDate:   req.EndDate,
				Name:      req.Name,
				PlaneNum:  req.PlaneNum,
				Types:     req.Types,
			})
			liberr.ErrIsNil(ctx, err, "添加失败")
		} else {
			// 如果数据存在，则更新
			_, err = dao.ProjectSchedule.Ctx(ctx).Where(dao.ProjectSchedule.Columns().ParentId, req.ConstructionId).Update(do.ProjectSchedule{
				ParentId:  req.ConstructionId,
				StartDate: req.StartDate,
				EndDate:   req.EndDate,
				Name:      req.Name,
				PlaneNum:  req.PlaneNum,
				Types:     req.Types,
			})
			liberr.ErrIsNil(ctx, err, "更新失败")
		}
	})

	return
}