operate.go 81 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642
  1. package db
  2. import (
  3. "context"
  4. "fmt"
  5. "log"
  6. "reflect"
  7. "time"
  8. "youngee_m_api/model/common_model"
  9. "youngee_m_api/model/gorm_model"
  10. "youngee_m_api/model/http_model"
  11. "youngee_m_api/util"
  12. "github.com/caixw/lib.go/conv"
  13. "github.com/sirupsen/logrus"
  14. )
  15. func CreatePricingStrategy(ctx context.Context, req *http_model.AddPricingRequest) (string, string, error) {
  16. db := GetReadDB(ctx)
  17. project_type, fee_form, platform, fans_low, fans_up, service_charge, base_offer := req.ProjectType, req.ManuscriptForm, req.Platform, req.FansLow, req.FansHigh, req.PlatformFee, req.BaseOffer
  18. var IsExclusive, IsExclusive2, IsExclusive3, IsExclusive1 gorm_model.InfoPricingStrategy
  19. err := db.Where(" fee_form = ? && platform = ? && fans_low <= ? && fans_up >= ?", fee_form, platform, fans_low, fans_up).First(&IsExclusive).Error
  20. if err == nil {
  21. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", "", nil
  22. }
  23. err = db.Where(" fee_form = ? && platform = ? && fans_low < ? && fans_up > ?", fee_form, platform, fans_low, fans_low).First(&IsExclusive1).Error
  24. if err == nil {
  25. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", "", nil
  26. }
  27. err = db.Where(" fee_form = ? && platform = ? && fans_low < ? && fans_up > ?", fee_form, platform, fans_up, fans_up).First(&IsExclusive2).Error
  28. if err == nil {
  29. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", "", nil
  30. }
  31. err = db.Where(" fee_form = ? && platform = ? && fans_low >= ? && fans_up <= ?", fee_form, platform, fans_low, fans_up).First(&IsExclusive3).Error
  32. if err == nil {
  33. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", "", nil
  34. }
  35. var newStrategy gorm_model.InfoPricingStrategy
  36. var strategyInfo gorm_model.InfoPricingStrategy
  37. getMonth := time.Now().Format("01") //获取月
  38. getDay := time.Now().Day() //获取日
  39. dayString := ""
  40. if getDay < 10 {
  41. dayString = conv.MustString(getDay, "")
  42. dayString = "0" + dayString
  43. } else {
  44. dayString = conv.MustString(getDay, "")
  45. }
  46. monthAndDay := conv.MustString(getMonth, "") + dayString
  47. err = db.Where(fmt.Sprintf(" strategyId like '%s%%'", monthAndDay)).Last(&strategyInfo).Error
  48. num := 1
  49. if strategyInfo.StrategyId != "" {
  50. num = conv.MustInt(strategyInfo.StrategyId[4:6], 0)
  51. num = num + 1 //获取日
  52. } else {
  53. num = 1
  54. }
  55. numString := ""
  56. if num < 10 {
  57. numString = conv.MustString(num, "")
  58. numString = "0" + numString
  59. } else {
  60. numString = conv.MustString(num, "")
  61. }
  62. newStrategy.StrategyId = monthAndDay + numString
  63. newStrategy.Status = 0
  64. newStrategy.FansLow = fans_low
  65. newStrategy.FansUp = fans_up
  66. newStrategy.ProjectType = conv.MustInt64(project_type, 0)
  67. newStrategy.BaseOffer = conv.MustFloat32(base_offer, 0)
  68. newStrategy.Platform = conv.MustInt64(platform, 0)
  69. newStrategy.FeeForm = conv.MustInt64(fee_form, 0)
  70. newStrategy.ServiceCharge = conv.MustInt64(service_charge, 0)
  71. newStrategy.CreateAt = time.Now()
  72. newStrategy.UpdateAt = time.Now()
  73. err = db.Create(&newStrategy).Error
  74. if err != nil {
  75. return "创建失败", "", err
  76. }
  77. return "创建成功", monthAndDay + numString, nil
  78. }
  79. func SearchPricing(ctx context.Context, pageSize, pageNum int32, conditions *common_model.PricingConditions) ([]*gorm_model.InfoPricingStrategy, int64, error) {
  80. db := GetReadDB(ctx)
  81. db = db.Model(gorm_model.InfoPricingStrategy{})
  82. // 根据 查询条件过滤
  83. conditionType := reflect.TypeOf(conditions).Elem()
  84. conditionValue := reflect.ValueOf(conditions).Elem()
  85. for i := 0; i < conditionType.NumField(); i++ {
  86. field := conditionType.Field(i)
  87. tag := field.Tag.Get("condition")
  88. value := conditionValue.FieldByName(field.Name)
  89. if !util.IsBlank(value) && tag != "update_at" {
  90. db = db.Where(fmt.Sprintf("%s = ?", tag), value.Interface())
  91. } else if tag == "update_at" && value.Interface() != nil {
  92. db = db.Where(fmt.Sprintf("update_at like '%s%%'", value.Interface()))
  93. }
  94. }
  95. // 查询总数
  96. var total int64
  97. var PricingDatas []*gorm_model.InfoPricingStrategy
  98. if err := db.Count(&total).Error; err != nil {
  99. logrus.WithContext(ctx).Errorf("[SearchPricing] error query mysql total, err:%+v", err)
  100. return nil, 0, err
  101. }
  102. // 查询该页数据
  103. limit := pageSize
  104. offset := pageSize * pageNum // assert pageNum start with 0
  105. err := db.Order("update_at desc").Limit(int(limit)).Offset(int(offset)).Find(&PricingDatas).Error
  106. if err != nil {
  107. logrus.WithContext(ctx).Errorf("[SearchPricing] error query mysql find, err:%+v", err)
  108. return nil, 0, err
  109. }
  110. return PricingDatas, total, nil
  111. }
  112. func ModifyPricing(ctx context.Context, req *http_model.ModifyPricingRequest) (string, error) {
  113. db := GetReadDB(ctx)
  114. strategyId := req.StrategyId
  115. db = db.Model(gorm_model.InfoPricingStrategy{}).Where("strategyId = ?", strategyId)
  116. project_type, fee_form, platform, fans_low, fans_up, service_charge, base_offer := req.ProjectType, req.FeeForm, req.Platform, req.FansLow, req.FansHigh, req.PlatformFee, req.BaseOffer
  117. var IsExclusive, IsExclusive2, IsExclusive3, IsExclusive1 gorm_model.InfoPricingStrategy
  118. err := db.Where(" fee_form = ? && platform = ? && fans_low <= ? && fans_up >= ?", fee_form, platform, fans_low, fans_up).First(&IsExclusive).Error
  119. if err == nil {
  120. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", nil
  121. }
  122. err = db.Where(" fee_form = ? && platform = ? && fans_low < ? && fans_up > ?", fee_form, platform, fans_low, fans_low).First(&IsExclusive1).Error
  123. if err == nil {
  124. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", nil
  125. }
  126. err = db.Where(" fee_form = ? && platform = ? && fans_low < ? && fans_up > ?", fee_form, platform, fans_up, fans_up).First(&IsExclusive2).Error
  127. if err == nil {
  128. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", nil
  129. }
  130. err = db.Where(" fee_form = ? && platform = ? && fans_low >= ? && fans_up <= ?", fee_form, platform, fans_low, fans_up).First(&IsExclusive3).Error
  131. if err == nil {
  132. return "该策略与已经运行的策略互斥,请修改相关字段后重新创建", nil
  133. }
  134. err = db.Updates(map[string]interface{}{
  135. "project_type": project_type, "base_offer": base_offer, "fee_form": fee_form,
  136. "platform": platform, "fans_low": fans_low, "fans_up": fans_up, "service_charge": service_charge,
  137. }).Error
  138. if err != nil {
  139. return "", err
  140. }
  141. return "修改成功", nil
  142. }
  143. func UpdateAutoTaskTime(ctx context.Context, time int32, num int32) error {
  144. db := GetReadDB(ctx)
  145. var autoTaskInfo gorm_model.InfoAutoTask
  146. err := db.Model(gorm_model.InfoAutoTask{}).Last(&autoTaskInfo).Error
  147. if err != nil {
  148. logrus.WithContext(ctx).Errorf("[UpdateAutoTaskTime] error get Last mysql, err:%+v", err)
  149. return err
  150. }
  151. switch num {
  152. case 1:
  153. autoTaskInfo.SignInOffline = time
  154. case 2:
  155. autoTaskInfo.SignInVirtual = time
  156. case 3:
  157. autoTaskInfo.ReviewInMv = time
  158. case 4:
  159. autoTaskInfo.ReviewUnlimited = time
  160. case 5:
  161. autoTaskInfo.Postreview = time
  162. case 6:
  163. autoTaskInfo.CaseClose = time
  164. case 7:
  165. autoTaskInfo.Invalid = time
  166. case 8:
  167. autoTaskInfo.DraftDefaultInPic = time
  168. case 9:
  169. autoTaskInfo.DraftDefaultInMv = time
  170. case 10:
  171. autoTaskInfo.ScriptDefault = time
  172. case 11:
  173. autoTaskInfo.LinkBreach = time
  174. case 12:
  175. autoTaskInfo.CaseCloseDefault = time
  176. }
  177. autoTaskInfo.AutoTaskID += 1
  178. err = db.Model(gorm_model.InfoAutoTask{}).Create(&autoTaskInfo).Error
  179. if err != nil {
  180. logrus.WithContext(ctx).Errorf("[UpdateAutoTaskTime] error Create mysql, err:%+v", err)
  181. return err
  182. }
  183. return nil
  184. }
  185. func UpdateAutoDefaultRate(ctx context.Context, rate int, num int32) error {
  186. db := GetReadDB(ctx)
  187. autoDefaultRateInfo := gorm_model.InfoAutoDefaultHandle{}
  188. err := db.Model(&gorm_model.InfoAutoDefaultHandle{}).Last(&autoDefaultRateInfo).Error
  189. if err != nil {
  190. logrus.WithContext(ctx).Errorf("[UpdateAutoDefaultRate] error get Last mysql, err:%+v", err)
  191. return err
  192. }
  193. switch num {
  194. case 1:
  195. autoDefaultRateInfo.SketchReplaceNotUpload = rate
  196. case 2:
  197. autoDefaultRateInfo.SketchReplaceTimeOut = rate
  198. case 3:
  199. autoDefaultRateInfo.SketchOtherNotUpload = rate
  200. case 4:
  201. autoDefaultRateInfo.SketchOtherTimeOut = rate
  202. case 5:
  203. autoDefaultRateInfo.ScriptReplaceNotUpload = rate
  204. case 6:
  205. autoDefaultRateInfo.ScriptReplaceTimeOut = rate
  206. case 7:
  207. autoDefaultRateInfo.ScriptOtherNotUpload = rate
  208. case 8:
  209. autoDefaultRateInfo.ScriptOtherTimeOut = rate
  210. case 9:
  211. autoDefaultRateInfo.LinkReplaceNotUpload = rate
  212. case 10:
  213. autoDefaultRateInfo.LinkReplaceTimeOut = rate
  214. case 11:
  215. autoDefaultRateInfo.LinkOtherNotUpload = rate
  216. case 12:
  217. autoDefaultRateInfo.LinkOtherTimeOut = rate
  218. case 13:
  219. autoDefaultRateInfo.DataReplaceNotUpload = rate
  220. case 14:
  221. autoDefaultRateInfo.DataReplaceTimeOut = rate
  222. case 15:
  223. autoDefaultRateInfo.DataOtherNotUpload = rate
  224. case 16:
  225. autoDefaultRateInfo.DataOtherTimeOut = rate
  226. }
  227. autoDefaultRateInfo.AutoDefaultID += 1
  228. err = db.Model(&gorm_model.InfoAutoDefaultHandle{}).Create(autoDefaultRateInfo).Error
  229. if err != nil {
  230. logrus.WithContext(ctx).Errorf("[UpdateAutoDefaultRate] error Create mysql, err:%+v", err)
  231. return err
  232. }
  233. return nil
  234. }
  235. func GetSignInOfflineTask(projectForm int32) error {
  236. db := GetReadDB(context.Background())
  237. var projectInfos []*gorm_model.ProjectInfo
  238. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND project_form = ?", 1, projectForm).Find(&projectInfos).Error
  239. if err != nil {
  240. return err
  241. }
  242. projectIdMap := make(map[int64]int64)
  243. taskIDs := make(map[int]int)
  244. var projectIds []int64
  245. var taskIds []int
  246. var t int32
  247. for _, projectInfo := range projectInfos {
  248. autoTaskId := projectInfo.AutoTaskID
  249. dbStart := GetReadDB(context.Background())
  250. var autoTaskInfo gorm_model.InfoAutoTask
  251. dbStart.Model(gorm_model.InfoAutoTask{}).Where("auto_task_id = ?", autoTaskId).First(&autoTaskInfo)
  252. //fmt.Println("t:", autoTaskInfo.SignInOffline)
  253. if projectForm == 3 {
  254. t = autoTaskInfo.SignInOffline
  255. } else {
  256. t = autoTaskInfo.SignInVirtual
  257. }
  258. projectIds = append(projectIds, projectInfo.ProjectID)
  259. if _, ok := projectIdMap[projectInfo.ProjectID]; !ok {
  260. projectIdMap[projectInfo.ProjectID] = projectInfo.ContentType
  261. db1 := GetReadDB(context.Background())
  262. var taskInfos []gorm_model.YoungeeTaskInfo
  263. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND logistics_status = ?", projectInfo.ProjectID, 2).Find(&taskInfos)
  264. //fmt.Println("taskInfos", taskInfos)
  265. for _, taskInfo := range taskInfos {
  266. //fmt.Println("TaskId", taskInfo.TaskId)
  267. if _, ok1 := taskIDs[taskInfo.TaskId]; !ok1 {
  268. taskIds = append(taskIds, taskInfo.TaskId)
  269. taskIDs[taskInfo.TaskId] = taskInfo.ProjectId
  270. var logisticInfo gorm_model.YoungeeTaskLogistics
  271. db2 := GetReadDB(context.Background())
  272. db2.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskInfo.TaskId).Find(&logisticInfo)
  273. fmt.Println("logisticID", logisticInfo.LogisticsID)
  274. dd, _ := time.ParseDuration(conv.MustString(t, "") + "h")
  275. if logisticInfo.AutoSignAt.IsZero() {
  276. db5 := GetReadDB(context.Background())
  277. db5.Model(gorm_model.YoungeeTaskLogistics{}).Where("logistics_id = ?", logisticInfo.LogisticsID).
  278. Updates(&gorm_model.YoungeeTaskLogistics{AutoSignAt: logisticInfo.DeliveryTime.Add(dd)})
  279. fmt.Println("已添加自动签收时间")
  280. }
  281. }
  282. }
  283. }
  284. }
  285. //fmt.Println("projectIdMap", projectIdMap)
  286. //fmt.Println("taskIds", taskIds)
  287. var logisticInfos []gorm_model.YoungeeTaskLogistics
  288. db3 := GetReadDB(context.Background())
  289. err = db3.Model(gorm_model.YoungeeTaskLogistics{}).Where("auto_sign_at <= ? AND task_id IN ?", time.Now(), taskIds).Find(&logisticInfos).Error
  290. if err != nil {
  291. log.Println("DB AutoGetSignInOfflineTask error :", err)
  292. return err
  293. }
  294. var signedTaskId []int64
  295. for _, logisticInfo := range logisticInfos {
  296. signedTaskId = append(signedTaskId, logisticInfo.TaskID)
  297. if logisticInfo.SignedTime.IsZero() {
  298. db6 := GetReadDB(context.Background())
  299. db6.Model(gorm_model.YoungeeTaskLogistics{}).Where("logistics_id = ?", logisticInfo.LogisticsID).Updates(&gorm_model.YoungeeTaskLogistics{SignedTime: time.Now(), Status: 1})
  300. fmt.Println("已更新签收时间")
  301. }
  302. }
  303. util.RemoveRepByMap(signedTaskId)
  304. //fmt.Println("signedTaskId", signedTaskId)
  305. for _, taskID := range signedTaskId {
  306. //fmt.Println("taskID", taskID)
  307. db4 := GetReadDB(context.Background())
  308. taskNeedMod := gorm_model.YoungeeTaskInfo{}
  309. db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskID).Find(&taskNeedMod)
  310. if projectIdMap[int64(taskIDs[int(taskID)])] == 1 {
  311. if taskNeedMod.LogisticsStatus == 2 && taskNeedMod.TaskStage == 5 {
  312. db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskID).Updates(&gorm_model.YoungeeTaskInfo{LogisticsStatus: 3, TaskStage: 9})
  313. fmt.Println("内容形式为图片的项目 已修改任务状态为待传初稿 及物流状态为已签收")
  314. }
  315. } else if projectIdMap[int64(taskIDs[int(taskID)])] == 2 {
  316. if taskNeedMod.LogisticsStatus == 2 && taskNeedMod.TaskStage == 5 {
  317. db4.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskID).Updates(&gorm_model.YoungeeTaskInfo{LogisticsStatus: 3, TaskStage: 7})
  318. fmt.Println("内容形式为视频的项目 已修改任务状态为待传脚本 及物流状态为已签收")
  319. }
  320. }
  321. }
  322. return nil
  323. }
  324. func GetAutoReviewTask(contentType int32) error {
  325. db := GetReadDB(context.Background())
  326. var projectInfos []*gorm_model.ProjectInfo
  327. if contentType == 2 {
  328. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ?", 1, contentType).Find(&projectInfos).Error
  329. if err != nil {
  330. return err
  331. }
  332. var projectIds []int64
  333. projectIdToAutoTaskIdMap := map[int64]int{}
  334. for _, projectInfo := range projectInfos {
  335. projectIds = append(projectIds, projectInfo.ProjectID)
  336. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  337. }
  338. //fmt.Println("projectIds:", projectIds)
  339. TaskIdToProjectId := make(map[int]int)
  340. var taskId []int
  341. db1 := GetReadDB(context.Background())
  342. for _, projectId := range projectIds {
  343. // 获取 autoTaskId 及其对应的限制时间、
  344. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  345. dbStart := GetReadDB(context.Background())
  346. var ReviewInMv int32
  347. dbStart.Model(gorm_model.InfoAutoTask{}).Select("review_in_mv").Where("auto_task_id = ?", autoTaskId).First(&ReviewInMv)
  348. //fmt.Println("t:", autoTaskInfo.ReviewInMv)
  349. // 获取 projectId 对应的 taskId 及添加脚本自动审核时间
  350. var taskInfos []gorm_model.YoungeeTaskInfo
  351. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  352. for _, taskInfo := range taskInfos {
  353. //fmt.Println("TaskID:", taskInfo.TaskId)
  354. taskId = append(taskId, taskInfo.TaskId)
  355. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  356. db2 := GetReadDB(context.Background())
  357. var scriptInfos gorm_model.YounggeeScriptInfo
  358. db2.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&scriptInfos)
  359. dd, _ := time.ParseDuration(conv.MustString(ReviewInMv, "") + "h")
  360. if scriptInfos.AutoAgreeAt.IsZero() && scriptInfos.IsSubmit == 1 {
  361. db3 := GetReadDB(context.Background())
  362. db3.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ?", taskInfo.TaskId).Updates(&gorm_model.YounggeeScriptInfo{AutoAgreeAt: scriptInfos.SubmitAt.Add(dd)})
  363. fmt.Println("已添加脚本自动审核时间")
  364. }
  365. }
  366. }
  367. //fmt.Println("TaskIdToProjectId:", TaskIdToProjectId)
  368. //fmt.Println("taskId:", taskId)
  369. var scriptInfos []gorm_model.YounggeeScriptInfo
  370. db4 := GetReadDB(context.Background())
  371. err = db4.Model(gorm_model.YounggeeScriptInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&scriptInfos).Error
  372. if err != nil {
  373. log.Println("DB GetAutoReviewTask error in script :", err)
  374. return err
  375. }
  376. for _, scriptInfo := range scriptInfos {
  377. if scriptInfo.IsReview == 0 && scriptInfo.IsSubmit == 1 {
  378. db7 := GetReadDB(context.Background())
  379. db7.Model(gorm_model.YounggeeScriptInfo{}).Where("script_id = ?", scriptInfo.ScriptID).Updates(&gorm_model.YounggeeScriptInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  380. fmt.Println("已更新脚本同意时间 及脚本审核状态")
  381. taskStatusNeedMod := gorm_model.YoungeeTaskInfo{}
  382. db6 := GetReadDB(context.Background())
  383. db6.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", scriptInfo.TaskID).Find(&taskStatusNeedMod)
  384. if (taskStatusNeedMod.ScriptStatus == 2 || taskStatusNeedMod.ScriptStatus == 4) && taskStatusNeedMod.TaskStage == 8 {
  385. db8 := GetReadDB(context.Background())
  386. db8.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", scriptInfo.TaskID).Updates(&gorm_model.YoungeeTaskInfo{ScriptStatus: 5, TaskStage: 9})
  387. fmt.Println("已更新任务阶段为待传初稿 及脚本上传状态为已通过")
  388. }
  389. }
  390. }
  391. return nil
  392. } else {
  393. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  394. if err != nil {
  395. return err
  396. }
  397. var projectIds []int64
  398. projectIdToAutoTaskIdMap := map[int64]int{}
  399. for _, projectInfo := range projectInfos {
  400. projectIds = append(projectIds, projectInfo.ProjectID)
  401. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  402. }
  403. //fmt.Println("projectIds:", projectIds)
  404. TaskIdToProjectId := make(map[int]int)
  405. var taskId []int
  406. db1 := GetReadDB(context.Background())
  407. for _, projectId := range projectIds {
  408. // 获取 autoTaskId 及其对应的限制时间、
  409. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  410. dbStart := GetReadDB(context.Background())
  411. var ReviewUnlimited int32
  412. dbStart.Model(gorm_model.InfoAutoTask{}).Select("review_unlimited").Where("auto_task_id = ?", autoTaskId).First(&ReviewUnlimited)
  413. //fmt.Println("t:", autoTaskInfo.ReviewUnlimited)
  414. var taskInfos []gorm_model.YoungeeTaskInfo
  415. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  416. for _, taskInfo := range taskInfos {
  417. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  418. taskId = append(taskId, taskInfo.TaskId)
  419. db2 := GetReadDB(context.Background())
  420. var scriptInfos gorm_model.YounggeeSketchInfo
  421. db2.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&scriptInfos)
  422. dd, _ := time.ParseDuration(conv.MustString(ReviewUnlimited, "") + "h")
  423. if scriptInfos.AutoAgreeAt.IsZero() && scriptInfos.IsSubmit == 1 {
  424. db3 := GetReadDB(context.Background())
  425. db3.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", scriptInfos.TaskID).Updates(&gorm_model.YounggeeSketchInfo{AutoAgreeAt: scriptInfos.SubmitAt.Add(dd)})
  426. fmt.Println("已添加初稿自动审核时间")
  427. }
  428. }
  429. }
  430. //fmt.Println("TaskIdToProjectId:", TaskIdToProjectId)
  431. //fmt.Println("taskId:", taskId)
  432. var sketchInfos []gorm_model.YounggeeSketchInfo
  433. db3 := GetReadDB(context.Background())
  434. err = db3.Model(gorm_model.YounggeeSketchInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&sketchInfos).Error
  435. if err != nil {
  436. log.Println("DB GetAutoReviewTask error in sketch:", err)
  437. return err
  438. }
  439. for _, sketchInfo := range sketchInfos {
  440. if sketchInfo.IsReview != 1 && sketchInfo.IsSubmit == 1 {
  441. db6 := GetReadDB(context.Background())
  442. db6.Model(gorm_model.YounggeeSketchInfo{}).Where("sketch_id = ?", sketchInfo.SketchID).Updates(&gorm_model.YounggeeSketchInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  443. fmt.Println("已更新初稿审核同意时间")
  444. sketchNeedMod := gorm_model.YoungeeTaskInfo{}
  445. db5 := GetReadDB(context.Background())
  446. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", sketchInfo.TaskID).Find(&sketchNeedMod)
  447. if (sketchNeedMod.SketchStatus == 2 || sketchNeedMod.SketchStatus == 4) && sketchNeedMod.TaskStage == 10 {
  448. db7 := GetReadDB(context.Background())
  449. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", sketchInfo.TaskID).Updates(&gorm_model.YoungeeTaskInfo{SketchStatus: 5, TaskStage: 11})
  450. fmt.Println("已更新任务阶段为待传链接 及初稿上传状态为已通过")
  451. }
  452. }
  453. }
  454. return nil
  455. }
  456. }
  457. func GetAutoPostReviewTask() error {
  458. db := GetReadDB(context.Background())
  459. var projectInfos []*gorm_model.ProjectInfo
  460. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  461. if err != nil {
  462. return err
  463. }
  464. var projectIds []int64
  465. projectIdToAutoTaskIdMap := map[int64]int{}
  466. for _, projectInfo := range projectInfos {
  467. projectIds = append(projectIds, projectInfo.ProjectID)
  468. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  469. }
  470. //fmt.Println("projectIds:", projectIds)
  471. TaskIdToProjectId := make(map[int]int)
  472. var taskId []int
  473. db1 := GetReadDB(context.Background())
  474. for _, projectId := range projectIds {
  475. // 获取 autoTaskId 及其对应的限制时间、
  476. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  477. dbStart := GetReadDB(context.Background())
  478. var Postreview int32
  479. dbStart.Model(gorm_model.InfoAutoTask{}).Select("postreview").Where("auto_task_id = ?", autoTaskId).First(&Postreview)
  480. //fmt.Println("t:", autoTaskInfo.Postreview)
  481. var taskInfos []gorm_model.YoungeeTaskInfo
  482. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  483. for _, taskInfo := range taskInfos {
  484. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  485. taskId = append(taskId, taskInfo.TaskId)
  486. db2 := GetReadDB(context.Background())
  487. var linkInfos gorm_model.YounggeeLinkInfo
  488. db2.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&linkInfos)
  489. dd, _ := time.ParseDuration(conv.MustString(Postreview, "") + "h")
  490. if linkInfos.AutoAgreeAt.IsZero() && linkInfos.IsSubmit == 1 {
  491. db3 := GetReadDB(context.Background())
  492. db3.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ?", linkInfos.TaskID).Updates(&gorm_model.YounggeeLinkInfo{AutoAgreeAt: linkInfos.SubmitAt.Add(dd)})
  493. fmt.Println("已添加链接自动审核时间")
  494. }
  495. }
  496. }
  497. var linkInfos []gorm_model.YounggeeLinkInfo
  498. db3 := GetReadDB(context.Background())
  499. err = db3.Model(gorm_model.YounggeeLinkInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&linkInfos).Error
  500. if err != nil {
  501. log.Println("DB GetAutoPostReviewTask error in sketch:", err)
  502. return err
  503. }
  504. for _, linkInfo := range linkInfos {
  505. if linkInfo.IsReview != 1 && linkInfo.IsSubmit == 1 {
  506. db6 := GetReadDB(context.Background())
  507. db6.Model(gorm_model.YounggeeLinkInfo{}).Where("link_id = ?", linkInfo.LinkID).Updates(&gorm_model.YounggeeLinkInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  508. fmt.Println("已更新链接审核同意时间")
  509. linkNeedMod := gorm_model.YoungeeTaskInfo{}
  510. db5 := GetReadDB(context.Background())
  511. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", linkInfo.TaskID).Find(&linkNeedMod)
  512. if (linkNeedMod.LinkStatus == 2 || linkNeedMod.LinkStatus == 4) && linkNeedMod.TaskStage == 12 {
  513. db7 := GetReadDB(context.Background())
  514. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", linkInfo.TaskID).Updates(&gorm_model.YoungeeTaskInfo{LinkStatus: 5, TaskStage: 13})
  515. fmt.Println("已更新任务阶段为待传数据 及 链接上传状态为已通过")
  516. }
  517. }
  518. }
  519. return nil
  520. }
  521. func GetAutoCaseCloseTask() error {
  522. db := GetReadDB(context.Background())
  523. var projectInfos []*gorm_model.ProjectInfo
  524. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  525. if err != nil {
  526. return err
  527. }
  528. var projectIds []int64
  529. projectIdToAutoTaskIdMap := map[int64]int{}
  530. for _, projectInfo := range projectInfos {
  531. projectIds = append(projectIds, projectInfo.ProjectID)
  532. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  533. }
  534. //fmt.Println("projectIds:", projectIds)
  535. TaskIdToProjectId := make(map[int]int)
  536. var taskId []int
  537. db1 := GetReadDB(context.Background())
  538. for _, projectId := range projectIds {
  539. // 获取 autoTaskId 及其对应的限制时间
  540. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  541. dbStart := GetReadDB(context.Background())
  542. var CaseClose int32
  543. dbStart.Model(gorm_model.InfoAutoTask{}).Select("case_close").Where("auto_task_id = ?", autoTaskId).First(&CaseClose)
  544. //fmt.Println("t:", autoTaskInfo.CaseClose)
  545. var taskInfos []gorm_model.YoungeeTaskInfo
  546. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  547. for _, taskInfo := range taskInfos {
  548. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  549. taskId = append(taskId, taskInfo.TaskId)
  550. db2 := GetReadDB(context.Background())
  551. var dataInfos gorm_model.YounggeeDataInfo
  552. db2.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&dataInfos)
  553. dd, _ := time.ParseDuration(conv.MustString(CaseClose, "") + "h")
  554. if dataInfos.AutoAgreeAt.IsZero() && dataInfos.IsSubmit == 1 {
  555. db3 := GetReadDB(context.Background())
  556. db3.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ?", dataInfos.TaskID).Updates(&gorm_model.YounggeeDataInfo{AutoAgreeAt: dataInfos.SubmitAt.Add(dd)})
  557. fmt.Println("已添加链接自动审核时间")
  558. }
  559. }
  560. }
  561. var dataInfos []gorm_model.YounggeeDataInfo
  562. db3 := GetReadDB(context.Background())
  563. err = db3.Model(gorm_model.YounggeeDataInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&dataInfos).Error
  564. if err != nil {
  565. log.Println("DB GetAutoCaseCloseTask error in sketch:", err)
  566. return err
  567. }
  568. for _, dataInfo := range dataInfos {
  569. if dataInfo.IsReview != 1 && dataInfo.IsSubmit == 1 {
  570. db6 := GetReadDB(context.Background())
  571. db6.Model(gorm_model.YounggeeDataInfo{}).Where("data_id = ?", dataInfo.DataID).Updates(&gorm_model.YounggeeDataInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  572. fmt.Println("已更新数据审核同意时间")
  573. dataNeedMod := gorm_model.YoungeeTaskInfo{}
  574. db5 := GetReadDB(context.Background())
  575. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", dataInfo.TaskID).Find(&dataNeedMod)
  576. if (dataNeedMod.DataStatus == 2 || dataNeedMod.DataStatus == 4) && dataNeedMod.TaskStage == 14 {
  577. db7 := GetReadDB(context.Background())
  578. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", dataInfo.TaskID).Updates(&gorm_model.YoungeeTaskInfo{DataStatus: 5, TaskStage: 15})
  579. fmt.Println("已更新任务阶段为已结案 及 数据上传状态为已通过")
  580. }
  581. }
  582. }
  583. return nil
  584. }
  585. func GetAutoInvalidTask() error {
  586. db := GetReadDB(context.Background())
  587. var projectInfos []*gorm_model.ProjectInfo
  588. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND project_status = ? ", 1, 6).Find(&projectInfos).Error
  589. if err != nil {
  590. return err
  591. }
  592. var projectIds []int64
  593. projectIdToAutoTaskIdMap := map[int64]int{}
  594. for _, projectInfo := range projectInfos {
  595. projectIds = append(projectIds, projectInfo.ProjectID)
  596. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  597. }
  598. for _, projectId := range projectIds {
  599. // 获取 autoTaskId 及其对应的限制时间、
  600. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  601. dbStart := GetReadDB(context.Background())
  602. var Invalid int32
  603. dbStart.Model(gorm_model.InfoAutoTask{}).Select("invalid").Where("auto_task_id = ?", autoTaskId).First(&Invalid)
  604. //fmt.Println("t:", autoTaskInfo.Invalid)
  605. db1 := GetReadDB(context.Background())
  606. project := gorm_model.ProjectInfo{}
  607. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&project)
  608. //if project.AutoFailAt.IsZero() {
  609. // dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  610. // t := project.RecruitDdl.Add(dd)
  611. // db2 := GetReadDB(context.Background())
  612. // db2.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  613. // fmt.Println("已添加失效自动处理时间")
  614. //}
  615. dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  616. project.RecruitDdl.Add(dd)
  617. t := project.RecruitDdl.Add(dd)
  618. if project.AutoFailAt != &t {
  619. db3 := GetReadDB(context.Background())
  620. db3.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  621. fmt.Println("已添加失效自动处理时间")
  622. }
  623. projectNeedMod := gorm_model.ProjectInfo{}
  624. db2 := GetReadDB(context.Background())
  625. db2.Where("project_id = ?", projectId).First(&projectNeedMod)
  626. if !projectNeedMod.AutoFailAt.IsZero() {
  627. if projectNeedMod.AutoFailAt.Sub(time.Now()) <= 0 && projectNeedMod.ProjectStatus < 8 {
  628. db4 := GetReadDB(context.Background())
  629. err1 := db4.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{ProjectStatus: 8, FailReason: 1}).Error
  630. if err1 != nil {
  631. return err1
  632. }
  633. fmt.Println("已更新项目状态")
  634. }
  635. }
  636. }
  637. return nil
  638. }
  639. func GetAutoDraftDefaultInPicTask() error {
  640. db := GetReadDB(context.Background())
  641. var projectInfos []*gorm_model.ProjectInfo
  642. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 1).Find(&projectInfos).Error
  643. if err != nil {
  644. return err
  645. }
  646. // 项目类型为 实体商品寄拍、虚拟产品测评、线下探店打卡 均为正常的 收发货情况,以签收状况作为判断条件
  647. var projectIds []int64
  648. // 存储素材微原创的项目id,此情况单独处理,它不涉及到物流,也不涉及到脚本,初稿上一个状态直接就是支付。
  649. var specialProjectIds []int64
  650. projectIdToAutoTaskIdMap := map[int64]int{}
  651. // 项目id 对 违约定时任务id 的map
  652. projectIdToAutoDefaultIdMap := map[int64]int{}
  653. // taskId 对 稿费形式的 map
  654. taskIdToFeeFormMap := make(map[int]int)
  655. TaskIdToProjectId := make(map[int]int)
  656. var taskNeedModIds []int
  657. var specialTaskNeedModIds []int
  658. // 已提交初稿的任务记录id
  659. var submitTaskNeedModIds []int
  660. for _, projectInfo := range projectInfos {
  661. if projectInfo.ProjectForm == 4 {
  662. specialProjectIds = append(specialProjectIds, projectInfo.ProjectID)
  663. specialProjectInfo := gorm_model.ProjectInfo{}
  664. db1 := GetReadDB(context.Background())
  665. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).First(&specialProjectInfo)
  666. dbStart := GetReadDB(context.Background())
  667. var DraftDefaultInPic int32
  668. dbStart.Model(gorm_model.InfoAutoTask{}).Select("draft_default_in_pic").Where("auto_task_id = ?", specialProjectInfo.AutoTaskID).First(&DraftDefaultInPic)
  669. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInPic, "") + "h")
  670. t1 := projectInfo.PayAt.Add(dd)
  671. if !projectInfo.PayAt.IsZero() && projectInfo.AutoSketchBreakAt.IsZero() {
  672. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).Updates(&gorm_model.ProjectInfo{AutoSketchBreakAt: &t1})
  673. fmt.Println("已添加素材微原创的初稿违约自动处理时间")
  674. }
  675. } else {
  676. projectIds = append(projectIds, projectInfo.ProjectID)
  677. }
  678. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  679. projectIdToAutoDefaultIdMap[projectInfo.ProjectID] = int(projectInfo.AutoDefaultID)
  680. }
  681. //fmt.Println("projectIds:", projectIds)
  682. //fmt.Println("specialProjectIds:", specialProjectIds)
  683. for _, specialProjectId := range specialProjectIds {
  684. db2 := GetReadDB(context.Background())
  685. var taskInfos []gorm_model.YoungeeTaskInfo
  686. db2.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ?", specialProjectId, 0).Find(&taskInfos)
  687. for _, taskInfo := range taskInfos {
  688. TaskIdToProjectId[taskInfo.TaskId] = int(specialProjectId)
  689. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  690. specialTaskNeedMod := gorm_model.YoungeeTaskInfo{}
  691. db3 := GetReadDB(context.Background())
  692. // 保存所有初稿上传状态为 1或 3 的任务记录的id
  693. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskInfo.TaskId).First(&specialTaskNeedMod)
  694. if specialTaskNeedMod.SketchStatus == 1 && taskInfo.TaskStage == 10 {
  695. specialTaskNeedModIds = append(specialTaskNeedModIds, taskInfo.TaskId)
  696. }
  697. if specialTaskNeedMod.SketchStatus == 3 && taskInfo.TaskStage == 10 {
  698. submitTaskNeedModIds = append(submitTaskNeedModIds, taskInfo.TaskId)
  699. }
  700. }
  701. }
  702. //fmt.Println("specialTaskNeedModIds:", specialTaskNeedModIds)
  703. for _, specialTaskNeedModId := range specialTaskNeedModIds {
  704. projectNeedMod := gorm_model.ProjectInfo{}
  705. db1 := GetReadDB(context.Background())
  706. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", TaskIdToProjectId[specialTaskNeedModId]).First(&projectNeedMod)
  707. var sketchDefaultRate int
  708. if taskIdToFeeFormMap[specialTaskNeedModId] == 1 {
  709. db6 := GetReadDB(context.Background())
  710. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[specialTaskNeedModId])]).Find(&sketchDefaultRate)
  711. } else {
  712. db6 := GetReadDB(context.Background())
  713. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[specialTaskNeedModId])]).Find(&sketchDefaultRate)
  714. }
  715. if projectNeedMod.AutoSketchBreakAt.Sub(time.Now()) <= 0 && !projectNeedMod.AutoSketchBreakAt.IsZero() {
  716. db2 := GetReadDB(context.Background())
  717. err1 := db2.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  718. TaskID: specialTaskNeedModId, ProjectID: TaskIdToProjectId[specialTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  719. if err1 != nil {
  720. return err1
  721. }
  722. db3 := GetReadDB(context.Background())
  723. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", specialTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate})
  724. fmt.Println("已创建素材微原创的任务初稿违约记录")
  725. }
  726. }
  727. for _, projectId := range projectIds {
  728. db1 := GetReadDB(context.Background())
  729. var taskInfos []gorm_model.YoungeeTaskInfo
  730. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  731. for _, taskInfo := range taskInfos {
  732. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  733. taskNeedMod := gorm_model.YoungeeTaskInfo{}
  734. db2 := GetReadDB(context.Background())
  735. // 保存所有满足物流状态为3 且 初稿上传状态为 1或 3 的任务记录的id
  736. db2.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskInfo.TaskId).First(&taskNeedMod)
  737. if taskNeedMod.LogisticsStatus == 3 && taskNeedMod.ScriptStatus == 1 {
  738. taskNeedModIds = append(taskNeedModIds, taskInfo.TaskId)
  739. }
  740. if taskNeedMod.LogisticsStatus == 3 && taskNeedMod.ScriptStatus == 3 {
  741. submitTaskNeedModIds = append(submitTaskNeedModIds, taskInfo.TaskId)
  742. }
  743. }
  744. }
  745. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  746. //fmt.Println("TaskIdToProjectId:", TaskIdToProjectId)
  747. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  748. for _, taskNeedModId := range taskNeedModIds {
  749. // 获取 autoTaskId 及其对应的限制时间、
  750. //fmt.Println("taskNeedModId:", taskNeedModId)
  751. autoTaskId := projectIdToAutoTaskIdMap[int64(TaskIdToProjectId[taskNeedModId])]
  752. dbStart := GetReadDB(context.Background())
  753. var DraftDefaultInPic int32
  754. dbStart.Model(gorm_model.InfoAutoTask{}).Select("draft_default_in_pic").Where("auto_task_id = ?", autoTaskId).First(&DraftDefaultInPic)
  755. //fmt.Println("t:", autoTaskInfo.DraftDefaultInPic)
  756. db3 := GetReadDB(context.Background())
  757. taskLogisticNeedMod := gorm_model.YoungeeTaskLogistics{}
  758. db3.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&taskLogisticNeedMod)
  759. var sketchDefaultRate int
  760. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  761. db6 := GetReadDB(context.Background())
  762. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[taskNeedModId])]).Find(&sketchDefaultRate)
  763. } else {
  764. db6 := GetReadDB(context.Background())
  765. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[taskNeedModId])]).Find(&sketchDefaultRate)
  766. }
  767. if taskLogisticNeedMod.AutoSketchBreakAt.IsZero() {
  768. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInPic, "") + "h")
  769. db4 := GetReadDB(context.Background())
  770. logisticNeedMod := gorm_model.YoungeeTaskLogistics{}
  771. db4.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&logisticNeedMod)
  772. if logisticNeedMod.ThingsType == 1 && !logisticNeedMod.SignedTime.IsZero() {
  773. db4.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{
  774. AutoSketchBreakAt: taskLogisticNeedMod.SignedTime.Add(dd),
  775. })
  776. fmt.Println("已添加实物的图片初稿违约自动处理时间")
  777. }
  778. if logisticNeedMod.ThingsType != 1 && !logisticNeedMod.AutoSignAt.IsZero() {
  779. db4.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{
  780. AutoSketchBreakAt: taskLogisticNeedMod.AutoSignAt.Add(dd),
  781. })
  782. fmt.Println("已添加虚拟产品及线下探店的图片初稿违约自动处理时间")
  783. }
  784. }
  785. db7 := GetReadDB(context.Background())
  786. taskLogisticNeedMod2 := gorm_model.YoungeeTaskLogistics{}
  787. db7.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&taskLogisticNeedMod2)
  788. if !taskLogisticNeedMod2.AutoSketchBreakAt.IsZero() && taskLogisticNeedMod2.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  789. db5 := GetReadDB(context.Background())
  790. err1 := db5.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  791. TaskID: taskNeedModId, ProjectID: TaskIdToProjectId[taskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  792. if err1 != nil {
  793. return err1
  794. }
  795. db8 := GetReadDB(context.Background())
  796. db8.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskLogisticNeedMod2.TaskID).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate})
  797. fmt.Println("已创建图文类型的初稿违约记录")
  798. }
  799. }
  800. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  801. db2 := GetReadDB(context.Background())
  802. var DraftDefaultInMv int32
  803. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).First(&DraftDefaultInMv)
  804. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  805. var sketchDefaultRate int
  806. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  807. db6 := GetReadDB(context.Background())
  808. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).Find(&sketchDefaultRate)
  809. } else {
  810. db6 := GetReadDB(context.Background())
  811. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).Find(&sketchDefaultRate)
  812. }
  813. db1 := GetReadDB(context.Background())
  814. var taskSketchInfo gorm_model.YounggeeSketchInfo
  815. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  816. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  817. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  818. if err4 != nil {
  819. return err4
  820. }
  821. fmt.Println("已添加图文形式的初稿违约自动处理时间")
  822. if taskSketchInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  823. db4 := GetReadDB(context.Background())
  824. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  825. TaskID: submitTaskNeedModId, ProjectID: TaskIdToProjectId[submitTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  826. if err1 != nil {
  827. return err1
  828. }
  829. db3 := GetReadDB(context.Background())
  830. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  831. if err2 != nil {
  832. return err2
  833. }
  834. fmt.Println("创建时已违约则创建已提交初稿的图文类型的初稿违约记录")
  835. }
  836. } else {
  837. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  838. db4 := GetReadDB(context.Background())
  839. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  840. TaskID: submitTaskNeedModId, ProjectID: TaskIdToProjectId[submitTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  841. if err1 != nil {
  842. return err1
  843. }
  844. db3 := GetReadDB(context.Background())
  845. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  846. if err2 != nil {
  847. return err2
  848. }
  849. fmt.Println("创建已提交初稿的图文类型的初稿违约记录")
  850. }
  851. }
  852. }
  853. return nil
  854. }
  855. func GetAutoDraftDefaultInMvTask() error {
  856. db := GetReadDB(context.Background())
  857. var VideoProjectIds []int64
  858. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 2).Find(&VideoProjectIds).Error
  859. if err != nil {
  860. return err
  861. }
  862. //fmt.Println("VideoProjectIds:", VideoProjectIds)
  863. // 任务id 对 项目id 的map
  864. taskIdToProjectIdMap := map[int]int64{}
  865. // 项目id 对 定时任务id 的map
  866. projectIdToAutoTaskIdMap := map[int64]int{}
  867. // 项目id 对 违约定时任务id 的map
  868. projectIdToAutoDefaultIdMap := map[int64]int{}
  869. // taskId 对 稿费形式的 map
  870. taskIdToFeeFormMap := make(map[int]int)
  871. var videoTaskIds []int
  872. for _, VideoProjectId := range VideoProjectIds {
  873. db2 := GetReadDB(context.Background())
  874. autoIds := gorm_model.ProjectInfo{}
  875. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", VideoProjectId).First(&autoIds)
  876. projectIdToAutoTaskIdMap[VideoProjectId] = int(autoIds.AutoTaskID)
  877. projectIdToAutoDefaultIdMap[VideoProjectId] = int(autoIds.AutoDefaultID)
  878. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  879. db1 := GetReadDB(context.Background())
  880. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", VideoProjectId, 0).Find(&videoTaskInfos)
  881. for _, videoTaskInfo := range videoTaskInfos {
  882. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  883. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  884. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  885. }
  886. }
  887. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  888. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  889. //fmt.Println("videoTaskIds:", videoTaskIds)
  890. // 首次提交初稿的任务记录id
  891. var videoTaskNeedModIds []int
  892. // 已提交初稿的任务记录id
  893. var submitVideoTaskNeedModIds []int
  894. for _, videoTaskId := range videoTaskIds {
  895. var taskInfo gorm_model.YoungeeTaskInfo
  896. db3 := GetReadDB(context.Background())
  897. // 保存所有初稿上传状态为 1 且任务状态为10 的任务记录的id
  898. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  899. if taskInfo.SketchStatus == 1 && taskInfo.TaskStage == 10 {
  900. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  901. }
  902. // 保存所有初稿上传状态为 3 且任务状态为10 的任务记录的id
  903. if taskInfo.SketchStatus == 3 && taskInfo.TaskStage == 10 {
  904. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  905. }
  906. }
  907. //fmt.Println("videoTaskNeedModIds:", videoTaskNeedModIds)
  908. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  909. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  910. db2 := GetReadDB(context.Background())
  911. var DraftDefaultInMv int32
  912. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&DraftDefaultInMv)
  913. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  914. var sketchDefaultRate int
  915. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  916. db6 := GetReadDB(context.Background())
  917. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  918. } else {
  919. db6 := GetReadDB(context.Background())
  920. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  921. }
  922. db1 := GetReadDB(context.Background())
  923. var taskScriptInfo gorm_model.YounggeeScriptInfo
  924. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ? AND is_ok = ?", videoTaskNeedModId, 1).Find(&taskScriptInfo)
  925. if taskScriptInfo.AutoSketchBreakAt.IsZero() {
  926. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoSketchBreakAt: taskScriptInfo.AutoAgreeAt.Add(dd)})
  927. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  928. if taskScriptInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  929. db4 := GetReadDB(context.Background())
  930. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  931. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  932. if err1 != nil {
  933. return err1
  934. }
  935. db3 := GetReadDB(context.Background())
  936. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  937. if err2 != nil {
  938. return err2
  939. }
  940. fmt.Println("创建时已违约则创建视频类型的初稿违约记录")
  941. }
  942. } else {
  943. if taskScriptInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  944. db4 := GetReadDB(context.Background())
  945. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  946. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  947. if err1 != nil {
  948. return err1
  949. }
  950. db3 := GetReadDB(context.Background())
  951. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  952. if err2 != nil {
  953. return err2
  954. }
  955. fmt.Println("已创建视频类型的初稿违约记录")
  956. }
  957. }
  958. }
  959. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  960. db2 := GetReadDB(context.Background())
  961. var DraftDefaultInMv int32
  962. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&DraftDefaultInMv)
  963. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  964. var sketchDefaultRate int
  965. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  966. db6 := GetReadDB(context.Background())
  967. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  968. } else {
  969. db6 := GetReadDB(context.Background())
  970. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  971. }
  972. db1 := GetReadDB(context.Background())
  973. var taskSketchInfo gorm_model.YounggeeSketchInfo
  974. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  975. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  976. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  977. if err4 != nil {
  978. return err4
  979. }
  980. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  981. if taskSketchInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  982. db4 := GetReadDB(context.Background())
  983. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  984. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  985. if err1 != nil {
  986. return err1
  987. }
  988. db3 := GetReadDB(context.Background())
  989. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  990. if err2 != nil {
  991. return err2
  992. }
  993. fmt.Println("创建时已违约则创建已提交初稿的视频类型的初稿违约记录")
  994. }
  995. } else {
  996. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  997. db4 := GetReadDB(context.Background())
  998. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  999. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1000. if err1 != nil {
  1001. return err1
  1002. }
  1003. db3 := GetReadDB(context.Background())
  1004. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  1005. if err2 != nil {
  1006. return err2
  1007. }
  1008. fmt.Println("创建已提交初稿的视频类型的初稿违约记录")
  1009. }
  1010. }
  1011. }
  1012. return nil
  1013. }
  1014. func GetAutoScriptDefaultTask() error {
  1015. db := GetReadDB(context.Background())
  1016. var VideoProjectIds []int64
  1017. var specialProjectIds []int
  1018. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? AND project_form != 4", 1, 2).Find(&VideoProjectIds).Error
  1019. if err != nil {
  1020. return err
  1021. }
  1022. db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? AND project_form = 4", 1, 2).Find(&specialProjectIds)
  1023. //fmt.Println("VideoProjectIds:", VideoProjectIds)
  1024. //fmt.Println("specialProjectIds:", specialProjectIds)
  1025. // 任务id 对 项目id 的map
  1026. taskIdToProjectIdMap := map[int]int64{}
  1027. // 项目id 对 定时任务id 的map
  1028. projectIdToAutoTaskIdMap := map[int64]int{}
  1029. // 项目id 对 违约定时任务id 的map
  1030. projectIdToAutoDefaultIdMap := map[int64]int{}
  1031. // taskId 对 稿费形式的 map
  1032. taskIdToFeeFormMap := make(map[int]int)
  1033. // 素材微原创的任务id
  1034. var specialVideoTaskIds []int
  1035. // 需要判断是否违约的素材微原创的任务id
  1036. var specialTaskNeedModIds []int
  1037. // 首次提交脚本的任务记录id
  1038. var videoTaskNeedModIds []int
  1039. // 已提交脚本的任务记录id
  1040. var submitVideoTaskNeedModIds []int
  1041. for _, specialProjectId := range specialProjectIds {
  1042. db2 := GetReadDB(context.Background())
  1043. var projectInfo gorm_model.ProjectInfo
  1044. db2.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", specialProjectId).First(&projectInfo)
  1045. projectIdToAutoTaskIdMap[int64(specialProjectId)] = int(projectInfo.AutoTaskID)
  1046. projectIdToAutoDefaultIdMap[int64(specialProjectId)] = int(projectInfo.AutoDefaultID)
  1047. var ScriptDefault int32
  1048. dbStart := GetReadDB(context.Background())
  1049. dbStart.Model(gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectInfo.AutoTaskID).First(&ScriptDefault)
  1050. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1051. t3 := projectInfo.PayAt.Add(dd)
  1052. if !projectInfo.PayAt.IsZero() && projectInfo.AutoScriptBreakAt.IsZero() {
  1053. db1 := GetReadDB(context.Background())
  1054. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).Updates(&gorm_model.ProjectInfo{AutoScriptBreakAt: &t3})
  1055. fmt.Println("已添加素材微原创的脚本违约自动处理时间")
  1056. }
  1057. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1058. db1 := GetReadDB(context.Background())
  1059. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", specialProjectId, 0).Find(&videoTaskInfos)
  1060. for _, videoTaskInfo := range videoTaskInfos {
  1061. taskIdToProjectIdMap[videoTaskInfo.TaskId] = int64(specialProjectId)
  1062. specialVideoTaskIds = append(specialVideoTaskIds, videoTaskInfo.TaskId)
  1063. specialTaskNeedMod := gorm_model.YoungeeTaskInfo{}
  1064. db3 := GetReadDB(context.Background())
  1065. // 保存所有脚本上传状态为 1 且任务状态为8 的任务记录的id
  1066. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskInfo.TaskId).First(&specialTaskNeedMod)
  1067. if specialTaskNeedMod.ScriptStatus == 1 && specialTaskNeedMod.TaskStage == 8 {
  1068. specialTaskNeedModIds = append(specialTaskNeedModIds, videoTaskInfo.TaskId)
  1069. }
  1070. // 保存所有脚本上传状态为 3 且任务状态为8 的任务记录的id
  1071. if specialTaskNeedMod.ScriptStatus == 3 && specialTaskNeedMod.TaskStage == 8 {
  1072. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskInfo.TaskId)
  1073. }
  1074. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  1075. }
  1076. }
  1077. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  1078. //fmt.Println("specialTaskNeedModIds:", specialTaskNeedModIds)
  1079. for _, specialTaskNeedModId := range specialTaskNeedModIds {
  1080. projectNeedMod := gorm_model.ProjectInfo{}
  1081. db1 := GetReadDB(context.Background())
  1082. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", taskIdToProjectIdMap[specialTaskNeedModId]).First(&projectNeedMod)
  1083. if projectNeedMod.AutoScriptBreakAt.Sub(time.Now()) <= 0 && !projectNeedMod.AutoScriptBreakAt.IsZero() {
  1084. var scriptDefaultRate int
  1085. if taskIdToFeeFormMap[specialTaskNeedModId] == 1 {
  1086. db6 := GetReadDB(context.Background())
  1087. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[specialTaskNeedModId]]).Find(&scriptDefaultRate)
  1088. } else {
  1089. db6 := GetReadDB(context.Background())
  1090. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[specialTaskNeedModId]]).Find(&scriptDefaultRate)
  1091. }
  1092. db2 := GetReadDB(context.Background())
  1093. err1 := db2.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1094. TaskID: specialTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[specialTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1095. if err1 != nil {
  1096. return err1
  1097. }
  1098. db3 := GetReadDB(context.Background())
  1099. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", specialTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate})
  1100. fmt.Println("已创建素材微原创的任务脚本违约记录")
  1101. }
  1102. }
  1103. var videoTaskIds []int
  1104. for _, VideoProjectId := range VideoProjectIds {
  1105. db2 := GetReadDB(context.Background())
  1106. var autoTaskId int
  1107. db2.Select("auto_task_id").Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", VideoProjectId).First(&autoTaskId)
  1108. projectIdToAutoTaskIdMap[VideoProjectId] = autoTaskId
  1109. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1110. db1 := GetReadDB(context.Background())
  1111. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", VideoProjectId, 0).Find(&videoTaskInfos)
  1112. for _, videoTaskInfo := range videoTaskInfos {
  1113. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  1114. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  1115. }
  1116. }
  1117. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1118. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  1119. //fmt.Println("videoTaskIds:", videoTaskIds)
  1120. for _, videoTaskId := range videoTaskIds {
  1121. var taskInfo gorm_model.YoungeeTaskInfo
  1122. db3 := GetReadDB(context.Background())
  1123. // 保存所有脚本上传状态为 1 且任务状态为8 的任务记录的id
  1124. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  1125. if taskInfo.ScriptStatus == 1 && taskInfo.TaskStage == 8 {
  1126. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  1127. }
  1128. // 保存所有脚本上传状态为 3 且任务状态为8 的任务记录的id
  1129. if taskInfo.ScriptStatus == 3 && taskInfo.TaskStage == 8 {
  1130. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  1131. }
  1132. }
  1133. //fmt.Println("videoTaskNeedModIds:", videoTaskNeedModIds)
  1134. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  1135. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  1136. db2 := GetReadDB(context.Background())
  1137. var ScriptDefault int32
  1138. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&ScriptDefault)
  1139. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1140. db1 := GetReadDB(context.Background())
  1141. var taskLogisticInfo gorm_model.YoungeeTaskLogistics
  1142. db1.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskLogisticInfo)
  1143. var scriptDefaultRate int
  1144. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  1145. db6 := GetReadDB(context.Background())
  1146. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1147. } else {
  1148. db6 := GetReadDB(context.Background())
  1149. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1150. }
  1151. if taskLogisticInfo.AutoScriptBreakAt.IsZero() {
  1152. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{AutoScriptBreakAt: taskLogisticInfo.AutoSignAt.Add(dd)})
  1153. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1154. if taskLogisticInfo.AutoSignAt.Add(dd).Sub(time.Now()) <= 0 {
  1155. db4 := GetReadDB(context.Background())
  1156. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1157. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1158. if err1 != nil {
  1159. return err1
  1160. }
  1161. db3 := GetReadDB(context.Background())
  1162. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1163. if err2 != nil {
  1164. return err2
  1165. }
  1166. fmt.Println("创建时已违约则创建视频类型的脚本违约记录")
  1167. }
  1168. } else {
  1169. if taskLogisticInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1170. db4 := GetReadDB(context.Background())
  1171. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1172. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1173. if err1 != nil {
  1174. return err1
  1175. }
  1176. db3 := GetReadDB(context.Background())
  1177. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1178. if err2 != nil {
  1179. return err2
  1180. }
  1181. fmt.Println("已创建视频类型的脚本违约记录")
  1182. }
  1183. }
  1184. }
  1185. // 根据最近一次拒绝时间判断当前是否违约
  1186. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  1187. db2 := GetReadDB(context.Background())
  1188. var ScriptDefault int32
  1189. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&ScriptDefault)
  1190. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1191. db1 := GetReadDB(context.Background())
  1192. var taskScriptInfo gorm_model.YounggeeScriptInfo
  1193. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskScriptInfo)
  1194. var scriptDefaultRate int
  1195. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  1196. db6 := GetReadDB(context.Background())
  1197. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1198. } else {
  1199. db6 := GetReadDB(context.Background())
  1200. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1201. }
  1202. if taskScriptInfo.AutoScriptBreakAt.IsZero() {
  1203. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoScriptBreakAt: taskScriptInfo.RejectAt.Add(dd)}).Error
  1204. if err4 != nil {
  1205. return err4
  1206. }
  1207. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1208. if taskScriptInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1209. db4 := GetReadDB(context.Background())
  1210. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1211. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1212. if err1 != nil {
  1213. return err1
  1214. }
  1215. db3 := GetReadDB(context.Background())
  1216. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1217. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1218. if err2 != nil {
  1219. return err2
  1220. }
  1221. fmt.Println("创建时已违约则创建已提交脚本的视频类型的脚本违约记录")
  1222. }
  1223. } else {
  1224. if taskScriptInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1225. db4 := GetReadDB(context.Background())
  1226. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1227. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1228. if err1 != nil {
  1229. return err1
  1230. }
  1231. db3 := GetReadDB(context.Background())
  1232. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1233. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1234. if err2 != nil {
  1235. return err2
  1236. }
  1237. fmt.Println("创建已提交脚本的视频类型的脚本违约记录")
  1238. }
  1239. }
  1240. }
  1241. return nil
  1242. }
  1243. func GetAutoLinkBreachTask() error {
  1244. db := GetReadDB(context.Background())
  1245. var projectIds []int64
  1246. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1247. if err != nil {
  1248. return err
  1249. }
  1250. //fmt.Println("projectIds:", projectIds)
  1251. // 任务id 对 项目id 的map
  1252. taskIdToProjectIdMap := map[int]int64{}
  1253. // 项目id 对 定时任务id 的map
  1254. projectIdToAutoTaskIdMap := map[int64]int{}
  1255. // 项目id 对 违约定时任务id 的map
  1256. projectIdToAutoDefaultIdMap := map[int64]int{}
  1257. var taskIds []int
  1258. // taskId 对 稿费形式的 map
  1259. taskIdToFeeFormMap := make(map[int]int)
  1260. for _, projectId := range projectIds {
  1261. db2 := GetReadDB(context.Background())
  1262. autoIds := gorm_model.ProjectInfo{}
  1263. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1264. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1265. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1266. var taskInfos []*gorm_model.YoungeeTaskInfo
  1267. db1 := GetReadDB(context.Background())
  1268. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  1269. for _, taskInfo := range taskInfos {
  1270. taskIdToProjectIdMap[taskInfo.TaskId] = projectId
  1271. taskIds = append(taskIds, taskInfo.TaskId)
  1272. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  1273. }
  1274. }
  1275. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1276. //fmt.Println("taskIds:", taskIds)
  1277. // 首次提交链接的任务记录id
  1278. var taskNeedModIds []int
  1279. // 已提交链接的任务记录id
  1280. var submitTaskNeedModIds []int
  1281. for _, taskId := range taskIds {
  1282. var taskInfo gorm_model.YoungeeTaskInfo
  1283. db3 := GetReadDB(context.Background())
  1284. // 保存所有链接上传状态为 1 且任务状态为12 的任务记录的id
  1285. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1286. if taskInfo.LinkStatus == 1 && taskInfo.TaskStage == 12 {
  1287. taskNeedModIds = append(taskNeedModIds, taskId)
  1288. }
  1289. // 保存所有链接上传状态为 3 且任务状态为12 的任务记录的id
  1290. if taskInfo.LinkStatus == 3 && taskInfo.TaskStage == 12 {
  1291. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1292. }
  1293. }
  1294. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  1295. //fmt.Println("submitTaskNeedModIds:", submitTaskNeedModIds)
  1296. for _, taskNeedModId := range taskNeedModIds {
  1297. db2 := GetReadDB(context.Background())
  1298. var linkBreach int32
  1299. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&linkBreach)
  1300. dd, _ := time.ParseDuration(conv.MustString(linkBreach, "") + "h")
  1301. var linkDefaultRate int
  1302. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1303. db6 := GetReadDB(context.Background())
  1304. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1305. } else {
  1306. db6 := GetReadDB(context.Background())
  1307. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1308. }
  1309. db1 := GetReadDB(context.Background())
  1310. var taskSketchInfo gorm_model.YounggeeSketchInfo
  1311. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskSketchInfo)
  1312. if taskSketchInfo.AutoLinkBreakAt.IsZero() {
  1313. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoLinkBreakAt: taskSketchInfo.AutoAgreeAt.Add(dd)})
  1314. fmt.Println("已添加链接违约自动处理时间")
  1315. if taskSketchInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1316. db4 := GetReadDB(context.Background())
  1317. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1318. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1319. if err1 != nil {
  1320. return err1
  1321. }
  1322. db3 := GetReadDB(context.Background())
  1323. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1324. if err2 != nil {
  1325. return err2
  1326. }
  1327. fmt.Println("创建时已违约则创建链接违约记录")
  1328. }
  1329. } else {
  1330. if taskSketchInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1331. db4 := GetReadDB(context.Background())
  1332. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1333. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1334. if err1 != nil {
  1335. return err1
  1336. }
  1337. db3 := GetReadDB(context.Background())
  1338. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1339. if err2 != nil {
  1340. return err2
  1341. }
  1342. fmt.Println("已创建链接违约记录")
  1343. }
  1344. }
  1345. }
  1346. // 根据最近一次拒绝时间判断当前是否违约
  1347. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1348. db2 := GetReadDB(context.Background())
  1349. var LinkBreach int32
  1350. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1351. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1352. var linkDefaultRate int
  1353. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1354. db6 := GetReadDB(context.Background())
  1355. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1356. } else {
  1357. db6 := GetReadDB(context.Background())
  1358. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1359. }
  1360. db1 := GetReadDB(context.Background())
  1361. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1362. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskLinkInfo)
  1363. if taskLinkInfo.AutoLinkBreakAt.IsZero() {
  1364. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoLinkBreakAt: taskLinkInfo.RejectAt.Add(dd)}).Error
  1365. if err4 != nil {
  1366. return err4
  1367. }
  1368. fmt.Println("已添加链接违约自动处理时间")
  1369. if taskLinkInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1370. db4 := GetReadDB(context.Background())
  1371. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1372. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1373. if err1 != nil {
  1374. return err1
  1375. }
  1376. db3 := GetReadDB(context.Background())
  1377. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1378. if err2 != nil {
  1379. return err2
  1380. }
  1381. fmt.Println("创建时已违约则创建已提交链接的链接违约记录")
  1382. }
  1383. } else {
  1384. if taskLinkInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1385. db4 := GetReadDB(context.Background())
  1386. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1387. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1388. if err1 != nil {
  1389. return err1
  1390. }
  1391. db3 := GetReadDB(context.Background())
  1392. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1393. if err2 != nil {
  1394. return err2
  1395. }
  1396. fmt.Println("创建已提交链接的链接违约记录")
  1397. }
  1398. }
  1399. }
  1400. return nil
  1401. }
  1402. func GetAutoCaseCloseDefaultTask() error {
  1403. db := GetReadDB(context.Background())
  1404. var projectIds []int64
  1405. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1406. if err != nil {
  1407. return err
  1408. }
  1409. //fmt.Println("projectIds:", projectIds)
  1410. // 任务id 对 项目id 的map
  1411. taskIdToProjectIdMap := map[int]int64{}
  1412. // 项目id 对 定时任务id 的map
  1413. projectIdToAutoTaskIdMap := map[int64]int{}
  1414. // 项目id 对 违约定时任务id 的map
  1415. projectIdToAutoDefaultIdMap := map[int64]int{}
  1416. var taskIds []int
  1417. for _, projectId := range projectIds {
  1418. db2 := GetReadDB(context.Background())
  1419. autoIds := gorm_model.ProjectInfo{}
  1420. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1421. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1422. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1423. var taskInfos []int32
  1424. db1 := GetReadDB(context.Background())
  1425. db1.Select("task_id").Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  1426. for _, taskInfo := range taskInfos {
  1427. taskIdToProjectIdMap[int(taskInfo)] = projectId
  1428. taskIds = append(taskIds, int(taskInfo))
  1429. }
  1430. }
  1431. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  1432. //fmt.Println("projectIdToAutoDefaultIdMap:", projectIdToAutoDefaultIdMap)
  1433. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1434. //fmt.Println("taskIds:", taskIds)
  1435. // 首次提交链接的任务记录id
  1436. var taskNeedModIds []int
  1437. // 已提交链接的任务记录id
  1438. var submitTaskNeedModIds []int
  1439. // taskId 对 稿费形式的 map
  1440. taskIdToFeeFormMap := make(map[int]int)
  1441. for _, taskId := range taskIds {
  1442. var taskInfo gorm_model.YoungeeTaskInfo
  1443. db3 := GetReadDB(context.Background())
  1444. // 保存所有数据上传状态为 1 且任务状态为14 的任务记录的id
  1445. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1446. if taskInfo.DataStatus == 1 && taskInfo.TaskStage == 14 {
  1447. taskNeedModIds = append(taskNeedModIds, taskId)
  1448. }
  1449. // 保存所有数据上传状态为 3 且任务状态为14 的任务记录的id
  1450. if taskInfo.DataStatus == 3 && taskInfo.TaskStage == 14 {
  1451. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1452. }
  1453. taskIdToFeeFormMap[taskId] = taskInfo.FeeForm
  1454. }
  1455. //fmt.Println("taskIdToFeeFormMap:", taskIdToFeeFormMap)
  1456. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  1457. //fmt.Println("submitTaskNeedModIds:", submitTaskNeedModIds)
  1458. for _, taskNeedModId := range taskNeedModIds {
  1459. db2 := GetReadDB(context.Background())
  1460. var CaseCloseDefault int32
  1461. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&CaseCloseDefault)
  1462. dd, _ := time.ParseDuration(conv.MustString(CaseCloseDefault, "") + "h")
  1463. var dataDefaultRate int
  1464. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1465. db6 := GetReadDB(context.Background())
  1466. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1467. } else {
  1468. db6 := GetReadDB(context.Background())
  1469. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1470. }
  1471. //fmt.Println("dataDefaultRate:", dataDefaultRate)
  1472. db1 := GetReadDB(context.Background())
  1473. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1474. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskLinkInfo)
  1475. if taskLinkInfo.AutoDataBreakAt.IsZero() {
  1476. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoDataBreakAt: taskLinkInfo.AutoAgreeAt.Add(dd)})
  1477. fmt.Println("已添加数据违约自动处理时间")
  1478. if taskLinkInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1479. db4 := GetReadDB(context.Background())
  1480. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1481. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1482. if err1 != nil {
  1483. return err1
  1484. }
  1485. db3 := GetReadDB(context.Background())
  1486. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1487. if err2 != nil {
  1488. return err2
  1489. }
  1490. fmt.Println("创建时已违约则创建数据违约记录")
  1491. }
  1492. } else {
  1493. if taskLinkInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1494. db4 := GetReadDB(context.Background())
  1495. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1496. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1497. if err1 != nil {
  1498. return err1
  1499. }
  1500. db3 := GetReadDB(context.Background())
  1501. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1502. if err2 != nil {
  1503. return err2
  1504. }
  1505. fmt.Println("已创建数据违约记录")
  1506. }
  1507. }
  1508. }
  1509. // 根据最近一次拒绝时间判断当前是否违约
  1510. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1511. db2 := GetReadDB(context.Background())
  1512. var LinkBreach int32
  1513. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1514. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1515. var dataDefaultRate int
  1516. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1517. db6 := GetReadDB(context.Background())
  1518. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1519. } else {
  1520. db6 := GetReadDB(context.Background())
  1521. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1522. }
  1523. db1 := GetReadDB(context.Background())
  1524. var taskDataInfo gorm_model.YounggeeDataInfo
  1525. db1.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskDataInfo)
  1526. if taskDataInfo.AutoDataBreakAt.IsZero() {
  1527. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeDataInfo{AutoDataBreakAt: taskDataInfo.RejectAt.Add(dd)}).Error
  1528. if err4 != nil {
  1529. return err4
  1530. }
  1531. fmt.Println("已添加数据违约自动处理时间")
  1532. if taskDataInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1533. db4 := GetReadDB(context.Background())
  1534. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1535. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1536. if err1 != nil {
  1537. return err1
  1538. }
  1539. db3 := GetReadDB(context.Background())
  1540. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1541. if err2 != nil {
  1542. return err2
  1543. }
  1544. fmt.Println("创建时已违约则创建已提交数据的数据违约记录")
  1545. }
  1546. } else {
  1547. if taskDataInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1548. db4 := GetReadDB(context.Background())
  1549. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1550. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1551. if err1 != nil {
  1552. return err1
  1553. }
  1554. db3 := GetReadDB(context.Background())
  1555. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1556. CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1557. if err2 != nil {
  1558. return err2
  1559. }
  1560. fmt.Println("创建已提交数据的数据违约记录")
  1561. }
  1562. }
  1563. }
  1564. return nil
  1565. }
  1566. func GetHours(ctx context.Context) (*http_model.Hours, error) {
  1567. db := GetReadDB(ctx)
  1568. infoAuto := gorm_model.InfoAutoTask{}
  1569. err := db.Model(gorm_model.InfoAutoTask{}).Last(&infoAuto).Error
  1570. if err != nil {
  1571. return nil, err
  1572. }
  1573. Hours := http_model.Hours{}
  1574. Hours.DraftDefaultInPic = infoAuto.DraftDefaultInPic
  1575. Hours.DraftDefaultInMv = infoAuto.DraftDefaultInMv
  1576. Hours.ScriptDefault = infoAuto.ScriptDefault
  1577. Hours.Invalid = infoAuto.Invalid
  1578. Hours.LinkBreach = infoAuto.LinkBreach
  1579. Hours.CaseClose = infoAuto.CaseClose
  1580. Hours.CaseCloseDefault = infoAuto.CaseCloseDefault
  1581. Hours.ReviewInMv = infoAuto.ReviewInMv
  1582. Hours.ReviewUnlimited = infoAuto.ReviewUnlimited
  1583. Hours.PostReview = infoAuto.Postreview
  1584. Hours.SignInVirtual = infoAuto.SignInVirtual
  1585. Hours.SignInOffline = infoAuto.SignInOffline
  1586. return &Hours, err
  1587. }
  1588. func GetPercents(ctx context.Context) (*http_model.Percents, error) {
  1589. db := GetReadDB(ctx)
  1590. percentsInfo := gorm_model.InfoAutoDefaultHandle{}
  1591. err := db.Model(gorm_model.InfoAutoDefaultHandle{}).Last(&percentsInfo).Error
  1592. if err != nil {
  1593. return nil, err
  1594. }
  1595. Percents := http_model.Percents{}
  1596. Percents.DataReplaceNotUpload = percentsInfo.DataReplaceNotUpload
  1597. Percents.DataReplaceTimeOut = percentsInfo.DataReplaceTimeOut
  1598. Percents.DataOtherTimeOut = percentsInfo.DataOtherTimeOut
  1599. Percents.DataOtherNotUpload = percentsInfo.DataOtherNotUpload
  1600. Percents.LinkReplaceTimeOut = percentsInfo.LinkReplaceTimeOut
  1601. Percents.LinkReplaceNotUpload = percentsInfo.LinkReplaceNotUpload
  1602. Percents.LinkOtherTimeOut = percentsInfo.LinkOtherTimeOut
  1603. Percents.LinkOtherNotUpload = percentsInfo.LinkOtherNotUpload
  1604. Percents.ScriptReplaceTimeOut = percentsInfo.ScriptReplaceTimeOut
  1605. Percents.ScriptReplaceNotUpload = percentsInfo.ScriptReplaceNotUpload
  1606. Percents.ScriptOtherNotUpload = percentsInfo.ScriptOtherNotUpload
  1607. Percents.ScriptOtherTimeOut = percentsInfo.ScriptOtherTimeOut
  1608. Percents.SketchReplaceTimeOut = percentsInfo.SketchReplaceTimeOut
  1609. Percents.SketchOtherNotUpload = percentsInfo.SketchOtherNotUpload
  1610. Percents.SketchOtherTimeOut = percentsInfo.SketchOtherTimeOut
  1611. Percents.SketchReplaceNotUpload = percentsInfo.SketchReplaceNotUpload
  1612. return &Percents, err
  1613. }