123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766 |
- package db
- import (
- "context"
- "errors"
- "fmt"
- "github.com/issue9/conv"
- "gorm.io/gorm"
- "strconv"
- "strings"
- "time"
- "youngee_b_api/model/gorm_model"
- "youngee_b_api/model/http_model"
- )
- func GetProjecttaskList(ctx context.Context, request http_model.GetTaskListRequest) (*http_model.GetTaskListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ?", request.ProjectId, request.CoopType)
- // 构建查询条件
- 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")
- }
- }
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.TaskInfo, 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
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterPriseId)
- response := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- Boperator: boperator,
- CreateAt: Time,
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- Sprojectid: task.SprojectId,
- City: task.City,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetTaskListData{
- TaskList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func getBOperator(db *gorm.DB, bOperatorID string, bOperatorType int) string {
- if bOperatorID == "" {
- return ""
- }
- if bOperatorType == 1 {
- var boperatorinfo gorm_model.Enterprise
- if err := db.Where(gorm_model.Enterprise{EnterpriseID: bOperatorID}).First(&boperatorinfo).Error; err != nil {
- return ""
- }
- return boperatorinfo.BusinessName
- }
- if bOperatorType == 2 {
- var bsubaccount gorm_model.YounggeeSubAccount
- if err := db.Where(gorm_model.YounggeeSubAccount{SubAccountId: conv.MustInt(bOperatorID)}).First(&bsubaccount).Error; err != nil {
- return ""
- }
- return bsubaccount.SubAccountName
- }
- if bOperatorType == 3 {
- return "管理后台"
- }
- return ""
- }
- func getTalentinfo(db *gorm.DB, talentID string) (string, string) {
- var talentinfo gorm_model.PlatformKuaishouUserInfo
- if err := db.Where(gorm_model.PlatformKuaishouUserInfo{TalentId: talentID}).First(&talentinfo).Error; err != nil {
- return "", ""
- }
- return talentinfo.NickName, talentinfo.HeadUri
- }
- func getIscoop(db *gorm.DB, talentid string, enterpriseid string) int {
- var etcoop gorm_model.EnterpriseTalentCooperate
- if err := db.Where(gorm_model.EnterpriseTalentCooperate{TalentId: talentid, EnterpriseId: enterpriseid}).First(&etcoop).Error; err != nil {
- return 0
- }
- return 1
- }
- func determineFrom(supplierID, supplierStatus int) int {
- if supplierID != 0 && supplierStatus == 2 {
- return 2
- }
- return 1
- }
- func stype(db *gorm.DB, supplierID, supplierStatus int) int {
- if supplierID != 0 && supplierStatus == 2 {
- var supplierinfo gorm_model.Supplier
- err := db.Model(gorm_model.Supplier{}).Where("supplier_id = ?", supplierID).First(&supplierinfo).Error
- if err != nil {
- return 0
- }
- return supplierinfo.SupplierType
- }
- return 0
- }
- func PassProTaskCoop(ctx context.Context, projectId string, taskIds []string, operatorid string, operatetype int, Isspecial int, req http_model.PassproTaskCoopRequest) (bool, error) {
- db := GetReadDB(ctx)
- var count int64
- fmt.Println("task_ids: ", taskIds)
- err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
- fmt.Println("count: ", count)
- if err != nil {
- return false, err
- }
- if int64(len(taskIds)) == 0 || count != int64(len(taskIds)) {
- return false, errors.New("任务id有误")
- }
- // 2. 查询任务对应达人id(用于生成达人消息)
- var talentIds []string
- err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
- if err != nil {
- return false, err
- }
- // 3. 查询任务对应project名称(用于生成达人消息)
- var project gorm_model.ProjectInfo
- err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&project).Error
- if err != nil {
- return false, err
- }
- var taskstages int
- if project.ProjectForm == 1 {
- taskstages = 4
- } else {
- taskstages = 9
- }
- nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
- if 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
- }
- }
- }
- //4.根据达人来源改变状态、阶段、数目
- var tasks []gorm_model.YoungeeTaskInfo
- err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", 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.YoungeeTaskInfo{
- TaskStage: taskstages,
- TaskStatus: 2,
- UpdateAt: time.Now(),
- BOperator: operatorid,
- BOperatorType: operatetype,
- SelectDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
- if err != nil {
- return err
- }
- //project招募数+1,预估成本+服务商价格
- var projectinfo gorm_model.ProjectInfo
- err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
- if err != nil {
- return err
- }
- projectinfo.RecruitNum += 1
- projectinfo.EstimatedCost += taskInfo.SupportFee
- err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
- if err != nil {
- return err
- }
- //服务商s_project已招募人数+1
- var sprojectinfo gorm_model.YounggeeSProjectInfo
- err = db.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", req.SProjectId).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 =?", req.SProjectId).Updates(sprojectinfo).Error
- if err != nil {
- return err
- }
- }
- if project.ProjectForm != 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.YoungeeTaskInfo{
- CurBreakAt: time.Now().Add(duration),
- }
- err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? ", taskInfo.TaskID).Updates(upbreakdate).Error
- if err != nil {
- return err
- }
- }
- if taskInfo.SupplierId == 0 && taskInfo.SupplierStatus == 0 { //来源于公海
- updateData := gorm_model.YoungeeTaskInfo{
- TaskStage: taskstages,
- TaskStatus: 2,
- UpdateAt: time.Now(),
- BOperator: operatorid,
- BOperatorType: operatetype,
- SelectDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
- if err != nil {
- return err
- }
- //project招募数+1
- var projectinfo gorm_model.ProjectInfo
- err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
- if err != nil {
- return err
- }
- projectinfo.RecruitNum += 1
- err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
- if err != nil {
- return err
- }
- }
- }
- // 4. 生成达人消息
- for _, talendId := range talentIds {
- err = CreateMessage(ctx, 1, 1, talendId, project.ProjectName)
- if err != nil {
- return err
- }
- }
- // 返回 nil 提交事务
- return nil
- })
- if err != nil {
- return false, err
- }
- return true, nil
- }
- func RefuseproTaskCoop(ctx context.Context, taskIds []string, operatorid string, operatetype int) (bool, error) {
- db := GetWriteDB(ctx)
- // 1. 校验
- var count int64
- err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
- if err != nil {
- return false, err
- }
- if count != int64(len(taskIds)) {
- return false, errors.New("任务id有误")
- }
- // 2. 查询任务对应达人id(用于生成达人消息)
- var talentIds []string
- err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
- if err != nil {
- return false, err
- }
- var tasks []gorm_model.YoungeeTaskInfo
- err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Find(&tasks).Error
- err = db.Transaction(func(tx *gorm.DB) error {
- // 2. 修改任务状态和任务阶段
- for _, taskInfo := range tasks {
- updateData := gorm_model.YoungeeTaskInfo{
- TaskStage: 3,
- TaskStatus: 3,
- UpdateAt: time.Now(),
- BOperator: operatorid,
- BOperatorType: operatetype,
- CompleteDate: time.Now(),
- }
- err = tx.Model(gorm_model.YoungeeTaskInfo{}).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 GetPreSketchList(ctx context.Context, request http_model.PreSketchListRequest) (*http_model.GetSketchTaskListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.TasksketchInfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.TasksketchInfo{
- Task: taskinfo,
- DDl: task.CurBreakAt,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetSketchTaskListData{
- TasksketchList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetSketchList(ctx context.Context, request http_model.TasksketchlistRequest) (*http_model.GetsketchtaskListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.Tasksketchinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- City: task.City,
- }
- 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
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.Tasksketchinfo{
- Task: taskinfo,
- SketchId: sketchinfo.SketchID,
- SubmitAt: sketchinfo.SubmitAt,
- AgreeAt: sketchinfo.AgreeAt,
- Operator: boperator,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetsketchtaskListData{
- TasksketchList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetPreLinkList(ctx context.Context, request http_model.PreLinkListRequest) (*http_model.GetprelinkListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.Tasklinkinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.Tasklinkinfo{
- Task: taskinfo,
- DDl: task.CurBreakAt,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetprelinkListData{
- TasklinkList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetLinkList(ctx context.Context, request http_model.TaskLinklistRequest) (*http_model.GettasklinkListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.TaskLinkinfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- City: task.City,
- }
- 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
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.TaskLinkinfo{
- Task: taskinfo,
- LinkId: linkinfo.LinkID,
- SubmitAt: linkinfo.SubmitAt,
- AgreeAt: linkinfo.AgreeAt,
- LinkUrl: linkinfo.LinkUrl,
- PhotoUrl: linkinfo.PhotoUrl,
- Operator: boperator,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GettasklinkListData{
- TaskLinklist: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetPreDataList(ctx context.Context, request http_model.PreDataListRequest) (*http_model.GetPreDataListData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.Taskdatainfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- City: task.City,
- }
- response := &http_model.Taskdatainfo{
- Task: taskinfo,
- DDl: task.CurBreakAt,
- }
- taskInfoPointers = append(taskInfoPointers, response)
- }
- return &http_model.GetPreDataListData{
- TaskdataList: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
- func GetDataList(ctx context.Context, request http_model.TaskDatalistRequest) (*http_model.GetTaskDatalistData, error) {
- db := GetReadDB(ctx)
- var projecrtaskinfo []gorm_model.YoungeeTaskInfo
- query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
- // 计算总数
- 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 err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
- return nil, err
- }
- taskInfoPointers := make([]*http_model.TaskDatainfo, 0, len(projecrtaskinfo))
- for _, task := range projecrtaskinfo {
- nickname, headurl := getTalentinfo(db, task.TalentID)
- Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
- taskinfo := &http_model.TaskInfo{
- TaskId: task.TaskID,
- ProjectId: task.ProjectID,
- TalentId: task.TalentID,
- FansNum: task.FansNum,
- DraftFee: task.DraftFee,
- Voteavg: task.VoteAvg,
- FeeFrom: task.FeeForm,
- TaskStage: task.TaskStage,
- Commentavg: task.CommitAvg,
- CurrentDefaultType: task.CurDefaultType,
- From: determineFrom(task.SupplierId, task.SupplierStatus),
- SType: stype(db, task.SupplierId, task.SupplierStatus),
- NickName: nickname,
- HeadUrl: headurl,
- ISCoop: Iscoop,
- City: task.City,
- }
- 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
- }
- boperator := getBOperator(db, task.BOperator, task.BOperatorType)
- response := &http_model.TaskDatainfo{
- Task: taskinfo,
- DataId: datainfo.DataID,
- SubmitAt: datainfo.SubmitAt,
- AgreeAt: datainfo.AgreeAt,
- 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.GetTaskDatalistData{
- TaskDatalist: taskInfoPointers,
- Total: conv.MustString(total, ""),
- }, nil
- }
|