123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796 |
- package db
- import (
- "context"
- "fmt"
- "reflect"
- "strings"
- "time"
- "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/pack"
- "youngee_m_api/util"
- "github.com/caixw/lib.go/conv"
- "github.com/sirupsen/logrus"
- )
- 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 != "" {
- 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 []string
- for _, contractInfo := range contractInfos {
- taskIds = append(taskIds, contractInfo.TaskID)
- }
- taskIds = util.RemoveStrRepByMap(taskIds)
- taskIdToProjectMap := make(map[string]string)
- taskIdToTalentIdMap := make(map[string]string)
- taskIdToTaskInfoMap := make(map[string]gorm_model.YoungeeTaskInfo)
- var projectIds []string
- 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)
- }
- projectIds = util.RemoveStrRepByMap(projectIds)
- var enterpriseIds []string
- projectIdToProjectInfoMap := make(map[string]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.RemoveStrRepByMap(enterpriseIds)
- enterpriseIdToUserId := make(map[string]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)
- }
- 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
- }
- 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[string]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
- }
- }
- var BreachPendingPreviews []*http_model.BreachPendingPreview
- for _, contractInfo := range contractInfos {
- BreachPendingPreview := new(http_model.BreachPendingPreview)
- BreachPendingPreview.ContractId = int32(contractInfo.ContractID)
- BreachPendingPreview.ProjectId = taskIdToProjectMap[contractInfo.TaskID]
- BreachPendingPreview.UserId = int32(enterpriseIdToUserId[projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].EnterpriseID])
- BreachPendingPreview.ProjectName = projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].ProjectName
- BreachPendingPreview.UserPhone = userIdToUserPhone[enterpriseIdToUserId[projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].EnterpriseID]]
- BreachPendingPreview.TaskId = 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)
- t := time.Now()
- err := db.Debug().Where("contract_id IN ?", req.ContractIds).Updates(&gorm_model.YoungeeContractInfo{DefaultStatus: int(req.DefaultStatus), HandleAt: &t}).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[BreachPending] error query mysql total, err:%+v", err)
- return err
- }
- var taskId string
- 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 string
- var 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
- data.Title = sketchInfo.Title
- 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 []string
- if req.ProjectName != "" {
- db1 := GetReadDB(ctx)
- db1.Model(gorm_model.ProjectInfo{}).Select("project_id").Where("project_name = ?", req.ProjectName).Find(&findProjectIds)
- var findTaskIds []string
- 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 []string
- for _, contractInfo := range contractInfos {
- taskIds = append(taskIds, contractInfo.TaskID)
- }
- taskIds = util.RemoveStrRepByMap(taskIds)
- taskIdToProjectMap := make(map[string]string)
- taskIdToTalentIdMap := make(map[string]string)
- taskIdToTaskInfoMap := make(map[string]gorm_model.YoungeeTaskInfo)
- var projectIds []string
- 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)
- }
- projectIds = util.RemoveStrRepByMap(projectIds)
- var enterpriseIds []string
- projectIdToProjectInfoMap := make(map[string]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.RemoveStrRepByMap(enterpriseIds)
- enterpriseIdToUserId := make(map[string]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)
- }
- 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
- }
- 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
- }
- }
- var BreachHandledPreviews []*http_model.BreachHandledPreview
- for _, contractInfo := range contractInfos {
- BreachHandledPreview := new(http_model.BreachHandledPreview)
- BreachHandledPreview.ContractId = int32(contractInfo.ContractID)
- BreachHandledPreview.ProjectId = taskIdToProjectMap[contractInfo.TaskID]
- BreachHandledPreview.UserId = int32(enterpriseIdToUserId[projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].EnterpriseID])
- BreachHandledPreview.ProjectName = projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].ProjectName
- BreachHandledPreview.UserPhone = userIdToUserPhone[enterpriseIdToUserId[projectIdToProjectInfoMap[taskIdToProjectMap[contractInfo.TaskID]].EnterpriseID]]
- BreachHandledPreview.TaskId = 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
- }
- func GetTaskDefaultReviewList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.TaskDefaultReviewInfo, int64, error) {
- db := GetReadDB(ctx)
- // 查询Task表信息
- db = db.Debug().Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2")
- // 根据Project条件过滤
- conditionType := reflect.TypeOf(conditions).Elem()
- conditionValue := reflect.ValueOf(conditions).Elem()
- var platform_nickname string = ""
- for i := 0; i < conditionType.NumField(); i++ {
- field := conditionType.Field(i)
- tag := field.Tag.Get("condition")
- value := conditionValue.FieldByName(field.Name)
- if tag == "default_status" {
- fmt.Printf("default %+v", value.Interface() == int64(0))
- if value.Interface() == int64(0) {
- db = db.Where("cur_default_type = 1")
- } else if value.Interface() == int64(1) {
- db = db.Where("cur_default_type = 3")
- } else if value.Interface() == int64(2) {
- db = db.Where("cur_default_type = 5")
- }
- continue
- } else if !util.IsBlank(value) {
- if tag == "platform_nickname" {
- platform_nickname = fmt.Sprintf("%v", value.Interface())
- continue
- } else if tag == "project_id" || tag == "strategy_id" {
- db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
- }
- }
- }
- var taskInfos []gorm_model.YoungeeTaskInfo
- db = db.Model(gorm_model.YoungeeTaskInfo{})
- // 查询总数
- var totalTask int64
- if err := db.Count(&totalTask).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- db.Order("task_id").Find(&taskInfos)
- // 查询任务id
- var taskIds []string
- taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
- for _, taskInfo := range taskInfos {
- taskIds = append(taskIds, taskInfo.TaskId)
- taskMap[taskInfo.TaskId] = taskInfo
- }
- db1 := GetReadDB(ctx)
- db1 = db1.Debug().Model(gorm_model.YoungeeContractInfo{})
- var DefaultReviewInfos []gorm_model.YoungeeContractInfo
- db1 = db1.Model(gorm_model.YoungeeContractInfo{}).Where("task_id IN ? AND (default_status = 1 OR default_status = 4)", taskIds)
- err := db1.Find(&DefaultReviewInfos).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- DefaultReviewMap := make(map[string]gorm_model.YoungeeContractInfo)
- for _, DefaultReviewInfo := range DefaultReviewInfos {
- DefaultReviewMap[DefaultReviewInfo.TaskID] = DefaultReviewInfo
- }
- // 查询总数
- var totalDefaultReview int64
- if err := db1.Count(&totalDefaultReview).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- // 查询该页数据
- limit := pageSize
- offset := pageSize * pageNum // assert pageNum start with 0
- err = db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- var TaskDefaultReviews []*http_model.TaskDefaultReview
- var taskDefaultReviews []*http_model.TaskDefaultReviewInfo
- var newTaskDefaultReviews []*http_model.TaskDefaultReviewInfo
- for _, taskId := range taskIds {
- TaskDefaultReview := new(http_model.TaskDefaultReview)
- TaskDefaultReview.Talent = taskMap[taskId]
- TaskDefaultReview.Default = DefaultReviewMap[taskId]
- TaskDefaultReviews = append(TaskDefaultReviews, TaskDefaultReview)
- }
- taskDefaultReviews = pack.TaskDefaultReviewToTaskInfo(TaskDefaultReviews)
- for _, v := range taskDefaultReviews {
- if platform_nickname == "" {
- newTaskDefaultReviews = append(newTaskDefaultReviews, v)
- } else if strings.Contains(v.PlatformNickname, platform_nickname) {
- newTaskDefaultReviews = append(newTaskDefaultReviews, v)
- } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
- newTaskDefaultReviews = append(newTaskDefaultReviews, v)
- } else {
- totalTask--
- }
- }
- return newTaskDefaultReviews, totalTask, nil
- }
- func GetTaskDefaultDataList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.TaskDefaultDataInfo, int64, error) {
- db := GetReadDB(ctx)
- // 查询Task表信息
- db = db.Debug().Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2")
- // 根据Project条件过滤
- conditionType := reflect.TypeOf(conditions).Elem()
- conditionValue := reflect.ValueOf(conditions).Elem()
- var platform_nickname string = ""
- for i := 0; i < conditionType.NumField(); i++ {
- field := conditionType.Field(i)
- tag := field.Tag.Get("condition")
- value := conditionValue.FieldByName(field.Name)
- if tag == "default_status" {
- if value.Interface() == int64(3) {
- db = db.Where("cur_default_type = 7")
- }
- continue
- } else if !util.IsBlank(value) {
- if tag == "platform_nickname" {
- platform_nickname = fmt.Sprintf("%v", value.Interface())
- continue
- } else if tag == "project_id" || tag == "strategy_id" {
- db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
- }
- }
- }
- var taskInfos []gorm_model.YoungeeTaskInfo
- db = db.Model(gorm_model.YoungeeTaskInfo{})
- // 查询总数
- var totalTask int64
- if err := db.Count(&totalTask).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- db.Order("task_id").Find(&taskInfos)
- // 查询任务id
- var taskIds []string
- taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
- for _, taskInfo := range taskInfos {
- taskIds = append(taskIds, taskInfo.TaskId)
- taskMap[taskInfo.TaskId] = taskInfo
- }
- db1 := GetReadDB(ctx)
- db1 = db1.Debug().Model(gorm_model.YoungeeContractInfo{})
- var DefaultDataInfos []gorm_model.YoungeeContractInfo
- db1 = db1.Model(gorm_model.YoungeeContractInfo{}).Where("task_id IN ? AND (default_status = 1 OR default_status = 4)", taskIds)
- err := db1.Find(&DefaultDataInfos).Error
- DefaultDataMap := make(map[string]gorm_model.YoungeeContractInfo)
- for _, DefaultDataInfo := range DefaultDataInfos {
- DefaultDataMap[DefaultDataInfo.TaskID] = DefaultDataInfo
- }
- var LinkInfos []gorm_model.YounggeeLinkInfo
- db2 := GetReadDB(ctx)
- db2 = db2.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id IN ? AND is_ok = 1", taskIds).Find(&LinkInfos)
- LinkMap := make(map[string]gorm_model.YounggeeLinkInfo)
- for _, LinkInfo := range LinkInfos {
- LinkMap[LinkInfo.TaskID] = LinkInfo
- }
- // 查询总数
- var totalDefaultData int64
- if err := db2.Count(&totalDefaultData).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- // 查询该页数据
- limit := pageSize
- offset := pageSize * pageNum // assert pageNum start with 0
- err = db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- var TaskDefaultDatas []*http_model.TaskDefaultData
- var taskDefaultDatas []*http_model.TaskDefaultDataInfo
- var newTaskDefaultDatas []*http_model.TaskDefaultDataInfo
- for _, taskId := range taskIds {
- TaskDefaultData := new(http_model.TaskDefaultData)
- TaskDefaultData.Talent = taskMap[taskId]
- TaskDefaultData.Default = DefaultDataMap[taskId]
- TaskDefaultData.Link = LinkMap[taskId]
- TaskDefaultDatas = append(TaskDefaultDatas, TaskDefaultData)
- }
- taskDefaultDatas = pack.TaskDefaultDataToTaskInfo(TaskDefaultDatas)
- for _, v := range taskDefaultDatas {
- if platform_nickname == "" {
- newTaskDefaultDatas = append(newTaskDefaultDatas, v)
- } else if strings.Contains(v.PlatformNickname, platform_nickname) {
- newTaskDefaultDatas = append(newTaskDefaultDatas, v)
- } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
- newTaskDefaultDatas = append(newTaskDefaultDatas, v)
- } else {
- totalTask--
- }
- }
- return newTaskDefaultDatas, totalTask, nil
- }
- func GetTaskTerminatingList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.TaskTerminatingInfo, int64, error) {
- db := GetReadDB(ctx)
- var taskIds1 []string
- var totalTerminating int64
- var TerminatingInfos []gorm_model.YoungeeContractInfo
- db = db.Model(gorm_model.YoungeeContractInfo{})
- err := db.Where("default_status = 3").Find(&TerminatingInfos).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- TerminatingMap := make(map[string]gorm_model.YoungeeContractInfo)
- for _, TerminatingInfo := range TerminatingInfos {
- taskIds1 = append(taskIds1, TerminatingInfo.TaskID)
- TerminatingMap[TerminatingInfo.TaskID] = TerminatingInfo
- }
- if err := db.Count(&totalTerminating).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- db1 := GetReadDB(ctx)
- // 查询Task表信息
- db1 = db1.Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2 and task_id in ?", taskIds1)
- // 根据Project条件过滤
- conditionType := reflect.TypeOf(conditions).Elem()
- conditionValue := reflect.ValueOf(conditions).Elem()
- var platform_nickname string = ""
- for i := 0; i < conditionType.NumField(); i++ {
- field := conditionType.Field(i)
- tag := field.Tag.Get("condition")
- value := conditionValue.FieldByName(field.Name)
- if tag == "default_status" {
- if value.Interface() == int64(4) {
- db = db.Where("cur_default_type = 9")
- }
- continue
- } else if !util.IsBlank(value) {
- if tag == "platform_nickname" {
- platform_nickname = fmt.Sprintf("%v", value.Interface())
- continue
- } else if tag == "project_id" || tag == "strategy_id" {
- db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
- }
- }
- }
- var taskInfos []gorm_model.YoungeeTaskInfo
- // db1 = db1.Model(gorm_model.YoungeeTaskInfo{})
- // 查询总数
- var totalTask int64
- if err := db1.Count(&totalTask).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- db1.Order("task_id").Find(&taskInfos)
- // 查询任务id
- var taskIds []string
- taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
- for _, taskInfo := range taskInfos {
- taskIds = append(taskIds, taskInfo.TaskId)
- taskMap[taskInfo.TaskId] = taskInfo
- }
- var misNum int64
- if totalTerminating > totalTask {
- misNum = totalTerminating - totalTask
- } else {
- misNum = totalTask - totalTerminating
- }
- //logrus.Println("totalTerminating,totalTalent,misNum:", totalTerminating, totalTask, misNum)
- // 查询该页数据
- limit := pageSize + misNum
- offset := pageSize * pageNum // assert pageNum start with 0
- err = db1.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- var TaskTerminatings []*http_model.TaskTerminating
- var taskTerminatings []*http_model.TaskTerminatingInfo
- var newTaskTerminatings []*http_model.TaskTerminatingInfo
- for _, taskId := range taskIds {
- TaskTerminating := new(http_model.TaskTerminating)
- TaskTerminating.Talent = taskMap[taskId]
- TaskTerminating.Default = TerminatingMap[taskId]
- TaskTerminatings = append(TaskTerminatings, TaskTerminating)
- }
- taskTerminatings = pack.TaskTerminatingToTaskInfo(TaskTerminatings)
- for _, v := range taskTerminatings {
- if platform_nickname == "" {
- newTaskTerminatings = append(newTaskTerminatings, v)
- } else if strings.Contains(v.PlatformNickname, platform_nickname) {
- newTaskTerminatings = append(newTaskTerminatings, v)
- } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
- newTaskTerminatings = append(newTaskTerminatings, v)
- } else {
- totalTask--
- }
- }
- return newTaskTerminatings, totalTask, nil
- }
- func GetTaskTerminatedList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.TaskTerminatedInfo, int64, error) {
- db := GetReadDB(ctx)
- // 查询Task表信息
- db = db.Debug().Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2")
- // 根据Project条件过滤
- conditionType := reflect.TypeOf(conditions).Elem()
- conditionValue := reflect.ValueOf(conditions).Elem()
- var platform_nickname string = ""
- for i := 0; i < conditionType.NumField(); i++ {
- field := conditionType.Field(i)
- tag := field.Tag.Get("condition")
- value := conditionValue.FieldByName(field.Name)
- if tag == "default_status" {
- fmt.Printf("default %+v", value.Interface() == int64(0))
- if value.Interface() == int64(5) {
- db = db.Where("complete_status = 4")
- }
- continue
- } else if !util.IsBlank(value) {
- if tag == "platform_nickname" {
- platform_nickname = fmt.Sprintf("%v", value.Interface())
- continue
- } else if tag == "project_id" || tag == "strategy_id" {
- db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
- }
- }
- }
- var taskInfos []gorm_model.YoungeeTaskInfo
- db = db.Model(gorm_model.YoungeeTaskInfo{})
- // 查询总数
- var totalTask int64
- if err := db.Count(&totalTask).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- db.Order("task_id").Find(&taskInfos)
- // 查询任务id
- var taskIds []string
- taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
- for _, taskInfo := range taskInfos {
- taskIds = append(taskIds, taskInfo.TaskId)
- taskMap[taskInfo.TaskId] = taskInfo
- }
- db1 := GetReadDB(ctx)
- db1 = db1.Debug().Model(gorm_model.YoungeeContractInfo{})
- var TerminatedInfos []gorm_model.YoungeeContractInfo
- db1 = db1.Model(gorm_model.YoungeeContractInfo{}).Where("task_id IN ? AND (default_status = 1 OR default_status = 4)", taskIds)
- err1 := db1.Find(&TerminatedInfos).Error
- if err1 != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql find, err:%+v", err1)
- return nil, 0, err1
- }
- TerminatedMap := make(map[string]gorm_model.YoungeeContractInfo)
- for _, TerminatedInfo := range TerminatedInfos {
- TerminatedMap[TerminatedInfo.TaskID] = TerminatedInfo
- }
- // 查询总数
- var totalTerminated int64
- if err := db1.Count(&totalTerminated).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- var misNum int64
- if totalTerminated > totalTask {
- misNum = totalTerminated - totalTask
- } else {
- misNum = totalTask - totalTerminated
- }
- //logrus.Println("totalTerminated,totalTalent,misNum:", totalTerminated, totalTask, misNum)
- // 查询该页数据
- limit := pageSize + misNum
- offset := pageSize * pageNum // assert pageNum start with 0
- err := db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
- if err != nil {
- logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
- return nil, 0, err
- }
- var TaskTerminateds []*http_model.TaskTerminated
- var taskTerminateds []*http_model.TaskTerminatedInfo
- var newTaskTerminateds []*http_model.TaskTerminatedInfo
- for _, taskId := range taskIds {
- TaskTerminated := new(http_model.TaskTerminated)
- TaskTerminated.Talent = taskMap[taskId]
- TaskTerminated.Default = TerminatedMap[taskId]
- TaskTerminateds = append(TaskTerminateds, TaskTerminated)
- }
- taskTerminateds = pack.TaskTerminatedToTaskInfo(TaskTerminateds)
- for _, v := range taskTerminateds {
- if platform_nickname == "" {
- newTaskTerminateds = append(newTaskTerminateds, v)
- } else if strings.Contains(v.PlatformNickname, platform_nickname) {
- newTaskTerminateds = append(newTaskTerminateds, v)
- } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
- newTaskTerminateds = append(newTaskTerminateds, v)
- } else {
- totalTask--
- }
- }
- return newTaskTerminateds, totalTask, nil
- }
|