younggee_s_api/db/local_life_task.go

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, orderBy []string, orderDesc []int, 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
		}
	}

	// 动态排序逻辑
	if len(orderBy) > 0 && len(orderDesc) > 0 && len(orderBy) == len(orderDesc) {
		for i := 0; i < len(orderBy); i++ {
			orderField := orderBy[i]
			isDesc := orderDesc[i] == 1 // 1=降序，其他值=升序

			switch orderField {
			case "fans_num":
				if isDesc {
					db = db.Order("fans_num desc")
				} else {
					db = db.Order("fans_num asc")
				}
			case "vote_avg":
				if isDesc {
					db = db.Order("vote_avg desc")
				} else {
					db = db.Order("vote_avg asc")
				}
			case "commit_avg":
				if isDesc {
					db = db.Order("commit_avg desc")
				} else {
					db = db.Order("commit_avg asc")
				}
			case "collect_avg":
				if isDesc {
					db = db.Order("collect_avg desc")
				} else {
					db = db.Order("collect_avg asc")
				}
			}
		}
	} else {
		// 默认排序
		db = db.Order("task_id desc")
	}

	var taskInfos []*gorm_model.YoungeeLocalTaskInfo
	db = db.Model(gorm_model.YoungeeLocalTaskInfo{})
	// 查询总数
	var totalTask int64
	if err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Count(&totalTask).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
		return nil, 0, err
	}

	// 查询该页数据
	limit := pageSize
	offset := pageSize * pageNum // assert pageNum start with 0
	if err := db.Limit(int(limit)).Offset(int(offset)).Find(&taskInfos).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql data, 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
}

// ChangeLocalTaskStatus 修改服务商本地生活子任务报名状态
func ChangeLocalTaskStatus(ctx context.Context, taskIds []string, supplierStatus int, supplierId int, subAccountId int) ([]int64, error) {
	db := GetReadDB(ctx)
	taskSta := supplierStatus
	var sOperate int
	var sOperateType int
	if subAccountId == 0 {
		sOperate = supplierId
		sOperateType = 1
	} else {
		sOperate = subAccountId
		sOperateType = 2
	}
	if err := db.Debug().Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ?", taskIds).
		Updates(gorm_model.YoungeeLocalTaskInfo{SupplierStatus: taskSta, SOperator: sOperate, SOperatorType: sOperateType}).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[ChangeTaskStatus]2 error query mysql total, err:%+v", err)
		return nil, err
	}

	var taskInfos []gorm_model.YoungeeLocalTaskInfo
	err := db.Debug().Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ?", taskIds).Find(&taskInfos).Error
	if err != nil {
		logrus.WithContext(ctx).Errorf("[ChangeTaskStatus]3 error query mysql total, err:%+v", err)
		return nil, err
	}

	var recruitStrategysIDs []int64
	recruitStrategys := gorm_model.RecruitStrategy{}
	for _, taskInfo := range taskInfos {
		err2 := db.Debug().Model(gorm_model.RecruitStrategy{}).Where("s_local_id=? AND strategy_id=?", taskInfo.SLocalID, taskInfo.StrategyID).Scan(&recruitStrategys).Error
		if err2 != nil {
			logrus.WithContext(ctx).Errorf("[ChangeTaskStatus]4 error query mysql total, err:%+v", err2)
			return nil, err2
		}
		recruitStrategysIDs = append(recruitStrategysIDs, recruitStrategys.RecruitStrategyID)
	}
	return recruitStrategysIDs, nil
}

// GetSLocalTaskList 服务商本地生活子任务账单列表
func GetSLocalTaskList(ctx context.Context, sLocalId int, condition string, pageSize, pageNum int32) ([]*gorm_model.YoungeeLocalTaskInfo, int64, error) {
	db := GetReadDB(ctx)

	// 基础查询条件 - 必须有s_local_id
	db = db.Debug().Model(gorm_model.YoungeeLocalTaskInfo{}).Where("s_local_id = ?", sLocalId)

	// 如果talentId不为空，则添加为查询条件
	if condition != "" {
		db = db.Where("talent_id = ?", condition)
	}

	var taskInfos []*gorm_model.YoungeeLocalTaskInfo

	// 查询总数
	var totalTask int64
	if err := db.Count(&totalTask).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetSLocalTaskList] error query mysql total, err:%+v", err)
		return nil, 0, err
	}

	// 查询该页数据
	limit := pageSize
	offset := pageSize * pageNum // 保持pageNum从0开始
	err := db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Find(&taskInfos).Error
	if err != nil {
		logrus.WithContext(ctx).Errorf("[GetSLocalTaskList] error query mysql data, err:%+v", err)
		return nil, 0, err
	}

	return taskInfos, totalTask, nil
}

// GetLocalTaskInfoById ID查找子任务
func GetLocalTaskInfoById(ctx context.Context, taskId string) (*gorm_model.YoungeeLocalTaskInfo, error) {
	db := GetReadDB(ctx)
	var taskInfo *gorm_model.YoungeeLocalTaskInfo
	whereCondition := gorm_model.YoungeeLocalTaskInfo{
		TaskID: taskId,
	}
	err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where(whereCondition).Find(&taskInfo).Error
	if err != nil {
		return nil, err
	}
	return taskInfo, nil
}

// CountLocalTaskNumByTaskStage 根据子任务状态查找对应的本地子任务数量
func CountLocalTaskNumByTaskStage(ctx context.Context, taskStage int, supplierStatus int, sLocalId int) (int64, error) {
	db := GetReadDB(ctx)
	var taskNum int64
	if taskStage == 0 && supplierStatus == 0 {
		err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("s_local_id = ?", sLocalId).Count(&taskNum).Error
		if err != nil {
			logrus.WithContext(ctx).Errorf("[CreateMessageByTask] error read mysql, err:%+v", err)
			return 0, err
		}
	} else if taskStage == 0 {
		err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("supplier_status = ? and s_local_id = ?", supplierStatus, sLocalId).Count(&taskNum).Error
		if err != nil {
			logrus.WithContext(ctx).Errorf("[CreateMessageByTask] error read mysql, err:%+v", err)
			return 0, err
		}
	} else if supplierStatus == 0 {
		err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_status = ? and s_local_id = ?", taskStage, sLocalId).Count(&taskNum).Error
		if err != nil {
			logrus.WithContext(ctx).Errorf("[CreateMessageByTask] error read mysql, err:%+v", err)
			return 0, err
		}
	} else {
		err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_status = ? and supplier_status = ? and s_local_id = ?", taskStage, supplierStatus, sLocalId).Count(&taskNum).Error
		if err != nil {
			logrus.WithContext(ctx).Errorf("[CreateMessageByTask] error read mysql, err:%+v", err)
			return 0, err
		}
	}
	return taskNum, nil
}