package db import ( "context" "fmt" "github.com/sirupsen/logrus" "reflect" "strings" "youngee_b_api/model/common_model" "youngee_b_api/model/gorm_model" "youngee_b_api/util" ) // GetLocalTaskList 本地生活子任务列表 func GetLocalTaskList(ctx context.Context, pageSize, pageNum int32, conditions *common_model.LocalTaskConditions) ([]*gorm_model.YoungeeLocalTaskInfo, int64, error) { db := GetReadDB(ctx) // 查询task表信息 db = db.Debug().Model(gorm_model.YoungeeLocalTaskInfo{}) // 根据Local条件过滤 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 !util.IsBlank(value) && tag != "platform_nickname" { db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface()) } else if tag == "platform_nickname" { platform_nickname = fmt.Sprintf("%v", value.Interface()) continue } } var taskInfos []*gorm_model.YoungeeLocalTaskInfo db = db.Model(gorm_model.YoungeeLocalTaskInfo{}) // 查询总数 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) // 查询该页数据 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 taskDatas []*gorm_model.YoungeeLocalTaskInfo var newTaskDatas []*gorm_model.YoungeeLocalTaskInfo taskDatas = taskInfos for _, v := range taskDatas { if platform_nickname == "" { newTaskDatas = append(newTaskDatas, v) } else if strings.Contains(v.TaskID, platform_nickname) { newTaskDatas = append(newTaskDatas, v) } else { totalTask-- } } return newTaskDatas, totalTask, nil }