link.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531
  1. package db
  2. import (
  3. "context"
  4. "fmt"
  5. "reflect"
  6. "strconv"
  7. "strings"
  8. "time"
  9. "youngee_m_api/model/common_model"
  10. "youngee_m_api/model/gorm_model"
  11. "youngee_m_api/model/http_model"
  12. "youngee_m_api/pack"
  13. "youngee_m_api/util"
  14. "github.com/caixw/lib.go/conv"
  15. "github.com/sirupsen/logrus"
  16. )
  17. // GetTaskLinkList 查询上传链接的task list
  18. func GetTaskLinkList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.TaskLinkInfo, int64, error) {
  19. db := GetReadDB(ctx)
  20. // 查询Task表信息
  21. db = db.Debug().Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2")
  22. // 根据Project条件过滤
  23. conditionType := reflect.TypeOf(conditions).Elem()
  24. conditionValue := reflect.ValueOf(conditions).Elem()
  25. var platform_nickname string = ""
  26. for i := 0; i < conditionType.NumField(); i++ {
  27. field := conditionType.Field(i)
  28. tag := field.Tag.Get("condition")
  29. value := conditionValue.FieldByName(field.Name)
  30. if tag == "link_status" {
  31. fmt.Printf("link %+v", value.Interface() == int64(0))
  32. if value.Interface() == int64(0) {
  33. db = db.Where("task_stage = 12")
  34. } else {
  35. db = db.Where("task_stage > 12 and task_stage <> 16")
  36. }
  37. continue
  38. } else if !util.IsBlank(value) {
  39. if tag == "platform_nickname" {
  40. platform_nickname = fmt.Sprintf("%v", value.Interface())
  41. continue
  42. } else if tag == "project_id" || tag == "strategy_id" {
  43. db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
  44. }
  45. }
  46. }
  47. var taskInfos []gorm_model.YoungeeTaskInfo
  48. db = db.Model(gorm_model.YoungeeTaskInfo{})
  49. // 查询总数
  50. var totalTask int64
  51. if err := db.Count(&totalTask).Error; err != nil {
  52. logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
  53. return nil, 0, err
  54. }
  55. db.Order("task_id").Find(&taskInfos)
  56. // 查询任务id
  57. var taskIds []string
  58. taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
  59. for _, taskInfo := range taskInfos {
  60. taskIds = append(taskIds, taskInfo.TaskId)
  61. taskMap[taskInfo.TaskId] = taskInfo
  62. }
  63. db1 := GetReadDB(ctx)
  64. db1 = db1.Debug().Model(gorm_model.YounggeeLinkInfo{})
  65. var LinkInfos []gorm_model.YounggeeLinkInfo
  66. db1 = db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id IN ? AND is_submit=? ", taskIds, 1)
  67. if conditions.LinkStatus == int64(0) {
  68. db1 = db1.Where("is_review = 0").Find(&LinkInfos)
  69. } else {
  70. db1 = db1.Where("is_ok = 1").Find(&LinkInfos)
  71. }
  72. LinkMap := make(map[string]gorm_model.YounggeeLinkInfo)
  73. for _, LinkInfo := range LinkInfos {
  74. LinkMap[LinkInfo.TaskID] = LinkInfo
  75. }
  76. // 查询总数
  77. var totalLink int64
  78. if err := db1.Count(&totalLink).Error; err != nil {
  79. logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
  80. return nil, 0, err
  81. }
  82. var misNum int64
  83. if totalLink > totalTask {
  84. misNum = totalLink - totalTask
  85. } else {
  86. misNum = totalTask - totalLink
  87. }
  88. logrus.Println("totalLink,totalTalent,misNum:", totalLink, totalTask, misNum)
  89. // 查询该页数据
  90. limit := pageSize + misNum
  91. offset := pageSize * pageNum // assert pageNum start with 0
  92. err := db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
  93. if err != nil {
  94. logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
  95. return nil, 0, err
  96. }
  97. var TaskLinks []*http_model.TaskLink
  98. var taskLinks []*http_model.TaskLinkInfo
  99. var newTaskLinks []*http_model.TaskLinkInfo
  100. for _, taskId := range taskIds {
  101. TaskLink := new(http_model.TaskLink)
  102. TaskLink.Talent = taskMap[taskId]
  103. TaskLink.Link = LinkMap[taskId]
  104. TaskLinks = append(TaskLinks, TaskLink)
  105. }
  106. taskLinks = pack.TaskLinkToTaskInfo(TaskLinks)
  107. for _, v := range taskLinks {
  108. if platform_nickname == "" {
  109. newTaskLinks = append(newTaskLinks, v)
  110. } else if strings.Contains(v.PlatformNickname, platform_nickname) {
  111. newTaskLinks = append(newTaskLinks, v)
  112. } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
  113. newTaskLinks = append(newTaskLinks, v)
  114. } else {
  115. totalTask--
  116. }
  117. }
  118. return newTaskLinks, totalTask, nil
  119. }
  120. //// LinkOpinion 提交意见
  121. //func LinkOpinion(ctx context.Context, TaskID string, ReviseOpinion string) error {
  122. // db := GetReadDB(ctx)
  123. // err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", TaskID).Updates(map[string]interface{}{"revise_opinion": ReviseOpinion, "reject_at": time.Now(), "is_review": 1}).Error
  124. // if err != nil {
  125. // logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  126. // return err
  127. // }
  128. // err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeTaskInfo{LinkStatus: 3}).Error
  129. // if err != nil {
  130. // logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  131. // return err
  132. // }
  133. // err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeTaskInfo{TaskStage: 11}).Error
  134. // if err != nil {
  135. // logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  136. // return err
  137. // }
  138. // return nil
  139. //}
  140. // LinkOpinion 提交意见
  141. func LinkOpinion(ctx context.Context, TaskID string, ReviseOpinion string, operatorid string, operatetype int, projectid string) error {
  142. db := GetReadDB(ctx)
  143. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", TaskID).Updates(map[string]interface{}{"revise_opinion": ReviseOpinion, "reject_at": time.Now(), "is_review": 1, "b_operator": operatorid, "b_operator_type": operatetype}).Error
  144. if err != nil {
  145. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  146. return err
  147. }
  148. var project gorm_model.ProjectInfo
  149. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectid).Find(&project).Error
  150. if err != nil {
  151. logrus.WithContext(ctx).Errorf("[Sketch db] Find ProjectInfo error,err:%+v", err)
  152. return err
  153. }
  154. var link_default gorm_model.InfoAutoTask
  155. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskID).Find(&link_default).Error
  156. if err != nil {
  157. return err
  158. }
  159. duration := time.Duration(link_default.LinkBreach) * time.Hour
  160. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeTaskInfo{LinkStatus: 3, CurBreakAt: time.Now().Add(duration)}).Error
  161. if err != nil {
  162. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  163. return err
  164. }
  165. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeTaskInfo{TaskStage: 11}).Error
  166. if err != nil {
  167. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  168. return err
  169. }
  170. return nil
  171. }
  172. // LinkOpinion 提交意见
  173. func LocalLinkOpinion(ctx context.Context, TaskID string, ReviseOpinion string, operatorid string, operatetype int, projectid string) error {
  174. db := GetReadDB(ctx)
  175. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? and is_review = 0", TaskID).Updates(map[string]interface{}{"revise_opinion": ReviseOpinion, "reject_at": time.Now(), "is_review": 1, "b_operator": operatorid, "b_operator_type": operatetype}).Error
  176. if err != nil {
  177. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  178. return err
  179. }
  180. var project gorm_model.YounggeeLocalLifeInfo
  181. err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", projectid).Find(&project).Error
  182. if err != nil {
  183. logrus.WithContext(ctx).Errorf("[Sketch db] Find ProjectInfo error,err:%+v", err)
  184. return err
  185. }
  186. var link_default gorm_model.InfoAutoTask
  187. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskId).Find(&link_default).Error
  188. if err != nil {
  189. return err
  190. }
  191. duration := time.Duration(link_default.LinkBreach) * time.Hour
  192. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeLocalTaskInfo{LinkStatus: 3, CurBreakAt: time.Now().Add(duration)}).Error
  193. if err != nil {
  194. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  195. return err
  196. }
  197. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id = ?", TaskID).Updates(gorm_model.YoungeeLocalTaskInfo{TaskStage: 11}).Error
  198. if err != nil {
  199. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  200. return err
  201. }
  202. return nil
  203. }
  204. //// AcceptLink 同意链接
  205. //func AcceptLink(ctx context.Context, TaskIDs []string) error {
  206. // db := GetReadDB(ctx)
  207. // err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id in ? and is_review = 0", TaskIDs).Updates(map[string]interface{}{"is_ok": 1, "is_review": 1, "agree_at": time.Now()}).Error
  208. // if err != nil {
  209. // logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  210. // return err
  211. // }
  212. // err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{LinkStatus: 5}).Error
  213. // if err != nil {
  214. // logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  215. // return err
  216. // }
  217. // err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{TaskStage: 13}).Error
  218. // if err != nil {
  219. // logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  220. // return err
  221. // }
  222. // return nil
  223. //}
  224. // AcceptLink 同意链接
  225. func AcceptLink(ctx context.Context, TaskIDs []string, operatorid string, operatetype int, projectid string, isspecial int) error {
  226. db := GetReadDB(ctx)
  227. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id in ? and is_review = 0", TaskIDs).Updates(map[string]interface{}{"is_ok": 1, "is_review": 1, "agree_at": time.Now(), "b_operator": operatorid, "b_operator_type": operatetype}).Error
  228. if err != nil {
  229. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  230. return err
  231. }
  232. var project gorm_model.ProjectInfo
  233. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectid).Find(&project).Error
  234. if err != nil {
  235. logrus.WithContext(ctx).Errorf("[Sketch db] Find ProjectInfo error,err:%+v", err)
  236. return err
  237. }
  238. taskstages := 13
  239. nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
  240. if isspecial == 1 {
  241. toolList := strings.Split(project.Tools, ",")
  242. for _, tool := range toolList {
  243. // 将 tool 转换为整数
  244. if t, err := strconv.Atoi(tool); err == nil && t >= 0 && t <= len(nextstage) {
  245. nextstage[t-1] = 1
  246. }
  247. }
  248. for i := 4; i < len(nextstage); i++ { // 从位置1开始
  249. if nextstage[i] == 1 {
  250. switch i {
  251. case 4:
  252. taskstages = 13
  253. case 5:
  254. taskstages = 15
  255. }
  256. break
  257. }
  258. }
  259. }
  260. var data_default gorm_model.InfoAutoTask
  261. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskID).Find(&data_default).Error
  262. if err != nil {
  263. return err
  264. }
  265. duration := time.Duration(data_default.CaseCloseDefault) * time.Hour
  266. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{LinkStatus: 5, CurBreakAt: time.Now().Add(duration)}).Error
  267. if err != nil {
  268. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  269. return err
  270. }
  271. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{TaskStage: taskstages}).Error
  272. if err != nil {
  273. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  274. return err
  275. }
  276. return nil
  277. }
  278. // AcceptLink 同意链接
  279. func LocalAcceptLink(ctx context.Context, TaskIDs []string, operatorid string, operatetype int, projectid string, isspecial int) error {
  280. db := GetReadDB(ctx)
  281. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id in ? and is_review = 0", TaskIDs).Updates(map[string]interface{}{"is_ok": 1, "is_review": 1, "agree_at": time.Now(), "b_operator": operatorid, "b_operator_type": operatetype}).Error
  282. if err != nil {
  283. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  284. return err
  285. }
  286. var project gorm_model.YounggeeLocalLifeInfo
  287. err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", projectid).Find(&project).Error
  288. if err != nil {
  289. logrus.WithContext(ctx).Errorf("[Sketch db] Find YounggeeLocalLifeInfo error,err:%+v", err)
  290. return err
  291. }
  292. taskstages := 13
  293. nextstage := [6]int{0, 0, 0, 0, 0, 0} // 数组长度为6,表示6种工具 4,9,11,13
  294. if isspecial == 1 {
  295. toolList := strings.Split(project.Tools, ",")
  296. for _, tool := range toolList {
  297. // 将 tool 转换为整数
  298. if t, err := strconv.Atoi(tool); err == nil && t >= 0 && t <= len(nextstage) {
  299. nextstage[t-1] = 1
  300. }
  301. }
  302. for i := 4; i < len(nextstage); i++ { // 从位置1开始
  303. if nextstage[i] == 1 {
  304. switch i {
  305. case 4:
  306. taskstages = 13
  307. case 5:
  308. taskstages = 15
  309. }
  310. break
  311. }
  312. }
  313. }
  314. var data_default gorm_model.InfoAutoTask
  315. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskId).Find(&data_default).Error
  316. if err != nil {
  317. return err
  318. }
  319. duration := time.Duration(data_default.CaseCloseDefault) * time.Hour
  320. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeLocalTaskInfo{LinkStatus: 5, CurBreakAt: time.Now().Add(duration)}).Error
  321. if err != nil {
  322. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeLocalTaskInfo error,err:%+v", err)
  323. return err
  324. }
  325. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeLocalTaskInfo{TaskStage: taskstages}).Error
  326. if err != nil {
  327. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeLocalTaskInfo error,err:%+v", err)
  328. return err
  329. }
  330. return nil
  331. }
  332. // RejectLink 不同意链接
  333. func RejectLink(ctx context.Context, TaskIDs []string, operatorid string, operatetype int, projectid string) error {
  334. db := GetReadDB(ctx)
  335. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id in ? and is_review = 0", TaskIDs).Updates(map[string]interface{}{"is_ok": 0, "is_review": 1, "reject_at": time.Now(), "b_operator": operatorid, "b_operator_type": operatetype}).Error
  336. if err != nil {
  337. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  338. return err
  339. }
  340. var project gorm_model.ProjectInfo
  341. err = db.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectid).Find(&project).Error
  342. if err != nil {
  343. logrus.WithContext(ctx).Errorf("[Sketch db] Find ProjectInfo error,err:%+v", err)
  344. return err
  345. }
  346. var link_default gorm_model.InfoAutoTask
  347. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskID).Find(&link_default).Error
  348. if err != nil {
  349. return err
  350. }
  351. duration := time.Duration(link_default.LinkBreach) * time.Hour
  352. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{LinkStatus: 3, CurBreakAt: time.Now().Add(duration)}).Error
  353. if err != nil {
  354. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  355. return err
  356. }
  357. err = db.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeTaskInfo{TaskStage: 11}).Error
  358. if err != nil {
  359. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeTaskInfo error,err:%+v", err)
  360. return err
  361. }
  362. return nil
  363. }
  364. // RejectLink 不同意链接
  365. func LocalRejectLink(ctx context.Context, TaskIDs []string, operatorid string, operatetype int, projectid string) error {
  366. db := GetReadDB(ctx)
  367. err := db.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id in ? and is_review = 0", TaskIDs).Updates(map[string]interface{}{"is_ok": 0, "is_review": 1, "reject_at": time.Now(), "b_operator": operatorid, "b_operator_type": operatetype}).Error
  368. if err != nil {
  369. logrus.WithContext(ctx).Errorf("[Link db] Update YounggeeLinkInfo error,err:%+v", err)
  370. return err
  371. }
  372. var project gorm_model.YounggeeLocalLifeInfo
  373. err = db.Model(gorm_model.YounggeeLocalLifeInfo{}).Where("local_id = ?", projectid).Find(&project).Error
  374. if err != nil {
  375. logrus.WithContext(ctx).Errorf("[Sketch db] Find ProjectInfo error,err:%+v", err)
  376. return err
  377. }
  378. var link_default gorm_model.InfoAutoTask
  379. err = db.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", project.AutoTaskId).Find(&link_default).Error
  380. if err != nil {
  381. return err
  382. }
  383. duration := time.Duration(link_default.LinkBreach) * time.Hour
  384. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeLocalTaskInfo{LinkStatus: 3, CurBreakAt: time.Now().Add(duration)}).Error
  385. if err != nil {
  386. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeLocalTaskInfo error,err:%+v", err)
  387. return err
  388. }
  389. err = db.Model(gorm_model.YoungeeLocalTaskInfo{}).Where("task_id in ?", TaskIDs).Updates(gorm_model.YoungeeLocalTaskInfo{TaskStage: 11}).Error
  390. if err != nil {
  391. logrus.WithContext(ctx).Errorf("[Link db] Update YoungeeLocalTaskInfo error,err:%+v", err)
  392. return err
  393. }
  394. return nil
  395. }
  396. // GetSpecialTaskLinkList 专项任务-查询上传链接的task list
  397. func GetSpecialTaskLinkList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) ([]*http_model.SpecialTaskLinkInfo, int64, error) {
  398. db := GetReadDB(ctx)
  399. // 查询Task表信息
  400. db = db.Debug().Model(gorm_model.YoungeeTaskInfo{}).Where("task_status = 2")
  401. // 根据Project条件过滤
  402. conditionType := reflect.TypeOf(conditions).Elem()
  403. conditionValue := reflect.ValueOf(conditions).Elem()
  404. var platform_nickname string = ""
  405. for i := 0; i < conditionType.NumField(); i++ {
  406. field := conditionType.Field(i)
  407. tag := field.Tag.Get("condition")
  408. value := conditionValue.FieldByName(field.Name)
  409. if tag == "link_status" {
  410. fmt.Printf("link %+v", value.Interface() == int64(0))
  411. if value.Interface() == int64(0) {
  412. db = db.Where("task_stage = 12")
  413. } else {
  414. db = db.Where("task_stage > 12 and task_stage <> 16")
  415. }
  416. continue
  417. } else if !util.IsBlank(value) {
  418. if tag == "platform_nickname" {
  419. platform_nickname = fmt.Sprintf("%v", value.Interface())
  420. continue
  421. } else if tag == "project_id" {
  422. db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
  423. }
  424. }
  425. }
  426. var taskInfos []gorm_model.YoungeeTaskInfo
  427. db = db.Model(gorm_model.YoungeeTaskInfo{})
  428. // 查询总数
  429. var totalTask int64
  430. if err := db.Count(&totalTask).Error; err != nil {
  431. logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
  432. return nil, 0, err
  433. }
  434. db.Order("task_id").Find(&taskInfos)
  435. // 查询任务id
  436. var taskIds []string
  437. taskMap := make(map[string]gorm_model.YoungeeTaskInfo)
  438. for _, taskInfo := range taskInfos {
  439. taskIds = append(taskIds, taskInfo.TaskId)
  440. taskMap[taskInfo.TaskId] = taskInfo
  441. }
  442. db1 := GetReadDB(ctx)
  443. db1 = db1.Debug().Model(gorm_model.YounggeeLinkInfo{})
  444. var LinkInfos []gorm_model.YounggeeLinkInfo
  445. db1 = db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id IN ? AND is_submit=? ", taskIds, 1)
  446. if conditions.LinkStatus == int64(0) {
  447. db1 = db1.Where("is_review = 0").Find(&LinkInfos)
  448. } else {
  449. db1 = db1.Where("is_ok = 1").Find(&LinkInfos)
  450. }
  451. LinkMap := make(map[string]gorm_model.YounggeeLinkInfo)
  452. for _, LinkInfo := range LinkInfos {
  453. LinkMap[LinkInfo.TaskID] = LinkInfo
  454. }
  455. // 查询总数
  456. var totalLink int64
  457. if err := db1.Count(&totalLink).Error; err != nil {
  458. logrus.WithContext(ctx).Errorf("[GetProjectTalentList] error query mysql total, err:%+v", err)
  459. return nil, 0, err
  460. }
  461. var misNum int64
  462. if totalLink > totalTask {
  463. misNum = totalLink - totalTask
  464. } else {
  465. misNum = totalTask - totalLink
  466. }
  467. //logrus.Println("totalLink,totalTalent,misNum:", totalLink, totalTask, misNum)
  468. // 查询该页数据
  469. limit := pageSize + misNum
  470. offset := pageSize * pageNum // assert pageNum start with 0
  471. err := db.Order("task_id").Limit(int(limit)).Offset(int(offset)).Error
  472. if err != nil {
  473. logrus.WithContext(ctx).Errorf("[GetProjectTaskList] error query mysql total, err:%+v", err)
  474. return nil, 0, err
  475. }
  476. var TaskLinks []*http_model.SpecialTaskLink
  477. var taskLinks []*http_model.SpecialTaskLinkInfo
  478. var newTaskLinks []*http_model.SpecialTaskLinkInfo
  479. for _, taskId := range taskIds {
  480. TaskLink := new(http_model.SpecialTaskLink)
  481. TaskLink.Talent = taskMap[taskId]
  482. TaskLink.Link = LinkMap[taskId]
  483. TaskLinks = append(TaskLinks, TaskLink)
  484. }
  485. taskLinks = pack.SpecialTaskLinkToTaskInfo(TaskLinks)
  486. for _, v := range taskLinks {
  487. if platform_nickname == "" {
  488. newTaskLinks = append(newTaskLinks, v)
  489. } else if strings.Contains(v.PlatformNickname, platform_nickname) {
  490. newTaskLinks = append(newTaskLinks, v)
  491. } else if strings.Contains(conv.MustString(v.TaskID, ""), platform_nickname) {
  492. newTaskLinks = append(newTaskLinks, v)
  493. } else {
  494. totalTask--
  495. }
  496. }
  497. return newTaskLinks, totalTask, nil
  498. }