project_task.go 43 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381
  1. package db
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "github.com/issue9/conv"
  7. "github.com/sirupsen/logrus"
  8. "gorm.io/gorm"
  9. "strconv"
  10. "strings"
  11. "time"
  12. "youngee_b_api/model/gorm_model"
  13. "youngee_b_api/model/http_model"
  14. )
  15. func GetRecruittime(ctx context.Context, request http_model.GetRecruitTimeRequest) (*http_model.GetRecruitTimeResponse, error) {
  16. db := GetReadDB(ctx)
  17. var info gorm_model.ProjectInfo
  18. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).Find(&info).Error
  19. var result http_model.GetRecruitTimeResponse
  20. if err != nil {
  21. return &result, err
  22. }
  23. result.RecruitTime = info.RecruitDdl.Format("2006-01-02 15:04:05")
  24. return &result, nil
  25. }
  26. func GetTalentstatusCount(db *gorm.DB, request http_model.GetTalentstatusNumRequest, status int) (int64, error) {
  27. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ?", request.ProjectId, status)
  28. // 计算总数
  29. var total int64
  30. if err := query.Count(&total).Error; err != nil {
  31. return 0, err
  32. }
  33. return total, nil
  34. }
  35. func GetTalentstatusNum(db *gorm.DB, request http_model.GetTalentstatusCountRequest, status int) (int64, error) {
  36. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, status)
  37. var total int64
  38. if err := query.Count(&total).Error; err != nil {
  39. return 0, err
  40. }
  41. return total, nil
  42. }
  43. func GetTalentstatusCountNum(ctx context.Context, request http_model.GetTalentstatusCountRequest) (*http_model.GetTalentstatusCountResponse, error) {
  44. db := GetReadDB(ctx)
  45. var unoperatenum, agreetalentnum, refusetalentnum int64
  46. unoperatenum, _ = GetTalentstatusNum(db, request, request.TaskStage)
  47. agreetalentnum, _ = GetTalentstatusNum(db, request, request.TaskStage+1)
  48. refusetalentnum, _ = GetTalentstatusNum(db, request, request.TaskStage+2)
  49. count := &http_model.GetTalentstatusCountResponse{
  50. UnoperateTalentnum: unoperatenum,
  51. AgreeTalentnum: agreetalentnum,
  52. RefuseTalentnum: refusetalentnum,
  53. }
  54. return count, nil
  55. }
  56. func GetTalentstatusNumCount(ctx context.Context, request http_model.GetTalentstatusNumRequest) (*http_model.GetTalentStatusNumResponse, error) {
  57. db := GetReadDB(ctx)
  58. var unoperatenum, agreetalentnum, refusetalentnum int64
  59. unoperatenum, _ = GetTalentstatusCount(db, request, 1)
  60. agreetalentnum, _ = GetTalentstatusCount(db, request, 2)
  61. refusetalentnum, _ = GetTalentstatusCount(db, request, 3)
  62. count := &http_model.GetTalentStatusNumResponse{
  63. UnoperateTalentnum: unoperatenum,
  64. AgreeTalentnum: agreetalentnum,
  65. RefuseTalentnum: refusetalentnum,
  66. }
  67. return count, nil
  68. }
  69. func GetProjecttaskList(ctx context.Context, request http_model.GetTaskListRequest) (*http_model.GetTaskListData, error) {
  70. db := GetReadDB(ctx)
  71. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  72. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ?", request.ProjectId, request.CoopType)
  73. var projectinfo gorm_model.ProjectInfo
  74. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  75. if err != nil {
  76. return &http_model.GetTaskListData{}, err
  77. }
  78. // 构建查询条件
  79. if request.FeeFrom != nil {
  80. query = query.Where("fee_form = ?", request.FeeFrom)
  81. }
  82. if request.Type != nil {
  83. switch *request.Type {
  84. case 1:
  85. query = query.Where("supplier_id = 0 AND supplier_status = 0") //服务商未选
  86. case 2:
  87. query = query.Where("supplier_id !=0 AND supplier_status = 2") //服务商已选
  88. }
  89. } else {
  90. query = query.Where("(supplier_id = 0 AND supplier_status = 0) OR (supplier_id != 0 AND supplier_status = 2)")
  91. }
  92. if request.Others != "" {
  93. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  94. }
  95. // 计算总数
  96. var total int64
  97. if err := query.Count(&total).Error; err != nil {
  98. return nil, err
  99. }
  100. // 添加分页逻辑
  101. pageSize := request.PageSize
  102. if pageSize == 0 {
  103. pageSize = 10
  104. }
  105. pageNum := request.PageNum
  106. if pageNum == 0 {
  107. pageNum = 1
  108. }
  109. offset := (pageNum - 1) * pageSize
  110. // 处理多字段排序逻辑
  111. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  112. for i := 0; i < len(request.SortField); i++ {
  113. field := request.SortField[i]
  114. order := request.SortOrder[i]
  115. switch field {
  116. case "fansnum":
  117. if order == "asc" {
  118. query = query.Order("fans_num asc")
  119. } else {
  120. query = query.Order("fans_num desc")
  121. }
  122. case "voteavg":
  123. if order == "asc" {
  124. query = query.Order("vote_avg asc")
  125. } else {
  126. query = query.Order("vote_avg desc")
  127. }
  128. case "commentavg":
  129. if order == "asc" {
  130. query = query.Order("commit_avg asc")
  131. } else {
  132. query = query.Order("commit_avg desc")
  133. }
  134. case "collectnum":
  135. if order == "asc" {
  136. query = query.Order("collect_num asc")
  137. } else {
  138. query = query.Order("collect_num desc")
  139. }
  140. }
  141. }
  142. } else {
  143. // Default sorting if no valid sort parameters
  144. query = query.Order("task_status asc").Order("task_stage asc")
  145. }
  146. // 执行分页查询
  147. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  148. return nil, err
  149. }
  150. taskInfoPointers := make([]*http_model.TaskInfo, 0, len(projecrtaskinfo))
  151. for _, task := range projecrtaskinfo {
  152. var Time time.Time
  153. switch request.CoopType {
  154. case 1:
  155. Time = task.CreateDate
  156. case 2:
  157. Time = task.SelectDate //2 signedtime
  158. case 3:
  159. Time = task.CompleteDate
  160. }
  161. var listtime, stageintro string
  162. switch task.TaskStage {
  163. case 4:
  164. {
  165. listtime = task.SelectDate.Format("2006-01-02 15:04:05")
  166. stageintro = "执行开始时间"
  167. }
  168. case 5:
  169. {
  170. listtime = task.DeliveryDate.Format("2006-01-02 15:04:05")
  171. stageintro = "商家寄样时间"
  172. }
  173. case 9:
  174. {
  175. listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
  176. stageintro = "初稿上传截止时间"
  177. }
  178. case 10:
  179. {
  180. var sketchinfo gorm_model.YounggeeSketchInfo
  181. err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&sketchinfo).Error
  182. if err != nil {
  183. return nil, err
  184. }
  185. listtime = sketchinfo.CreateAt.Format("2006-01-02 15:04:05")
  186. stageintro = "达人上传初稿时间" //
  187. }
  188. case 11:
  189. {
  190. listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
  191. stageintro = " 链接上传截止时间"
  192. }
  193. case 12:
  194. {
  195. var linkinfo gorm_model.YounggeeLinkInfo
  196. err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&linkinfo).Error
  197. if err != nil {
  198. return nil, err
  199. }
  200. listtime = linkinfo.CreateAt.Format("2006-01-02 15:04:05")
  201. stageintro = "达人上传链接时间"
  202. }
  203. case 13:
  204. {
  205. listtime = task.CurBreakAt.Format("2006-01-02 15:04:05")
  206. stageintro = "数据待传截止时间"
  207. }
  208. case 14:
  209. {
  210. var datainfo gorm_model.YounggeeDataInfo
  211. err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&datainfo).Error
  212. if err != nil {
  213. return nil, err
  214. }
  215. var auto_task_finish gorm_model.InfoAutoTask
  216. db.Model(&gorm_model.InfoAutoTask{}).Where("enterprise_id = ?", request.EnterPriseId).First(&auto_task_finish)
  217. // 将小时数转换为 Duration 后相加
  218. resultTime := datainfo.CreateAt.Add(time.Duration(auto_task_finish.TaskFinishAuto) * time.Hour)
  219. listtime = resultTime.Format("2006-01-02 15:04:05")
  220. stageintro = "品牌最晚结算时间"
  221. }
  222. case 15:
  223. {
  224. listtime = task.CompleteDate.Format("2006-01-02 15:04:05")
  225. stageintro = "品牌结算时间"
  226. }
  227. case 16:
  228. {
  229. listtime = task.CancelTime.Format("2006-01-02 15:04:05")
  230. stageintro = "解约时间"
  231. }
  232. }
  233. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  234. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  235. Iscoop := getIscoop(db, task.TalentID, request.EnterPriseId)
  236. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  237. response := &http_model.TaskInfo{
  238. TaskId: task.TaskID,
  239. ProjectId: task.ProjectID,
  240. TalentId: task.TalentID,
  241. FansNum: fansnum,
  242. DraftFee: task.DraftFee,
  243. Voteavg: voteavg,
  244. FeeFrom: task.FeeForm,
  245. TaskStage: task.TaskStage,
  246. Commentavg: 0, //暂时获取不到
  247. CollectNum: 0,
  248. CurrentDefaultType: task.CurDefaultType,
  249. From: determineFrom(task.SupplierId, task.SupplierStatus),
  250. SType: s_type,
  251. SName: s_name,
  252. Boperator: boperator,
  253. CreateAt: Time.Format("2006-01-02 15:04:05"),
  254. NickName: nickname,
  255. Gender: gender,
  256. HeadUrl: headurl,
  257. ISCoop: Iscoop,
  258. Sprojectid: task.SprojectId,
  259. City: task.City,
  260. ListTime: listtime,
  261. StageIntro: stageintro,
  262. }
  263. taskInfoPointers = append(taskInfoPointers, response)
  264. }
  265. return &http_model.GetTaskListData{
  266. TaskList: taskInfoPointers,
  267. Total: conv.MustString(total, ""),
  268. }, nil
  269. }
  270. func getBOperator(db *gorm.DB, bOperatorID string, bOperatorType int) string {
  271. if bOperatorID == "" {
  272. return ""
  273. }
  274. if bOperatorType == 1 {
  275. var boperatorinfo gorm_model.Enterprise
  276. if err := db.Where(gorm_model.Enterprise{EnterpriseID: bOperatorID}).First(&boperatorinfo).Error; err != nil {
  277. return ""
  278. }
  279. return boperatorinfo.BusinessName
  280. }
  281. if bOperatorType == 2 {
  282. var bsubaccount gorm_model.YounggeeSubAccount
  283. if err := db.Where(gorm_model.YounggeeSubAccount{SubAccountId: conv.MustInt(bOperatorID)}).First(&bsubaccount).Error; err != nil {
  284. return ""
  285. }
  286. return bsubaccount.SubAccountName
  287. }
  288. if bOperatorType == 3 {
  289. return "管理后台"
  290. }
  291. return ""
  292. }
  293. func getTalentinfo(db *gorm.DB, OpenId string) (string, string, string, int, int) {
  294. var talentinfo gorm_model.PlatformKuaishouUserInfo
  295. if err := db.Where(gorm_model.PlatformKuaishouUserInfo{OpenId: OpenId}).First(&talentinfo).Error; err != nil {
  296. return "", "", "", 0, 0
  297. }
  298. fan, _ := conv.Int(talentinfo.Fan)
  299. return talentinfo.NickName, talentinfo.HeadUri, talentinfo.Gender, fan, talentinfo.LikeNum
  300. }
  301. func getIscoop(db *gorm.DB, talentid string, enterpriseid string) int {
  302. var etcoop gorm_model.EnterpriseTalentCooperate
  303. if err := db.Where(gorm_model.EnterpriseTalentCooperate{TalentId: talentid, EnterpriseId: enterpriseid}).First(&etcoop).Error; err != nil {
  304. return 0
  305. }
  306. return 1
  307. }
  308. func determineFrom(supplierID, supplierStatus int) int {
  309. if supplierID != 0 && supplierStatus == 2 {
  310. return 2
  311. }
  312. return 1
  313. }
  314. func stype(db *gorm.DB, supplierID, supplierStatus int, projecttype int64) (int, string) {
  315. if projecttype == 1 {
  316. if supplierID != 0 && supplierStatus == 2 {
  317. var supplierinfo gorm_model.Supplier
  318. err := db.Model(gorm_model.Supplier{}).Where("supplier_id = ?", supplierID).First(&supplierinfo).Error
  319. if err != nil {
  320. return 0, "公海"
  321. }
  322. return supplierinfo.SupplierType, supplierinfo.SupplierName
  323. }
  324. return 0, "公海"
  325. } else {
  326. if supplierID != 0 && supplierStatus == 2 {
  327. var supplierinfo gorm_model.Supplier
  328. err := db.Model(gorm_model.Supplier{}).Where("supplier_id = ?", supplierID).First(&supplierinfo).Error
  329. if err != nil {
  330. return 0, "私域"
  331. }
  332. return supplierinfo.SupplierType, supplierinfo.SupplierName
  333. }
  334. return 0, "私域"
  335. }
  336. }
  337. func PassProTaskCoop(ctx context.Context, projectId string, taskIds []string, operatorid string, operatetype int, Isspecial int, Enterpriseid string, req http_model.PassproTaskCoopRequest) (bool, error) {
  338. db := GetReadDB(ctx)
  339. var count int64
  340. err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
  341. if err != nil {
  342. return false, err
  343. }
  344. if int64(len(taskIds)) == 0 || count != int64(len(taskIds)) {
  345. return false, errors.New("任务id有误")
  346. }
  347. // 2. 查询任务对应达人id(用于生成达人消息)
  348. var talentIds []string
  349. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
  350. if err != nil {
  351. return false, err
  352. }
  353. // 3. 查询任务对应project名称(用于生成达人消息)
  354. var project gorm_model.ProjectInfo
  355. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&project).Error
  356. if err != nil {
  357. return false, err
  358. }
  359. var taskstages int
  360. if project.ProjectForm == 1 {
  361. taskstages = 4
  362. } else {
  363. taskstages = 9
  364. }
  365. nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
  366. if Isspecial == 1 {
  367. toolList := strings.Split(project.Tools, ",")
  368. for _, tool := range toolList {
  369. // 将 tool 转换为整数
  370. if t, err := strconv.Atoi(tool); err == nil && t >= 0 && t <= len(nextstage) {
  371. nextstage[t-1] = 1
  372. }
  373. }
  374. fmt.Println(nextstage)
  375. for i := 1; i < len(nextstage); i++ { // 从位置1开始
  376. if nextstage[i] == 1 {
  377. switch i {
  378. case 1:
  379. taskstages = 4
  380. case 2:
  381. taskstages = 9
  382. case 3:
  383. taskstages = 11
  384. case 4:
  385. taskstages = 13
  386. case 5:
  387. taskstages = 15
  388. }
  389. break
  390. }
  391. }
  392. }
  393. var opid string
  394. var optype int
  395. switch operatetype {
  396. case 3:
  397. {
  398. opid = Enterpriseid
  399. optype = 1
  400. }
  401. case 4:
  402. {
  403. opid = operatorid
  404. optype = 2
  405. }
  406. }
  407. //4.根据达人来源改变状态、阶段、数目
  408. var tasks []gorm_model.YoungeeTaskInfo
  409. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Find(&tasks).Error
  410. if err != nil {
  411. return false, err
  412. }
  413. err = db.Transaction(func(tx *gorm.DB) error {
  414. // 2. 修改任务状态和任务阶段
  415. for _, taskInfo := range tasks {
  416. if taskInfo.SupplierId != 0 && taskInfo.SupplierStatus == 2 { //服务商招募数量+1
  417. updateData := gorm_model.YoungeeTaskInfo{
  418. TaskStage: taskstages,
  419. TaskStatus: 2,
  420. UpdateAt: time.Now(),
  421. BOperator: opid,
  422. BOperatorType: optype,
  423. SelectDate: time.Now(),
  424. }
  425. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  426. if err != nil {
  427. return err
  428. }
  429. //project招募数+1,预估成本+服务商价格
  430. var projectinfo gorm_model.ProjectInfo
  431. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
  432. if err != nil {
  433. return err
  434. }
  435. projectinfo.RecruitNum += 1
  436. projectinfo.NeedPay += taskInfo.DraftFee + taskInfo.ServiceCharge
  437. //recruit_strategy total_offer
  438. err = db.Model(gorm_model.RecruitStrategy{}).Where("recruit_strategy_id = ?", taskInfo.StrategyID).Update("total_offer", gorm.Expr("total_offer + ?", taskInfo.DraftFee+taskInfo.ServiceCharge)).Error
  439. if err != nil {
  440. return nil
  441. }
  442. err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
  443. if err != nil {
  444. return err
  445. }
  446. //服务商s_project已招募人数+1
  447. var sprojectinfo gorm_model.YounggeeSProjectInfo
  448. err = db.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", taskInfo.SprojectId).Find(&sprojectinfo).Error
  449. if err != nil {
  450. return err
  451. }
  452. sprojectinfo.RecruitNum += 1
  453. sprojectinfo.ServiceCharge += taskInfo.ServiceCharge
  454. sprojectinfo.EstimateSupportFee += taskInfo.SupportFee
  455. sprojectinfo.EstimateDraftFee += taskInfo.DraftFee
  456. err = tx.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", taskInfo.SprojectId).Updates(sprojectinfo).Error
  457. if err != nil {
  458. return err
  459. }
  460. }
  461. if project.ProjectForm != 1 { //非签收截止时间
  462. var draft_default gorm_model.InfoAutoTask
  463. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskID).Find(&draft_default).Error
  464. if err != nil {
  465. return err
  466. }
  467. duration := time.Duration(draft_default.DraftDefault) * time.Hour
  468. upbreakdate := gorm_model.YoungeeTaskInfo{
  469. CurBreakAt: time.Now().Add(duration),
  470. }
  471. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? ", taskInfo.TaskID).Updates(upbreakdate).Error
  472. if err != nil {
  473. return err
  474. }
  475. }
  476. if taskInfo.SupplierId == 0 && taskInfo.SupplierStatus == 0 { //来源于公海
  477. updateData := gorm_model.YoungeeTaskInfo{
  478. TaskStage: taskstages,
  479. TaskStatus: 2,
  480. UpdateAt: time.Now(),
  481. BOperator: opid,
  482. BOperatorType: optype,
  483. SelectDate: time.Now(),
  484. }
  485. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  486. if err != nil {
  487. return err
  488. }
  489. //project招募数+1,成本
  490. var projectinfo gorm_model.ProjectInfo
  491. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
  492. if err != nil {
  493. return err
  494. }
  495. projectinfo.RecruitNum += 1
  496. projectinfo.NeedPay += taskInfo.DraftFee + taskInfo.ServiceCharge
  497. err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
  498. if err != nil {
  499. return err
  500. }
  501. }
  502. }
  503. // 4. 生成达人消息
  504. for _, talendId := range talentIds {
  505. err = CreateMessage(ctx, 1, 1, talendId, project.ProjectName)
  506. if err != nil {
  507. return err
  508. }
  509. }
  510. // 5. tasklog表插入信息
  511. for _, taskid := range taskIds {
  512. err = CreateTaskLog(ctx, taskid, "已确认合作")
  513. if err != nil {
  514. return err
  515. }
  516. }
  517. // 返回 nil 提交事务
  518. return nil
  519. })
  520. if err != nil {
  521. return false, err
  522. }
  523. return true, nil
  524. }
  525. func RefuseproTaskCoop(ctx context.Context, taskIds []string, operatorid string, operatetype int, Enterpriseid string) (bool, error) {
  526. db := GetWriteDB(ctx)
  527. // 1. 校验
  528. var count int64
  529. err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
  530. if err != nil {
  531. return false, err
  532. }
  533. if count != int64(len(taskIds)) {
  534. return false, errors.New("任务id有误")
  535. }
  536. var opid string
  537. var optype int
  538. switch operatetype {
  539. case 3:
  540. {
  541. opid = Enterpriseid
  542. optype = 1
  543. }
  544. case 4:
  545. {
  546. opid = operatorid
  547. optype = 2
  548. }
  549. }
  550. // 2. 查询任务对应达人id(用于生成达人消息)
  551. var talentIds []string
  552. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
  553. if err != nil {
  554. return false, err
  555. }
  556. var tasks []gorm_model.YoungeeTaskInfo
  557. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Find(&tasks).Error
  558. fmt.Println(taskIds, talentIds)
  559. err = db.Transaction(func(tx *gorm.DB) error {
  560. // 2. 修改任务状态和任务阶段
  561. for _, taskInfo := range tasks {
  562. updateData := gorm_model.YoungeeTaskInfo{
  563. TaskStage: 3,
  564. TaskStatus: 3,
  565. UpdateAt: time.Now(),
  566. BOperator: opid,
  567. BOperatorType: optype,
  568. CompleteDate: time.Now(),
  569. }
  570. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  571. }
  572. if err != nil {
  573. return err
  574. }
  575. // 5. tasklog表插入信息
  576. for _, taskid := range taskIds {
  577. err = CreateTaskLog(ctx, taskid, "已拒绝合作")
  578. if err != nil {
  579. return err
  580. }
  581. }
  582. // 返回 nil 提交事务
  583. return nil
  584. })
  585. if err != nil {
  586. return false, err
  587. }
  588. return true, nil
  589. }
  590. func GetPreSketchList(ctx context.Context, request http_model.PreSketchListRequest) (*http_model.GetSketchTaskListData, error) {
  591. db := GetReadDB(ctx)
  592. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  593. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
  594. var projectinfo gorm_model.ProjectInfo
  595. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  596. if err != nil {
  597. return &http_model.GetSketchTaskListData{}, err
  598. }
  599. if request.Others != "" {
  600. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  601. }
  602. // 计算总数
  603. var total int64
  604. if err := query.Count(&total).Error; err != nil {
  605. return nil, err
  606. }
  607. // 添加分页逻辑
  608. pageSize := request.PageSize
  609. if pageSize == 0 {
  610. pageSize = 10
  611. }
  612. pageNum := request.PageNum
  613. if pageNum == 0 {
  614. pageNum = 1
  615. }
  616. offset := (pageNum - 1) * pageSize
  617. // 执行分页查询
  618. // 处理多字段排序逻辑
  619. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  620. for i := 0; i < len(request.SortField); i++ {
  621. field := request.SortField[i]
  622. order := request.SortOrder[i]
  623. switch field {
  624. case "fansnum":
  625. if order == "asc" {
  626. query = query.Order("fans_num asc")
  627. } else {
  628. query = query.Order("fans_num desc")
  629. }
  630. case "voteavg":
  631. if order == "asc" {
  632. query = query.Order("vote_avg asc")
  633. } else {
  634. query = query.Order("vote_avg desc")
  635. }
  636. case "commentavg":
  637. if order == "asc" {
  638. query = query.Order("commit_avg asc")
  639. } else {
  640. query = query.Order("commit_avg desc")
  641. }
  642. case "collectnum":
  643. if order == "asc" {
  644. query = query.Order("view_num asc")
  645. } else {
  646. query = query.Order("view_num desc")
  647. }
  648. }
  649. }
  650. } else {
  651. // Default sorting if no valid sort parameters
  652. query = query.Order("task_status asc").Order("task_stage asc")
  653. }
  654. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  655. return nil, err
  656. }
  657. taskInfoPointers := make([]*http_model.TasksketchInfo, 0, len(projecrtaskinfo))
  658. for _, task := range projecrtaskinfo {
  659. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  660. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  661. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  662. taskinfo := &http_model.TaskInfo{
  663. TaskId: task.TaskID,
  664. ProjectId: task.ProjectID,
  665. TalentId: task.TalentID,
  666. FansNum: fansnum,
  667. DraftFee: task.DraftFee,
  668. Voteavg: voteavg,
  669. FeeFrom: task.FeeForm,
  670. TaskStage: task.TaskStage,
  671. Commentavg: 0,
  672. CollectNum: 0,
  673. CurrentDefaultType: task.CurDefaultType,
  674. From: determineFrom(task.SupplierId, task.SupplierStatus),
  675. SType: s_type,
  676. SName: s_name,
  677. NickName: nickname,
  678. Gender: gender,
  679. HeadUrl: headurl,
  680. ISCoop: Iscoop,
  681. City: task.City,
  682. }
  683. response := &http_model.TasksketchInfo{
  684. Task: taskinfo,
  685. DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
  686. }
  687. taskInfoPointers = append(taskInfoPointers, response)
  688. }
  689. return &http_model.GetSketchTaskListData{
  690. TasksketchList: taskInfoPointers,
  691. Total: conv.MustString(total, ""),
  692. }, nil
  693. }
  694. func GetSketchList(ctx context.Context, request http_model.TasksketchlistRequest) (*http_model.GetsketchtaskListData, error) {
  695. db := GetReadDB(ctx)
  696. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  697. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? ", request.ProjectId)
  698. switch request.ScriptStatus {
  699. case 10:
  700. query = query.Where("task_stage = 10")
  701. case 11:
  702. query = query.Where(" task_stage >= 11")
  703. }
  704. var projectinfo gorm_model.ProjectInfo
  705. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  706. if err != nil {
  707. return &http_model.GetsketchtaskListData{}, err
  708. }
  709. if request.Others != "" {
  710. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  711. }
  712. // 计算总数
  713. var total int64
  714. if err := query.Count(&total).Error; err != nil {
  715. return nil, err
  716. }
  717. // 添加分页逻辑
  718. pageSize := request.PageSize
  719. if pageSize == 0 {
  720. pageSize = 10
  721. }
  722. pageNum := request.PageNum
  723. if pageNum == 0 {
  724. pageNum = 1
  725. }
  726. offset := (pageNum - 1) * pageSize
  727. // 执行分页查询
  728. // 处理多字段排序逻辑
  729. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  730. for i := 0; i < len(request.SortField); i++ {
  731. field := request.SortField[i]
  732. order := request.SortOrder[i]
  733. switch field {
  734. case "fansnum":
  735. if order == "asc" {
  736. query = query.Order("fans_num asc")
  737. } else {
  738. query = query.Order("fans_num desc")
  739. }
  740. case "voteavg":
  741. if order == "asc" {
  742. query = query.Order("vote_avg asc")
  743. } else {
  744. query = query.Order("vote_avg desc")
  745. }
  746. case "commentavg":
  747. if order == "asc" {
  748. query = query.Order("commit_avg asc")
  749. } else {
  750. query = query.Order("commit_avg desc")
  751. }
  752. case "collectnum":
  753. if order == "asc" {
  754. query = query.Order("view_num asc")
  755. } else {
  756. query = query.Order("view_num desc")
  757. }
  758. }
  759. }
  760. } else {
  761. // Default sorting if no valid sort parameters
  762. query = query.Order("task_status asc").Order("task_stage asc")
  763. }
  764. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  765. return nil, err
  766. }
  767. taskInfoPointers := make([]*http_model.Tasksketchinfo, 0, len(projecrtaskinfo))
  768. for _, task := range projecrtaskinfo {
  769. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  770. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  771. taskinfo := &http_model.TaskInfo{
  772. TaskId: task.TaskID,
  773. ProjectId: task.ProjectID,
  774. TalentId: task.TalentID,
  775. FansNum: fansnum,
  776. DraftFee: task.DraftFee,
  777. Voteavg: voteavg,
  778. FeeFrom: task.FeeForm,
  779. TaskStage: task.TaskStage,
  780. Commentavg: 0,
  781. CollectNum: 0,
  782. CurrentDefaultType: task.CurDefaultType,
  783. From: determineFrom(task.SupplierId, task.SupplierStatus),
  784. SType: s_type,
  785. SName: s_name,
  786. NickName: nickname,
  787. Gender: gender,
  788. HeadUrl: headurl,
  789. City: task.City,
  790. }
  791. var sketchinfo gorm_model.YounggeeSketchInfo
  792. switch request.ScriptStatus {
  793. case 10:
  794. err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskID).First(&sketchinfo).Error
  795. if err != nil {
  796. return nil, err
  797. }
  798. case 11:
  799. err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_ok = 1", task.TaskID).First(&sketchinfo).Error
  800. if err != nil {
  801. return nil, err
  802. }
  803. }
  804. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  805. response := &http_model.Tasksketchinfo{
  806. Task: taskinfo,
  807. SketchId: sketchinfo.SketchID,
  808. SubmitAt: sketchinfo.CreateAt.Format("2006-01-02 15:04:05"),
  809. AgreeAt: sketchinfo.AgreeAt.Format("2006-01-02 15:04:05"),
  810. SketchType: sketchinfo.Type,
  811. Operator: boperator,
  812. }
  813. taskInfoPointers = append(taskInfoPointers, response)
  814. }
  815. return &http_model.GetsketchtaskListData{
  816. TasksketchList: taskInfoPointers,
  817. Total: conv.MustString(total, ""),
  818. }, nil
  819. }
  820. func GetPreLinkList(ctx context.Context, request http_model.PreLinkListRequest) (*http_model.GetprelinkListData, error) {
  821. db := GetReadDB(ctx)
  822. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  823. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
  824. var projectinfo gorm_model.ProjectInfo
  825. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  826. if err != nil {
  827. return &http_model.GetprelinkListData{}, err
  828. }
  829. if request.Others != "" {
  830. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  831. }
  832. // 计算总数
  833. var total int64
  834. if err := query.Count(&total).Error; err != nil {
  835. return nil, err
  836. }
  837. // 添加分页逻辑
  838. pageSize := request.PageSize
  839. if pageSize == 0 {
  840. pageSize = 10
  841. }
  842. pageNum := request.PageNum
  843. if pageNum == 0 {
  844. pageNum = 1
  845. }
  846. offset := (pageNum - 1) * pageSize
  847. // 执行分页查询
  848. // 处理多字段排序逻辑
  849. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  850. for i := 0; i < len(request.SortField); i++ {
  851. field := request.SortField[i]
  852. order := request.SortOrder[i]
  853. switch field {
  854. case "fansnum":
  855. if order == "asc" {
  856. query = query.Order("fans_num asc")
  857. } else {
  858. query = query.Order("fans_num desc")
  859. }
  860. case "voteavg":
  861. if order == "asc" {
  862. query = query.Order("vote_avg asc")
  863. } else {
  864. query = query.Order("vote_avg desc")
  865. }
  866. case "commentavg":
  867. if order == "asc" {
  868. query = query.Order("commit_avg asc")
  869. } else {
  870. query = query.Order("commit_avg desc")
  871. }
  872. case "collectnum":
  873. if order == "asc" {
  874. query = query.Order("view_num asc")
  875. } else {
  876. query = query.Order("view_num desc")
  877. }
  878. }
  879. }
  880. } else {
  881. // Default sorting if no valid sort parameters
  882. query = query.Order("task_status asc").Order("task_stage asc")
  883. }
  884. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  885. return nil, err
  886. }
  887. taskInfoPointers := make([]*http_model.Tasklinkinfo, 0, len(projecrtaskinfo))
  888. for _, task := range projecrtaskinfo {
  889. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  890. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  891. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  892. taskinfo := &http_model.TaskInfo{
  893. TaskId: task.TaskID,
  894. ProjectId: task.ProjectID,
  895. TalentId: task.TalentID,
  896. FansNum: fansnum,
  897. DraftFee: task.DraftFee,
  898. Voteavg: voteavg,
  899. FeeFrom: task.FeeForm,
  900. TaskStage: task.TaskStage,
  901. Commentavg: 0,
  902. CollectNum: 0,
  903. CurrentDefaultType: task.CurDefaultType,
  904. From: determineFrom(task.SupplierId, task.SupplierStatus),
  905. SType: s_type,
  906. SName: s_name,
  907. NickName: nickname,
  908. Gender: gender,
  909. HeadUrl: headurl,
  910. ISCoop: Iscoop,
  911. City: task.City,
  912. }
  913. response := &http_model.Tasklinkinfo{
  914. Task: taskinfo,
  915. DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
  916. }
  917. taskInfoPointers = append(taskInfoPointers, response)
  918. }
  919. return &http_model.GetprelinkListData{
  920. TasklinkList: taskInfoPointers,
  921. Total: conv.MustString(total, ""),
  922. }, nil
  923. }
  924. func GetLinkList(ctx context.Context, request http_model.TaskLinklistRequest) (*http_model.GettasklinkListData, error) {
  925. db := GetReadDB(ctx)
  926. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  927. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? ", request.ProjectId)
  928. switch request.LinkStatus {
  929. case "12":
  930. query = query.Where("task_stage = 12")
  931. case "13":
  932. query = query.Where(" task_stage >= 13")
  933. }
  934. var projectinfo gorm_model.ProjectInfo
  935. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  936. if err != nil {
  937. return &http_model.GettasklinkListData{}, err
  938. }
  939. if request.Others != "" {
  940. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  941. }
  942. // 计算总数
  943. var total int64
  944. if err := query.Count(&total).Error; err != nil {
  945. return nil, err
  946. }
  947. // 添加分页逻辑
  948. pageSize := request.PageSize
  949. if pageSize == 0 {
  950. pageSize = 10
  951. }
  952. pageNum := request.PageNum
  953. if pageNum == 0 {
  954. pageNum = 1
  955. }
  956. offset := (pageNum - 1) * pageSize
  957. // 执行分页查询
  958. // 处理多字段排序逻辑
  959. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  960. for i := 0; i < len(request.SortField); i++ {
  961. field := request.SortField[i]
  962. order := request.SortOrder[i]
  963. switch field {
  964. case "fansnum":
  965. if order == "asc" {
  966. query = query.Order("fans_num asc")
  967. } else {
  968. query = query.Order("fans_num desc")
  969. }
  970. case "voteavg":
  971. if order == "asc" {
  972. query = query.Order("vote_avg asc")
  973. } else {
  974. query = query.Order("vote_avg desc")
  975. }
  976. case "commentavg":
  977. if order == "asc" {
  978. query = query.Order("commit_avg asc")
  979. } else {
  980. query = query.Order("commit_avg desc")
  981. }
  982. case "collectnum":
  983. if order == "asc" {
  984. query = query.Order("view_num asc")
  985. } else {
  986. query = query.Order("view_num desc")
  987. }
  988. }
  989. }
  990. } else {
  991. // Default sorting if no valid sort parameters
  992. query = query.Order("task_status asc").Order("task_stage asc")
  993. }
  994. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  995. return nil, err
  996. }
  997. taskInfoPointers := make([]*http_model.TaskLinkinfo, 0, len(projecrtaskinfo))
  998. for _, task := range projecrtaskinfo {
  999. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  1000. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  1001. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  1002. taskinfo := &http_model.TaskInfo{
  1003. TaskId: task.TaskID,
  1004. ProjectId: task.ProjectID,
  1005. TalentId: task.TalentID,
  1006. FansNum: fansnum,
  1007. DraftFee: task.DraftFee,
  1008. Voteavg: voteavg,
  1009. FeeFrom: task.FeeForm,
  1010. TaskStage: task.TaskStage,
  1011. Commentavg: 0,
  1012. CollectNum: 0,
  1013. CurrentDefaultType: task.CurDefaultType,
  1014. From: determineFrom(task.SupplierId, task.SupplierStatus),
  1015. SType: s_type,
  1016. SName: s_name,
  1017. NickName: nickname,
  1018. Gender: gender,
  1019. HeadUrl: headurl,
  1020. ISCoop: Iscoop,
  1021. City: task.City,
  1022. }
  1023. var linkinfo gorm_model.YounggeeLinkInfo
  1024. switch request.LinkStatus {
  1025. case "12":
  1026. err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskID).First(&linkinfo).Error
  1027. if err != nil {
  1028. return nil, err
  1029. }
  1030. case "13":
  1031. err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_ok = 1", task.TaskID).First(&linkinfo).Error
  1032. if err != nil {
  1033. return nil, err
  1034. }
  1035. }
  1036. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  1037. response := &http_model.TaskLinkinfo{
  1038. Task: taskinfo,
  1039. LinkId: linkinfo.LinkID,
  1040. SubmitAt: linkinfo.CreateAt.Format("2006-01-02 15:04:05"),
  1041. AgreeAt: linkinfo.AgreeAt.Format("2006-01-02 15:04:05"),
  1042. LinkUrl: linkinfo.LinkUrl,
  1043. PhotoUrl: linkinfo.PhotoUrl,
  1044. Operator: boperator,
  1045. }
  1046. taskInfoPointers = append(taskInfoPointers, response)
  1047. }
  1048. return &http_model.GettasklinkListData{
  1049. TaskLinklist: taskInfoPointers,
  1050. Total: conv.MustString(total, ""),
  1051. }, nil
  1052. }
  1053. func GetPreDataList(ctx context.Context, request http_model.PreDataListRequest) (*http_model.GetPreDataListData, error) {
  1054. db := GetReadDB(ctx)
  1055. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  1056. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
  1057. var projectinfo gorm_model.ProjectInfo
  1058. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  1059. if err != nil {
  1060. return &http_model.GetPreDataListData{}, err
  1061. }
  1062. if request.Others != "" {
  1063. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  1064. }
  1065. // 计算总数
  1066. var total int64
  1067. if err := query.Count(&total).Error; err != nil {
  1068. return nil, err
  1069. }
  1070. // 添加分页逻辑
  1071. pageSize := request.PageSize
  1072. if pageSize == 0 {
  1073. pageSize = 10
  1074. }
  1075. pageNum := request.PageNum
  1076. if pageNum == 0 {
  1077. pageNum = 1
  1078. }
  1079. offset := (pageNum - 1) * pageSize
  1080. // 执行分页查询
  1081. // 处理多字段排序逻辑
  1082. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  1083. for i := 0; i < len(request.SortField); i++ {
  1084. field := request.SortField[i]
  1085. order := request.SortOrder[i]
  1086. switch field {
  1087. case "fansnum":
  1088. if order == "asc" {
  1089. query = query.Order("fans_num asc")
  1090. } else {
  1091. query = query.Order("fans_num desc")
  1092. }
  1093. case "voteavg":
  1094. if order == "asc" {
  1095. query = query.Order("vote_avg asc")
  1096. } else {
  1097. query = query.Order("vote_avg desc")
  1098. }
  1099. case "commentavg":
  1100. if order == "asc" {
  1101. query = query.Order("commit_avg asc")
  1102. } else {
  1103. query = query.Order("commit_avg desc")
  1104. }
  1105. case "collectnum":
  1106. if order == "asc" {
  1107. query = query.Order("view_num asc")
  1108. } else {
  1109. query = query.Order("view_num desc")
  1110. }
  1111. }
  1112. }
  1113. } else {
  1114. // Default sorting if no valid sort parameters
  1115. query = query.Order("task_status asc").Order("task_stage asc")
  1116. }
  1117. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  1118. return nil, err
  1119. }
  1120. taskInfoPointers := make([]*http_model.Taskdatainfo, 0, len(projecrtaskinfo))
  1121. for _, task := range projecrtaskinfo {
  1122. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  1123. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  1124. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  1125. taskinfo := &http_model.TaskInfo{
  1126. TaskId: task.TaskID,
  1127. ProjectId: task.ProjectID,
  1128. TalentId: task.TalentID,
  1129. FansNum: fansnum,
  1130. DraftFee: task.DraftFee,
  1131. Voteavg: voteavg,
  1132. FeeFrom: task.FeeForm,
  1133. TaskStage: task.TaskStage,
  1134. Commentavg: 0,
  1135. CollectNum: 0,
  1136. CurrentDefaultType: task.CurDefaultType,
  1137. From: determineFrom(task.SupplierId, task.SupplierStatus),
  1138. SType: s_type,
  1139. SName: s_name,
  1140. NickName: nickname,
  1141. Gender: gender,
  1142. HeadUrl: headurl,
  1143. ISCoop: Iscoop,
  1144. City: task.City,
  1145. }
  1146. response := &http_model.Taskdatainfo{
  1147. Task: taskinfo,
  1148. DDl: task.CurBreakAt.Format("2006-01-02 15:04:05"),
  1149. }
  1150. taskInfoPointers = append(taskInfoPointers, response)
  1151. }
  1152. return &http_model.GetPreDataListData{
  1153. TaskdataList: taskInfoPointers,
  1154. Total: conv.MustString(total, ""),
  1155. }, nil
  1156. }
  1157. func GetDataList(ctx context.Context, request http_model.TaskDatalistRequest) (*http_model.GetTaskDatalistData, error) {
  1158. db := GetReadDB(ctx)
  1159. var projectinfo gorm_model.ProjectInfo
  1160. err := db.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", request.ProjectId).First(&projectinfo).Error
  1161. if err != nil {
  1162. return &http_model.GetTaskDatalistData{}, err
  1163. }
  1164. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  1165. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
  1166. if request.Others != "" {
  1167. query = query.Where("talent_name LIKE ? OR s_operate_name LIKE ?", "%"+request.Others+"%", "%"+request.Others+"%")
  1168. }
  1169. // 计算总数
  1170. fmt.Println(projecrtaskinfo)
  1171. var total int64
  1172. if err := query.Count(&total).Error; err != nil {
  1173. return nil, err
  1174. }
  1175. // 添加分页逻辑
  1176. pageSize := request.PageSize
  1177. if pageSize == 0 {
  1178. pageSize = 10
  1179. }
  1180. pageNum := request.PageNum
  1181. if pageNum == 0 {
  1182. pageNum = 1
  1183. }
  1184. offset := (pageNum - 1) * pageSize
  1185. // 执行分页查询
  1186. // 处理多字段排序逻辑
  1187. if len(request.SortField) > 0 && len(request.SortOrder) > 0 && len(request.SortField) == len(request.SortOrder) {
  1188. for i := 0; i < len(request.SortField); i++ {
  1189. field := request.SortField[i]
  1190. order := request.SortOrder[i]
  1191. switch field {
  1192. case "fansnum":
  1193. if order == "asc" {
  1194. query = query.Order("fans_num asc")
  1195. } else {
  1196. query = query.Order("fans_num desc")
  1197. }
  1198. case "voteavg":
  1199. if order == "asc" {
  1200. query = query.Order("vote_avg asc")
  1201. } else {
  1202. query = query.Order("vote_avg desc")
  1203. }
  1204. case "commentavg":
  1205. if order == "asc" {
  1206. query = query.Order("commit_avg asc")
  1207. } else {
  1208. query = query.Order("commit_avg desc")
  1209. }
  1210. case "collectnum":
  1211. if order == "asc" {
  1212. query = query.Order("view_num asc")
  1213. } else {
  1214. query = query.Order("view_num desc")
  1215. }
  1216. }
  1217. }
  1218. } else {
  1219. // Default sorting if no valid sort parameters
  1220. query = query.Order("task_status asc").Order("task_stage asc")
  1221. }
  1222. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  1223. return nil, err
  1224. }
  1225. taskInfoPointers := make([]*http_model.TaskDatainfo, 0, len(projecrtaskinfo))
  1226. for _, task := range projecrtaskinfo {
  1227. nickname, headurl, gender, fansnum, voteavg := getTalentinfo(db, task.OpenId)
  1228. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  1229. s_type, s_name := stype(db, task.SupplierId, task.SupplierStatus, projectinfo.ProjectType)
  1230. taskinfo := &http_model.TaskInfo{
  1231. TaskId: task.TaskID,
  1232. ProjectId: task.ProjectID,
  1233. TalentId: task.TalentID,
  1234. FansNum: fansnum,
  1235. DraftFee: task.DraftFee,
  1236. Voteavg: voteavg,
  1237. FeeFrom: task.FeeForm,
  1238. TaskStage: task.TaskStage,
  1239. Commentavg: 0,
  1240. CollectNum: 0,
  1241. CurrentDefaultType: task.CurDefaultType,
  1242. From: determineFrom(task.SupplierId, task.SupplierStatus),
  1243. SType: s_type,
  1244. SName: s_name,
  1245. NickName: nickname,
  1246. Gender: gender,
  1247. HeadUrl: headurl,
  1248. ISCoop: Iscoop,
  1249. City: task.City,
  1250. }
  1251. var datainfo gorm_model.YounggeeDataInfo
  1252. switch request.DataStatus {
  1253. case "14":
  1254. err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 0 ", task.TaskID).First(&datainfo).Error
  1255. if err != nil {
  1256. return nil, err
  1257. }
  1258. case "15":
  1259. err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_ok = 1", task.TaskID).First(&datainfo).Error
  1260. if err != nil {
  1261. return nil, err
  1262. }
  1263. }
  1264. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  1265. response := &http_model.TaskDatainfo{
  1266. Task: taskinfo,
  1267. DataId: datainfo.DataID,
  1268. SubmitAt: datainfo.CreateAt.Format("2006-01-02 15:04:05"),
  1269. AgreeAt: datainfo.AgreeAt.Format("2006-01-02 15:04:05"),
  1270. PhotoUrl: datainfo.PhotoUrl,
  1271. PlayNumber: datainfo.PlayNumber,
  1272. LikeNumber: datainfo.LikeNumber,
  1273. CollectNumber: datainfo.CollectNumber,
  1274. CommentNumber: datainfo.CommentNumber,
  1275. SettleAmount: task.SettleAmount,
  1276. Breakrate: task.SketchBreakRate + task.LinkBreakRate + task.DataBreakRate,
  1277. Operator: boperator,
  1278. }
  1279. taskInfoPointers = append(taskInfoPointers, response)
  1280. }
  1281. return &http_model.GetTaskDatalistData{
  1282. TaskDatalist: taskInfoPointers,
  1283. Total: conv.MustString(total, ""),
  1284. }, nil
  1285. }
  1286. // GetProjectTaskIdList 子任务ID列表
  1287. func GetProjectTaskIdList(ctx context.Context, projectId string) ([]string, int64, error) {
  1288. db := GetReadDB(ctx)
  1289. // Get count first
  1290. var total int64
  1291. if err := db.Model(gorm_model.YoungeeTaskInfo{}).
  1292. Where("link_status = ? and project_id = ?", 5, projectId).
  1293. Count(&total).
  1294. Error; err != nil {
  1295. logrus.WithContext(ctx).Errorf("[GetProjectTaskIdList] error counting task, err:%+v", err)
  1296. return nil, 0, err
  1297. }
  1298. // Query only project_id with the given status
  1299. var taskIds []string
  1300. err := db.Debug().
  1301. Model(gorm_model.YoungeeTaskInfo{}).
  1302. Where("link_status = ? and project_id = ?", 5, projectId).
  1303. Order("update_at desc").
  1304. Pluck("task_id", &taskIds).
  1305. Error
  1306. if err != nil {
  1307. logrus.WithContext(ctx).Errorf("[GetProjectTaskIdList] error querying taskIDs, err:%+v", err)
  1308. return nil, 0, err
  1309. }
  1310. return taskIds, total, nil
  1311. }