younggee_m_api/db/default.go

package db

import (
	"context"
	"fmt"
	"github.com/caixw/lib.go/conv"
	"github.com/sirupsen/logrus"
	"reflect"
	"youngee_m_api/consts"
	"youngee_m_api/model/common_model"
	"youngee_m_api/model/gorm_model"
	"youngee_m_api/model/http_model"
	"youngee_m_api/util"
)

func CountDefaultNum(ctx context.Context) (error, *http_model.CountNumOfDefaultsResponse) {
	db := GetReadDB(ctx)
	var contractInfos []gorm_model.YoungeeContractInfo
	err := db.Debug().Model(gorm_model.YoungeeContractInfo{}).Where("default_status = 3").Find(&contractInfos).Error
	if err != nil {
		return err, nil
	}
	DraftDefaultNum, ScriptDefaultNum, LinkDefaultNum, DataDefaultNum := 0, 0, 0, 0
	for _, contractInfo := range contractInfos {
		if contractInfo.BreakType == 1 {
			ScriptDefaultNum++
		} else if contractInfo.BreakType == 2 {
			DraftDefaultNum++
		} else if contractInfo.BreakType == 3 {
			LinkDefaultNum++
		} else {
			DataDefaultNum++
		}
	}
	data := &http_model.CountNumOfDefaultsResponse{}
	data.ScriptDefaultNum = int32(ScriptDefaultNum)
	data.DraftDefaultNum = int32(DraftDefaultNum)
	data.LinkDefaultNum = int32(LinkDefaultNum)
	data.DataDefaultNum = int32(DataDefaultNum)
	return nil, data
}

func BreachPending(ctx context.Context, pageSize, pageNum int32, req *http_model.BreachPendingRequest) (*http_model.BreachPendingData, error) {
	db := GetReadDB(ctx)
	var contractInfos []*gorm_model.YoungeeContractInfo
	db = db.Model(gorm_model.YoungeeContractInfo{}).Where("default_status = 3")
	if req.DefaultType == 1 || req.DefaultType == 2 {
		db = db.Where("break_type = 1 OR break_type = 2")
	} else {
		db = db.Where("break_type = ?", req.DefaultType)
	}
	if req.TaskId != 0 {
		db = db.Where("task_id = ?", req.TaskId)
	}
	var findProjectIds []int64
	if req.ProjectName != "" {
		db1 := GetReadDB(ctx)
		db1.Model(gorm_model.ProjectInfo{}).Select("project_id").Where("project_name = ?", req.ProjectName).Find(&findProjectIds)
		var findTaskIds []int
		db2 := GetReadDB(ctx)
		db2.Model(gorm_model.YoungeeTaskInfo{}).Select("task_id").Where("project_id IN ?", findProjectIds).Find(&findTaskIds)
		db = db.Where("task_id IN ?", findTaskIds)
	}
	// 查询总数
	var total int64
	if err := db.Count(&total).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[BreachPending] error query mysql total, err:%+v", err)
		return nil, err
	}
	// 查询该页数据
	limit := pageSize
	offset := pageSize * pageNum // assert pageNum start with 0
	err := db.Order("terminate_at").Limit(int(limit)).Offset(int(offset)).Find(&contractInfos).Error
	if err != nil {
		logrus.WithContext(ctx).Errorf("[BreachPending] error query mysql total, err:%+v", err)
		return nil, err
	}
	var taskIds []int
	for _, contractInfo := range contractInfos {
		taskIds = append(taskIds, contractInfo.TaskID)
	}
	taskIds = util.RemoveIntRepByMap(taskIds)
	taskIdToProjectMap := make(map[int]int)
	taskIdToTalentIdMap := make(map[int]string)
	taskIdToTaskInfoMap := make(map[int]gorm_model.YoungeeTaskInfo)
	var projectIds []int
	for _, taskId := range taskIds {
		db1 := GetReadDB(ctx)
		var taskInfo gorm_model.YoungeeTaskInfo
		db1.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Find(&taskInfo)
		taskIdToProjectMap[taskId] = taskInfo.ProjectId
		taskIdToTalentIdMap[taskId] = taskInfo.TalentId
		taskIdToTaskInfoMap[taskId] = taskInfo
		projectIds = append(projectIds, taskInfo.ProjectId)
	}
	//fmt.Println("TaskID:", taskIds)
	//fmt.Println("projectIds:", projectIds)
	//fmt.Println("taskIdToProjectMap:", taskIdToProjectMap)
	//fmt.Println("taskIdToTalentIdMap:", taskIdToTalentIdMap)
	projectIds = util.RemoveIntRepByMap(projectIds)
	var enterpriseIds []int64
	projectIdToProjectInfoMap := make(map[int64]gorm_model.ProjectInfo)
	for _, projectId := range projectIds {
		db1 := GetReadDB(ctx)
		projectInfo := gorm_model.ProjectInfo{}
		db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectInfo)
		projectIdToProjectInfoMap[projectInfo.ProjectID] = projectInfo
		enterpriseIds = append(enterpriseIds, projectInfo.EnterpriseID)
	}
	enterpriseIds = util.RemoveRepByMap(enterpriseIds)
	//fmt.Println("enterpriseIds:", enterpriseIds)
	enterpriseIdToUserId := make(map[int64]int64)
	var userIds []int64
	for _, enterpriseId := range enterpriseIds {
		db1 := GetReadDB(ctx)
		var userId int64
		db1.Model(gorm_model.Enterprise{}).Select("user_id").Where("enterprise_id = ?", enterpriseId).Find(&userId)
		enterpriseIdToUserId[enterpriseId] = userId
		userIds = append(userIds, userId)
	}
	//fmt.Println("projectIdToProjectInfoMap:", projectIdToProjectInfoMap)
	//fmt.Println("enterpriseIdToUserId:", enterpriseIdToUserId)
	//fmt.Println("userIds:", userIds)
	userIdToUserPhone := make(map[int64]string)
	for _, userId := range userIds {
		db1 := GetReadDB(ctx)
		var userPhone string
		db1.Model(gorm_model.YounggeeUser{}).Select("phone").Where("id = ?", userId).Find(&userPhone)
		userIdToUserPhone[userId] = userPhone
	}
	//fmt.Println("userIdToUserPhone:", userIdToUserPhone)
	talentIdToTalentPhoneMap := make(map[string]string)
	for _, v := range taskIdToTalentIdMap {
		if len(talentIdToTalentPhoneMap) == 0 {
			db1 := GetReadDB(ctx)
			var talentPhoneNumber string
			db1.Model(gorm_model.YoungeeTalentInfo{}).Select("talent_phone_number").Where("id = ?", v).Find(&talentPhoneNumber)
			talentIdToTalentPhoneMap[v] = talentPhoneNumber
		}
		if _, ok := talentIdToTalentPhoneMap[v]; !ok {
			db1 := GetReadDB(ctx)
			var talentPhoneNumber string
			db1.Model(gorm_model.YoungeeTalentInfo{}).Select("talent_phone_number").Where("id = ?", v).Find(&talentPhoneNumber)
			talentIdToTalentPhoneMap[v] = talentPhoneNumber
		}
	}
	taskIdToDefaultInfo := make(map[int]string)
	if req.DefaultType == 4 {
		for _, taskId := range taskIds {
			db1 := GetReadDB(ctx)
			var link string
			db1.Debug().Model(gorm_model.YounggeeLinkInfo{}).Select("link_url").Where("task_id = ? AND is_ok = 1", taskId).Find(&link)
			taskIdToDefaultInfo[taskId] = link
		}
	}
	//fmt.Println("talentIdToTalentPhoneMap:", talentIdToTalentPhoneMap)
	var BreachPendingPreviews []*http_model.BreachPendingPreview
	for _, contractInfo := range contractInfos {
		BreachPendingPreview := new(http_model.BreachPendingPreview)
		BreachPendingPreview.ContractId = int32(contractInfo.ContractID)
		BreachPendingPreview.ProjectId = int32(taskIdToProjectMap[contractInfo.TaskID])
		BreachPendingPreview.UserId = int32(enterpriseIdToUserId[projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].EnterpriseID])
		BreachPendingPreview.ProjectName = projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].ProjectName
		BreachPendingPreview.UserPhone = userIdToUserPhone[enterpriseIdToUserId[projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].EnterpriseID]]
		BreachPendingPreview.TaskId = int32(contractInfo.TaskID)
		BreachPendingPreview.TalentId = taskIdToTalentIdMap[contractInfo.TaskID]
		BreachPendingPreview.TalentPhone = talentIdToTalentPhoneMap[taskIdToTalentIdMap[contractInfo.TaskID]]
		BreachPendingPreview.LinkInfo = taskIdToDefaultInfo[contractInfo.TaskID]
		BreachPendingPreview.Price = taskIdToTaskInfoMap[contractInfo.TaskID].AllPayment
		BreachPendingPreview.SettlementAmount = taskIdToTaskInfoMap[contractInfo.TaskID].SettleAmount
		BreachPendingPreview.DefaultAt = conv.MustString(contractInfo.BreakAt, "")[0:19]
		BreachPendingPreview.TerminateAt = conv.MustString(contractInfo.TerminateAt, "")[0:19]
		BreachPendingPreviews = append(BreachPendingPreviews, BreachPendingPreview)
	}
	var BreachPendingData http_model.BreachPendingData
	BreachPendingData.BreachPendingPreview = BreachPendingPreviews
	BreachPendingData.Total = total
	return &BreachPendingData, nil
}

func ContractBreach(ctx context.Context, req *http_model.ContractBreachRequest) error {
	db := GetReadDB(ctx)
	var breakType int
	db.Model(gorm_model.YoungeeContractInfo{}).Select("break_type").Where("contract_id IN ?", req.ContractIds).Find(&breakType)
	err := db.Debug().Where("contract_id IN ?", req.ContractIds).Updates(&gorm_model.YoungeeContractInfo{DefaultStatus: int(req.DefaultStatus)}).Error
	if err != nil {
		logrus.WithContext(ctx).Errorf("[BreachPending] error query mysql total, err:%+v", err)
		return err
	}
	var taskId int
	db.Model(gorm_model.YoungeeContractInfo{}).Select("task_id").Where("contract_id IN ?", req.ContractIds).Find(&taskId)
	if req.DefaultStatus == 5 {
		db1 := GetReadDB(ctx)
		var projectId, autoDefaultId, feeForm int
		db1.Model(gorm_model.YoungeeTaskInfo{}).Select("project_id").Where("task_id = ?", taskId).Find(&projectId)
		db2 := GetReadDB(ctx)
		db2.Model(gorm_model.ProjectInfo{}).Select("auto_default_id").Where("project_id = ?", projectId).Find(&autoDefaultId)
		var rateInfo gorm_model.InfoAutoDefaultHandle
		db3 := GetReadDB(ctx)
		db3.Model(gorm_model.InfoAutoDefaultHandle{}).Where("auto_default_id = ?", autoDefaultId).Find(&rateInfo)
		db4 := GetReadDB(ctx)
		db4.Model(gorm_model.YoungeeTaskInfo{}).Select("fee_form").Where("task_id = ?", taskId).Find(&feeForm)
		if feeForm == 1 {
			if breakType == 1 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.ScriptReplaceNotUpload, CurDefaultType: 2})
			} else if breakType == 2 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.SketchReplaceNotUpload, CurDefaultType: 4})
			} else if breakType == 3 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.LinkReplaceNotUpload, CurDefaultType: 6})
			} else if breakType == 4 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.DataReplaceNotUpload, CurDefaultType: 8})
			}
		} else {
			if breakType == 1 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.ScriptOtherNotUpload, CurDefaultType: 2})
			} else if breakType == 2 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.SketchOtherNotUpload, CurDefaultType: 4})
			} else if breakType == 3 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.LinkOtherNotUpload, CurDefaultType: 6})
			} else if breakType == 4 {
				db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Updates(gorm_model.YoungeeTaskInfo{
					ErrBreakRate: rateInfo.DataOtherNotUpload, CurDefaultType: 8})
			}
		}
	}
	return nil
}

func GetSketchInfoByTaskId(ctx context.Context, request *http_model.GetSketchInfoByTaskIdRequest) (*http_model.SketchInfoResponse, error) {
	db := GetReadDB(ctx)
	var sketchInfo gorm_model.YounggeeSketchInfo
	db.Debug().Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_ok = 1", request.TaskId).Find(&sketchInfo)
	db2 := GetReadDB(ctx)
	sketchPhotoInfo := gorm_model.YounggeeSketchPhoto{}
	db2.Debug().Model(gorm_model.YounggeeSketchPhoto{}).Where("sketch_id = ?", sketchInfo.SketchID).Find(&sketchPhotoInfo)
	data := new(http_model.SketchInfoResponse)
	data.Type = int32(sketchPhotoInfo.Symbol)
	data.PhotoUrl = sketchPhotoInfo.PhotoUrl
	data.Content = sketchInfo.Content
	return data, nil
}

func BreachHandled(ctx context.Context, pageSize, pageNum int32, req *http_model.BreachHandledRequest, conditions *common_model.BreachHandledConditions) (*http_model.BreachHandledData, error) {
	db := GetReadDB(ctx)
	var contractInfos []*gorm_model.YoungeeContractInfo
	db = db.Model(gorm_model.YoungeeContractInfo{}).Where("default_status = 4 OR default_status = 5")
	// 根据Project条件过滤
	conditionType := reflect.TypeOf(conditions).Elem()
	conditionValue := reflect.ValueOf(conditions).Elem()
	for i := 0; i < conditionType.NumField(); i++ {
		field := conditionType.Field(i)
		tag := field.Tag.Get("condition")
		value := conditionValue.FieldByName(field.Name)
		if !util.IsBlank(value) {
			db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
		}
	}
	if req.TaskId != 0 {
		db = db.Where("task_id = ?", req.TaskId)
	}
	var findProjectIds []int64
	if req.ProjectName != "" {
		db1 := GetReadDB(ctx)
		db1.Model(gorm_model.ProjectInfo{}).Select("project_id").Where("project_name = ?", req.ProjectName).Find(&findProjectIds)
		var findTaskIds []int
		db2 := GetReadDB(ctx)
		db2.Model(gorm_model.YoungeeTaskInfo{}).Select("task_id").Where("project_id IN ?", findProjectIds).Find(&findTaskIds)
		db = db.Where("task_id IN ?", findTaskIds)
	}
	// 查询总数
	var total int64
	if err := db.Count(&total).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[BreachHandled] error query mysql total, err:%+v", err)
		return nil, err
	}
	// 查询该页数据
	limit := pageSize
	offset := pageSize * pageNum // assert pageNum start with 0
	err := db.Order("terminate_at").Limit(int(limit)).Offset(int(offset)).Find(&contractInfos).Error
	if err != nil {
		logrus.WithContext(ctx).Errorf("[BreachHandled] error query mysql total, err:%+v", err)
		return nil, err
	}

	var taskIds []int
	for _, contractInfo := range contractInfos {
		taskIds = append(taskIds, contractInfo.TaskID)
	}
	taskIds = util.RemoveIntRepByMap(taskIds)
	taskIdToProjectMap := make(map[int]int)
	taskIdToTalentIdMap := make(map[int]string)
	taskIdToTaskInfoMap := make(map[int]gorm_model.YoungeeTaskInfo)
	var projectIds []int
	for _, taskId := range taskIds {
		db1 := GetReadDB(ctx)
		var taskInfo gorm_model.YoungeeTaskInfo
		db1.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Find(&taskInfo)
		taskIdToProjectMap[taskId] = taskInfo.ProjectId
		taskIdToTalentIdMap[taskId] = taskInfo.TalentId
		taskIdToTaskInfoMap[taskId] = taskInfo
		projectIds = append(projectIds, taskInfo.ProjectId)
	}
	//fmt.Println("TaskID:", taskIds)
	//fmt.Println("projectIds:", projectIds)
	//fmt.Println("taskIdToProjectMap:", taskIdToProjectMap)
	//fmt.Println("taskIdToTalentIdMap:", taskIdToTalentIdMap)
	projectIds = util.RemoveIntRepByMap(projectIds)
	var enterpriseIds []int64
	projectIdToProjectInfoMap := make(map[int64]gorm_model.ProjectInfo)
	for _, projectId := range projectIds {
		db1 := GetReadDB(ctx)
		projectInfo := gorm_model.ProjectInfo{}
		db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectInfo)
		projectIdToProjectInfoMap[projectInfo.ProjectID] = projectInfo
		enterpriseIds = append(enterpriseIds, projectInfo.EnterpriseID)
	}
	enterpriseIds = util.RemoveRepByMap(enterpriseIds)
	//fmt.Println("enterpriseIds:", enterpriseIds)
	enterpriseIdToUserId := make(map[int64]int64)
	var userIds []int64
	for _, enterpriseId := range enterpriseIds {
		db1 := GetReadDB(ctx)
		var userId int64
		db1.Model(gorm_model.Enterprise{}).Select("user_id").Where("enterprise_id = ?", enterpriseId).Find(&userId)
		enterpriseIdToUserId[enterpriseId] = userId
		userIds = append(userIds, userId)
	}
	//fmt.Println("projectIdToProjectInfoMap:", projectIdToProjectInfoMap)
	//fmt.Println("enterpriseIdToUserId:", enterpriseIdToUserId)
	//fmt.Println("userIds:", userIds)
	userIdToUserPhone := make(map[int64]string)
	for _, userId := range userIds {
		db1 := GetReadDB(ctx)
		var userPhone string
		db1.Model(gorm_model.YounggeeUser{}).Select("phone").Where("id = ?", userId).Find(&userPhone)
		userIdToUserPhone[userId] = userPhone
	}
	//fmt.Println("userIdToUserPhone:", userIdToUserPhone)
	talentIdToTalentPhoneMap := make(map[string]string)
	for _, v := range taskIdToTalentIdMap {
		if len(talentIdToTalentPhoneMap) == 0 {
			db1 := GetReadDB(ctx)
			var talentPhoneNumber string
			db1.Model(gorm_model.YoungeeTalentInfo{}).Select("talent_phone_number").Where("id = ?", v).Find(&talentPhoneNumber)
			talentIdToTalentPhoneMap[v] = talentPhoneNumber
		}
		if _, ok := talentIdToTalentPhoneMap[v]; !ok {
			db1 := GetReadDB(ctx)
			var talentPhoneNumber string
			db1.Model(gorm_model.YoungeeTalentInfo{}).Select("talent_phone_number").Where("id = ?", v).Find(&talentPhoneNumber)
			talentIdToTalentPhoneMap[v] = talentPhoneNumber
		}
	}
	//fmt.Println("talentIdToTalentPhoneMap:", talentIdToTalentPhoneMap)
	var BreachHandledPreviews []*http_model.BreachHandledPreview
	for _, contractInfo := range contractInfos {
		BreachHandledPreview := new(http_model.BreachHandledPreview)
		BreachHandledPreview.ContractId = int32(contractInfo.ContractID)
		BreachHandledPreview.ProjectId = int32(taskIdToProjectMap[contractInfo.TaskID])
		BreachHandledPreview.UserId = int32(enterpriseIdToUserId[projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].EnterpriseID])
		BreachHandledPreview.ProjectName = projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].ProjectName
		BreachHandledPreview.UserPhone = userIdToUserPhone[enterpriseIdToUserId[projectIdToProjectInfoMap[int64(taskIdToProjectMap[contractInfo.TaskID])].EnterpriseID]]
		BreachHandledPreview.TaskId = int32(contractInfo.TaskID)
		BreachHandledPreview.TalentId = taskIdToTalentIdMap[contractInfo.TaskID]
		BreachHandledPreview.TalentPhone = talentIdToTalentPhoneMap[taskIdToTalentIdMap[contractInfo.TaskID]]
		BreachHandledPreview.TerminateReason = consts.GetBreakType(contractInfo.BreakType)
		BreachHandledPreview.HandleResult = consts.GetHandleResult(contractInfo.DefaultStatus)
		BreachHandledPreview.HandleAt = conv.MustString(contractInfo.HandleAt, "")[0:19]
		BreachHandledPreviews = append(BreachHandledPreviews, BreachHandledPreview)
	}
	var BreachHandledData http_model.BreachHandledData
	BreachHandledData.BreachHandledPreview = BreachHandledPreviews
	BreachHandledData.Total = total
	return &BreachHandledData, nil
}