youngee_b_api/service/data.go

package service

import (
	"context"
	"errors"
	"fmt"
	"strings"
	"youngee_b_api/db"
	"youngee_b_api/model/gorm_model"
	"youngee_b_api/model/http_model"

	"github.com/sirupsen/logrus"
)

var Data *data

type data struct {
}

// DataOpinion 在上传脚本表上提交修改意见
func (*data) DataOpinion(ctx context.Context, request http_model.DataOpinionRequest) (*http_model.DataOpinionData, error) {
	Data := gorm_model.YounggeeDataInfo{
		TaskID:        request.TaskID,
		ReviseOpinion: request.DataOpinion,
	}
	err := db.DataOpinion(ctx, Data.TaskID, Data.ReviseOpinion, request.OperatorId, request.OperateType, request.ProjectId, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}
	// 记录任务日志
	err = db.CreateTaskLog(ctx, Data.TaskID, "数据驳回")
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Link service] call CreateTaskLog error,err:%+v", err)
		return nil, err
	}

	err = db.CreateMessageByTaskId(ctx, 19, 3, Data.TaskID)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateMessageByTaskId error,err:%+v", err)
		return nil, err
	}

	res := &http_model.DataOpinionData{
		TaskID: Data.TaskID,
	}
	return res, nil
}

// DataOpinion 在上传脚本表上提交修改意见
func (*data) LocalDataOpinion(ctx context.Context, request http_model.LocalDataOpinionRequest) (*http_model.LocalDataOpinionData, error) {
	Data := gorm_model.YounggeeDataInfo{
		TaskID:        request.TaskID,
		ReviseOpinion: request.LocalDataOpinion,
	}
	err := db.LocalDataOpinion(ctx, Data.TaskID, Data.ReviseOpinion, request.OperatorId, request.OperateType, request.ProjectId, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}
	// 记录任务日志
	err = db.CreateTaskLog(ctx, Data.TaskID, "数据驳回")
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Link service] call CreateTaskLog error,err:%+v", err)
		return nil, err
	}

	err = db.CreateMessageByTaskId(ctx, 19, 3, Data.TaskID)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateMessageByTaskId error,err:%+v", err)
		return nil, err
	}

	res := &http_model.LocalDataOpinionData{
		TaskID: Data.TaskID,
	}
	return res, nil
}

// 任务结案service
func (*data) AcceptData(ctx context.Context, request http_model.AcceptDataRequest) (*http_model.AcceptDataData, error) {
	var TaskIDList []string
	TaskIDs := strings.Split(request.TaskIds, ",")
	for _, taskId := range TaskIDs {
		TaskIDList = append(TaskIDList, taskId)
	}

	// 1. 更新YounggeeDataInfo表，通过数据
	//fmt.Printf("acc request %+v", TaskIDList)
	err := db.AcceptData(ctx, TaskIDList, request.OperatorId, request.OperateType, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call AcceptData error,err:%+v", err)
		return nil, err
	}

	// 记录任务日志和达人消息
	for _, taskId := range TaskIDList {
		err = db.CreateTaskLog(ctx, taskId, "数据通过")
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
		err = db.CreateMessageByTaskId(ctx, 5, 1, taskId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Script service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
	}

	res := &http_model.AcceptDataData{
		TaskIds: TaskIDList,
		//IsEnd:   0,
	}

	if request.IsSpecial == 1 {
		// 2. 更新YoungeeTaskInfo表，将任务结案
		err = db.SetSpecialTaskFinish(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTaskFinish error,err:%+v", err)
			return nil, err
		}

		err = db.SetSpecialTaskSettleNum(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSpecialTaskSettleNum error,err:%+v", err)
			return nil, err
		}
		if request.SettleAmount == nil {
			return nil, errors.New("SettleAmount is nil")
		}
		err = db.SetSettleAmount(ctx, request.SettleAmount, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSettleAmount error,err:%+v", err)
			return nil, err
		}
		err = db.SetProjectSettleFlag(ctx, request.ProjectId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetProjectSettleFlag error,err:%+v", err)
			return nil, err
		}

	} else {
		// 2. 更新YoungeeTaskInfo表，将任务结案
		//fmt.Printf("acc request %+v", TaskIDList)
		err = db.SetTaskFinish(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTaskFinish error,err:%+v", err)
			return nil, err
		}
		err = db.SetTalentIncome(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTalentIncome error,err:%+v", err)
			return nil, err
		}
		err = db.SetSpecialTaskSettleNum(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSpecialTaskSettleNum error,err:%+v", err)
			return nil, err
		}
		err = db.SetProjectSettleFlag(ctx, request.ProjectId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetProjectSettleFlag error,err:%+v", err)
			return nil, err
		}
	}
	return res, nil
}

// 任务结案service
func (*data) LocalAcceptData(ctx context.Context, request http_model.LocalAcceptDataRequest) (*http_model.LocalAcceptDataData, error) {
	var TaskIDList []string
	TaskIDs := strings.Split(request.TaskIds, ",")
	for _, taskId := range TaskIDs {
		TaskIDList = append(TaskIDList, taskId)
	}

	// 1. 更新YounggeeDataInfo表，通过数据
	fmt.Printf("acc request %+v", TaskIDList)
	err := db.AcceptData(ctx, TaskIDList, request.OperatorId, request.OperateType, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call AcceptData error,err:%+v", err)
		return nil, err
	}

	// 记录任务日志和达人消息
	for _, taskId := range TaskIDList {
		err = db.CreateTaskLog(ctx, taskId, "数据通过")
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
		err = db.CreateMessageByTaskId(ctx, 5, 1, taskId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Script service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
	}

	res := &http_model.LocalAcceptDataData{
		TaskIds: TaskIDList,
		//IsEnd:   0,
	}

	if request.IsSpecial == 1 {
		// 2. 更新YoungeeTaskInfo表，将任务结案
		err = db.SetSpecialLocalTaskFinish(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTaskFinish error,err:%+v", err)
			return nil, err
		}
		err = db.SetSpecialLocalTaskSettleNum(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSpecialTaskSettleNum error,err:%+v", err)
			return nil, err
		}
		err = db.SetLocalSettleAmount(ctx, request.SettleAmount, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSettleAmount error,err:%+v", err)
			return nil, err
		}
		err = db.SetLocalLifeSettleFlag(ctx, request.ProjectId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetLocalLifeSettleFlag error,err:%+v", err)
			return nil, err
		}

	} else {
		// 2. 更新YoungeeTaskInfo表，将任务结案
		fmt.Printf("acc request %+v", TaskIDList)
		err = db.SetLocalTaskFinish(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTaskFinish error,err:%+v", err)
			return nil, err
		}
		err = db.SetLocalTalentIncome(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetTalentIncome error,err:%+v", err)
			return nil, err
		}
		err = db.SetSpecialLocalTaskSettleNum(ctx, TaskIDList)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetSpecialTaskSettleNum error,err:%+v", err)
			return nil, err
		}
		err = db.SetLocalLifeSettleFlag(ctx, request.ProjectId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Data service] call SetLocalLifeSettleFlag error,err:%+v", err)
			return nil, err
		}
	}
	return res, nil
}

// RejectData
func (*data) RejectData(ctx context.Context, request http_model.RejectDataRequest) (*http_model.RejectDataData, error) {
	var TaskIDList []string
	TaskIDs := strings.Split(request.TaskIds, ",")
	for _, taskId := range TaskIDs {
		TaskIDList = append(TaskIDList, taskId)
	}
	//fmt.Printf("acc request %+v", TaskIDList)
	err := db.RejectData(ctx, TaskIDList, request.OperatorId, request.OperateType, request.ProjectId, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Link service] call CreateLink error,err:%+v", err)
		return nil, err
	}
	// 记录任务日志
	for _, taskId := range TaskIDList {
		err = db.CreateTaskLog(ctx, taskId, "数据驳回")
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Link service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
		err = db.CreateMessageByTaskId(ctx, 19, 3, taskId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Script service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
	}

	res := &http_model.RejectDataData{
		TaskIds: TaskIDList,
	}
	return res, nil
}

// RejectData
func (*data) LocalRejectData(ctx context.Context, request http_model.LocalRejectDataRequest) (*http_model.LocalRejectDataData, error) {
	var TaskIDList []string
	TaskIDs := strings.Split(request.TaskIds, ",")
	for _, taskId := range TaskIDs {
		TaskIDList = append(TaskIDList, taskId)
	}
	fmt.Printf("acc request %+v", TaskIDList)
	err := db.LocalRejectData(ctx, TaskIDList, request.OperatorId, request.OperateType, request.ProjectId, request.EnterpriseId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Link service] call CreateLink error,err:%+v", err)
		return nil, err
	}
	// 记录任务日志
	for _, taskId := range TaskIDList {
		err = db.CreateTaskLog(ctx, taskId, "数据驳回")
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Link service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
		err = db.CreateMessageByTaskId(ctx, 19, 3, taskId)
		if err != nil {
			logrus.WithContext(ctx).Errorf("[Script service] call CreateTaskLog error,err:%+v", err)
			return nil, err
		}
	}

	res := &http_model.LocalRejectDataData{
		TaskIds: TaskIDList,
	}
	return res, nil
}