project.go 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991
  1. package service
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "math/rand"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "youngee_b_api/db"
  11. "youngee_b_api/model/common_model"
  12. "youngee_b_api/model/gorm_model"
  13. "youngee_b_api/model/http_model"
  14. "youngee_b_api/pack"
  15. "youngee_b_api/util"
  16. "github.com/gin-gonic/gin"
  17. "github.com/issue9/conv"
  18. "github.com/sirupsen/logrus"
  19. )
  20. var Project *project
  21. type project struct {
  22. }
  23. func (*project) Create(ctx context.Context, newProject http_model.CreateProjectRequest, enterpriseID string) (*http_model.CreateProjectData, error) {
  24. // build gorm_model.ProjectInfo
  25. // 查询关联商品信息
  26. product, err := db.GetProductByID(ctx, newProject.ProductID)
  27. if err != nil {
  28. return nil, err
  29. }
  30. productPhotos, err := db.GetProductPhotoByProductID(ctx, newProject.ProductID)
  31. productInfoToJson, _ := json.Marshal(product)
  32. productPhotosToJson, _ := json.Marshal(productPhotos)
  33. //fmt.Println("productPhotosToJson:", productPhotosToJson)
  34. AutoTaskID, err := db.GetLastAutoTaskID()
  35. if err != nil {
  36. return nil, err
  37. }
  38. AutoDefaultID, err := db.GetLastAutoDefaultID()
  39. if err != nil {
  40. return nil, err
  41. }
  42. // 按照品牌名-商品名对项目进行命名
  43. projectName := product.BrandName + "-" + product.ProductName
  44. //feeForm := fmt.Sprintf("[")
  45. feeFrom := []string{}
  46. for _, strategy := range newProject.RecruitStrategys {
  47. feeFrom = append(feeFrom, strconv.FormatInt(strategy.FeeForm, 10))
  48. }
  49. var ECost float64 = 0
  50. if newProject.ProjectType == int64(1) {
  51. for _, strategy := range newProject.RecruitStrategys {
  52. // 计算预估成本
  53. var tmpCost float64 = 0
  54. if strategy.FeeForm == 1 {
  55. tmpCost = strategy.ServiceCharge * float64(strategy.RecruitNumber)
  56. } else if strategy.FeeForm == 2 {
  57. tmpCost = strategy.Offer * float64(strategy.RecruitNumber)
  58. }
  59. ECost += tmpCost
  60. }
  61. }
  62. feeFroms := strings.Join(feeFrom, ",")
  63. //fmt.Printf("创建项目new %+v", newProject)
  64. RecruitDdl := time.Time{} //赋零值
  65. if newProject.RecruitDdl != "" {
  66. RecruitDdl, _ = time.ParseInLocation("2006-01-02 15:04:05", newProject.RecruitDdl, time.Local)
  67. }
  68. fmt.Println("Create RecruitDdl:", newProject.RecruitDdl, RecruitDdl)
  69. projectInfo := gorm_model.ProjectInfo{}
  70. rand.Seed(time.Now().UnixNano())
  71. td := conv.MustString(time.Now().Day())
  72. for {
  73. if len(td) == 3 {
  74. break
  75. }
  76. td = "0" + td
  77. }
  78. if newProject.ProjectType == int64(1) {
  79. // 没有填写DDL
  80. if newProject.RecruitDdl == "" {
  81. projectInfo = gorm_model.ProjectInfo{
  82. ProjectID: conv.MustString(time.Now().Year())[2:] + td + conv.MustString(rand.Intn(100000-10000)+10000),
  83. ProjectName: projectName,
  84. ProjectStatus: 1,
  85. ProjectType: newProject.ProjectType,
  86. TalentType: newProject.TalentType,
  87. ProjectPlatform: newProject.ProjectPlatform,
  88. ProjectForm: newProject.ProjectForm,
  89. ProjectDetail: newProject.ProjectDetail,
  90. ContentType: newProject.ContentType,
  91. EnterpriseID: enterpriseID,
  92. ProductID: newProject.ProductID,
  93. FeeForm: feeFroms,
  94. AutoTaskID: conv.MustInt64(AutoTaskID),
  95. AutoDefaultID: conv.MustInt64(AutoDefaultID),
  96. EstimatedCost: ECost,
  97. IsRead: 0,
  98. ProductSnap: string(productInfoToJson),
  99. ProductPhotoSnap: string(productPhotosToJson),
  100. }
  101. } else {
  102. projectInfo = gorm_model.ProjectInfo{
  103. ProjectID: conv.MustString(time.Now().Year())[2:] + td + conv.MustString(rand.Intn(100000-10000)+10000),
  104. ProjectName: projectName,
  105. ProjectStatus: 1,
  106. ProjectType: newProject.ProjectType,
  107. TalentType: newProject.TalentType,
  108. ProjectPlatform: newProject.ProjectPlatform,
  109. ProjectForm: newProject.ProjectForm,
  110. RecruitDdl: &RecruitDdl,
  111. ProjectDetail: newProject.ProjectDetail,
  112. ContentType: newProject.ContentType,
  113. EnterpriseID: enterpriseID,
  114. ProductID: newProject.ProductID,
  115. FeeForm: feeFroms,
  116. AutoTaskID: conv.MustInt64(AutoTaskID),
  117. AutoDefaultID: conv.MustInt64(AutoDefaultID),
  118. EstimatedCost: ECost,
  119. IsRead: 0,
  120. ProductSnap: string(productInfoToJson),
  121. ProductPhotoSnap: string(productPhotosToJson),
  122. }
  123. }
  124. } else {
  125. projectInfo = gorm_model.ProjectInfo{
  126. ProjectID: conv.MustString(time.Now().Year())[2:] + td + conv.MustString(rand.Intn(100000-10000)+10000),
  127. ProjectName: projectName,
  128. ProjectStatus: 1,
  129. ProjectType: newProject.ProjectType,
  130. TalentType: "[]",
  131. ProjectPlatform: newProject.ProjectPlatform,
  132. ProjectForm: newProject.ProjectForm,
  133. //RecruitDdl: &RecruitDdl,
  134. ProjectDetail: newProject.ProjectDetail,
  135. ContentType: newProject.ContentType,
  136. EnterpriseID: enterpriseID,
  137. ProductID: newProject.ProductID,
  138. FeeForm: feeFroms,
  139. AutoTaskID: conv.MustInt64(AutoTaskID),
  140. AutoDefaultID: conv.MustInt64(AutoDefaultID),
  141. EstimatedCost: ECost,
  142. ProductSnap: string(productInfoToJson),
  143. ProductPhotoSnap: string(productPhotosToJson),
  144. }
  145. }
  146. // db create ProjectInfo
  147. projectID, err := db.CreateProject(ctx, projectInfo)
  148. if err != nil {
  149. return nil, err
  150. }
  151. if len(newProject.ProjectPhotos) != 0 {
  152. // build []gorm_model.ProjectPhoto
  153. projectPhotos := []gorm_model.ProjectPhoto{}
  154. for _, photo := range newProject.ProjectPhotos {
  155. projectPhoto := gorm_model.ProjectPhoto{
  156. PhotoUrl: photo.PhotoUrl,
  157. PhotoUid: photo.PhotoUid,
  158. ProjectID: *projectID,
  159. }
  160. projectPhotos = append(projectPhotos, projectPhoto)
  161. }
  162. // db create ProjectPhoto
  163. err = db.CreateProjectPhoto(ctx, projectPhotos)
  164. if err != nil {
  165. return nil, err
  166. }
  167. }
  168. // build
  169. if newProject.ProjectType == int64(1) && newProject.RecruitStrategys != nil {
  170. recruitStrategys := []gorm_model.RecruitStrategy{}
  171. for _, strategy := range newProject.RecruitStrategys {
  172. // 查询对应定价策略
  173. pricingStrategy, err := db.GetPricingStrategy(ctx, strategy.FollowersLow, strategy.FollowersUp, strategy.FeeForm, newProject.ProjectPlatform)
  174. if err != nil {
  175. return nil, err
  176. }
  177. // 根据定价策略计算达人所见报价
  178. if strategy.FeeForm == 2 {
  179. strategy.TOffer = strategy.Offer * (1 - conv.MustFloat64(pricingStrategy.ServiceRate)/1000)
  180. }
  181. recruitStrategy := gorm_model.RecruitStrategy{
  182. FeeForm: strategy.FeeForm,
  183. StrategyID: strategy.StrategyID,
  184. FollowersLow: strategy.FollowersLow,
  185. FollowersUp: strategy.FollowersUp,
  186. RecruitNumber: strategy.RecruitNumber,
  187. ServiceCharge: strategy.ServiceCharge,
  188. Offer: strategy.Offer,
  189. TOffer: strategy.TOffer,
  190. ProjectID: *projectID,
  191. }
  192. recruitStrategys = append(recruitStrategys, recruitStrategy)
  193. }
  194. err = db.CreateRecruitStrategy(ctx, recruitStrategys)
  195. if err != nil {
  196. return nil, err
  197. }
  198. }
  199. res := &http_model.CreateProjectData{
  200. ProjectID: *projectID,
  201. }
  202. //fmt.Printf("%+v", res)
  203. return res, nil
  204. }
  205. func (*project) Update(ctx context.Context, newProject http_model.UpdateProjectRequest, enterpriseID string) (*http_model.UpdateProjectData, error) {
  206. RecruitDdl := time.Time{} //赋零值
  207. if newProject.RecruitDdl != "" {
  208. RecruitDdl, _ = time.ParseInLocation("2006-01-02 15:04:05", newProject.RecruitDdl, time.Local)
  209. }
  210. fmt.Println("Update RecruitDdl:", newProject.RecruitDdl, RecruitDdl)
  211. oldProject, err3 := db.GetProjectDetail(ctx, newProject.ProjectID)
  212. if err3 != nil {
  213. return nil, err3
  214. }
  215. feeFrom := []string{}
  216. for _, strategy := range newProject.RecruitStrategys {
  217. //if strategy.StrategyID
  218. feeFrom = append(feeFrom, strconv.FormatInt(strategy.FeeForm, 10))
  219. //feeForm += string(strategy.StrategyID)
  220. }
  221. var ECost float64 = 0
  222. if newProject.ProjectType == int64(1) && newProject.RecruitStrategys != nil {
  223. for _, strategy := range newProject.RecruitStrategys {
  224. // 计算预估成本
  225. var tmpCost float64 = 0
  226. if strategy.FeeForm == 1 {
  227. tmpCost = strategy.ServiceCharge * float64(strategy.RecruitNumber)
  228. } else if strategy.FeeForm == 2 {
  229. tmpCost = strategy.Offer * float64(strategy.RecruitNumber)
  230. }
  231. ECost += tmpCost
  232. }
  233. }
  234. feeFroms := strings.Join(feeFrom, ",")
  235. t := time.Now()
  236. project := gorm_model.ProjectInfo{}
  237. if newProject.RecruitDdl == "" {
  238. project = gorm_model.ProjectInfo{
  239. ProjectID: newProject.ProjectID,
  240. TalentType: newProject.TalentType,
  241. ContentType: conv.MustInt64(newProject.ContentType),
  242. ProjectDetail: newProject.ProjectDetail,
  243. ProjectForm: conv.MustInt64(newProject.ProjectForm),
  244. EnterpriseID: enterpriseID,
  245. ProjectStatus: conv.MustInt64(newProject.ProjectStatus),
  246. FeeForm: feeFroms,
  247. EstimatedCost: ECost,
  248. SubmitAt: &t,
  249. }
  250. } else {
  251. project = gorm_model.ProjectInfo{
  252. ProjectID: newProject.ProjectID,
  253. RecruitDdl: &RecruitDdl,
  254. TalentType: newProject.TalentType,
  255. ContentType: conv.MustInt64(newProject.ContentType),
  256. ProjectDetail: newProject.ProjectDetail,
  257. ProjectForm: conv.MustInt64(newProject.ProjectForm),
  258. EnterpriseID: enterpriseID,
  259. ProjectStatus: conv.MustInt64(newProject.ProjectStatus),
  260. FeeForm: feeFroms,
  261. EstimatedCost: ECost,
  262. SubmitAt: &t,
  263. }
  264. }
  265. projectID, err := db.UpdateProject(ctx, project)
  266. if err != nil {
  267. return nil, err
  268. }
  269. // 删除该项目之前的所有图片
  270. err = db.DeleteProjectPhotoByProjecttID(ctx, *projectID)
  271. if err != nil {
  272. return nil, err
  273. }
  274. //fmt.Printf("照片:\t %+v", newProject.ProjectPhotos)
  275. if len(newProject.ProjectPhotos) != 0 {
  276. // 新增图片
  277. projectPhotos := []gorm_model.ProjectPhoto{}
  278. for _, photo := range newProject.ProjectPhotos {
  279. projectPhoto := gorm_model.ProjectPhoto{
  280. ProjectID: project.ProjectID,
  281. PhotoUrl: photo.PhotoUrl,
  282. PhotoUid: photo.PhotoUid,
  283. FileName: photo.FileName,
  284. }
  285. projectPhotos = append(projectPhotos, projectPhoto)
  286. }
  287. err = db.CreateProjectPhoto(ctx, projectPhotos)
  288. if err != nil {
  289. return nil, err
  290. }
  291. }
  292. // 删除该项目之前的所有策略
  293. err = db.DeleteRecruitStrategyByProjectID(ctx, *projectID)
  294. if err != nil {
  295. return nil, err
  296. }
  297. fmt.Printf("策略:\t %+v,%+v", newProject.RecruitStrategys, len(newProject.RecruitStrategys) == 0)
  298. if len(newProject.RecruitStrategys) != 0 && newProject.ProjectType == int64(1) {
  299. // 新增策略
  300. RecruitStrategys := []gorm_model.RecruitStrategy{}
  301. for _, Strategy := range newProject.RecruitStrategys {
  302. // 查询对应定价策略
  303. pricingStrategy, err := db.GetPricingStrategy(ctx, Strategy.FollowersLow, Strategy.FollowersUp, Strategy.FeeForm, oldProject.ProjectPlatform)
  304. if err != nil {
  305. return nil, err
  306. }
  307. // 根据定价策略计算达人所见报价
  308. if Strategy.FeeForm == 2 {
  309. Strategy.TOffer = Strategy.Offer * (1 - conv.MustFloat64(pricingStrategy.ServiceRate)/1000)
  310. }
  311. RecruitStrategy := gorm_model.RecruitStrategy{
  312. FeeForm: conv.MustInt64(Strategy.FeeForm),
  313. StrategyID: conv.MustInt64(Strategy.StrategyID),
  314. FollowersLow: conv.MustInt64(Strategy.FollowersLow),
  315. FollowersUp: conv.MustInt64(Strategy.FollowersUp),
  316. RecruitNumber: conv.MustInt64(Strategy.RecruitNumber),
  317. ServiceCharge: Strategy.ServiceCharge,
  318. Offer: Strategy.Offer,
  319. TOffer: Strategy.TOffer,
  320. ProjectID: project.ProjectID,
  321. }
  322. //fmt.Printf("Offer:\t %+v", Strategy.Offer)
  323. RecruitStrategys = append(RecruitStrategys, RecruitStrategy)
  324. }
  325. err = db.CreateRecruitStrategy(ctx, RecruitStrategys)
  326. if err != nil {
  327. return nil, err
  328. }
  329. }
  330. res := &http_model.UpdateProjectData{
  331. ProjectID: *projectID,
  332. }
  333. return res, nil
  334. }
  335. func (*project) Delete(ctx context.Context, projectID string) (*http_model.DeleteProjectData, error) {
  336. // 删除该项目之前的所有图片
  337. err := db.DeleteProjectPhotoByProjecttID(ctx, projectID)
  338. if err != nil {
  339. return nil, err
  340. }
  341. // 删除该项目之前的所有策略
  342. err = db.DeleteRecruitStrategyByProjectID(ctx, projectID)
  343. if err != nil {
  344. return nil, err
  345. }
  346. //删除项目
  347. NewProjectID, err1 := db.DeleteProject(ctx, projectID)
  348. if err1 != nil {
  349. return nil, err1
  350. }
  351. res := &http_model.DeleteProjectData{
  352. ProjectID: *NewProjectID,
  353. }
  354. return res, nil
  355. }
  356. func (*project) GetFullProjectList(ctx context.Context, pageSize, pageNum int32, supplierId int, condition *common_model.ProjectCondition) (*http_model.FullProjectListData, error) {
  357. // 1. 查询种草任务基本信息
  358. fullProjects, total, err := db.GetFullProjectList(ctx, pageSize, pageNum, condition)
  359. if err != nil {
  360. logrus.WithContext(ctx).Errorf("[project service] call GetFullProjectList error,err:%+v", err)
  361. return nil, err
  362. }
  363. fullProjectListData := new(http_model.FullProjectListData)
  364. fullProjectListData.FullProjectPreview = pack.MGormFullProjectToHttpFullProjectPreview(fullProjects)
  365. fullProjectListData.Total = total
  366. // 2. 查询种草任务补充信息:商品信息,招募策略
  367. for _, project := range fullProjectListData.FullProjectPreview {
  368. // 2.1. 商品信息
  369. productInfo, productErr := db.GetProductByID(ctx, project.ProductId)
  370. if productErr != nil {
  371. return nil, productErr
  372. }
  373. if productInfo != nil {
  374. project.ProductId = productInfo.ProductID
  375. project.ProductPrice = productInfo.ProductPrice
  376. project.ProductName = productInfo.ProductName
  377. }
  378. // 2.2. 商品图片信息
  379. productPhotoInfo, productPhotoErr := db.GetProductPhotoByProductID(ctx, project.ProductId)
  380. if productPhotoErr != nil {
  381. return nil, productPhotoErr
  382. }
  383. if productPhotoInfo != nil {
  384. for _, photo := range productPhotoInfo {
  385. fmt.Println(photo)
  386. if photo.Symbol == 1 {
  387. project.ProductPhotoSymbol = 1
  388. project.ProductPhotoUrl = photo.PhotoUrl
  389. project.ProductPhotoUid = photo.PhotoUid
  390. }
  391. }
  392. }
  393. // 2.3. 招募策略信息
  394. recruitStrategyInfo, recruitErr := db.GetRecruitStrategyByProjectId(ctx, project.ProjectId)
  395. if recruitErr != nil {
  396. return nil, recruitErr
  397. }
  398. if recruitStrategyInfo != nil {
  399. for _, strategy := range recruitStrategyInfo {
  400. var recruitStrategy *http_model.EasyRecruitStrategy
  401. recruitStrategy = &http_model.EasyRecruitStrategy{}
  402. recruitStrategy.StrategyId = strategy.StrategyID
  403. recruitStrategy.FeeForm = strategy.FeeForm
  404. recruitStrategy.RecruitNumber = strategy.RecruitNumber
  405. project.RecruitStrategy = append(project.RecruitStrategy, recruitStrategy)
  406. }
  407. }
  408. // 2.4. 判断是否加入商单
  409. sProjectCount, sProjectErr := db.FindSProjectByProjectIdAndSupplierId(ctx, project.ProjectId, supplierId)
  410. if sProjectErr != nil {
  411. return nil, sProjectErr
  412. }
  413. if sProjectCount > 0 {
  414. project.AddToListStatus = 1
  415. } else {
  416. project.AddToListStatus = 2
  417. }
  418. }
  419. return fullProjectListData, nil
  420. }
  421. func (*project) GetProjectDraftList(ctx context.Context, enterpriseID string, pageSize, pageNum int32, condition *common_model.ProjectCondition) (*http_model.ProjectDraftListData, error) {
  422. ProjectDrafts, total, err := db.GetProjectDraftList(ctx, enterpriseID, pageSize, pageNum, condition)
  423. if err != nil {
  424. logrus.WithContext(ctx).Errorf("[project service] call GetProjectDraftList error,err:%+v", err)
  425. return nil, err
  426. }
  427. ProjectDraftListData := new(http_model.ProjectDraftListData)
  428. ProjectDraftListData.ProjectDraftPreview = pack.MGormProjectDraftToHttpProjectDraftPreview(ProjectDrafts)
  429. ProjectDraftListData.Total = conv.MustString(total)
  430. return ProjectDraftListData, nil
  431. }
  432. func (*project) GetProjectTaskList(ctx context.Context, pageSize, pageNum int64, orderBy []string, orderDesc []int, conditions *common_model.TaskConditions) (*http_model.ProjectTaskListData, error) {
  433. // 1. 根据SProject和condition查询所有Task
  434. projectTasks, total, err := db.GetProjectTaskList(ctx, pageSize, pageNum, orderBy, orderDesc, conditions)
  435. if err != nil {
  436. logrus.WithContext(ctx).Errorf("[project service] call GetProjectTaskList error,err:%+v", err)
  437. return nil, err
  438. }
  439. // projectTaskListData := new(http_model.ProjectTaskListData)
  440. var projectTaskListData *http_model.ProjectTaskListData
  441. projectTaskListData = &http_model.ProjectTaskListData{}
  442. // fmt.Println(projectTasks)
  443. // projectTaskListData.ProjectTaskPreview = pack.MGormProjectTaskToHttpProjectTaskPreview(projectTasks)
  444. if projectTasks != nil {
  445. projectTaskListData.Total = total
  446. for _, projectTask := range projectTasks {
  447. var projectTaskInfo *http_model.ProjectTaskPreview
  448. projectTaskInfo = &http_model.ProjectTaskPreview{}
  449. // 2. 补充查询达人身份信息
  450. platformUserInfo, platformUserErr := db.FindUserInfoByOpenId(ctx, projectTask.OpenId)
  451. if platformUserErr != nil {
  452. return nil, platformUserErr
  453. }
  454. if platformUserInfo != nil {
  455. projectTaskInfo.TaskId = projectTask.TaskID
  456. projectTaskInfo.TaskStage = projectTask.TaskStage
  457. projectTaskInfo.ServiceCharge = projectTask.ServiceCharge
  458. projectTaskInfo.DraftFee = projectTask.DraftFee
  459. projectTaskInfo.SupportFee = projectTask.SupportFee
  460. projectTaskInfo.StrategyId = projectTask.StrategyID
  461. projectTaskInfo.TaskStatus = projectTask.TaskStatus
  462. projectTaskInfo.CreateDate = conv.MustString(projectTask.CreateDate)
  463. projectTaskInfo.SelectDate = conv.MustString(projectTask.SelectDate)
  464. projectTaskInfo.DeliveryDate = conv.MustString(projectTask.DeliveryDate)
  465. projectTaskInfo.SignedTime = conv.MustString(projectTask.SignedTime)
  466. projectTaskInfo.CurBreakAt = conv.MustString(projectTask.CurBreakAt)
  467. projectTaskInfo.FansNum = projectTask.FansNum
  468. projectTaskInfo.VoteAvg = projectTask.VoteAvg
  469. projectTaskInfo.CommitAvg = projectTask.CommitAvg
  470. projectTaskInfo.BOperator = projectTask.BOperator
  471. projectTaskInfo.BOperatorType = projectTask.BOperatorType
  472. projectTaskInfo.SOperator = projectTask.SOperator
  473. projectTaskInfo.SOperatorType = projectTask.SOperatorType
  474. projectTaskInfo.PlatformNickname = platformUserInfo.NickName
  475. projectTaskInfo.FansCount = platformUserInfo.Fan
  476. projectTaskInfo.AvatarUrl = platformUserInfo.HeadUri
  477. projectTaskInfo.Location = platformUserInfo.City
  478. projectTaskInfo.Gender = platformUserInfo.Gender
  479. projectTaskListData.ProjectTaskPreview = append(projectTaskListData.ProjectTaskPreview, projectTaskInfo)
  480. }
  481. }
  482. }
  483. // 2. 查找SProject中其他信息
  484. sProjectInfo, sProjectErr := db.GetSProjectDetail(ctx, conditions.SProjectId)
  485. if sProjectErr != nil {
  486. return nil, sProjectErr
  487. }
  488. if sProjectInfo != nil {
  489. projectTaskListData.SettleNum = sProjectInfo.SettleNum
  490. projectTaskListData.RecruitNum = sProjectInfo.RecruitNum
  491. projectTaskListData.QuitNum = sProjectInfo.QuitNum
  492. projectTaskListData.ServiceChargeSettle = sProjectInfo.ServiceChargeSettle
  493. projectTaskListData.ServiceChargeActual = sProjectInfo.ServiceChargeActual
  494. projectTaskListData.EstimateServiceCharge = sProjectInfo.ServiceChargeActual
  495. projectTaskListData.EstimateDraftFee = sProjectInfo.EstimateDraftFee
  496. projectTaskListData.EstimateSupportFee = sProjectInfo.EstimateSupportFee
  497. }
  498. return projectTaskListData, nil
  499. }
  500. func (*project) GetPorjectDetail(ctx context.Context, projectID string) (*http_model.ShowProjectData, error) {
  501. project, err := db.GetProjectDetail(ctx, projectID)
  502. if err != nil {
  503. logrus.WithContext(ctx).Errorf("[project service] call GetPorjectDetail error,err:%+v", err)
  504. return nil, err
  505. }
  506. enterprise, err := db.GetEnterpriseByEnterpriseID(ctx, project.EnterpriseID)
  507. if err != nil {
  508. logrus.WithContext(ctx).Errorf("[project service] call GetEnterpriseByEnterpriseID error,err:%+v", err)
  509. return nil, err
  510. }
  511. user, err := db.GetUserByID(ctx, enterprise.UserID)
  512. if err != nil {
  513. logrus.WithContext(ctx).Errorf("[project service] call GetUserByID error,err:%+v", err)
  514. return nil, err
  515. }
  516. ProjectDetail := http_model.ShowProjectData{
  517. ProjectID: conv.MustString(project.ProjectID),
  518. ProjectName: conv.MustString(project.ProjectName),
  519. ProjectStatus: conv.MustString(project.ProjectStatus),
  520. ProjectType: conv.MustString(project.ProjectType),
  521. ProjectPlatform: conv.MustString(project.ProjectPlatform),
  522. ProjectForm: conv.MustString(project.ProjectForm),
  523. TalentType: conv.MustString(project.TalentType),
  524. RecruitDdl: util.GetTimePoionter(project.RecruitDdl),
  525. ContentType: conv.MustString(project.ContentType),
  526. ProjectDetail: conv.MustString(project.ProjectDetail),
  527. ProductID: conv.MustString(project.ProductID),
  528. EnterpriseID: conv.MustString(project.EnterpriseID),
  529. Balance: conv.MustString(enterprise.Balance),
  530. AvailableBalance: conv.MustString(enterprise.AvailableBalance),
  531. EstimatedCost: conv.MustString(project.EstimatedCost),
  532. FailReason: conv.MustString(project.FailReason),
  533. CreateAt: util.GetTimePoionter(project.CreatedAt),
  534. UpdateAt: util.GetTimePoionter(project.UpdatedAt),
  535. Phone: user.Phone,
  536. FinishAt: util.GetTimePoionter(project.FinishAt),
  537. PassAt: util.GetTimePoionter(project.PassAt),
  538. SubmitAt: util.GetTimePoionter(project.SubmitAt),
  539. PayAt: util.GetTimePoionter(project.PayAt),
  540. ProductInfo: conv.MustString(project.ProductSnap),
  541. ProductPhotoInfo: conv.MustString(project.ProductPhotoSnap),
  542. AutoFailAt: util.GetTimePoionter(project.AutoFailAt),
  543. }
  544. Strategys, err := db.GetRecruitStrategys(ctx, projectID)
  545. if err != nil {
  546. logrus.WithContext(ctx).Error()
  547. return nil, err
  548. }
  549. for _, strategy := range Strategys {
  550. RecruitStrategy := http_model.ShowRecruitStrategy{
  551. RecruitStrategyID: conv.MustString(strategy.RecruitStrategyID),
  552. FeeForm: conv.MustString(strategy.FeeForm),
  553. StrategyID: conv.MustString(strategy.StrategyID),
  554. FollowersLow: conv.MustString(strategy.FollowersLow),
  555. FollowersUp: conv.MustString(strategy.FollowersUp),
  556. RecruitNumber: conv.MustString(strategy.RecruitNumber),
  557. Offer: conv.MustString(strategy.Offer),
  558. ServiceCharge: conv.MustString(strategy.ServiceCharge),
  559. SelectedNumber: strategy.SelectedNumber,
  560. WaitingNumber: strategy.WaitingNumber,
  561. DeliveredNumber: strategy.DeliveredNumber,
  562. SignedNumber: strategy.SignedNumber,
  563. }
  564. ProjectDetail.RecruitStrategys = append(ProjectDetail.RecruitStrategys, RecruitStrategy)
  565. }
  566. Photos, err := db.GetProjectPhoto(ctx, projectID)
  567. if err != nil {
  568. logrus.WithContext(ctx).Error()
  569. return nil, err
  570. }
  571. for _, Photo := range Photos {
  572. ProjectPhoto := http_model.ShowProjectPhoto{
  573. PhotoUrl: Photo.PhotoUrl,
  574. PhotoUid: Photo.PhotoUid,
  575. FileName: Photo.FileName,
  576. }
  577. ProjectDetail.ProjectPhotos = append(ProjectDetail.ProjectPhotos, ProjectPhoto)
  578. }
  579. return &ProjectDetail, nil
  580. }
  581. func (*project) GetTaskLogisticsList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskLogisticsListData, error) {
  582. TaskLogisticss, total, err := db.GetTaskLogisticsList(ctx, projectID, pageSize, pageNum, conditions)
  583. if err != nil {
  584. logrus.WithContext(ctx).Errorf("[project service] call GetTaskLogisticsList error,err:%+v", err)
  585. return nil, err
  586. }
  587. TaskLogisticsListData := new(http_model.TaskLogisticsListData)
  588. TaskLogisticsListData.TaskLogisticsPreview = pack.MGormTaskLogisticsInfoListToHttpTaskLogisticsPreviewList(TaskLogisticss)
  589. TaskLogisticsListData.Total = conv.MustString(total)
  590. return TaskLogisticsListData, nil
  591. }
  592. // ChangeTaskStatus 提报达人或拒绝提报达人
  593. func (*project) ChangeTaskStatus(ctx *gin.Context, data http_model.ProjectChangeTaskStatusRequest) interface{} {
  594. var sOperatorType int
  595. if data.SubAccountId == 0 {
  596. sOperatorType = 1
  597. } else {
  598. sOperatorType = 2
  599. }
  600. RecruitStrategyIDs, err := db.ChangeTaskStatus(ctx, data.TaskIds, data.SupplierStatus, data.SupplierId, data.SubAccountId, sOperatorType)
  601. if err != nil {
  602. logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  603. return err
  604. }
  605. fmt.Println(RecruitStrategyIDs)
  606. // 已选数量
  607. //if data.SupplierStatus == 2 {
  608. // for _, RecruitStrategyID := range RecruitStrategyIDs {
  609. // err = db.CalculateSelectedNumberByRecruitStrategyID(ctx, RecruitStrategyID, 1)
  610. // if err != nil {
  611. // logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  612. // return err
  613. // }
  614. // }
  615. //} else if data.SupplierStatus == 3 {
  616. // for _, RecruitStrategyID := range RecruitStrategyIDs {
  617. // err = db.CalculateSelectedNumberByRecruitStrategyID(ctx, RecruitStrategyID, 1)
  618. // if err != nil {
  619. // logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  620. // return err
  621. // }
  622. // }
  623. //}
  624. return nil
  625. }
  626. // ChangeSpecialTaskStatus 定向种草任务 提报达人,拒绝提报
  627. /*
  628. func (*project) ChangeSpecialTaskStatus(ctx *gin.Context, data http_model.ProjectChangeTaskStatusRequest) interface{} {
  629. err := db.ChangeSpecialTaskStatus(ctx, data.TaskIds, data.SupplierStatus, data.SupplierId, data.SubAccountId, data.SOperatorType)
  630. if err != nil {
  631. logrus.WithContext(ctx).Errorf("[project service] call ChangeSpecialTaskStatus error,err:%+v", err)
  632. return err
  633. }
  634. return nil
  635. }
  636. */
  637. func (p *project) GetTaskScriptList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskScriptListData, error) {
  638. TaskScripts, total, err := db.GetTaskScriptList(ctx, projectID, pageSize, pageNum, conditions)
  639. if err != nil {
  640. logrus.WithContext(ctx).Errorf("[project service] call GetTaskScriptList error,err:%+v", err)
  641. return nil, err
  642. }
  643. TaskScriptListData := new(http_model.TaskScriptListData)
  644. TaskScriptListData.TaskScriptPreview = pack.MGormTaskScriptInfoListToHttpTaskScriptPreviewList(TaskScripts)
  645. TaskScriptListData.Total = conv.MustString(total)
  646. return TaskScriptListData, nil
  647. }
  648. func (p *project) GetTaskSketchList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskSketchListData, error) {
  649. TaskSketchs, total, err := db.GetTaskSketchList(ctx, projectID, pageSize, pageNum, conditions)
  650. if err != nil {
  651. logrus.WithContext(ctx).Errorf("[project service] call GetTaskSketchList error,err:%+v", err)
  652. return nil, err
  653. }
  654. TaskSketchListData := new(http_model.TaskSketchListData)
  655. TaskSketchListData.TaskSketchPreview = pack.MGormTaskSketchInfoListToHttpTaskSketchPreviewList(TaskSketchs)
  656. TaskSketchListData.Total = conv.MustString(total)
  657. return TaskSketchListData, nil
  658. }
  659. func (p *project) GetTaskLinkList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskLinkListData, error) {
  660. TaskLinks, total, err := db.GetTaskLinkList(ctx, projectID, pageSize, pageNum, conditions)
  661. if err != nil {
  662. logrus.WithContext(ctx).Errorf("[project service] call GetTaskLinkList error,err:%+v", err)
  663. return nil, err
  664. }
  665. TaskLinkListData := new(http_model.TaskLinkListData)
  666. TaskLinkListData.TaskLinkPreview = pack.MGormTaskLinkInfoListToHttpTaskLinkPreviewList(TaskLinks)
  667. TaskLinkListData.Total = conv.MustString(total)
  668. return TaskLinkListData, nil
  669. }
  670. func (p *project) GetTaskDataList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskDataListData, error) {
  671. TaskDatas, total, err := db.GetTaskDataList(ctx, projectID, pageSize, pageNum, conditions)
  672. if err != nil {
  673. logrus.WithContext(ctx).Errorf("[project service] call GetTaskDataList error,err:%+v", err)
  674. return nil, err
  675. }
  676. TaskDataListData := new(http_model.TaskDataListData)
  677. TaskDataListData.TaskDataPreview = pack.MGormTaskDataInfoListToHttpTaskDataPreviewList(TaskDatas)
  678. TaskDataListData.Total = conv.MustString(total)
  679. return TaskDataListData, nil
  680. }
  681. func (p *project) GetTaskFinishList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskFinishListData, error) {
  682. TaskFinishs, total, err := db.GetTaskFinishList(ctx, pageSize, pageNum, conditions)
  683. if err != nil {
  684. logrus.WithContext(ctx).Errorf("[project service] call GetTaskFinishList error,err:%+v", err)
  685. return nil, err
  686. }
  687. TaskFinishListData := new(http_model.TaskFinishListData)
  688. TaskFinishListData.TaskFinishPreview = pack.MGormTaskFinishInfoListToHttpTaskFinishPreviewList(TaskFinishs)
  689. TaskFinishListData.Total = conv.MustString(total)
  690. return TaskFinishListData, nil
  691. }
  692. func (p *project) GetServiceCharge(ctx *gin.Context, data http_model.GetServiceChargeRequest) (*http_model.ServiceChargeData, error) {
  693. pricingStrategy, err := db.GetPricingStrategy(ctx, data.FollowersLow, data.FollowersUp, data.FeeForm, data.Platform)
  694. if err != nil {
  695. return nil, err
  696. }
  697. serviceFee := http_model.ServiceChargeData{
  698. ServiceCharge: pricingStrategy.ServiceCharge,
  699. }
  700. return &serviceFee, nil
  701. }
  702. func (*project) GetSpecialProjectTaskList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.SpecialTaskLogisticsListData, error) {
  703. SpecialTaskLogisticsListDatas, total, err := db.GetSpecialTaskLogisticsList(ctx, projectID, pageSize, pageNum, conditions)
  704. if err != nil {
  705. logrus.WithContext(ctx).Errorf("[project service] call GetSpecialProjectTaskList error,err:%+v", err)
  706. return nil, err
  707. }
  708. TaskLogisticsListData := new(http_model.SpecialTaskLogisticsListData)
  709. TaskLogisticsListData.SpecialTaskLogisticsPreview = pack.MGormSpecialTaskLogisticsInfoListToHttpTaskLogisticsPreviewList(SpecialTaskLogisticsListDatas)
  710. TaskLogisticsListData.Total = conv.MustString(total, "")
  711. return TaskLogisticsListData, nil
  712. }
  713. // ShowTaskProgress ToDo
  714. // ShowTaskProgress 展示种草子任务进度
  715. func (p *project) ShowTaskProgress(ctx *gin.Context, data http_model.ShowTaskProgressRequest) (*http_model.ShowTaskProgressData, error) {
  716. // 1. 初始化返回数据结构体
  717. var taskProgressData *http_model.ShowTaskProgressData
  718. taskProgressData = &http_model.ShowTaskProgressData{}
  719. // 2. 根据task_id筛选出对应的子任务基础信息
  720. taskInfo, taskErr := db.GetTaskByTaskId(ctx, data.TaskId)
  721. if taskErr != nil {
  722. return nil, taskErr
  723. }
  724. if taskInfo != nil {
  725. taskProgressData.TaskId = data.TaskId
  726. taskProgressData.CreateData = taskInfo.CreateDate
  727. taskProgressData.SelectDate = taskInfo.SelectDate
  728. taskProgressData.DeliveryDate = taskInfo.DeliveryDate
  729. taskProgressData.SignedTime = taskInfo.SignedTime
  730. taskProgressData.ServiceCharge = taskInfo.ServiceCharge
  731. taskProgressData.DraftFee = taskInfo.DraftFee
  732. }
  733. // 3. 筛选出上传初稿时间和初稿状态
  734. taskSketchs, taskSketchErr := db.FindSketchInfoByTaskId(ctx, data.TaskId)
  735. if taskSketchErr != nil {
  736. return nil, taskSketchErr
  737. }
  738. if taskSketchs != nil {
  739. taskProgressData.SketchInfo = taskSketchs
  740. }
  741. // 4. 筛选作品链接上传时间和状态
  742. taskLinks, taskLinksErr := db.GetLinkByTaskId(ctx, data.TaskId)
  743. if taskLinksErr != nil {
  744. return nil, taskLinksErr
  745. }
  746. if taskLinks != nil {
  747. taskProgressData.LinkInfo = taskLinks
  748. }
  749. // 5. 若有违约,筛选出违约情况
  750. if taskInfo != nil {
  751. taskProgressData.TerminateTime = taskInfo.TerminateTime
  752. taskProgressData.TerminateReason = taskInfo.TerminateReason
  753. taskProgressData.CancelTime = taskInfo.CancelTime
  754. taskProgressData.CancelReason = taskInfo.CancelReason
  755. }
  756. return taskProgressData, nil
  757. }
  758. func (p *project) GetProjectStrategys(ctx *gin.Context, projectId string) ([]gorm_model.RecruitStrategy, error) {
  759. recruitStrategys, getProjectStrategysErr := db.GetRecruitStrategyByProjectId(ctx, projectId)
  760. if getProjectStrategysErr != nil {
  761. return nil, getProjectStrategysErr
  762. }
  763. return recruitStrategys, nil
  764. }
  765. // GetProjectDetail 查找种草任务详情
  766. func (*project) GetProjectDetail(ctx context.Context, ProjectId string) (*http_model.ShowNewProjectData, error) {
  767. var sProjectData *http_model.ShowNewProjectData
  768. sProjectData = &http_model.ShowNewProjectData{}
  769. // 1. 取出种草表中的信息
  770. projectInfo, projectErr := db.GetProjectDetail(ctx, ProjectId)
  771. if projectErr != nil {
  772. return nil, projectErr
  773. }
  774. if projectInfo != nil {
  775. sProjectData.ProjectName = projectInfo.ProjectName
  776. sProjectData.ProjectID = projectInfo.ProjectID
  777. sProjectData.ProjectType = projectInfo.ProjectType
  778. sProjectData.ProjectPlatform = projectInfo.ProjectPlatform
  779. sProjectData.ProjectForm = projectInfo.ProjectForm
  780. sProjectData.ContentType = projectInfo.ContentType
  781. sProjectData.EnterpriseID = projectInfo.EnterpriseID
  782. sProjectData.TalentType = projectInfo.TalentType
  783. sProjectData.RecruitDdl = conv.MustString(projectInfo.RecruitDdl)[0:19]
  784. sProjectData.ProjectDetail = projectInfo.ProjectDetail
  785. sProjectData.EstimatedCost = projectInfo.EstimatedCost
  786. sProjectData.PassAt = conv.MustString(projectInfo.PassAt)[0:19]
  787. sProjectData.Tools = projectInfo.Tools
  788. // 3. 取出招募策略并聚合达人数量信息
  789. recruitStrategy, recruitErr := db.GetRecruitStrategyByProjectId(ctx, ProjectId)
  790. if recruitErr != nil {
  791. return nil, recruitErr
  792. }
  793. if recruitStrategy != nil {
  794. for _, strategy := range recruitStrategy {
  795. showStrategy := http_model.ShowNewRecruitStrategy{
  796. StrategyID: strategy.StrategyID,
  797. FeeForm: strategy.FeeForm,
  798. RecruitStrategyID: strategy.RecruitStrategyID,
  799. FollowersLow: strategy.FollowersLow,
  800. FollowersUp: strategy.FollowersUp,
  801. RecruitNumber: strategy.RecruitNumber,
  802. ServiceCharge: projectInfo.ServiceChargeRate,
  803. Offer: strategy.Offer,
  804. }
  805. sProjectData.SRecruitStrategys = append(sProjectData.SRecruitStrategys, showStrategy)
  806. }
  807. }
  808. // 4. 取出种草任务创建者用户信息
  809. if projectInfo.OperatorType == 1 {
  810. // fmt.Println("商家用户")
  811. enterpriseInfo, enterpriseErr := db.GetEnterpriseByEnterpriseID(ctx, projectInfo.EnterpriseID)
  812. if enterpriseErr != nil {
  813. return nil, enterpriseErr
  814. }
  815. sProjectData.CreatorName = enterpriseInfo.BusinessName
  816. sProjectData.CreatorCompany = enterpriseInfo.BusinessName
  817. sProjectData.CreatorType = 1
  818. sProjectData.Phone = enterpriseInfo.BusinessName
  819. } else if projectInfo.OperatorType == 2 {
  820. // fmt.Println("商家子账号")
  821. enterpriseInfo, enterpriseErr := db.GetEnterpriseByEnterpriseID(ctx, projectInfo.EnterpriseID)
  822. if enterpriseErr != nil {
  823. return nil, enterpriseErr
  824. }
  825. sProjectData.CreatorCompany = enterpriseInfo.BusinessName
  826. subAccountInfo, SubAccountErr := db.FindSubAccountById(ctx, projectInfo.SubAccountId)
  827. if SubAccountErr != nil {
  828. return nil, SubAccountErr
  829. }
  830. sProjectData.Phone = subAccountInfo.PhoneNumber
  831. jobInfo, jobErr := db.FindJobByJobId(ctx, subAccountInfo.JobId)
  832. if jobErr != nil {
  833. return nil, jobErr
  834. }
  835. sProjectData.CreatorType = 2
  836. sProjectData.CreatorName = jobInfo.JobName
  837. }
  838. // 5. 商品信息
  839. // 5.1. 取出商品信息并聚合
  840. productInfo, productErr := db.GetProductByID(ctx, projectInfo.ProductID)
  841. if productErr != nil {
  842. return nil, productErr
  843. }
  844. if productInfo != nil {
  845. sProjectData.ProductID = productInfo.ProductID
  846. sProjectData.ProductName = productInfo.ProductName
  847. sProjectData.ProductType = productInfo.ProductType
  848. sProjectData.ProductPrice = productInfo.ProductPrice
  849. sProjectData.ProductCategory = productInfo.ProductCategory
  850. }
  851. // 5.2. 聚合商品图片信息
  852. productPhotoInfo, productPhotoErr := db.GetProductPhotoByProductID(ctx, projectInfo.ProductID)
  853. if productPhotoErr != nil {
  854. return nil, productPhotoErr
  855. }
  856. if productPhotoInfo != nil {
  857. for _, p := range productPhotoInfo {
  858. if p.Symbol == 1 {
  859. sProjectData.ProductMainPhotoUrl = p.PhotoUrl
  860. sProjectData.ProductMainPhotoUid = p.PhotoUid
  861. sProjectData.Symbol = 1
  862. }
  863. }
  864. }
  865. // 6. 执行要求Brief和素材
  866. // 6.1. Brief
  867. projectBrief, briefErr := db.FindProjectBriefByProjectId(ctx, sProjectData.ProjectID)
  868. if briefErr != nil {
  869. return nil, briefErr
  870. }
  871. if projectBrief != nil {
  872. sProjectData.ProjectBriefInfo = projectBrief
  873. }
  874. // 6.2. 素材
  875. projectMaterial, materialErr := db.FindProjectMaterialByProjectId(ctx, sProjectData.ProjectID)
  876. if materialErr != nil {
  877. return nil, materialErr
  878. }
  879. if projectMaterial != nil {
  880. sProjectData.ProjectMaterial = projectMaterial
  881. }
  882. }
  883. return sProjectData, nil
  884. }
  885. // CountProjectTask 按照状态统计子任务数量
  886. func (*project) CountProjectTask(ctx context.Context, sProjectId int) (*http_model.ProjectTaskCountData, error) {
  887. var counter *http_model.ProjectTaskCountData
  888. counter = &http_model.ProjectTaskCountData{}
  889. // 1. Stage1
  890. stage1, stage1Err := db.CountTaskNumByTaskStage(ctx, 0, 1, sProjectId)
  891. if stage1Err != nil {
  892. return nil, stage1Err
  893. }
  894. counter.Stage1 = stage1
  895. // 2. Stage2
  896. stage2, stage2Err := db.CountTaskNumByTaskStage(ctx, 0, 2, sProjectId)
  897. if stage2Err != nil {
  898. return nil, stage2Err
  899. }
  900. counter.Stage2 = stage2
  901. // 3. Stage3
  902. stage3, stage3Err := db.CountTaskNumByTaskStage(ctx, 0, 3, sProjectId)
  903. if stage3Err != nil {
  904. return nil, stage3Err
  905. }
  906. counter.Stage3 = stage3
  907. // 4. Stage4
  908. stage4, stage4Err := db.CountTaskNumByTaskStage(ctx, 1, 2, sProjectId)
  909. if stage4Err != nil {
  910. return nil, stage4Err
  911. }
  912. counter.Stage4 = stage4
  913. // 5. Stage5
  914. stage5, stage5Err := db.CountTaskNumByTaskStage(ctx, 2, 2, sProjectId)
  915. if stage5Err != nil {
  916. return nil, stage5Err
  917. }
  918. counter.Stage5 = stage5
  919. // 6. Stage6
  920. stage6, stage6Err := db.CountTaskNumByTaskStage(ctx, 3, 2, sProjectId)
  921. if stage6Err != nil {
  922. return nil, stage6Err
  923. }
  924. counter.Stage6 = stage6
  925. // 7. Stage7
  926. stage7, stage7Err := db.CountTaskNumByTaskStage(ctx, 0, 0, sProjectId)
  927. if stage7Err != nil {
  928. return nil, stage7Err
  929. }
  930. counter.Stage7 = stage7
  931. return counter, nil
  932. }