project_task.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766
  1. package db
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "github.com/issue9/conv"
  7. "gorm.io/gorm"
  8. "strconv"
  9. "strings"
  10. "time"
  11. "youngee_b_api/model/gorm_model"
  12. "youngee_b_api/model/http_model"
  13. )
  14. func GetProjecttaskList(ctx context.Context, request http_model.GetTaskListRequest) (*http_model.GetTaskListData, error) {
  15. db := GetReadDB(ctx)
  16. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  17. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ?", request.ProjectId, request.CoopType)
  18. // 构建查询条件
  19. if request.FeeFrom != nil {
  20. query = query.Where("fee_form = ?", request.FeeFrom)
  21. }
  22. if request.Type != nil {
  23. switch *request.Type {
  24. case 1:
  25. query = query.Where("supplier_id = 0 AND supplier_status = 0")
  26. case 2:
  27. query = query.Where("supplier_id !=0 AND supplier_status = 2")
  28. }
  29. }
  30. // 计算总数
  31. var total int64
  32. if err := query.Count(&total).Error; err != nil {
  33. return nil, err
  34. }
  35. // 添加分页逻辑
  36. pageSize := request.PageSize
  37. if pageSize == 0 {
  38. pageSize = 10
  39. }
  40. pageNum := request.PageNum
  41. if pageNum == 0 {
  42. pageNum = 1
  43. }
  44. offset := (pageNum - 1) * pageSize
  45. // 执行分页查询
  46. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  47. return nil, err
  48. }
  49. taskInfoPointers := make([]*http_model.TaskInfo, 0, len(projecrtaskinfo))
  50. for _, task := range projecrtaskinfo {
  51. var Time time.Time
  52. switch request.CoopType {
  53. case 1:
  54. Time = task.CreateDate
  55. case 2:
  56. Time = task.SelectDate //2 signedtime
  57. case 3:
  58. Time = task.CompleteDate
  59. }
  60. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  61. nickname, headurl := getTalentinfo(db, task.TalentID)
  62. Iscoop := getIscoop(db, task.TalentID, request.EnterPriseId)
  63. response := &http_model.TaskInfo{
  64. TaskId: task.TaskID,
  65. ProjectId: task.ProjectID,
  66. TalentId: task.TalentID,
  67. FansNum: task.FansNum,
  68. DraftFee: task.DraftFee,
  69. Voteavg: task.VoteAvg,
  70. FeeFrom: task.FeeForm,
  71. TaskStage: task.TaskStage,
  72. Commentavg: task.CommitAvg,
  73. CurrentDefaultType: task.CurDefaultType,
  74. From: determineFrom(task.SupplierId, task.SupplierStatus),
  75. SType: stype(db, task.SupplierId, task.SupplierStatus),
  76. Boperator: boperator,
  77. CreateAt: Time,
  78. NickName: nickname,
  79. HeadUrl: headurl,
  80. ISCoop: Iscoop,
  81. Sprojectid: task.SprojectId,
  82. City: task.City,
  83. }
  84. taskInfoPointers = append(taskInfoPointers, response)
  85. }
  86. return &http_model.GetTaskListData{
  87. TaskList: taskInfoPointers,
  88. Total: conv.MustString(total, ""),
  89. }, nil
  90. }
  91. func getBOperator(db *gorm.DB, bOperatorID string, bOperatorType int) string {
  92. if bOperatorID == "" {
  93. return ""
  94. }
  95. if bOperatorType == 1 {
  96. var boperatorinfo gorm_model.Enterprise
  97. if err := db.Where(gorm_model.Enterprise{EnterpriseID: bOperatorID}).First(&boperatorinfo).Error; err != nil {
  98. return ""
  99. }
  100. return boperatorinfo.BusinessName
  101. }
  102. if bOperatorType == 2 {
  103. var bsubaccount gorm_model.YounggeeSubAccount
  104. if err := db.Where(gorm_model.YounggeeSubAccount{SubAccountId: conv.MustInt(bOperatorID)}).First(&bsubaccount).Error; err != nil {
  105. return ""
  106. }
  107. return bsubaccount.SubAccountName
  108. }
  109. if bOperatorType == 3 {
  110. return "管理后台"
  111. }
  112. return ""
  113. }
  114. func getTalentinfo(db *gorm.DB, talentID string) (string, string) {
  115. var talentinfo gorm_model.PlatformKuaishouUserInfo
  116. if err := db.Where(gorm_model.PlatformKuaishouUserInfo{TalentId: talentID}).First(&talentinfo).Error; err != nil {
  117. return "", ""
  118. }
  119. return talentinfo.NickName, talentinfo.HeadUri
  120. }
  121. func getIscoop(db *gorm.DB, talentid string, enterpriseid string) int {
  122. var etcoop gorm_model.EnterpriseTalentCooperate
  123. if err := db.Where(gorm_model.EnterpriseTalentCooperate{TalentId: talentid, EnterpriseId: enterpriseid}).First(&etcoop).Error; err != nil {
  124. return 0
  125. }
  126. return 1
  127. }
  128. func determineFrom(supplierID, supplierStatus int) int {
  129. if supplierID != 0 && supplierStatus == 2 {
  130. return 2
  131. }
  132. return 1
  133. }
  134. func stype(db *gorm.DB, supplierID, supplierStatus int) int {
  135. if supplierID != 0 && supplierStatus == 2 {
  136. var supplierinfo gorm_model.Supplier
  137. err := db.Model(gorm_model.Supplier{}).Where("supplier_id = ?", supplierID).First(&supplierinfo).Error
  138. if err != nil {
  139. return 0
  140. }
  141. return supplierinfo.SupplierType
  142. }
  143. return 0
  144. }
  145. func PassProTaskCoop(ctx context.Context, projectId string, taskIds []string, operatorid string, operatetype int, Isspecial int, req http_model.PassproTaskCoopRequest) (bool, error) {
  146. db := GetReadDB(ctx)
  147. var count int64
  148. fmt.Println("task_ids: ", taskIds)
  149. err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
  150. fmt.Println("count: ", count)
  151. if err != nil {
  152. return false, err
  153. }
  154. if int64(len(taskIds)) == 0 || count != int64(len(taskIds)) {
  155. return false, errors.New("任务id有误")
  156. }
  157. // 2. 查询任务对应达人id(用于生成达人消息)
  158. var talentIds []string
  159. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
  160. if err != nil {
  161. return false, err
  162. }
  163. // 3. 查询任务对应project名称(用于生成达人消息)
  164. var project gorm_model.ProjectInfo
  165. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&project).Error
  166. if err != nil {
  167. return false, err
  168. }
  169. var taskstages int
  170. if project.ProjectForm == 1 {
  171. taskstages = 4
  172. } else {
  173. taskstages = 9
  174. }
  175. nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
  176. if Isspecial == 1 {
  177. toolList := strings.Split(project.Tools, ",")
  178. for _, tool := range toolList {
  179. // 将 tool 转换为整数
  180. if t, err := strconv.Atoi(tool); err == nil && t >= 0 && t <= len(nextstage) {
  181. nextstage[t-1] = 1
  182. }
  183. }
  184. fmt.Println(nextstage)
  185. for i := 1; i < len(nextstage); i++ { // 从位置1开始
  186. if nextstage[i] == 1 {
  187. switch i {
  188. case 1:
  189. taskstages = 4
  190. case 2:
  191. taskstages = 9
  192. case 3:
  193. taskstages = 11
  194. case 4:
  195. taskstages = 13
  196. case 5:
  197. taskstages = 15
  198. }
  199. break
  200. }
  201. }
  202. }
  203. //4.根据达人来源改变状态、阶段、数目
  204. var tasks []gorm_model.YoungeeTaskInfo
  205. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Find(&tasks).Error
  206. if err != nil {
  207. return false, err
  208. }
  209. err = db.Transaction(func(tx *gorm.DB) error {
  210. // 2. 修改任务状态和任务阶段
  211. for _, taskInfo := range tasks {
  212. if taskInfo.SupplierId != 0 && taskInfo.SupplierStatus == 2 { //服务商招募数量+1
  213. updateData := gorm_model.YoungeeTaskInfo{
  214. TaskStage: taskstages,
  215. TaskStatus: 2,
  216. UpdateAt: time.Now(),
  217. BOperator: operatorid,
  218. BOperatorType: operatetype,
  219. SelectDate: time.Now(),
  220. }
  221. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  222. if err != nil {
  223. return err
  224. }
  225. //project招募数+1,预估成本+服务商价格
  226. var projectinfo gorm_model.ProjectInfo
  227. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
  228. if err != nil {
  229. return err
  230. }
  231. projectinfo.RecruitNum += 1
  232. projectinfo.EstimatedCost += taskInfo.SupportFee
  233. err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
  234. if err != nil {
  235. return err
  236. }
  237. //服务商s_project已招募人数+1
  238. var sprojectinfo gorm_model.YounggeeSProjectInfo
  239. err = db.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", req.SProjectId).Find(&sprojectinfo).Error
  240. if err != nil {
  241. return err
  242. }
  243. sprojectinfo.RecruitNum += 1
  244. fmt.Println("......", projectinfo.RecruitNum)
  245. err = tx.Model(gorm_model.YounggeeSProjectInfo{}).Where(" s_project_id =?", req.SProjectId).Updates(sprojectinfo).Error
  246. if err != nil {
  247. return err
  248. }
  249. }
  250. if project.ProjectForm != 1 { //非签收截止时间
  251. var draft_default gorm_model.InfoAutoTask
  252. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskID).Find(&draft_default).Error
  253. if err != nil {
  254. return err
  255. }
  256. duration := time.Duration(draft_default.DraftDefault) * time.Hour
  257. upbreakdate := gorm_model.YoungeeTaskInfo{
  258. CurBreakAt: time.Now().Add(duration),
  259. }
  260. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? ", taskInfo.TaskID).Updates(upbreakdate).Error
  261. if err != nil {
  262. return err
  263. }
  264. }
  265. if taskInfo.SupplierId == 0 && taskInfo.SupplierStatus == 0 { //来源于公海
  266. updateData := gorm_model.YoungeeTaskInfo{
  267. TaskStage: taskstages,
  268. TaskStatus: 2,
  269. UpdateAt: time.Now(),
  270. BOperator: operatorid,
  271. BOperatorType: operatetype,
  272. SelectDate: time.Now(),
  273. }
  274. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  275. if err != nil {
  276. return err
  277. }
  278. //project招募数+1
  279. var projectinfo gorm_model.ProjectInfo
  280. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Find(&projectinfo).Error
  281. if err != nil {
  282. return err
  283. }
  284. projectinfo.RecruitNum += 1
  285. err = tx.Model(gorm_model.ProjectInfo{}).Where("project_id = ? ", projectId).Updates(projectinfo).Error
  286. if err != nil {
  287. return err
  288. }
  289. }
  290. }
  291. // 4. 生成达人消息
  292. for _, talendId := range talentIds {
  293. err = CreateMessage(ctx, 1, 1, talendId, project.ProjectName)
  294. if err != nil {
  295. return err
  296. }
  297. }
  298. // 返回 nil 提交事务
  299. return nil
  300. })
  301. if err != nil {
  302. return false, err
  303. }
  304. return true, nil
  305. }
  306. func RefuseproTaskCoop(ctx context.Context, taskIds []string, operatorid string, operatetype int) (bool, error) {
  307. db := GetWriteDB(ctx)
  308. // 1. 校验
  309. var count int64
  310. err := db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Count(&count).Error
  311. if err != nil {
  312. return false, err
  313. }
  314. if count != int64(len(taskIds)) {
  315. return false, errors.New("任务id有误")
  316. }
  317. // 2. 查询任务对应达人id(用于生成达人消息)
  318. var talentIds []string
  319. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ?", taskIds).Select("talent_id").Find(&talentIds).Error
  320. if err != nil {
  321. return false, err
  322. }
  323. var tasks []gorm_model.YoungeeTaskInfo
  324. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id IN ? AND task_stage = 1", taskIds).Find(&tasks).Error
  325. err = db.Transaction(func(tx *gorm.DB) error {
  326. // 2. 修改任务状态和任务阶段
  327. for _, taskInfo := range tasks {
  328. updateData := gorm_model.YoungeeTaskInfo{
  329. TaskStage: 3,
  330. TaskStatus: 3,
  331. UpdateAt: time.Now(),
  332. BOperator: operatorid,
  333. BOperatorType: operatetype,
  334. CompleteDate: time.Now(),
  335. }
  336. err = tx.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ? AND task_stage = 1", taskInfo.TaskID).Updates(updateData).Error
  337. }
  338. if err != nil {
  339. return err
  340. }
  341. // 返回 nil 提交事务
  342. return nil
  343. })
  344. if err != nil {
  345. return false, err
  346. }
  347. return true, nil
  348. }
  349. func GetPreSketchList(ctx context.Context, request http_model.PreSketchListRequest) (*http_model.GetSketchTaskListData, error) {
  350. db := GetReadDB(ctx)
  351. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  352. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
  353. // 计算总数
  354. var total int64
  355. if err := query.Count(&total).Error; err != nil {
  356. return nil, err
  357. }
  358. // 添加分页逻辑
  359. pageSize := request.PageSize
  360. if pageSize == 0 {
  361. pageSize = 10
  362. }
  363. pageNum := request.PageNum
  364. if pageNum == 0 {
  365. pageNum = 1
  366. }
  367. offset := (pageNum - 1) * pageSize
  368. // 执行分页查询
  369. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  370. return nil, err
  371. }
  372. taskInfoPointers := make([]*http_model.TasksketchInfo, 0, len(projecrtaskinfo))
  373. for _, task := range projecrtaskinfo {
  374. nickname, headurl := getTalentinfo(db, task.TalentID)
  375. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  376. taskinfo := &http_model.TaskInfo{
  377. TaskId: task.TaskID,
  378. ProjectId: task.ProjectID,
  379. TalentId: task.TalentID,
  380. FansNum: task.FansNum,
  381. DraftFee: task.DraftFee,
  382. Voteavg: task.VoteAvg,
  383. FeeFrom: task.FeeForm,
  384. TaskStage: task.TaskStage,
  385. Commentavg: task.CommitAvg,
  386. CurrentDefaultType: task.CurDefaultType,
  387. From: determineFrom(task.SupplierId, task.SupplierStatus),
  388. SType: stype(db, task.SupplierId, task.SupplierStatus),
  389. NickName: nickname,
  390. HeadUrl: headurl,
  391. ISCoop: Iscoop,
  392. City: task.City,
  393. }
  394. response := &http_model.TasksketchInfo{
  395. Task: taskinfo,
  396. DDl: task.CurBreakAt,
  397. }
  398. taskInfoPointers = append(taskInfoPointers, response)
  399. }
  400. return &http_model.GetSketchTaskListData{
  401. TasksketchList: taskInfoPointers,
  402. Total: conv.MustString(total, ""),
  403. }, nil
  404. }
  405. func GetSketchList(ctx context.Context, request http_model.TasksketchlistRequest) (*http_model.GetsketchtaskListData, error) {
  406. db := GetReadDB(ctx)
  407. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  408. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.ScriptStatus)
  409. // 计算总数
  410. var total int64
  411. if err := query.Count(&total).Error; err != nil {
  412. return nil, err
  413. }
  414. // 添加分页逻辑
  415. pageSize := request.PageSize
  416. if pageSize == 0 {
  417. pageSize = 10
  418. }
  419. pageNum := request.PageNum
  420. if pageNum == 0 {
  421. pageNum = 1
  422. }
  423. offset := (pageNum - 1) * pageSize
  424. // 执行分页查询
  425. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  426. return nil, err
  427. }
  428. taskInfoPointers := make([]*http_model.Tasksketchinfo, 0, len(projecrtaskinfo))
  429. for _, task := range projecrtaskinfo {
  430. nickname, headurl := getTalentinfo(db, task.TalentID)
  431. taskinfo := &http_model.TaskInfo{
  432. TaskId: task.TaskID,
  433. ProjectId: task.ProjectID,
  434. TalentId: task.TalentID,
  435. FansNum: task.FansNum,
  436. DraftFee: task.DraftFee,
  437. Voteavg: task.VoteAvg,
  438. FeeFrom: task.FeeForm,
  439. TaskStage: task.TaskStage,
  440. Commentavg: task.CommitAvg,
  441. CurrentDefaultType: task.CurDefaultType,
  442. From: determineFrom(task.SupplierId, task.SupplierStatus),
  443. SType: stype(db, task.SupplierId, task.SupplierStatus),
  444. NickName: nickname,
  445. HeadUrl: headurl,
  446. City: task.City,
  447. }
  448. var sketchinfo gorm_model.YounggeeSketchInfo
  449. err := db.Model(&gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&sketchinfo).Error
  450. if err != nil {
  451. return nil, err
  452. }
  453. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  454. response := &http_model.Tasksketchinfo{
  455. Task: taskinfo,
  456. SketchId: sketchinfo.SketchID,
  457. SubmitAt: sketchinfo.SubmitAt,
  458. AgreeAt: sketchinfo.AgreeAt,
  459. Operator: boperator,
  460. }
  461. taskInfoPointers = append(taskInfoPointers, response)
  462. }
  463. return &http_model.GetsketchtaskListData{
  464. TasksketchList: taskInfoPointers,
  465. Total: conv.MustString(total, ""),
  466. }, nil
  467. }
  468. func GetPreLinkList(ctx context.Context, request http_model.PreLinkListRequest) (*http_model.GetprelinkListData, error) {
  469. db := GetReadDB(ctx)
  470. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  471. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
  472. // 计算总数
  473. var total int64
  474. if err := query.Count(&total).Error; err != nil {
  475. return nil, err
  476. }
  477. // 添加分页逻辑
  478. pageSize := request.PageSize
  479. if pageSize == 0 {
  480. pageSize = 10
  481. }
  482. pageNum := request.PageNum
  483. if pageNum == 0 {
  484. pageNum = 1
  485. }
  486. offset := (pageNum - 1) * pageSize
  487. // 执行分页查询
  488. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  489. return nil, err
  490. }
  491. taskInfoPointers := make([]*http_model.Tasklinkinfo, 0, len(projecrtaskinfo))
  492. for _, task := range projecrtaskinfo {
  493. nickname, headurl := getTalentinfo(db, task.TalentID)
  494. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  495. taskinfo := &http_model.TaskInfo{
  496. TaskId: task.TaskID,
  497. ProjectId: task.ProjectID,
  498. TalentId: task.TalentID,
  499. FansNum: task.FansNum,
  500. DraftFee: task.DraftFee,
  501. Voteavg: task.VoteAvg,
  502. FeeFrom: task.FeeForm,
  503. TaskStage: task.TaskStage,
  504. Commentavg: task.CommitAvg,
  505. CurrentDefaultType: task.CurDefaultType,
  506. From: determineFrom(task.SupplierId, task.SupplierStatus),
  507. SType: stype(db, task.SupplierId, task.SupplierStatus),
  508. NickName: nickname,
  509. HeadUrl: headurl,
  510. ISCoop: Iscoop,
  511. City: task.City,
  512. }
  513. response := &http_model.Tasklinkinfo{
  514. Task: taskinfo,
  515. DDl: task.CurBreakAt,
  516. }
  517. taskInfoPointers = append(taskInfoPointers, response)
  518. }
  519. return &http_model.GetprelinkListData{
  520. TasklinkList: taskInfoPointers,
  521. Total: conv.MustString(total, ""),
  522. }, nil
  523. }
  524. func GetLinkList(ctx context.Context, request http_model.TaskLinklistRequest) (*http_model.GettasklinkListData, error) {
  525. db := GetReadDB(ctx)
  526. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  527. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.LinkStatus)
  528. // 计算总数
  529. var total int64
  530. if err := query.Count(&total).Error; err != nil {
  531. return nil, err
  532. }
  533. // 添加分页逻辑
  534. pageSize := request.PageSize
  535. if pageSize == 0 {
  536. pageSize = 10
  537. }
  538. pageNum := request.PageNum
  539. if pageNum == 0 {
  540. pageNum = 1
  541. }
  542. offset := (pageNum - 1) * pageSize
  543. // 执行分页查询
  544. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  545. return nil, err
  546. }
  547. taskInfoPointers := make([]*http_model.TaskLinkinfo, 0, len(projecrtaskinfo))
  548. for _, task := range projecrtaskinfo {
  549. nickname, headurl := getTalentinfo(db, task.TalentID)
  550. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  551. taskinfo := &http_model.TaskInfo{
  552. TaskId: task.TaskID,
  553. ProjectId: task.ProjectID,
  554. TalentId: task.TalentID,
  555. FansNum: task.FansNum,
  556. DraftFee: task.DraftFee,
  557. Voteavg: task.VoteAvg,
  558. FeeFrom: task.FeeForm,
  559. TaskStage: task.TaskStage,
  560. Commentavg: task.CommitAvg,
  561. CurrentDefaultType: task.CurDefaultType,
  562. From: determineFrom(task.SupplierId, task.SupplierStatus),
  563. SType: stype(db, task.SupplierId, task.SupplierStatus),
  564. NickName: nickname,
  565. HeadUrl: headurl,
  566. ISCoop: Iscoop,
  567. City: task.City,
  568. }
  569. var linkinfo gorm_model.YounggeeLinkInfo
  570. err := db.Model(&gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&linkinfo).Error
  571. if err != nil {
  572. return nil, err
  573. }
  574. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  575. response := &http_model.TaskLinkinfo{
  576. Task: taskinfo,
  577. LinkId: linkinfo.LinkID,
  578. SubmitAt: linkinfo.SubmitAt,
  579. AgreeAt: linkinfo.AgreeAt,
  580. LinkUrl: linkinfo.LinkUrl,
  581. PhotoUrl: linkinfo.PhotoUrl,
  582. Operator: boperator,
  583. }
  584. taskInfoPointers = append(taskInfoPointers, response)
  585. }
  586. return &http_model.GettasklinkListData{
  587. TaskLinklist: taskInfoPointers,
  588. Total: conv.MustString(total, ""),
  589. }, nil
  590. }
  591. func GetPreDataList(ctx context.Context, request http_model.PreDataListRequest) (*http_model.GetPreDataListData, error) {
  592. db := GetReadDB(ctx)
  593. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  594. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
  595. // 计算总数
  596. var total int64
  597. if err := query.Count(&total).Error; err != nil {
  598. return nil, err
  599. }
  600. // 添加分页逻辑
  601. pageSize := request.PageSize
  602. if pageSize == 0 {
  603. pageSize = 10
  604. }
  605. pageNum := request.PageNum
  606. if pageNum == 0 {
  607. pageNum = 1
  608. }
  609. offset := (pageNum - 1) * pageSize
  610. // 执行分页查询
  611. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  612. return nil, err
  613. }
  614. taskInfoPointers := make([]*http_model.Taskdatainfo, 0, len(projecrtaskinfo))
  615. for _, task := range projecrtaskinfo {
  616. nickname, headurl := getTalentinfo(db, task.TalentID)
  617. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  618. taskinfo := &http_model.TaskInfo{
  619. TaskId: task.TaskID,
  620. ProjectId: task.ProjectID,
  621. TalentId: task.TalentID,
  622. FansNum: task.FansNum,
  623. DraftFee: task.DraftFee,
  624. Voteavg: task.VoteAvg,
  625. FeeFrom: task.FeeForm,
  626. TaskStage: task.TaskStage,
  627. Commentavg: task.CommitAvg,
  628. CurrentDefaultType: task.CurDefaultType,
  629. From: determineFrom(task.SupplierId, task.SupplierStatus),
  630. SType: stype(db, task.SupplierId, task.SupplierStatus),
  631. NickName: nickname,
  632. HeadUrl: headurl,
  633. ISCoop: Iscoop,
  634. City: task.City,
  635. }
  636. response := &http_model.Taskdatainfo{
  637. Task: taskinfo,
  638. DDl: task.CurBreakAt,
  639. }
  640. taskInfoPointers = append(taskInfoPointers, response)
  641. }
  642. return &http_model.GetPreDataListData{
  643. TaskdataList: taskInfoPointers,
  644. Total: conv.MustString(total, ""),
  645. }, nil
  646. }
  647. func GetDataList(ctx context.Context, request http_model.TaskDatalistRequest) (*http_model.GetTaskDatalistData, error) {
  648. db := GetReadDB(ctx)
  649. var projecrtaskinfo []gorm_model.YoungeeTaskInfo
  650. query := db.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_stage = ?", request.ProjectId, request.DataStatus)
  651. // 计算总数
  652. var total int64
  653. if err := query.Count(&total).Error; err != nil {
  654. return nil, err
  655. }
  656. // 添加分页逻辑
  657. pageSize := request.PageSize
  658. if pageSize == 0 {
  659. pageSize = 10
  660. }
  661. pageNum := request.PageNum
  662. if pageNum == 0 {
  663. pageNum = 1
  664. }
  665. offset := (pageNum - 1) * pageSize
  666. // 执行分页查询
  667. if err := query.Offset(offset).Limit(pageSize).Find(&projecrtaskinfo).Error; err != nil {
  668. return nil, err
  669. }
  670. taskInfoPointers := make([]*http_model.TaskDatainfo, 0, len(projecrtaskinfo))
  671. for _, task := range projecrtaskinfo {
  672. nickname, headurl := getTalentinfo(db, task.TalentID)
  673. Iscoop := getIscoop(db, task.TalentID, request.EnterpriseId)
  674. taskinfo := &http_model.TaskInfo{
  675. TaskId: task.TaskID,
  676. ProjectId: task.ProjectID,
  677. TalentId: task.TalentID,
  678. FansNum: task.FansNum,
  679. DraftFee: task.DraftFee,
  680. Voteavg: task.VoteAvg,
  681. FeeFrom: task.FeeForm,
  682. TaskStage: task.TaskStage,
  683. Commentavg: task.CommitAvg,
  684. CurrentDefaultType: task.CurDefaultType,
  685. From: determineFrom(task.SupplierId, task.SupplierStatus),
  686. SType: stype(db, task.SupplierId, task.SupplierStatus),
  687. NickName: nickname,
  688. HeadUrl: headurl,
  689. ISCoop: Iscoop,
  690. City: task.City,
  691. }
  692. var datainfo gorm_model.YounggeeDataInfo
  693. err := db.Model(&gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 0", task.TaskID).First(&datainfo).Error
  694. if err != nil {
  695. return nil, err
  696. }
  697. boperator := getBOperator(db, task.BOperator, task.BOperatorType)
  698. response := &http_model.TaskDatainfo{
  699. Task: taskinfo,
  700. DataId: datainfo.DataID,
  701. SubmitAt: datainfo.SubmitAt,
  702. AgreeAt: datainfo.AgreeAt,
  703. PhotoUrl: datainfo.PhotoUrl,
  704. PlayNumber: datainfo.PlayNumber,
  705. LikeNumber: datainfo.LikeNumber,
  706. CollectNumber: datainfo.CollectNumber,
  707. CommentNumber: datainfo.CommentNumber,
  708. SettleAmount: task.SettleAmount,
  709. Breakrate: task.SketchBreakRate + task.LinkBreakRate + task.DataBreakRate,
  710. Operator: boperator,
  711. }
  712. taskInfoPointers = append(taskInfoPointers, response)
  713. }
  714. return &http_model.GetTaskDatalistData{
  715. TaskDatalist: taskInfoPointers,
  716. Total: conv.MustString(total, ""),
  717. }, nil
  718. }