operate.go 79 KB

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