youngee_mc_api/app/service/youngee_task_service/task_data.go

package youngee_task_service

import (
	"youngmini_server/app/dao"
	"youngmini_server/app/model"
	"youngmini_server/app/model/youngee_talent_model"

	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/net/ghttp"
	"github.com/gogf/gf/os/gtime"
)

// 添加数据service
func AddTaskData(r *ghttp.Request) *TalentHttpResult {
	var DataInfoReq *youngee_talent_model.AddTaskDataRequest
	err := r.ParseForm(&DataInfoReq)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}
	// 查询该任务id是否有未提交数据
	// 查询该任务是否有未提交脚本，无则插入，有则更新
	unSubmitData, err := g.DB().Model(model.YounggeeDataInfo{}).Where("task_id = ? and is_submit = 0", DataInfoReq.TaskId).One()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "YounggeeDataInfo find failed"}
	}
	if unSubmitData != nil {
		_, err = g.DB().Model(model.YounggeeDataInfo{}).Where("Data_id = ?", unSubmitData[dao.YounggeeDataInfo.Columns.DataId]).Delete()
		if err != nil {
			return &TalentHttpResult{Code: -3, Msg: "YounggeeDataInfo delete failed"}
		}
	}

	// 如果无未提交数据，则直接插入
	DataInfo := model.YounggeeDataInfo{
		TaskId:        DataInfoReq.TaskId,
		LikeNumber:    DataInfoReq.LikeNumber,
		PlayNumber:    DataInfoReq.PlayNumber,
		CommentNumber: DataInfoReq.CommentNumber,
		CollectNumber: DataInfoReq.CollectNumber,
		PhotoUrl:      DataInfoReq.PhotoUrl,
		CreateAt:      gtime.Now(),
		IsReview:      0,
		IsSubmit:      0,
		IsOk:          0,
	}

	_, err = g.DB().Model(model.YounggeeDataInfo{}).Data(DataInfo).Insert()
	if err != nil {
		return &TalentHttpResult{Code: -3, Msg: "YounggeeDataInfo insert failed"}
	}

	DataStatus, err := g.DB().Model(model.YoungeeTaskInfo{}).Fields("data_status").Where("task_id = ?", DataInfoReq.TaskId).Value()
	if err != nil {
		return &TalentHttpResult{Code: -5, Msg: "Get task info failed"}
	}
	if DataStatus.Int64() == 1 {
		_, err = g.Model(dao.YoungeeTaskInfo.Table).Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 2})
		if err != nil {
			return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
		}
	} else if DataStatus.Int64() == 3 {
		_, err = g.Model(dao.YoungeeTaskInfo.Table).Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 4})
		if err != nil {
			return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
		}
	}

	return &TalentHttpResult{Code: 0, Msg: "success"}
}

// 提交数据service
func SubmitTaskData(r *ghttp.Request) *TalentHttpResult {
	taskId, _ := r.Get("task_id").(string)
	taskInfo := model.YoungeeTaskInfo{}
	// 查询是否处于违约状态
	err1 := g.DB().Model(model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Scan(&taskInfo)
	if err1 != nil {
		return &TalentHttpResult{Code: -1, Msg: "YoungeeTaskInfo find failed"}
	}
	if taskInfo.CurDefaultType == 7 || taskInfo.CurDefaultType == 9 {
		// 若处于违约状态则解除并更新企业应支付金额
		_, err1 = g.DB().Model(model.YoungeeTaskInfo{}).Data(g.Map{"cur_default_type": "0", "real_payment": taskInfo.AllPayment, "err_break_rate": 0}).Where("task_id = ?", taskId).Update()
		if err1 != nil {
			return &TalentHttpResult{Code: -2, Msg: "YoungeeTaskInfo update failed"}
		}

		// 更新违约记录表
		_, err1 = g.DB().Model(model.YoungeeContractInfo{}).Data(g.Map{"default_status": 2}).Where("task_id = ? and default_status in (?)", taskId, g.Slice{1, 3, 4}).Update()
		if err1 != nil {
			return &TalentHttpResult{Code: -2, Msg: "YoungeeContractInfo update failed"}
		}
	}

	// 查询该任务是否有已添加或已修改数据
	res, err := g.DB().Model(model.YounggeeDataInfo{}).Where("task_id = ? and is_submit = 0", taskId).Count()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "YounggeeDataInfo find failed"}
	}
	if res == 1 && (taskInfo.DataStatus == 2 || taskInfo.DataStatus == 4) {
		_, err = g.DB().Model(model.YounggeeDataInfo{}).Data(g.Map{"is_submit": "1", "submit_at": gtime.Now()}).Where("task_id = ? and is_submit = 0", taskId).Update()
		if err != nil {
			return &TalentHttpResult{Code: -3, Msg: "YounggeeDataInfo update failed"}
		}
		_, err = g.DB().Model(model.YoungeeTaskInfo{}).Data(g.Map{"task_stage": "14"}).Where("task_id = ?", taskId).Update()
		if err != nil {
			return &TalentHttpResult{Code: -4, Msg: "YoungeeTaskInfo update failed"}
		}
		// 记录任务日志-上传数据
		taskLog := model.YounggeeTaskLog{
			TaskId:  taskId,
			Content: "上传数据",
			LogAt:   gtime.Now(),
		}
		_, err = g.DB().Model(dao.YounggeeTaskLog.Table).Data(&taskLog).Insert()
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "YounggeeTaskLog insert failed"}
		}
		projectInfo := model.ProjectInfo{}
		err1 := g.DB().Model(model.ProjectInfo{}).Where("project_id = ?", taskInfo.ProjectId).Scan(&projectInfo)
		if err1 != nil {
			return &TalentHttpResult{Code: -8, Msg: "ProjectInfo find failed"}
		}
		messageInfo := model.YounggeeMessageInfo{
			MessageId:   13,
			MessageType: 2,
			CreatedAt:   gtime.Now(),
			TalentId:    taskInfo.TalentId,
			ProjectName: projectInfo.ProjectName,
			IsReaded:    0,
			IsDeleted:   0,
		}
		_, err = g.DB().Model(dao.YounggeeMessageInfo.Table).Data(&messageInfo).Insert()
		if err != nil {
			return &TalentHttpResult{Code: -9, Msg: "YounggeeMessageInfo insert failed"}
		}
	}

	return &TalentHttpResult{Code: 0, Msg: "success"}
}

// 查询数据提交审阅记录service
func GetTaskData(r *ghttp.Request) *TalentHttpResult {
	taskId := r.Get("task_id")

	res, err := g.DB().Model(dao.YounggeeDataInfo.Table).Where("is_submit = 1 and is_review = 1 and task_id = ?", taskId).OrderAsc("submit_at").All()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}

	return &TalentHttpResult{Code: 0, Msg: "success", Data: res}
}

// 查询未提交数据service
func GetUnSubmitTaskData(r *ghttp.Request) *TalentHttpResult {
	taskId := r.Get("task_id")

	res, err := g.DB().Model(dao.YounggeeDataInfo.Table).Where("is_review = 0 and task_id = ?", taskId).One()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}

	return &TalentHttpResult{Code: 0, Msg: "success", Data: res}
}