|
@@ -0,0 +1,491 @@
|
|
|
|
+package db
|
|
|
|
+
|
|
|
|
+import (
|
|
|
|
+ "context"
|
|
|
|
+ "fmt"
|
|
|
|
+ "reflect"
|
|
|
|
+ "strings"
|
|
|
|
+ "youngee_b_api/model/common_model"
|
|
|
|
+ "youngee_b_api/model/gorm_model"
|
|
|
|
+ "youngee_b_api/model/http_model"
|
|
|
|
+ "youngee_b_api/pack"
|
|
|
|
+ "youngee_b_api/util"
|
|
|
|
+
|
|
|
|
+ "github.com/issue9/conv"
|
|
|
|
+ "github.com/sirupsen/logrus"
|
|
|
|
+)
|
|
|
|
+
|
|
|
|
+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
|
|
|
|
+ }
|
|
|
|
+ if !util.IsBlank(value) {
|
|
|
|
+ logrus.Println("tag: ", tag)
|
|
|
|
+ if tag == "platform_nickname" {
|
|
|
|
+ platform_nickname = fmt.Sprintf("%v", value.Interface())
|
|
|
|
+ continue
|
|
|
|
+ } else if tag == "project_id" {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
|
|
|
|
+ } else if tag == "strategy_ids" {
|
|
|
|
+ strategyIds := strings.Split(fmt.Sprintf("%v", value.Interface()), ",")
|
|
|
|
+ var strategyIdList []int
|
|
|
|
+ for _, strategyId := range strategyIds {
|
|
|
|
+ strategyIdList = append(strategyIdList, conv.MustInt(strategyId))
|
|
|
|
+ }
|
|
|
|
+ db = db.Where("strategy_id in ?", strategyIdList)
|
|
|
|
+ } else {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s like '%%%v%%'", 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 []int
|
|
|
|
+ taskMap := make(map[int]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", 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[int]gorm_model.YoungeeContractInfo)
|
|
|
|
+ for _, DefaultReviewInfo := range DefaultReviewInfos {
|
|
|
|
+ DefaultReviewMap[conv.MustInt(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
|
|
|
|
+ }
|
|
|
|
+ var misNum int64
|
|
|
|
+ if totalDefaultReview > totalTask {
|
|
|
|
+ misNum = totalDefaultReview - totalTask
|
|
|
|
+ } else {
|
|
|
|
+ misNum = totalTask - totalDefaultReview
|
|
|
|
+ }
|
|
|
|
+ logrus.Println("totalDefaultReview,totalTalent,misNum:", totalDefaultReview, 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 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 {
|
|
|
|
+ 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) {
|
|
|
|
+ logrus.Println("tag: ", tag)
|
|
|
|
+ if tag == "platform_nickname" {
|
|
|
|
+ platform_nickname = fmt.Sprintf("%v", value.Interface())
|
|
|
|
+ continue
|
|
|
|
+ } else if tag == "project_id" {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
|
|
|
|
+ } else if tag == "strategy_ids" {
|
|
|
|
+ strategyIds := strings.Split(fmt.Sprintf("%v", value.Interface()), ",")
|
|
|
|
+ var strategyIdList []int
|
|
|
|
+ for _, strategyId := range strategyIds {
|
|
|
|
+ strategyIdList = append(strategyIdList, conv.MustInt(strategyId))
|
|
|
|
+ }
|
|
|
|
+ db = db.Where("strategy_id in ?", strategyIdList)
|
|
|
|
+ } else {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s like '%%%v%%'", 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 []int
|
|
|
|
+ taskMap := make(map[int]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", taskIds)
|
|
|
|
+ err := db1.Find(&DefaultDataInfos).Error
|
|
|
|
+ DefaultDataMap := make(map[int]gorm_model.YoungeeContractInfo)
|
|
|
|
+ for _, DefaultDataInfo := range DefaultDataInfos {
|
|
|
|
+ DefaultDataMap[conv.MustInt(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[int]gorm_model.YounggeeLinkInfo)
|
|
|
|
+ for _, LinkInfo := range LinkInfos {
|
|
|
|
+ LinkMap[conv.MustInt(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
|
|
|
|
+ }
|
|
|
|
+ var misNum int64
|
|
|
|
+ if totalDefaultData > totalTask {
|
|
|
|
+ misNum = totalDefaultData - totalTask
|
|
|
|
+ } else {
|
|
|
|
+ misNum = totalTask - totalDefaultData
|
|
|
|
+ }
|
|
|
|
+ logrus.Println("totalDefaultData,totalTalent,misNum:", totalDefaultData, 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 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 {
|
|
|
|
+ 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 []int
|
|
|
|
+ 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[int]gorm_model.YoungeeContractInfo)
|
|
|
|
+ for _, TerminatingInfo := range TerminatingInfos {
|
|
|
|
+ taskIds1 = append(taskIds1, TerminatingInfo.TaskID)
|
|
|
|
+ TerminatingMap[conv.MustInt(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) {
|
|
|
|
+ logrus.Println("tag: ", tag)
|
|
|
|
+ if tag == "platform_nickname" {
|
|
|
|
+ platform_nickname = fmt.Sprintf("%v", value.Interface())
|
|
|
|
+ continue
|
|
|
|
+ } else if tag == "project_id" {
|
|
|
|
+ db1 = db1.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
|
|
|
|
+ } else if tag == "strategy_ids" {
|
|
|
|
+ strategyIds := strings.Split(fmt.Sprintf("%v", value.Interface()), ",")
|
|
|
|
+ var strategyIdList []int
|
|
|
|
+ for _, strategyId := range strategyIds {
|
|
|
|
+ strategyIdList = append(strategyIdList, conv.MustInt(strategyId))
|
|
|
|
+ }
|
|
|
|
+ db1 = db1.Where("strategy_id in ?", strategyIdList)
|
|
|
|
+ } else {
|
|
|
|
+ db1 = db1.Where(fmt.Sprintf("%s like '%%%v%%'", 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 []int
|
|
|
|
+ taskMap := make(map[int]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 {
|
|
|
|
+ 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) {
|
|
|
|
+ logrus.Println("tag: ", tag)
|
|
|
|
+ if tag == "platform_nickname" {
|
|
|
|
+ platform_nickname = fmt.Sprintf("%v", value.Interface())
|
|
|
|
+ continue
|
|
|
|
+ } else if tag == "project_id" {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
|
|
|
|
+ } else if tag == "strategy_ids" {
|
|
|
|
+ strategyIds := strings.Split(fmt.Sprintf("%v", value.Interface()), ",")
|
|
|
|
+ var strategyIdList []int
|
|
|
|
+ for _, strategyId := range strategyIds {
|
|
|
|
+ strategyIdList = append(strategyIdList, conv.MustInt(strategyId))
|
|
|
|
+ }
|
|
|
|
+ db = db.Where("strategy_id in ?", strategyIdList)
|
|
|
|
+ } else {
|
|
|
|
+ db = db.Where(fmt.Sprintf("%s like '%%%v%%'", 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 []int
|
|
|
|
+ taskMap := make(map[int]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", taskIds)
|
|
|
|
+ if conditions.DefaultStatus == int64(0) {
|
|
|
|
+ db1 = db1.Where("is_review = 0").Find(&TerminatedInfos)
|
|
|
|
+ } else {
|
|
|
|
+ db1 = db1.Where("is_ok = 1").Find(&TerminatedInfos)
|
|
|
|
+ }
|
|
|
|
+ TerminatedMap := make(map[int]gorm_model.YoungeeContractInfo)
|
|
|
|
+ for _, TerminatedInfo := range TerminatedInfos {
|
|
|
|
+ TerminatedMap[conv.MustInt(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 {
|
|
|
|
+ totalTask--
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ return newTaskTerminateds, totalTask, nil
|
|
|
|
+}
|