project.go 37 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012
  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, addToListStatus int) (*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 _, projectInfo := range fullProjectListData.FullProjectPreview {
  368. // 2.1. 商品信息
  369. productInfo, productErr := db.GetProductByID(ctx, projectInfo.ProductId)
  370. if productErr != nil {
  371. return nil, productErr
  372. }
  373. if productInfo != nil {
  374. projectInfo.ProductId = productInfo.ProductID
  375. projectInfo.ProductPrice = productInfo.ProductPrice
  376. projectInfo.ProductName = productInfo.ProductName
  377. }
  378. // 2.2. 商品图片信息
  379. productPhotoInfo, productPhotoErr := db.GetProductPhotoByProductID(ctx, projectInfo.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. projectInfo.ProductPhotoSymbol = 1
  388. projectInfo.ProductPhotoUrl = photo.PhotoUrl
  389. projectInfo.ProductPhotoUid = photo.PhotoUid
  390. }
  391. }
  392. }
  393. // 2.3. 招募策略信息
  394. recruitStrategyInfo, recruitErr := db.GetRecruitStrategyByProjectId(ctx, projectInfo.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. projectInfo.RecruitStrategy = append(projectInfo.RecruitStrategy, recruitStrategy)
  406. }
  407. }
  408. // 2.4. 判断是否加入商单
  409. sProjectCount, sProjectErr := db.FindSProjectByProjectIdAndSupplierId(ctx, projectInfo.ProjectId, supplierId)
  410. if sProjectErr != nil {
  411. return nil, sProjectErr
  412. }
  413. if sProjectCount > 0 {
  414. projectInfo.AddToListStatus = 1
  415. } else {
  416. projectInfo.AddToListStatus = 2
  417. }
  418. }
  419. var returnList *http_model.FullProjectListData
  420. returnList = &http_model.FullProjectListData{}
  421. returnList.Total = total
  422. for _, projectInfo := range fullProjectListData.FullProjectPreview {
  423. if addToListStatus == 1 {
  424. if projectInfo.AddToListStatus == 1 {
  425. returnList.FullProjectPreview = append(returnList.FullProjectPreview, projectInfo)
  426. } else {
  427. returnList.Total--
  428. }
  429. } else if projectInfo.AddToListStatus == 2 {
  430. if projectInfo.AddToListStatus == 2 {
  431. returnList.FullProjectPreview = append(returnList.FullProjectPreview, projectInfo)
  432. } else {
  433. returnList.Total--
  434. }
  435. }
  436. }
  437. return returnList, nil
  438. }
  439. func (*project) GetProjectDraftList(ctx context.Context, enterpriseID string, pageSize, pageNum int32, condition *common_model.ProjectCondition) (*http_model.ProjectDraftListData, error) {
  440. ProjectDrafts, total, err := db.GetProjectDraftList(ctx, enterpriseID, pageSize, pageNum, condition)
  441. if err != nil {
  442. logrus.WithContext(ctx).Errorf("[project service] call GetProjectDraftList error,err:%+v", err)
  443. return nil, err
  444. }
  445. ProjectDraftListData := new(http_model.ProjectDraftListData)
  446. ProjectDraftListData.ProjectDraftPreview = pack.MGormProjectDraftToHttpProjectDraftPreview(ProjectDrafts)
  447. ProjectDraftListData.Total = conv.MustString(total)
  448. return ProjectDraftListData, nil
  449. }
  450. func (*project) GetProjectTaskList(ctx context.Context, pageSize, pageNum int64, orderBy []string, orderDesc []int, conditions *common_model.TaskConditions) (*http_model.ProjectTaskListData, error) {
  451. // 1. 根据SProject和condition查询所有Task
  452. projectTasks, total, err := db.GetProjectTaskList(ctx, pageSize, pageNum, orderBy, orderDesc, conditions)
  453. if err != nil {
  454. logrus.WithContext(ctx).Errorf("[project service] call GetProjectTaskList error,err:%+v", err)
  455. return nil, err
  456. }
  457. // projectTaskListData := new(http_model.ProjectTaskListData)
  458. var projectTaskListData *http_model.ProjectTaskListData
  459. projectTaskListData = &http_model.ProjectTaskListData{}
  460. // fmt.Println(projectTasks)
  461. // projectTaskListData.ProjectTaskPreview = pack.MGormProjectTaskToHttpProjectTaskPreview(projectTasks)
  462. if projectTasks != nil {
  463. projectTaskListData.Total = total
  464. for _, projectTask := range projectTasks {
  465. var projectTaskInfo *http_model.ProjectTaskPreview
  466. projectTaskInfo = &http_model.ProjectTaskPreview{}
  467. // 2. 补充查询达人身份信息
  468. platformUserInfo, platformUserErr := db.FindUserInfoByOpenId(ctx, projectTask.OpenId)
  469. if platformUserErr != nil {
  470. return nil, platformUserErr
  471. }
  472. if platformUserInfo != nil {
  473. projectTaskInfo.TaskId = projectTask.TaskID
  474. projectTaskInfo.TaskStage = projectTask.TaskStage
  475. projectTaskInfo.ServiceCharge = projectTask.ServiceCharge
  476. projectTaskInfo.DraftFee = projectTask.DraftFee
  477. projectTaskInfo.SupportFee = projectTask.SupportFee
  478. projectTaskInfo.StrategyId = projectTask.StrategyID
  479. projectTaskInfo.TaskStatus = projectTask.TaskStatus
  480. projectTaskInfo.CreateDate = conv.MustString(projectTask.CreateDate)
  481. projectTaskInfo.SelectDate = conv.MustString(projectTask.SelectDate)
  482. projectTaskInfo.DeliveryDate = conv.MustString(projectTask.DeliveryDate)
  483. projectTaskInfo.SignedTime = conv.MustString(projectTask.SignedTime)
  484. projectTaskInfo.CurBreakAt = conv.MustString(projectTask.CurBreakAt)
  485. projectTaskInfo.FansNum = projectTask.FansNum
  486. projectTaskInfo.VoteAvg = projectTask.VoteAvg
  487. projectTaskInfo.CommitAvg = projectTask.CommitAvg
  488. projectTaskInfo.BOperator = projectTask.BOperator
  489. projectTaskInfo.BOperatorType = projectTask.BOperatorType
  490. projectTaskInfo.SOperator = projectTask.SOperator
  491. projectTaskInfo.SOperatorType = projectTask.SOperatorType
  492. projectTaskInfo.PlatformNickname = platformUserInfo.NickName
  493. projectTaskInfo.FansCount = platformUserInfo.Fan
  494. projectTaskInfo.AvatarUrl = platformUserInfo.HeadUri
  495. projectTaskInfo.Location = platformUserInfo.City
  496. projectTaskInfo.Gender = platformUserInfo.Gender
  497. projectTaskListData.ProjectTaskPreview = append(projectTaskListData.ProjectTaskPreview, projectTaskInfo)
  498. }
  499. }
  500. }
  501. // 2. 查找SProject中其他信息
  502. sProjectInfo, sProjectErr := db.GetSProjectDetail(ctx, conditions.SProjectId)
  503. if sProjectErr != nil {
  504. return nil, sProjectErr
  505. }
  506. if sProjectInfo != nil {
  507. projectTaskListData.SettleNum = sProjectInfo.SettleNum
  508. projectTaskListData.RecruitNum = sProjectInfo.RecruitNum
  509. projectTaskListData.QuitNum = sProjectInfo.QuitNum
  510. projectTaskListData.ServiceChargeSettle = sProjectInfo.ServiceChargeSettle
  511. projectTaskListData.ServiceChargeActual = sProjectInfo.ServiceChargeActual
  512. projectTaskListData.EstimateServiceCharge = sProjectInfo.ServiceChargeActual
  513. projectTaskListData.EstimateDraftFee = sProjectInfo.EstimateDraftFee
  514. projectTaskListData.EstimateSupportFee = sProjectInfo.EstimateSupportFee
  515. }
  516. return projectTaskListData, nil
  517. }
  518. func (*project) GetPorjectDetail(ctx context.Context, projectID string) (*http_model.ShowProjectData, error) {
  519. project, err := db.GetProjectDetail(ctx, projectID)
  520. if err != nil {
  521. logrus.WithContext(ctx).Errorf("[project service] call GetPorjectDetail error,err:%+v", err)
  522. return nil, err
  523. }
  524. enterprise, err := db.GetEnterpriseByEnterpriseID(ctx, project.EnterpriseID)
  525. if err != nil {
  526. logrus.WithContext(ctx).Errorf("[project service] call GetEnterpriseByEnterpriseID error,err:%+v", err)
  527. return nil, err
  528. }
  529. user, err := db.GetUserByID(ctx, enterprise.UserID)
  530. if err != nil {
  531. logrus.WithContext(ctx).Errorf("[project service] call GetUserByID error,err:%+v", err)
  532. return nil, err
  533. }
  534. ProjectDetail := http_model.ShowProjectData{
  535. ProjectID: conv.MustString(project.ProjectID),
  536. ProjectName: conv.MustString(project.ProjectName),
  537. ProjectStatus: conv.MustString(project.ProjectStatus),
  538. ProjectType: conv.MustString(project.ProjectType),
  539. ProjectPlatform: conv.MustString(project.ProjectPlatform),
  540. ProjectForm: conv.MustString(project.ProjectForm),
  541. TalentType: conv.MustString(project.TalentType),
  542. RecruitDdl: util.GetTimePoionter(project.RecruitDdl),
  543. ContentType: conv.MustString(project.ContentType),
  544. ProjectDetail: conv.MustString(project.ProjectDetail),
  545. ProductID: conv.MustString(project.ProductID),
  546. EnterpriseID: conv.MustString(project.EnterpriseID),
  547. Balance: conv.MustString(enterprise.Balance),
  548. AvailableBalance: conv.MustString(enterprise.AvailableBalance),
  549. EstimatedCost: conv.MustString(project.EstimatedCost),
  550. FailReason: conv.MustString(project.FailReason),
  551. CreateAt: util.GetTimePoionter(project.CreatedAt),
  552. UpdateAt: util.GetTimePoionter(project.UpdatedAt),
  553. Phone: user.Phone,
  554. FinishAt: util.GetTimePoionter(project.FinishAt),
  555. PassAt: util.GetTimePoionter(project.PassAt),
  556. SubmitAt: util.GetTimePoionter(project.SubmitAt),
  557. PayAt: util.GetTimePoionter(project.PayAt),
  558. ProductInfo: conv.MustString(project.ProductSnap),
  559. ProductPhotoInfo: conv.MustString(project.ProductPhotoSnap),
  560. AutoFailAt: util.GetTimePoionter(project.AutoFailAt),
  561. }
  562. Strategys, err := db.GetRecruitStrategys(ctx, projectID)
  563. if err != nil {
  564. logrus.WithContext(ctx).Error()
  565. return nil, err
  566. }
  567. for _, strategy := range Strategys {
  568. RecruitStrategy := http_model.ShowRecruitStrategy{
  569. RecruitStrategyID: conv.MustString(strategy.RecruitStrategyID),
  570. FeeForm: conv.MustString(strategy.FeeForm),
  571. StrategyID: conv.MustString(strategy.StrategyID),
  572. FollowersLow: conv.MustString(strategy.FollowersLow),
  573. FollowersUp: conv.MustString(strategy.FollowersUp),
  574. RecruitNumber: conv.MustString(strategy.RecruitNumber),
  575. Offer: conv.MustString(strategy.Offer),
  576. ServiceCharge: conv.MustString(strategy.ServiceCharge),
  577. SelectedNumber: strategy.SelectedNumber,
  578. WaitingNumber: strategy.WaitingNumber,
  579. DeliveredNumber: strategy.DeliveredNumber,
  580. SignedNumber: strategy.SignedNumber,
  581. }
  582. ProjectDetail.RecruitStrategys = append(ProjectDetail.RecruitStrategys, RecruitStrategy)
  583. }
  584. Photos, err := db.GetProjectPhoto(ctx, projectID)
  585. if err != nil {
  586. logrus.WithContext(ctx).Error()
  587. return nil, err
  588. }
  589. for _, Photo := range Photos {
  590. ProjectPhoto := http_model.ShowProjectPhoto{
  591. PhotoUrl: Photo.PhotoUrl,
  592. PhotoUid: Photo.PhotoUid,
  593. FileName: Photo.FileName,
  594. }
  595. ProjectDetail.ProjectPhotos = append(ProjectDetail.ProjectPhotos, ProjectPhoto)
  596. }
  597. return &ProjectDetail, nil
  598. }
  599. func (*project) GetTaskLogisticsList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskLogisticsListData, error) {
  600. TaskLogisticss, total, err := db.GetTaskLogisticsList(ctx, projectID, pageSize, pageNum, conditions)
  601. if err != nil {
  602. logrus.WithContext(ctx).Errorf("[project service] call GetTaskLogisticsList error,err:%+v", err)
  603. return nil, err
  604. }
  605. TaskLogisticsListData := new(http_model.TaskLogisticsListData)
  606. TaskLogisticsListData.TaskLogisticsPreview = pack.MGormTaskLogisticsInfoListToHttpTaskLogisticsPreviewList(TaskLogisticss)
  607. TaskLogisticsListData.Total = conv.MustString(total)
  608. return TaskLogisticsListData, nil
  609. }
  610. // ChangeTaskStatus 提报达人或拒绝提报达人
  611. func (*project) ChangeTaskStatus(ctx *gin.Context, data http_model.ProjectChangeTaskStatusRequest) interface{} {
  612. var sOperatorType int
  613. if data.SubAccountId == 0 {
  614. sOperatorType = 1
  615. } else {
  616. sOperatorType = 2
  617. }
  618. RecruitStrategyIDs, err := db.ChangeTaskStatus(ctx, data.TaskIds, data.SupplierStatus, data.SupplierId, data.SubAccountId, sOperatorType)
  619. if err != nil {
  620. logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  621. return err
  622. }
  623. fmt.Println(RecruitStrategyIDs)
  624. // 已选数量
  625. //if data.SupplierStatus == 2 {
  626. // for _, RecruitStrategyID := range RecruitStrategyIDs {
  627. // err = db.CalculateSelectedNumberByRecruitStrategyID(ctx, RecruitStrategyID, 1)
  628. // if err != nil {
  629. // logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  630. // return err
  631. // }
  632. // }
  633. //} else if data.SupplierStatus == 3 {
  634. // for _, RecruitStrategyID := range RecruitStrategyIDs {
  635. // err = db.CalculateSelectedNumberByRecruitStrategyID(ctx, RecruitStrategyID, 1)
  636. // if err != nil {
  637. // logrus.WithContext(ctx).Errorf("[project service] call ChangeTaskStatus error,err:%+v", err)
  638. // return err
  639. // }
  640. // }
  641. //}
  642. return nil
  643. }
  644. // ChangeSpecialTaskStatus 定向种草任务 提报达人,拒绝提报
  645. /*
  646. func (*project) ChangeSpecialTaskStatus(ctx *gin.Context, data http_model.ProjectChangeTaskStatusRequest) interface{} {
  647. err := db.ChangeSpecialTaskStatus(ctx, data.TaskIds, data.SupplierStatus, data.SupplierId, data.SubAccountId, data.SOperatorType)
  648. if err != nil {
  649. logrus.WithContext(ctx).Errorf("[project service] call ChangeSpecialTaskStatus error,err:%+v", err)
  650. return err
  651. }
  652. return nil
  653. }
  654. */
  655. func (p *project) GetTaskScriptList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskScriptListData, error) {
  656. TaskScripts, total, err := db.GetTaskScriptList(ctx, projectID, pageSize, pageNum, conditions)
  657. if err != nil {
  658. logrus.WithContext(ctx).Errorf("[project service] call GetTaskScriptList error,err:%+v", err)
  659. return nil, err
  660. }
  661. TaskScriptListData := new(http_model.TaskScriptListData)
  662. TaskScriptListData.TaskScriptPreview = pack.MGormTaskScriptInfoListToHttpTaskScriptPreviewList(TaskScripts)
  663. TaskScriptListData.Total = conv.MustString(total)
  664. return TaskScriptListData, nil
  665. }
  666. func (p *project) GetTaskSketchList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskSketchListData, error) {
  667. TaskSketchs, total, err := db.GetTaskSketchList(ctx, projectID, pageSize, pageNum, conditions)
  668. if err != nil {
  669. logrus.WithContext(ctx).Errorf("[project service] call GetTaskSketchList error,err:%+v", err)
  670. return nil, err
  671. }
  672. TaskSketchListData := new(http_model.TaskSketchListData)
  673. TaskSketchListData.TaskSketchPreview = pack.MGormTaskSketchInfoListToHttpTaskSketchPreviewList(TaskSketchs)
  674. TaskSketchListData.Total = conv.MustString(total)
  675. return TaskSketchListData, nil
  676. }
  677. func (p *project) GetTaskLinkList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskLinkListData, error) {
  678. TaskLinks, total, err := db.GetTaskLinkList(ctx, projectID, pageSize, pageNum, conditions)
  679. if err != nil {
  680. logrus.WithContext(ctx).Errorf("[project service] call GetTaskLinkList error,err:%+v", err)
  681. return nil, err
  682. }
  683. TaskLinkListData := new(http_model.TaskLinkListData)
  684. TaskLinkListData.TaskLinkPreview = pack.MGormTaskLinkInfoListToHttpTaskLinkPreviewList(TaskLinks)
  685. TaskLinkListData.Total = conv.MustString(total)
  686. return TaskLinkListData, nil
  687. }
  688. func (p *project) GetTaskDataList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskDataListData, error) {
  689. TaskDatas, total, err := db.GetTaskDataList(ctx, projectID, pageSize, pageNum, conditions)
  690. if err != nil {
  691. logrus.WithContext(ctx).Errorf("[project service] call GetTaskDataList error,err:%+v", err)
  692. return nil, err
  693. }
  694. TaskDataListData := new(http_model.TaskDataListData)
  695. TaskDataListData.TaskDataPreview = pack.MGormTaskDataInfoListToHttpTaskDataPreviewList(TaskDatas)
  696. TaskDataListData.Total = conv.MustString(total)
  697. return TaskDataListData, nil
  698. }
  699. func (p *project) GetTaskFinishList(ctx *gin.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.TaskFinishListData, error) {
  700. TaskFinishs, total, err := db.GetTaskFinishList(ctx, pageSize, pageNum, conditions)
  701. if err != nil {
  702. logrus.WithContext(ctx).Errorf("[project service] call GetTaskFinishList error,err:%+v", err)
  703. return nil, err
  704. }
  705. TaskFinishListData := new(http_model.TaskFinishListData)
  706. TaskFinishListData.TaskFinishPreview = pack.MGormTaskFinishInfoListToHttpTaskFinishPreviewList(TaskFinishs)
  707. TaskFinishListData.Total = conv.MustString(total)
  708. return TaskFinishListData, nil
  709. }
  710. func (p *project) GetServiceCharge(ctx *gin.Context, data http_model.GetServiceChargeRequest) (*http_model.ServiceChargeData, error) {
  711. pricingStrategy, err := db.GetPricingStrategy(ctx, data.FollowersLow, data.FollowersUp, data.FeeForm, data.Platform)
  712. if err != nil {
  713. return nil, err
  714. }
  715. serviceFee := http_model.ServiceChargeData{
  716. ServiceCharge: pricingStrategy.ServiceCharge,
  717. }
  718. return &serviceFee, nil
  719. }
  720. func (*project) GetSpecialProjectTaskList(ctx context.Context, projectID string, pageSize, pageNum int64, conditions *common_model.TalentConditions) (*http_model.SpecialTaskLogisticsListData, error) {
  721. SpecialTaskLogisticsListDatas, total, err := db.GetSpecialTaskLogisticsList(ctx, projectID, pageSize, pageNum, conditions)
  722. if err != nil {
  723. logrus.WithContext(ctx).Errorf("[project service] call GetSpecialProjectTaskList error,err:%+v", err)
  724. return nil, err
  725. }
  726. TaskLogisticsListData := new(http_model.SpecialTaskLogisticsListData)
  727. TaskLogisticsListData.SpecialTaskLogisticsPreview = pack.MGormSpecialTaskLogisticsInfoListToHttpTaskLogisticsPreviewList(SpecialTaskLogisticsListDatas)
  728. TaskLogisticsListData.Total = conv.MustString(total, "")
  729. return TaskLogisticsListData, nil
  730. }
  731. // ShowTaskProgress ToDo
  732. // ShowTaskProgress 展示种草子任务进度
  733. func (p *project) ShowTaskProgress(ctx *gin.Context, data http_model.ShowTaskProgressRequest) (*http_model.ShowTaskProgressData, error) {
  734. // 1. 初始化返回数据结构体
  735. var taskProgressData *http_model.ShowTaskProgressData
  736. taskProgressData = &http_model.ShowTaskProgressData{}
  737. // 2. 根据task_id筛选出对应的子任务基础信息
  738. taskInfo, taskErr := db.GetTaskByTaskId(ctx, data.TaskId)
  739. if taskErr != nil {
  740. return nil, taskErr
  741. }
  742. if taskInfo != nil {
  743. taskProgressData.TaskId = data.TaskId
  744. taskProgressData.CreateData = taskInfo.CreateDate
  745. taskProgressData.SelectDate = taskInfo.SelectDate
  746. taskProgressData.DeliveryDate = taskInfo.DeliveryDate
  747. taskProgressData.SignedTime = taskInfo.SignedTime
  748. taskProgressData.ServiceCharge = taskInfo.ServiceCharge
  749. taskProgressData.DraftFee = taskInfo.DraftFee
  750. }
  751. // 3. 筛选出上传初稿时间和初稿状态
  752. taskSketchs, taskSketchErr := db.FindSketchInfoByTaskId(ctx, data.TaskId)
  753. if taskSketchErr != nil {
  754. return nil, taskSketchErr
  755. }
  756. if taskSketchs != nil {
  757. taskProgressData.SketchInfo = taskSketchs
  758. }
  759. // 4. 筛选作品链接上传时间和状态
  760. taskLinks, taskLinksErr := db.GetLinkByTaskId(ctx, data.TaskId)
  761. if taskLinksErr != nil {
  762. return nil, taskLinksErr
  763. }
  764. if taskLinks != nil {
  765. taskProgressData.LinkInfo = taskLinks
  766. }
  767. // 5. 若有违约,筛选出违约情况
  768. if taskInfo != nil {
  769. taskProgressData.TerminateTime = taskInfo.TerminateTime
  770. taskProgressData.TerminateReason = taskInfo.TerminateReason
  771. taskProgressData.CancelTime = taskInfo.CancelTime
  772. taskProgressData.CancelReason = taskInfo.CancelReason
  773. }
  774. return taskProgressData, nil
  775. }
  776. func (p *project) GetProjectStrategys(ctx *gin.Context, projectId string) ([]gorm_model.RecruitStrategy, error) {
  777. recruitStrategys, getProjectStrategysErr := db.GetRecruitStrategyByProjectId(ctx, projectId)
  778. if getProjectStrategysErr != nil {
  779. return nil, getProjectStrategysErr
  780. }
  781. return recruitStrategys, nil
  782. }
  783. // GetProjectDetail 查找种草任务详情
  784. func (*project) GetProjectDetail(ctx context.Context, ProjectId string) (*http_model.ShowNewProjectData, error) {
  785. var sProjectData *http_model.ShowNewProjectData
  786. sProjectData = &http_model.ShowNewProjectData{}
  787. // 1. 取出种草表中的信息
  788. projectInfo, projectErr := db.GetProjectDetail(ctx, ProjectId)
  789. if projectErr != nil {
  790. return nil, projectErr
  791. }
  792. if projectInfo != nil {
  793. sProjectData.ProjectName = projectInfo.ProjectName
  794. sProjectData.ProjectID = projectInfo.ProjectID
  795. sProjectData.ProjectType = projectInfo.ProjectType
  796. sProjectData.ProjectPlatform = projectInfo.ProjectPlatform
  797. sProjectData.ProjectForm = projectInfo.ProjectForm
  798. sProjectData.ContentType = projectInfo.ContentType
  799. sProjectData.EnterpriseID = projectInfo.EnterpriseID
  800. sProjectData.TalentType = projectInfo.TalentType
  801. sProjectData.RecruitDdl = conv.MustString(projectInfo.RecruitDdl)[0:19]
  802. sProjectData.ProjectDetail = projectInfo.ProjectDetail
  803. sProjectData.EstimatedCost = projectInfo.EstimatedCost
  804. sProjectData.PassAt = conv.MustString(projectInfo.PassAt)[0:19]
  805. sProjectData.Tools = projectInfo.Tools
  806. // 3. 取出招募策略并聚合达人数量信息
  807. recruitStrategy, recruitErr := db.GetRecruitStrategyByProjectId(ctx, ProjectId)
  808. if recruitErr != nil {
  809. return nil, recruitErr
  810. }
  811. if recruitStrategy != nil {
  812. for _, strategy := range recruitStrategy {
  813. showStrategy := http_model.ShowNewRecruitStrategy{
  814. StrategyID: strategy.StrategyID,
  815. FeeForm: strategy.FeeForm,
  816. RecruitStrategyID: strategy.RecruitStrategyID,
  817. FollowersLow: strategy.FollowersLow,
  818. FollowersUp: strategy.FollowersUp,
  819. RecruitNumber: strategy.RecruitNumber,
  820. ServiceCharge: projectInfo.ServiceChargeRate,
  821. Offer: strategy.Offer,
  822. }
  823. sProjectData.SRecruitStrategys = append(sProjectData.SRecruitStrategys, showStrategy)
  824. }
  825. }
  826. // 4. 取出种草任务创建者用户信息
  827. if projectInfo.OperatorType == 1 {
  828. // fmt.Println("商家用户")
  829. enterpriseInfo, enterpriseErr := db.GetEnterpriseByEnterpriseID(ctx, projectInfo.EnterpriseID)
  830. if enterpriseErr != nil {
  831. return nil, enterpriseErr
  832. }
  833. sProjectData.CreatorName = enterpriseInfo.BusinessName
  834. sProjectData.CreatorCompany = enterpriseInfo.BusinessName
  835. sProjectData.CreatorType = 1
  836. sProjectData.Phone = enterpriseInfo.BusinessName
  837. } else if projectInfo.OperatorType == 2 {
  838. // fmt.Println("商家子账号")
  839. enterpriseInfo, enterpriseErr := db.GetEnterpriseByEnterpriseID(ctx, projectInfo.EnterpriseID)
  840. if enterpriseErr != nil {
  841. return nil, enterpriseErr
  842. }
  843. sProjectData.CreatorCompany = enterpriseInfo.BusinessName
  844. subAccountInfo, SubAccountErr := db.FindSubAccountById(ctx, projectInfo.SubAccountId)
  845. if SubAccountErr != nil {
  846. return nil, SubAccountErr
  847. }
  848. sProjectData.Phone = subAccountInfo.PhoneNumber
  849. jobInfo, jobErr := db.FindJobByJobId(ctx, subAccountInfo.JobId)
  850. if jobErr != nil {
  851. return nil, jobErr
  852. }
  853. sProjectData.CreatorType = 2
  854. sProjectData.CreatorName = jobInfo.JobName
  855. }
  856. // 5. 商品信息
  857. // 5.1. 取出商品信息并聚合
  858. productInfo, productErr := db.GetProductByID(ctx, projectInfo.ProductID)
  859. if productErr != nil {
  860. return nil, productErr
  861. }
  862. if productInfo != nil {
  863. sProjectData.ProductID = productInfo.ProductID
  864. sProjectData.ProductName = productInfo.ProductName
  865. sProjectData.ProductType = productInfo.ProductType
  866. sProjectData.ProductPrice = productInfo.ProductPrice
  867. sProjectData.ProductCategory = productInfo.ProductCategory
  868. }
  869. // 5.2. 聚合商品图片信息
  870. productPhotoInfo, productPhotoErr := db.GetProductPhotoByProductID(ctx, projectInfo.ProductID)
  871. if productPhotoErr != nil {
  872. return nil, productPhotoErr
  873. }
  874. if productPhotoInfo != nil {
  875. for _, p := range productPhotoInfo {
  876. if p.Symbol == 1 {
  877. sProjectData.ProductMainPhotoUrl = p.PhotoUrl
  878. sProjectData.ProductMainPhotoUid = p.PhotoUid
  879. sProjectData.Symbol = 1
  880. }
  881. }
  882. }
  883. // 6. 执行要求Brief和素材
  884. // 6.1. Brief
  885. projectBrief, briefErr := db.FindProjectBriefByProjectId(ctx, sProjectData.ProjectID)
  886. if briefErr != nil {
  887. return nil, briefErr
  888. }
  889. if projectBrief != nil {
  890. sProjectData.ProjectBriefInfo = projectBrief
  891. }
  892. // 6.2. 素材
  893. projectMaterial, materialErr := db.FindProjectMaterialByProjectId(ctx, sProjectData.ProjectID)
  894. if materialErr != nil {
  895. return nil, materialErr
  896. }
  897. if projectMaterial != nil {
  898. sProjectData.ProjectMaterial = projectMaterial
  899. }
  900. }
  901. return sProjectData, nil
  902. }
  903. // CountProjectTask 按照状态统计子任务数量
  904. func (*project) CountProjectTask(ctx context.Context, sProjectId int) (*http_model.ProjectTaskCountData, error) {
  905. var counter *http_model.ProjectTaskCountData
  906. counter = &http_model.ProjectTaskCountData{}
  907. // 1. Stage1
  908. stage1, stage1Err := db.CountTaskNumByTaskStage(ctx, 0, 1, sProjectId)
  909. if stage1Err != nil {
  910. return nil, stage1Err
  911. }
  912. counter.Stage1 = stage1
  913. // 2. Stage2
  914. stage2, stage2Err := db.CountTaskNumByTaskStage(ctx, 0, 2, sProjectId)
  915. if stage2Err != nil {
  916. return nil, stage2Err
  917. }
  918. counter.Stage2 = stage2
  919. // 3. Stage3
  920. stage3, stage3Err := db.CountTaskNumByTaskStage(ctx, 0, 3, sProjectId)
  921. if stage3Err != nil {
  922. return nil, stage3Err
  923. }
  924. counter.Stage3 = stage3
  925. // 4. Stage4
  926. stage4, stage4Err := db.CountTaskNumByTaskStage(ctx, 1, 2, sProjectId)
  927. if stage4Err != nil {
  928. return nil, stage4Err
  929. }
  930. counter.Stage4 = stage4
  931. // 5. Stage5
  932. stage5, stage5Err := db.CountTaskNumByTaskStage(ctx, 2, 2, sProjectId)
  933. if stage5Err != nil {
  934. return nil, stage5Err
  935. }
  936. counter.Stage5 = stage5
  937. // 6. Stage6
  938. stage6, stage6Err := db.CountTaskNumByTaskStage(ctx, 3, 2, sProjectId)
  939. if stage6Err != nil {
  940. return nil, stage6Err
  941. }
  942. counter.Stage6 = stage6
  943. // 7. Stage7
  944. stage7, stage7Err := db.CountTaskNumByTaskStage(ctx, 0, 0, sProjectId)
  945. if stage7Err != nil {
  946. return nil, stage7Err
  947. }
  948. counter.Stage7 = stage7
  949. return counter, nil
  950. }