1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366 |
- package db
- import (
- "context"
- "errors"
- "fmt"
- "github.com/issue9/conv"
- "github.com/sirupsen/logrus"
- "gorm.io/gorm"
- "strconv"
- "strings"
- "time"
- "youngee_b_api/model/gorm_model"
- "youngee_b_api/model/http_model"
- )
- func GetLocalRecruittime(ctx context.Context, request http_model.GetLocalRecruitTimeRequest) (*http_model.GetLocalRecruitTimeResponse, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).Find(&localinfo).Error
- var result http_model.GetLocalRecruitTimeResponse
- if err != nil {
- return &result, err
- }
- result.RecruitTime = localinfo.RecruitDdl.Format("2006-01-02 15:04:05")
- return &result, nil
- }
- func GetLocalTalentstatusCount(db *gorm.DB, request http_model.GetLocalTalentstatusNumRequest, status int) (int64, error) {
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_status = ?", request.ProjectId, status)
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return 0, err
- }
- return total, nil
- }
- func GetLocalTalentstatusNumCount(ctx context.Context, request http_model.GetLocalTalentstatusNumRequest) (*http_model.GetLocalTalentstatusNumResponse, error) {
- db := GetReadDB(ctx)
- var unoperatenum, agreetalentnum, refusetalentnum int64
- unoperatenum, _ = GetLocalTalentstatusCount(db, request, 1)
- agreetalentnum, _ = GetLocalTalentstatusCount(db, request, 2)
- refusetalentnum, _ = GetLocalTalentstatusCount(db, request, 3)
- count := &http_model.GetLocalTalentstatusNumResponse{
- UnoperateTalentnum: unoperatenum,
- AgreeTalentnum: agreetalentnum,
- RefuseTalentnum: refusetalentnum,
- }
- return count, nil
- }
- func GetLocalTalentstatusNum(db *gorm.DB, request http_model.GetLocalTalentstatusCountRequest, status int) (int64, error) {
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, status)
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return 0, err
- }
- return total, nil
- }
- func GetLocalTalentstatusCountNum(ctx context.Context, request http_model.GetLocalTalentstatusCountRequest) (*http_model.GetLocalTalentstatusCountResponse, error) {
- db := GetReadDB(ctx)
- var unoperatenum, agreetalentnum, refusetalentnum int64
- unoperatenum, _ = GetLocalTalentstatusNum(db, request, request.TaskStage)
- agreetalentnum, _ = GetLocalTalentstatusNum(db, request, request.TaskStage+1)
- refusetalentnum, _ = GetLocalTalentstatusNum(db, request, request.TaskStage+2)
- count := &http_model.GetLocalTalentstatusCountResponse{
- UnoperateTalentnum: unoperatenum,
- AgreeTalentnum: agreetalentnum,
- RefuseTalentnum: refusetalentnum,
- }
- return count, nil
- }
- func GetLocallifetaskList(ctx context.Context, request http_model.GetLocalTaskListRequest) (*http_model.GetLocalTaskListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_status = ?", request.ProjectId, request.CoopType)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalTaskListData{}, err
- }
- // 构建查询条件
- if request.FeeFrom != nil {
- query = query.Where("fee_form = ?", request.FeeFrom)
- }
- if request.Type != nil {
- switch *request.Type {
- case 1:
- query = query.Where("supplier_id = 0 AND supplier_status = 0")
- case 2:
- query = query.Where("supplier_id !=0 AND supplier_status = 2")
- }
- }
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- if request.TalentFromList != "" {
- provinceList := strings.Split(request.TalentFromList, ",") // 解析传入的省份列表
- query = query.Where("SUBSTRING_INDEX(city, ' ', 1) IN (?)", provinceList) // 提取城市字段中的省份并进行匹配
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("collect_num asc")
- } else {
- query = query.Order("collect_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- // 执行分页查询
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocaLTaskInfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- var Time time.Time
- switch request.CoopType {
- case 1:
- Time = task.CreateDate
- case 2:
- Time = task.SelectDate //2 signedtime
- case 3:
- Time = task.CompleteDate
- }
- var listtime, stageintro string
- switch task.TaskStage {
- case 4:
- {
- listtime = task.SelectDate.Format("2006-01-02 15:04:05")
- stageintro = "执行开始时间"
- }
- case 5:
- {
- listtime = task.ReserveTime.Format("2006-01-02 15:04:05")
- stageintro = "达人探店时间预约申请时间"
- }
- case 6:
- {
- listtime = task.ExploreTime.Format("2006-01-02 15:04:05")
- stageintro = "达人探店时间预约申请时间"
- }
- case 9:
- {
- listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
- stageintro = "初稿上传截止时间"
- }
- case 10:
- {
- var sketchinfo gorm_model.YounggeeSketchInfo
- err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 0", task.TaskId).First(&sketchinfo).Error
- if err != nil {
- return nil, err
- }
- listtime = sketchinfo.CreateAt.Format("2006-01-02 15:04:05")
- stageintro = "达人上传初稿时间" //
- }
- case 11:
- {
- listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
- stageintro = " 链接上传截止时间"
- }
- case 12:
- {
- var linkinfo gorm_model.YounggeeLinkInfo
- err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", task.TaskId).First(&linkinfo).Error
- if err != nil {
- return nil, err
- }
- listtime = linkinfo.CreateAt.Format("2006-01-02 15:04:05")
- stageintro = "达人上传链接时间"
- }
- case 13:
- {
- listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
- stageintro = "数据待传截止时间"
- }
- case 14:
- {
- var datainfo gorm_model.YounggeeDataInfo
- err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 0", task.TaskId).First(&datainfo).Error
- if err != nil {
- return nil, err
- }
- var auto_task_finish gorm_model.InfoAutoTask
- db.Model(&gorm_model.InfoAutoTask{}).Where("enterprise_id = ?", request.EnterPriseId).First(&auto_task_finish)
- // 将小时数转换为 Duration 后相加
- resultTime := datainfo.CreateAt.Add(time.Duration(auto_task_finish.TaskFinishAuto) * time.Hour)
- listtime = resultTime.Format("2006-01-02 15:04:05")
- stageintro = "品牌最晚结算时间"
- }
- case 15:
- {
- listtime = task.CompleteDate.Format("2006-01-02 15:04:05")
- stageintro = "品牌结算时间"
- }
- case 16:
- {
- listtime = task.CancelTime.Format("2006-01-02 15:04:05")
- stageintro = "解约时间"
- }
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterPriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- response := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- Boperator: boperator,
- CreateAt: Time.Format("2006-01-02 15:04:05"),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- Gender: gender,
- Sprojectid: task.SLocalLifeId,
- City: task.City,
- StageIntro: stageintro,
- ListTime: listtime,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalTaskListData{
- TaskList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func PassLocalTaskCoop(ctx context.Context, req http_model.PasslocalTaskCoopRequest) (bool, error) {
- db := GetReadDB(ctx)
- var count int64
- err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ? AND task_stage = 1", req.TaskIds).Count(&count).Error
- if err != nil {
- return false, err
- }
- if int64(len(req.TaskIds)) == 0 || count != int64(len(req.TaskIds)) {
- return false, errors.New("任务id有误")
- }
- // 2. 查询任务对应达人id(用于生成达人消息)
- var talentIds []string
- err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ?", req.TaskIds).Select("talent_id").Find(&talentIds).Error
- if err != nil {
- return false, err
- }
- // 3. 查询任务对应project名称(用于生成达人消息)
- var project gorm_model.YounggeeLocalLifeInfo
- err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", req.ProjectId).Find(&project).Error
- if err != nil {
- return false, err
- }
- var taskstages int
- if project.TaskForm == 1 {
- taskstages = 4
- } else {
- taskstages = 9
- }
- nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
- if req.IsSpecial == 1 {
- toolList := strings.Split(project.Tools, ",")
- for _, tool := range toolList {
- // 将 tool 转换为整数
- if t, err := strconv.Atoi(tool); err == nil && t >= 0 && t <= len(nextstage) {
- nextstage[t-1] = 1
- }
- }
- fmt.Println(nextstage)
- for i := 1; i < len(nextstage); i++ { // 从位置1开始
- if nextstage[i] == 1 {
- switch i {
- case 1:
- taskstages = 4
- case 2:
- taskstages = 9
- case 3:
- taskstages = 11
- case 4:
- taskstages = 13
- case 5:
- taskstages = 15
- }
- break
- }
- }
- }
- var opid string
- var optype int
- switch req.OperateType {
- case 3:
- {
- opid = req.EnterpriseId
- optype = 1
- }
- case 4:
- {
- opid = req.OperatorId
- optype = 2
- }
- }
- //4.根据达人来源改变状态、阶段、数目
- var tasks []gorm_model.YoungeeLocalTaskInfo
- err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ? AND task_stage = 1", req.TaskIds).Find(&tasks).Error
- if err != nil {
- return false, err
- }
- err = db.Transaction(func(tx *gorm.DB) error {
- // 2. 修改任务状态和任务阶段
- for _, taskInfo := range tasks {
- if taskInfo.SupplierId != 0 && taskInfo.SupplierStatus == 2 { //服务商招募数量+1
- updateData := gorm_model.YoungeeLocalTaskInfo{
- TaskStage: taskstages,
- TaskStatus: 2,
- UpdateAt: time.Now(),
- BOperator: opid,
- BOperatorType: optype,
- SelectDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskId).Updates(updateData).Error
- if err != nil {
- return err
- }
- //project招募数+1,预估成本+服务商价格
- var projectinfo gorm_model.YounggeeLocalLifeInfo
- err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", req.ProjectId).Find(&projectinfo).Error
- if err != nil {
- return err
- }
- projectinfo.RecruitNum += 1
- projectinfo.NeedPay += taskInfo.SupportFee
- err = tx.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ? ", req.ProjectId).Updates(projectinfo).Error
- if err != nil {
- return err
- }
- //服务商s_project已招募人数+1改成s_locallife
- var sprojectinfo gorm_model.YounggeeSProjectInfo
- err = db.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", taskInfo.SLocalLifeId).Find(&sprojectinfo).Error
- if err != nil {
- return err
- }
- sprojectinfo.RecruitNum += 1
- fmt.Println("......", projectinfo.RecruitNum)
- err = tx.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", taskInfo.SLocalLifeId).Updates(sprojectinfo).Error
- if err != nil {
- return err
- } //
- }
- //if project.TaskForm != 1 { //素材分发
- // var draft_default gorm_model.InfoAutoTask
- // err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskId).Find(&draft_default).Error
- // if err != nil {
- // return err
- // }
- // duration := time.Duration(draft_default.DraftDefault) * time.Hour
- // upbreakdate := gorm_model.YoungeeLocalTaskInfo{
- // CurBreakAt: time.Now().Add(duration),
- // }
- // err = tx.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ? ", taskInfo.TaskId).Updates(upbreakdate).Error
- // if err != nil {
- // return err
- // }
- //}
- if taskInfo.SupplierId == 0 && taskInfo.SupplierStatus == 0 { //来源于公海
- updateData := gorm_model.YoungeeLocalTaskInfo{
- TaskStage: taskstages,
- TaskStatus: 2,
- UpdateAt: time.Now(),
- BOperator: opid,
- BOperatorType: optype,
- SelectDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskId).Updates(updateData).Error
- if err != nil {
- return err
- }
- //project招募数+1
- var projectinfo gorm_model.YounggeeLocalLifeInfo
- err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", req.ProjectId).Find(&projectinfo).Error
- if err != nil {
- return err
- }
- projectinfo.RecruitNum += 1
- projectinfo.NeedPay += taskInfo.SupportFee
- err = tx.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ? ", req.ProjectId).Updates(projectinfo).Error
- if err != nil {
- return err
- }
- }
- }
- // 4. 生成达人消息
- for _, talendId := range talentIds {
- err = CreateMessage(ctx, 1, 1, talendId, project.LocalName)
- if err != nil {
- return err
- }
- }
- // 返回 nil 提交事务
- return nil
- })
- if err != nil {
- return false, err
- }
- return true, nil
- }
- func RefuseLocalTaskCoop(ctx context.Context, req http_model.RefuselocalTaskCoopRequest) (bool, error) {
- db := GetWriteDB(ctx)
- // 1. 校验
- var count int64
- err := db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ? AND task_stage = 1", req.TaskIds).Count(&count).Error
- if err != nil {
- return false, err
- }
- if count != int64(len(req.TaskIds)) {
- return false, errors.New("任务id有误")
- }
- var opid string
- var optype int
- switch req.OperateType {
- case 3:
- {
- opid = req.EnterpriseId
- optype = 1
- }
- case 4:
- {
- opid = req.OperatorId
- optype = 2
- }
- }
- // 2. 查询任务对应达人id(用于生成达人消息)
- var talentIds []string
- err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ?", req.TaskIds).Select("talent_id").Find(&talentIds).Error
- if err != nil {
- return false, err
- }
- var tasks []gorm_model.YoungeeLocalTaskInfo
- err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id IN ? AND task_stage = 1", req.TaskIds).Find(&tasks).Error
- err = db.Transaction(func(tx *gorm.DB) error {
- // 2. 修改任务状态和任务阶段
- for _, taskInfo := range tasks {
- updateData := gorm_model.YoungeeLocalTaskInfo{
- TaskStage: 3,
- TaskStatus: 3,
- UpdateAt: time.Now(),
- BOperator: opid,
- BOperatorType: optype,
- CompleteDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskId).Updates(updateData).Error
- }
- if err != nil {
- return err
- }
- // 返回 nil 提交事务
- return nil
- })
- if err != nil {
- return false, err
- }
- return true, nil
- }
- func GetLocallifeData(ctx context.Context, req http_model.LocallifeDataRequest) (http_model.LocallifeDataResponse, error) {
- db := GetReadDB(ctx)
- var projectInfo gorm_model.YounggeeLocalLifeInfo
- // 查询项目基本信息
- if err := db.Where("local_id = ?", req.LocallifeId).First(&projectInfo).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[Locallife db] GetLocallifeData error, err: %+v", err)
- return http_model.LocallifeDataResponse{}, err
- }
- fmt.Println("projectInfo---", projectInfo.LocalId)
- var projectTasks []gorm_model.YoungeeLocalTaskInfo
- if err := db.Where("local_id = ? AND task_status = ?", req.LocallifeId, 2).Find(&projectTasks).Error; err != nil {
- return http_model.LocallifeDataResponse{}, err
- }
- var totalTasks int64
- if err := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_status = ?", req.LocallifeId, 2).Count(&totalTasks).Error; err != nil {
- return http_model.LocallifeDataResponse{}, err
- }
- // 初始化统计数据
- dateStats := make(map[string]int)
- fansCounts := map[int64]int{0: 0, 1: 0, 2: 0, 3: 0}
- feeCounts := map[int]int{1: 0, 2: 0, 3: 0}
- startDate := *projectInfo.CreatedAt
- endDate := time.Now()
- // 填充日期统计
- for d := startDate; d.Before(endDate) || d.Equal(endDate); d = d.Add(24 * time.Hour) {
- dateStats[d.Format("2006-01-02")] = 0
- }
- var totalCost float64
- var totalFans int
- var serverTaskCount float64
- var maleCount int64
- // 处理每个任务
- for _, task := range projectTasks {
- totalCost += task.DraftFee
- var talentInfo gorm_model.YoungeeTalentInfo
- // 查询达人信息
- if err := db.Where("id = ?", task.TalentId).First(&talentInfo).Error; err != nil {
- logrus.WithContext(ctx).Errorf("[Locallife db] GetLocallifeData error, err: %+v", err)
- return http_model.LocallifeDataResponse{}, err
- }
- // 统计性别
- if talentInfo.Sex == 1 {
- maleCount++
- }
- totalFans += task.FansNum
- // 统计来自服务商的任务
- if task.SupplierId != 0 && task.SupplierStatus == 2 {
- serverTaskCount++
- }
- // 稿费形式统计
- feeCounts[task.FeeForm]++
- // 统计粉丝数
- switch {
- case task.FansNum < 10000:
- fansCounts[0]++
- case task.FansNum < 100000:
- fansCounts[1]++
- case task.FansNum < 1000000:
- fansCounts[2]++
- default:
- fansCounts[3]++
- }
- SelectDate := task.SelectDate.Format("2006-01-02")
- if _, exists := dateStats[SelectDate]; exists {
- dateStats[SelectDate]++
- }
- }
- // 计算平均值
- var avgFansNum int
- if totalTasks > 0 {
- avgFansNum = totalFans / int(totalTasks)
- } else {
- avgFansNum = 0
- }
- var fromType float64
- if totalTasks > 0 {
- fromType = serverTaskCount / float64(totalTasks)
- } else {
- fromType = 0
- }
- var manRatio float64
- if totalTasks > 0 {
- manRatio = float64(maleCount) / float64(totalTasks)
- } else {
- manRatio = 0
- }
- // 构建响应
- res := http_model.LocallifeDataResponse{
- Recruittime: *projectInfo.PassAt,
- RecruitDDl: *projectInfo.RecruitDdl,
- AgreeNum: projectInfo.RecruitNum,
- ApplyNum: projectInfo.ApplyNum,
- DraftFee: totalCost,
- AvgFansNum: avgFansNum,
- FromType: fromType,
- ManRatio: manRatio,
- DateStats: dateStats,
- FeeMap: feeCounts,
- FansMap: fansCounts,
- }
- return res, nil
- }
- func GetLocalPreSketchList(ctx context.Context, request http_model.LocalPreSketchListRequest) (*http_model.GetLocalSketchTaskListData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalSketchTaskListData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTasksketchInfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterpriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- NickName: nickname,
- HeadUrl: headurl,
- Gender: gender,
- CollectNum: 0,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.LocalTasksketchInfo{
- Task: taskinfo,
- DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalSketchTaskListData{
- TasksketchList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLocalSketchList(ctx context.Context, request http_model.LocalTasksketchlistRequest) (*http_model.GetsketchlocaltaskListData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetsketchlocaltaskListData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTasksketchinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- SType: s_type,
- SName: s_name,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- Gender: gender,
- HeadUrl: headurl,
- City: task.City,
- }
- var sketchinfo gorm_model.YounggeeSketchInfo
- switch request.ScriptStatus {
- case 10:
- err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskId).First(&sketchinfo).Error
- if err != nil {
- return nil, err
- }
- case 11:
- err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_ok = 1", task.TaskId).First(&sketchinfo).Error
- if err != nil {
- return nil, err
- }
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.LocalTasksketchinfo{
- Task: taskinfo,
- SketchId: sketchinfo.SketchID,
- SubmitAt: sketchinfo.CreateAt.Format("2006-01-02 15:04:05"),
- AgreeAt: sketchinfo.AgreeAt.Format("2006-01-02 15:04:05"),
- SketchType: sketchinfo.Type,
- Operator: boperator,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetsketchlocaltaskListData{
- LocalTasksketchList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLocalPreLinkList(ctx context.Context, request http_model.LocalPreLinkListRequest) (*http_model.GetLocalPreLinkListData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalPreLinkListData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTasklinkinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterpriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- NickName: nickname,
- HeadUrl: headurl,
- Gender: gender,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.LocalTasklinkinfo{
- Task: taskinfo,
- DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalPreLinkListData{
- TasklinkList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLocalLinkList(ctx context.Context, request http_model.LocalTaskLinklistRequest) (*http_model.GetLocalTaskLinkListData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalTaskLinkListData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTaskLinkinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterpriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- NickName: nickname,
- HeadUrl: headurl,
- Gender: gender,
- ISCoop: Iscoop,
- City: task.City,
- }
- var linkinfo gorm_model.YounggeeLinkInfo
- switch request.LinkStatus {
- case "12":
- err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskId).First(&linkinfo).Error
- if err != nil {
- return nil, err
- }
- case "13":
- err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_ok = 1", task.TaskId).First(&linkinfo).Error
- if err != nil {
- return nil, err
- }
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.LocalTaskLinkinfo{
- Task: taskinfo,
- LinkId: linkinfo.LinkID,
- SubmitAt: linkinfo.CreateAt.Format("2006-01-02 15:04:05"),
- AgreeAt: linkinfo.AgreeAt.Format("2006-01-02 15:04:05"),
- LinkUrl: linkinfo.LinkUrl,
- PhotoUrl: linkinfo.PhotoUrl,
- Operator: boperator,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalTaskLinkListData{
- LocalTaskLinklist: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLocalPreDataList(ctx context.Context, request http_model.LocalPreDataListRequest) (*http_model.GetLocalPreDataListData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalPreDataListData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTaskdatainfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterpriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- NickName: nickname,
- HeadUrl: headurl,
- Gender: gender,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.LocalTaskdatainfo{
- Task: taskinfo,
- DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalPreDataListData{
- TaskdataList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLocalDataList(ctx context.Context, request http_model.LocalTaskDatalistRequest) (*http_model.GetLocalTaskDatalistData, error) {
- db := GetReadDB(ctx)
- var localinfo gorm_model.YounggeeLocalLifeInfo
- err := db.Model(&gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", request.ProjectId).First(&localinfo).Error
- if err != nil {
- return &http_model.GetLocalTaskDatalistData{}, err
- }
- var projecrtaskinfo []gorm_model.YoungeeLocalTaskInfo
- query := db.Model(&gorm_model.YoungeeLocalTaskInfo{}).Where("local_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
- if request.Others != "" {
- query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
- }
- // 计算总数
- var total int64
- if err := query.Count(&total).Error; err != nil {
- return nil, err
- }
- // 添加分页逻辑
- pageSize := request.PageSize
- if pageSize == 0 {
- pageSize = 10
- }
- pageNum := request.PageNum
- if pageNum == 0 {
- pageNum = 1
- }
- offset := (pageNum - 1) * pageSize
- // 执行分页查询
- // 处理多字段排序逻辑
- if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
- for i := 0; i < len(request.SortField); i++ {
- field := request.SortField[i]
- order := request.SortOrder[i]
- switch field {
- case "fansnum":
- if order == "asc" {
- query = query.Order("fans_num asc")
- } else {
- query = query.Order("fans_num desc")
- }
- case "voteavg":
- if order == "asc" {
- query = query.Order("vote_avg asc")
- } else {
- query = query.Order("vote_avg desc")
- }
- case "commentavg":
- if order == "asc" {
- query = query.Order("commit_avg asc")
- } else {
- query = query.Order("commit_avg desc")
- }
- case "collectnum":
- if order == "asc" {
- query = query.Order("view_num asc")
- } else {
- query = query.Order("view_num desc")
- }
- }
- }
- } else {
- // Default sorting if no valid sort parameters
- query = query.Order("task_status asc").Order("task_stage asc")
- }
- if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.LocalTaskDatainfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl, gender, fans, voteavg := getTalentinfo(db, task.TalentId)
- Iscoop := getIscoop(db, task.TalentId, request.EnterpriseId)
- s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, localinfo.LocalType)
- taskinfo := &http_model.LocaLTaskInfo{
- TaskId: task.TaskId,
- ProjectId: task.LocalId,
- TalentId: task.TalentId,
- FansNum: fans,
- DraftFee: task.DraftFee,
- Voteavg: voteavg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: 0,
- CollectNum: 0,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: s_type,
- SName: s_name,
- NickName: nickname,
- HeadUrl: headurl,
- Gender: gender,
- ISCoop: Iscoop,
- City: task.City,
- }
- var datainfo gorm_model.YounggeeDataInfo
- switch request.DataStatus {
- case "12":
- err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskId).First(&datainfo).Error
- if err != nil {
- return nil, err
- }
- case "13":
- err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_ok = 1", task.TaskId).First(&datainfo).Error
- if err != nil {
- return nil, err
- }
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.LocalTaskDatainfo{
- Task: taskinfo,
- DataId: datainfo.DataID,
- SubmitAt: datainfo.CreateAt.Format("2006-01-02 15:04:05"),
- AgreeAt: datainfo.AgreeAt.Format("2006-01-02 15:04:05"),
- PhotoUrl: datainfo.PhotoUrl,
- PlayNumber: datainfo.PlayNumber,
- LikeNumber: datainfo.LikeNumber,
- CollectNumber: datainfo.CollectNumber,
- CommentNumber: datainfo.CommentNumber,
- SettleAmount: task.SettleAmount,
- Breakrate: task.SketchBreakRate + task.LinkBreakRate + task.DataBreakRate,
- Operator: boperator,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetLocalTaskDatalistData{
- LocalTaskDatalist: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
|