opms_backend/opms_parent/app/service/plat/plat_task.go

package plat

import (
	"context"
	"dashoo.cn/micro/app/dao/plat"
	model "dashoo.cn/micro/app/model/plat"
	"dashoo.cn/micro/app/service"
	"database/sql"
	"fmt"
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gconv"
	"strconv"
)

type taskService struct {
	*service.ContextService

	Dao *plat.PlatTaskDao
}

func NewTaskService(ctx context.Context) (svc *taskService, err error) {
	svc = new(taskService)
	if svc.ContextService, err = svc.Init(ctx); err != nil {
		return nil, err
	}
	svc.Dao = plat.NewPlatTaskDao(svc.Tenant)
	return svc, nil
}

// 任务信息列表
func (s *taskService) GetList(req *model.SearchPlatTaskReq) (total int, TaskList []*model.PlatTask, err error) {
	TaskModel := s.Dao.M
	if req.TaskTitle != "" {
		TaskModel = TaskModel.Where("task_title LIKE ?", "%" + req.TaskTitle + "%")
	}
	if req.TaskType != "" {
		TaskModel = TaskModel.Where("task_type", req.TaskType)
	}
	if req.TaskStatus != "" {
		TaskModel = TaskModel.Where("task_status", req.TaskStatus)
	}
	if req.IsOverdue != "" {
		TaskModel = TaskModel.Where("is_overdue", req.IsOverdue)
	}
	if req.MainUserId != "" {
		TaskModel = TaskModel.Where("main_user_id", req.MainUserId)
	}
	if req.TargetId != "" {
		TaskModel = TaskModel.Where("target_id", req.TargetId)
	}
	if req.TargetType != "" {
		TaskModel = TaskModel.Where("target_type", req.TargetType)
	}
	total, err = TaskModel.Count()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取总行数失败")
		return
	}

	err = TaskModel.Page(req.PageNum, req.PageSize).Order("created_time DESC").Scan(&TaskList)
	return
}

// 添加信息
func (s *taskService) Create(req *model.AddPlatTaskReq) (err error) {
	platTask := new(model.PlatTask)
	if err = gconv.Struct(req, platTask); err != nil {
		return
	}
	platTask.IsOverdue = "10" // 是否超期（10否20是）
	if platTask.TaskStartDate == nil {
		platTask.TaskStartDate = gtime.Now()
	}
	// 填充创建信息
	service.SetCreatedInfo(platTask, s.GetCxtUserId(), s.GetCxtUserName())
	// 填充更新信息
	//service.SetUpdatedInfo(platTask, s.GetCxtUserId(), s.GetCxtUserName())
	Model := s.Dao.M
	res, err := Model.Insert(platTask)
	if err != nil {
		return
	}
	// 流程日志
	id, _ := res.LastInsertId()
	err = CreateTaskLog(s, nil, int(id), s.GetCxtUserId(), s.GetCxtUserName(), "创建任务", "创建任务成功", "")

	return
}

// 修改状态
func (s *taskService) ChangeStatus(req *model.ChangeStatusReq) (err error) {
	data := ""
	where := fmt.Sprintf("id='%v'", req.TaskId)
	nodeName := "状态修改"
	desc := "修改成功"
	// 类型合规判断
	if req.Type == "TaskStatus" {
		data = fmt.Sprintf("task_status='%v'", req.NowStatus)
		nodeName = "任务状态修改"
		if req.NowStatus == "20" {
			desc = "任务关闭"
		} else if req.NowStatus == "30" {
			desc = "任务完成"
			data += fmt.Sprintf(",task_end_date='%v'", gtime.Now().Format("Y-m-d H:i:s"))
		}
	} else if req.Type == "IsOverdue" {
		nodeName = "超期状态修改"
		if req.NowStatus == "20" {
			desc = "任务超期"
		}
		data = fmt.Sprintf("is_overdue='%v'", req.NowStatus)
	} else {
		return fmt.Errorf("类型不匹配，请仔细检查")
	}
	// 读取现有数据
	task, err := s.Dao.FindOne(where)
	if err != nil {
		return err
	}
	// 状态数据一致性判断
	if (req.Type == "TaskStatus" && task.TaskStatus != req.OldStatus) || (req.Type == "IsOverdue" && task.IsOverdue != req.OldStatus) {
		return fmt.Errorf("状态不匹配，进行该操作")
	}
	// 更新数据
	_, err = s.Dao.Update(data, where)
	if err != nil {
		return
	}

	// 流程日志
	taskId, _ := strconv.Atoi(req.TaskId)
	err = CreateTaskLog(s, nil, taskId, s.GetCxtUserId(), s.GetCxtUserName(), nodeName, desc, "")

	return
}

// 任务日志创建方法
func CreateTaskLog(s1 *taskService, s2 *taskProgressService, taskId, userId int, userName, nodeName, desc, remark string) (err error) {
	var log model.PlatTaskLog
	startTime := gtime.Now()
	endTime := startTime
	// 默认为当前时间，然后随已有日志情况进行更新
	if s1 != nil {
		logs, err := s1.Dao.DB.Model(plat.PlatTaskLog.Table).Where(fmt.Sprintf("task_id='%v'", taskId)).Order("created_time DESC").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return err
		}
		if len(logs) > 0 {
			startTime = logs[0]["end_time"].GTime()
		}
	} else {
		logs, err := s2.Dao.DB.Model(plat.PlatTaskLog.Table).Where(fmt.Sprintf("task_id='%v'", taskId)).Order("created_time DESC").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return err
		}
		if len(logs) > 0 {
			startTime = logs[0]["end_time"].GTime()
		}
	}

	// 填入日志相关数据
	log.TaskId = taskId
	log.NodeName = nodeName
	log.Desc = desc
	log.StartTime = startTime
	log.EndTime = endTime
	log.Remark = remark
	log.CreatedTime = endTime
	log.CreatedName = userName
	log.CreatedBy = userId

	if s1 != nil {
		_, err = s1.Dao.DB.Save(plat.PlatTaskLog.Table, log)
	} else {
		_, err = s2.Dao.DB.Save(plat.PlatTaskLog.Table, log)
	}

	return err
}