project_task.go 44 KB

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