operate.go 82 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680
  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, 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()})
  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. //fmt.Println("scriptInfos:", scriptInfos)
  377. var scriptTaskIds []int
  378. for _, scriptInfo := range scriptInfos {
  379. scriptTaskIds = append(scriptTaskIds, scriptInfo.TaskID)
  380. scriptReviewNeedMod := gorm_model.YounggeeScriptInfo{}
  381. db5 := GetReadDB(context.Background())
  382. db5.Model(gorm_model.YounggeeScriptInfo{}).Where("script_id = ?", scriptInfo.ScriptID).Find(&scriptReviewNeedMod)
  383. if scriptInfo.AgreeAt.IsZero() && scriptInfo.IsSubmit == 1 {
  384. if scriptReviewNeedMod.IsReview != 1 {
  385. db7 := GetReadDB(context.Background())
  386. db7.Model(gorm_model.YounggeeScriptInfo{}).Where("script_id = ?", scriptInfo.ScriptID).Updates(&gorm_model.YounggeeScriptInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  387. fmt.Println("已更新脚本同意时间 及脚本审核状态")
  388. }
  389. }
  390. }
  391. //fmt.Println("scriptTaskIds:", scriptTaskIds)
  392. for _, scriptTaskId := range scriptTaskIds {
  393. //fmt.Println("scriptTaskId", scriptTaskId)
  394. taskStatusNeedMod := gorm_model.YoungeeTaskInfo{}
  395. db6 := GetReadDB(context.Background())
  396. db6.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", scriptTaskId).Find(&taskStatusNeedMod)
  397. if (taskStatusNeedMod.ScriptStatus == 2 || taskStatusNeedMod.ScriptStatus == 4) && taskStatusNeedMod.TaskStage == 7 {
  398. db8 := GetReadDB(context.Background())
  399. db8.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", scriptTaskId).Updates(&gorm_model.YoungeeTaskInfo{ScriptStatus: 5, TaskStage: 9})
  400. fmt.Println("已更新任务阶段为待传初稿 及脚本上传状态为已通过")
  401. }
  402. }
  403. return nil
  404. } else {
  405. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  406. if err != nil {
  407. return err
  408. }
  409. var projectIds []int64
  410. projectIdToAutoTaskIdMap := map[int64]int{}
  411. for _, projectInfo := range projectInfos {
  412. projectIds = append(projectIds, projectInfo.ProjectID)
  413. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  414. }
  415. //fmt.Println("projectIds:", projectIds)
  416. TaskIdToProjectId := make(map[int]int)
  417. var taskId []int
  418. db1 := GetReadDB(context.Background())
  419. for _, projectId := range projectIds {
  420. // 获取 autoTaskId 及其对应的限制时间、
  421. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  422. dbStart := GetReadDB(context.Background())
  423. var ReviewUnlimited int32
  424. dbStart.Model(gorm_model.InfoAutoTask{}).Select("review_unlimited").Where("auto_task_id = ?", autoTaskId).First(&ReviewUnlimited)
  425. //fmt.Println("t:", autoTaskInfo.ReviewUnlimited)
  426. var taskInfos []gorm_model.YoungeeTaskInfo
  427. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  428. for _, taskInfo := range taskInfos {
  429. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  430. taskId = append(taskId, taskInfo.TaskId)
  431. db2 := GetReadDB(context.Background())
  432. var scriptInfos gorm_model.YounggeeSketchInfo
  433. db2.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&scriptInfos)
  434. dd, _ := time.ParseDuration(conv.MustString(ReviewUnlimited, "") + "h")
  435. if scriptInfos.AutoAgreeAt.IsZero() && scriptInfos.IsSubmit == 1 {
  436. db3 := GetReadDB(context.Background())
  437. db3.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", scriptInfos.TaskID).Updates(&gorm_model.YounggeeSketchInfo{AutoAgreeAt: scriptInfos.SubmitAt.Add(dd)})
  438. fmt.Println("已添加脚本自动审核时间")
  439. }
  440. }
  441. }
  442. //fmt.Println("TaskIdToProjectId:", TaskIdToProjectId)
  443. //fmt.Println("taskId:", taskId)
  444. var sketchInfos []gorm_model.YounggeeSketchInfo
  445. db3 := GetReadDB(context.Background())
  446. err = db3.Model(gorm_model.YounggeeSketchInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&sketchInfos).Error
  447. if err != nil {
  448. log.Println("DB GetAutoReviewTask error in sketch:", err)
  449. return err
  450. }
  451. var sketchTaskIds []int
  452. for _, scriptInfo := range sketchInfos {
  453. sketchTaskIds = append(sketchTaskIds, scriptInfo.TaskID)
  454. sketchNeedMod := gorm_model.YounggeeSketchInfo{}
  455. db4 := GetReadDB(context.Background())
  456. db4.Model(gorm_model.YounggeeSketchInfo{}).Where("sketch_id = ?", scriptInfo.SketchID).Find(&sketchNeedMod)
  457. if scriptInfo.AgreeAt.IsZero() && scriptInfo.IsSubmit == 1 {
  458. if sketchNeedMod.IsReview != 1 {
  459. db6 := GetReadDB(context.Background())
  460. db6.Model(gorm_model.YounggeeSketchInfo{}).Where("sketch_id = ?", scriptInfo.SketchID).Updates(&gorm_model.YounggeeSketchInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  461. fmt.Println("已更新初稿审核同意时间")
  462. }
  463. }
  464. }
  465. for _, sketchTaskId := range sketchTaskIds {
  466. //fmt.Println("taskID", sketchTaskId)
  467. sketchNeedMod := gorm_model.YoungeeTaskInfo{}
  468. db5 := GetReadDB(context.Background())
  469. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", sketchTaskId).Find(&sketchNeedMod)
  470. if (sketchNeedMod.SketchStatus == 2 || sketchNeedMod.SketchStatus == 4) && sketchNeedMod.TaskStage == 9 {
  471. db7 := GetReadDB(context.Background())
  472. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", sketchTaskId).Updates(&gorm_model.YoungeeTaskInfo{SketchStatus: 5, TaskStage: 11})
  473. fmt.Println("已更新任务阶段为待传链接 及初稿上传状态为已通过")
  474. }
  475. }
  476. return nil
  477. }
  478. }
  479. func GetAutoPostReviewTask() error {
  480. db := GetReadDB(context.Background())
  481. var projectInfos []*gorm_model.ProjectInfo
  482. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  483. if err != nil {
  484. return err
  485. }
  486. var projectIds []int64
  487. projectIdToAutoTaskIdMap := map[int64]int{}
  488. for _, projectInfo := range projectInfos {
  489. projectIds = append(projectIds, projectInfo.ProjectID)
  490. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  491. }
  492. //fmt.Println("projectIds:", projectIds)
  493. TaskIdToProjectId := make(map[int]int)
  494. var taskId []int
  495. db1 := GetReadDB(context.Background())
  496. for _, projectId := range projectIds {
  497. // 获取 autoTaskId 及其对应的限制时间、
  498. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  499. dbStart := GetReadDB(context.Background())
  500. var Postreview int32
  501. dbStart.Model(gorm_model.InfoAutoTask{}).Select("postreview").Where("auto_task_id = ?", autoTaskId).First(&Postreview)
  502. //fmt.Println("t:", autoTaskInfo.Postreview)
  503. var taskInfos []gorm_model.YoungeeTaskInfo
  504. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  505. for _, taskInfo := range taskInfos {
  506. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  507. taskId = append(taskId, taskInfo.TaskId)
  508. db2 := GetReadDB(context.Background())
  509. var linkInfos gorm_model.YounggeeLinkInfo
  510. db2.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&linkInfos)
  511. dd, _ := time.ParseDuration(conv.MustString(Postreview, "") + "h")
  512. if linkInfos.AutoAgreeAt.IsZero() && linkInfos.IsSubmit == 1 {
  513. db3 := GetReadDB(context.Background())
  514. db3.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ?", linkInfos.TaskID).Updates(&gorm_model.YounggeeLinkInfo{AutoAgreeAt: linkInfos.SubmitAt.Add(dd)})
  515. fmt.Println("已添加链接自动审核时间")
  516. }
  517. }
  518. }
  519. var linkInfos []gorm_model.YounggeeLinkInfo
  520. db3 := GetReadDB(context.Background())
  521. err = db3.Model(gorm_model.YounggeeLinkInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&linkInfos).Error
  522. if err != nil {
  523. log.Println("DB GetAutoPostReviewTask error in sketch:", err)
  524. return err
  525. }
  526. var linkTaskIds []int
  527. for _, linkInfo := range linkInfos {
  528. linkTaskIds = append(linkTaskIds, linkInfo.TaskID)
  529. linkNeedMod := gorm_model.YounggeeLinkInfo{}
  530. db4 := GetReadDB(context.Background())
  531. db4.Model(gorm_model.YounggeeLinkInfo{}).Where("link_id = ?", linkInfo.LinkID).Find(&linkNeedMod)
  532. if linkInfo.AgreeAt.IsZero() && linkInfo.IsSubmit == 1 {
  533. if linkNeedMod.IsReview != 1 {
  534. db6 := GetReadDB(context.Background())
  535. db6.Model(gorm_model.YounggeeLinkInfo{}).Where("link_id = ?", linkInfo.LinkID).Updates(&gorm_model.YounggeeLinkInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  536. fmt.Println("已更新链接审核同意时间")
  537. }
  538. }
  539. }
  540. for _, linkTaskId := range linkTaskIds {
  541. //fmt.Println("taskID", linkTaskId)
  542. linkNeedMod := gorm_model.YoungeeTaskInfo{}
  543. db5 := GetReadDB(context.Background())
  544. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", linkTaskId).Find(&linkNeedMod)
  545. if (linkNeedMod.LinkStatus == 2 || linkNeedMod.LinkStatus == 4) && linkNeedMod.TaskStage == 11 {
  546. db7 := GetReadDB(context.Background())
  547. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", linkTaskId).Updates(&gorm_model.YoungeeTaskInfo{LinkStatus: 5, TaskStage: 13})
  548. fmt.Println("已更新任务阶段为待传数据 及 链接上传状态为已通过")
  549. }
  550. }
  551. return nil
  552. }
  553. func GetAutoCaseCloseTask() error {
  554. db := GetReadDB(context.Background())
  555. var projectInfos []*gorm_model.ProjectInfo
  556. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? ", 1).Find(&projectInfos).Error
  557. if err != nil {
  558. return err
  559. }
  560. var projectIds []int64
  561. projectIdToAutoTaskIdMap := map[int64]int{}
  562. for _, projectInfo := range projectInfos {
  563. projectIds = append(projectIds, projectInfo.ProjectID)
  564. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  565. }
  566. //fmt.Println("projectIds:", projectIds)
  567. TaskIdToProjectId := make(map[int]int)
  568. var taskId []int
  569. db1 := GetReadDB(context.Background())
  570. for _, projectId := range projectIds {
  571. // 获取 autoTaskId 及其对应的限制时间、
  572. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  573. dbStart := GetReadDB(context.Background())
  574. var CaseClose int32
  575. dbStart.Model(gorm_model.InfoAutoTask{}).Select("case_close").Where("auto_task_id = ?", autoTaskId).First(&CaseClose)
  576. //fmt.Println("t:", autoTaskInfo.CaseClose)
  577. var taskInfos []gorm_model.YoungeeTaskInfo
  578. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ?", projectId).Find(&taskInfos)
  579. for _, taskInfo := range taskInfos {
  580. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  581. taskId = append(taskId, taskInfo.TaskId)
  582. db2 := GetReadDB(context.Background())
  583. var dataInfos gorm_model.YounggeeDataInfo
  584. db2.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ? AND is_submit = ? AND is_review = ? ", taskInfo.TaskId, 1, 0).Find(&dataInfos)
  585. dd, _ := time.ParseDuration(conv.MustString(CaseClose, "") + "h")
  586. if dataInfos.AutoAgreeAt.IsZero() && dataInfos.IsSubmit == 1 {
  587. db3 := GetReadDB(context.Background())
  588. db3.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ?", dataInfos.TaskID).Updates(&gorm_model.YounggeeDataInfo{AutoAgreeAt: dataInfos.SubmitAt.Add(dd)})
  589. fmt.Println("已添加链接自动审核时间")
  590. }
  591. }
  592. }
  593. var dataInfos []gorm_model.YounggeeDataInfo
  594. db3 := GetReadDB(context.Background())
  595. err = db3.Model(gorm_model.YounggeeDataInfo{}).Where("auto_agree_at <= ? AND task_id IN ?", time.Now(), taskId).Find(&dataInfos).Error
  596. if err != nil {
  597. log.Println("DB GetAutoCaseCloseTask error in sketch:", err)
  598. return err
  599. }
  600. var dataTaskIds []int
  601. for _, dataInfo := range dataInfos {
  602. dataTaskIds = append(dataTaskIds, dataInfo.TaskID)
  603. dataNeedMod := gorm_model.YounggeeDataInfo{}
  604. db4 := GetReadDB(context.Background())
  605. db4.Model(gorm_model.YounggeeDataInfo{}).Where("data_id = ?", dataInfo.DataID).Find(&dataNeedMod)
  606. if dataInfo.AgreeAt.IsZero() && dataInfo.IsSubmit == 1 {
  607. if dataNeedMod.IsReview != 1 {
  608. db6 := GetReadDB(context.Background())
  609. db6.Model(gorm_model.YounggeeDataInfo{}).Where("data_id = ?", dataInfo.DataID).Updates(&gorm_model.YounggeeDataInfo{AgreeAt: time.Now(), IsReview: 1, IsOk: 1})
  610. fmt.Println("已更新数据审核同意时间")
  611. }
  612. }
  613. }
  614. for _, dataTaskId := range dataTaskIds {
  615. dataNeedMod := gorm_model.YoungeeTaskInfo{}
  616. db5 := GetReadDB(context.Background())
  617. db5.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", dataTaskId).Find(&dataNeedMod)
  618. if (dataNeedMod.DataStatus == 2 || dataNeedMod.DataStatus == 4) && dataNeedMod.TaskStage == 13 {
  619. db7 := GetReadDB(context.Background())
  620. db7.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", dataTaskId).Updates(&gorm_model.YoungeeTaskInfo{DataStatus: 5, TaskStage: 15})
  621. fmt.Println("已更新任务阶段为已结案 及 数据上传状态为已通过")
  622. }
  623. }
  624. return nil
  625. }
  626. func GetAutoInvalidTask() error {
  627. db := GetReadDB(context.Background())
  628. var projectInfos []*gorm_model.ProjectInfo
  629. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND project_status = ? ", 1, 6).Find(&projectInfos).Error
  630. if err != nil {
  631. return err
  632. }
  633. var projectIds []int64
  634. projectIdToAutoTaskIdMap := map[int64]int{}
  635. for _, projectInfo := range projectInfos {
  636. projectIds = append(projectIds, projectInfo.ProjectID)
  637. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  638. }
  639. for _, projectId := range projectIds {
  640. // 获取 autoTaskId 及其对应的限制时间、
  641. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  642. dbStart := GetReadDB(context.Background())
  643. var Invalid int32
  644. dbStart.Model(gorm_model.InfoAutoTask{}).Select("invalid").Where("auto_task_id = ?", autoTaskId).First(&Invalid)
  645. //fmt.Println("t:", autoTaskInfo.Invalid)
  646. db1 := GetReadDB(context.Background())
  647. project := gorm_model.ProjectInfo{}
  648. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&project)
  649. //if project.AutoFailAt.IsZero() {
  650. // dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  651. // t := project.RecruitDdl.Add(dd)
  652. // db2 := GetReadDB(context.Background())
  653. // db2.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  654. // fmt.Println("已添加失效自动处理时间")
  655. //}
  656. dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  657. project.RecruitDdl.Add(dd)
  658. t := project.RecruitDdl.Add(dd)
  659. if project.AutoFailAt != &t {
  660. db3 := GetReadDB(context.Background())
  661. db3.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  662. fmt.Println("已添加失效自动处理时间")
  663. }
  664. projectNeedMod := gorm_model.ProjectInfo{}
  665. db2 := GetReadDB(context.Background())
  666. db2.Where("project_id = ?", projectId).First(&projectNeedMod)
  667. if !projectNeedMod.AutoFailAt.IsZero() {
  668. if projectNeedMod.AutoFailAt.Sub(time.Now()) <= 0 && projectNeedMod.ProjectStatus < 8 {
  669. db4 := GetReadDB(context.Background())
  670. err1 := db4.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{ProjectStatus: 8}).Error
  671. if err1 != nil {
  672. return err1
  673. }
  674. fmt.Println("已更新项目状态")
  675. }
  676. }
  677. }
  678. return nil
  679. }
  680. func GetAutoDraftDefaultInPicTask() error {
  681. db := GetReadDB(context.Background())
  682. var projectInfos []*gorm_model.ProjectInfo
  683. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 1).Find(&projectInfos).Error
  684. if err != nil {
  685. return err
  686. }
  687. // 项目类型为 实体商品寄拍、虚拟产品测评、线下探店打卡 均为正常的 收发货情况,以签收状况作为判断条件
  688. var projectIds []int64
  689. // 存储素材微原创的项目id,此情况单独处理,它不涉及到物流,也不涉及到脚本,初稿上一个状态直接就是支付。
  690. var specialProjectIds []int64
  691. projectIdToAutoTaskIdMap := map[int64]int{}
  692. // 项目id 对 违约定时任务id 的map
  693. projectIdToAutoDefaultIdMap := map[int64]int{}
  694. // taskId 对 稿费形式的 map
  695. taskIdToFeeFormMap := make(map[int]int)
  696. TaskIdToProjectId := make(map[int]int)
  697. var taskNeedModIds []int
  698. var specialTaskNeedModIds []int
  699. // 已提交初稿的任务记录id
  700. var submitTaskNeedModIds []int
  701. for _, projectInfo := range projectInfos {
  702. if projectInfo.ProjectForm == 4 {
  703. specialProjectIds = append(specialProjectIds, projectInfo.ProjectID)
  704. specialProjectInfo := gorm_model.ProjectInfo{}
  705. db1 := GetReadDB(context.Background())
  706. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).First(&specialProjectInfo)
  707. dbStart := GetReadDB(context.Background())
  708. var DraftDefaultInPic int32
  709. dbStart.Model(gorm_model.InfoAutoTask{}).Select("draft_default_in_pic").Where("auto_task_id = ?", specialProjectInfo.AutoTaskID).First(&DraftDefaultInPic)
  710. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInPic, "") + "h")
  711. t1 := projectInfo.PayAt.Add(dd)
  712. if !projectInfo.PayAt.IsZero() && projectInfo.AutoSketchBreakAt.IsZero() {
  713. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).Updates(&gorm_model.ProjectInfo{AutoSketchBreakAt: &t1})
  714. fmt.Println("已添加素材微原创的初稿违约自动处理时间")
  715. }
  716. } else {
  717. projectIds = append(projectIds, projectInfo.ProjectID)
  718. }
  719. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  720. projectIdToAutoDefaultIdMap[projectInfo.ProjectID] = int(projectInfo.AutoDefaultID)
  721. }
  722. //fmt.Println("projectIds:", projectIds)
  723. //fmt.Println("specialProjectIds:", specialProjectIds)
  724. for _, specialProjectId := range specialProjectIds {
  725. db2 := GetReadDB(context.Background())
  726. var taskInfos []gorm_model.YoungeeTaskInfo
  727. db2.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ?", specialProjectId, 0).Find(&taskInfos)
  728. for _, taskInfo := range taskInfos {
  729. TaskIdToProjectId[taskInfo.TaskId] = int(specialProjectId)
  730. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  731. specialTaskNeedMod := gorm_model.YoungeeTaskInfo{}
  732. db3 := GetReadDB(context.Background())
  733. // 保存所有初稿上传状态为 1或 3 的任务记录的id
  734. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskInfo.TaskId).First(&specialTaskNeedMod)
  735. if specialTaskNeedMod.SketchStatus == 1 && taskInfo.TaskStage == 10 {
  736. specialTaskNeedModIds = append(specialTaskNeedModIds, taskInfo.TaskId)
  737. }
  738. if specialTaskNeedMod.SketchStatus == 3 && taskInfo.TaskStage == 10 {
  739. submitTaskNeedModIds = append(submitTaskNeedModIds, taskInfo.TaskId)
  740. }
  741. }
  742. }
  743. //fmt.Println("specialTaskNeedModIds:", specialTaskNeedModIds)
  744. for _, specialTaskNeedModId := range specialTaskNeedModIds {
  745. projectNeedMod := gorm_model.ProjectInfo{}
  746. db1 := GetReadDB(context.Background())
  747. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", TaskIdToProjectId[specialTaskNeedModId]).First(&projectNeedMod)
  748. var sketchDefaultRate int
  749. if taskIdToFeeFormMap[specialTaskNeedModId] == 1 {
  750. db6 := GetReadDB(context.Background())
  751. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[specialTaskNeedModId])]).Find(&sketchDefaultRate)
  752. } else {
  753. db6 := GetReadDB(context.Background())
  754. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[specialTaskNeedModId])]).Find(&sketchDefaultRate)
  755. }
  756. if projectNeedMod.AutoSketchBreakAt.Sub(time.Now()) <= 0 && !projectNeedMod.AutoSketchBreakAt.IsZero() {
  757. db2 := GetReadDB(context.Background())
  758. err1 := db2.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  759. TaskID: specialTaskNeedModId, ProjectID: TaskIdToProjectId[specialTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  760. if err1 != nil {
  761. return err1
  762. }
  763. db3 := GetReadDB(context.Background())
  764. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", specialTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate})
  765. fmt.Println("已创建素材微原创的任务初稿违约记录")
  766. }
  767. }
  768. for _, projectId := range projectIds {
  769. db1 := GetReadDB(context.Background())
  770. var taskInfos []gorm_model.YoungeeTaskInfo
  771. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  772. for _, taskInfo := range taskInfos {
  773. TaskIdToProjectId[taskInfo.TaskId] = int(projectId)
  774. taskNeedMod := gorm_model.YoungeeTaskInfo{}
  775. db2 := GetReadDB(context.Background())
  776. // 保存所有满足物流状态为3 且 初稿上传状态为 1或 3 的任务记录的id
  777. db2.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskInfo.TaskId).First(&taskNeedMod)
  778. if taskNeedMod.LogisticsStatus == 3 && taskNeedMod.ScriptStatus == 1 {
  779. taskNeedModIds = append(taskNeedModIds, taskInfo.TaskId)
  780. }
  781. if taskNeedMod.LogisticsStatus == 3 && taskNeedMod.ScriptStatus == 3 {
  782. submitTaskNeedModIds = append(submitTaskNeedModIds, taskInfo.TaskId)
  783. }
  784. }
  785. }
  786. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  787. //fmt.Println("TaskIdToProjectId:", TaskIdToProjectId)
  788. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  789. for _, taskNeedModId := range taskNeedModIds {
  790. // 获取 autoTaskId 及其对应的限制时间、
  791. //fmt.Println("taskNeedModId:", taskNeedModId)
  792. autoTaskId := projectIdToAutoTaskIdMap[int64(TaskIdToProjectId[taskNeedModId])]
  793. dbStart := GetReadDB(context.Background())
  794. var DraftDefaultInPic int32
  795. dbStart.Model(gorm_model.InfoAutoTask{}).Select("draft_default_in_pic").Where("auto_task_id = ?", autoTaskId).First(&DraftDefaultInPic)
  796. //fmt.Println("t:", autoTaskInfo.DraftDefaultInPic)
  797. db3 := GetReadDB(context.Background())
  798. taskLogisticNeedMod := gorm_model.YoungeeTaskLogistics{}
  799. db3.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&taskLogisticNeedMod)
  800. var sketchDefaultRate int
  801. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  802. db6 := GetReadDB(context.Background())
  803. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[taskNeedModId])]).Find(&sketchDefaultRate)
  804. } else {
  805. db6 := GetReadDB(context.Background())
  806. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[taskNeedModId])]).Find(&sketchDefaultRate)
  807. }
  808. if taskLogisticNeedMod.AutoSketchBreakAt.IsZero() {
  809. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInPic, "") + "h")
  810. db4 := GetReadDB(context.Background())
  811. logisticNeedMod := gorm_model.YoungeeTaskLogistics{}
  812. db4.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&logisticNeedMod)
  813. if logisticNeedMod.ThingsType == 1 && !logisticNeedMod.SignedTime.IsZero() {
  814. db4.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{
  815. AutoSketchBreakAt: taskLogisticNeedMod.SignedTime.Add(dd),
  816. })
  817. fmt.Println("已添加实物的图片初稿违约自动处理时间")
  818. }
  819. if logisticNeedMod.ThingsType != 1 && !logisticNeedMod.AutoSignAt.IsZero() {
  820. db4.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{
  821. AutoSketchBreakAt: taskLogisticNeedMod.AutoSignAt.Add(dd),
  822. })
  823. fmt.Println("已添加虚拟产品及线下探店的图片初稿违约自动处理时间")
  824. }
  825. }
  826. db7 := GetReadDB(context.Background())
  827. taskLogisticNeedMod2 := gorm_model.YoungeeTaskLogistics{}
  828. db7.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&taskLogisticNeedMod2)
  829. if !taskLogisticNeedMod2.AutoSketchBreakAt.IsZero() && taskLogisticNeedMod2.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  830. db5 := GetReadDB(context.Background())
  831. err1 := db5.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  832. TaskID: taskNeedModId, ProjectID: TaskIdToProjectId[taskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  833. if err1 != nil {
  834. return err1
  835. }
  836. db8 := GetReadDB(context.Background())
  837. db8.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskLogisticNeedMod2.TaskID).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate})
  838. fmt.Println("已创建图文类型的初稿违约记录")
  839. }
  840. }
  841. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  842. db2 := GetReadDB(context.Background())
  843. var DraftDefaultInMv int32
  844. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).First(&DraftDefaultInMv)
  845. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  846. var sketchDefaultRate int
  847. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  848. db6 := GetReadDB(context.Background())
  849. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).Find(&sketchDefaultRate)
  850. } else {
  851. db6 := GetReadDB(context.Background())
  852. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[int64(TaskIdToProjectId[submitTaskNeedModId])]).Find(&sketchDefaultRate)
  853. }
  854. db1 := GetReadDB(context.Background())
  855. var taskSketchInfo gorm_model.YounggeeSketchInfo
  856. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  857. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  858. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  859. if err4 != nil {
  860. return err4
  861. }
  862. fmt.Println("已添加图文形式的初稿违约自动处理时间")
  863. if taskSketchInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  864. db4 := GetReadDB(context.Background())
  865. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  866. TaskID: submitTaskNeedModId, ProjectID: TaskIdToProjectId[submitTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  867. if err1 != nil {
  868. return err1
  869. }
  870. db3 := GetReadDB(context.Background())
  871. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  872. if err2 != nil {
  873. return err2
  874. }
  875. fmt.Println("创建时已违约则创建已提交初稿的图文类型的初稿违约记录")
  876. }
  877. } else {
  878. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  879. db4 := GetReadDB(context.Background())
  880. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  881. TaskID: submitTaskNeedModId, ProjectID: TaskIdToProjectId[submitTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  882. if err1 != nil {
  883. return err1
  884. }
  885. db3 := GetReadDB(context.Background())
  886. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  887. if err2 != nil {
  888. return err2
  889. }
  890. fmt.Println("创建已提交初稿的图文类型的初稿违约记录")
  891. }
  892. }
  893. }
  894. return nil
  895. }
  896. func GetAutoDraftDefaultInMvTask() error {
  897. db := GetReadDB(context.Background())
  898. var VideoProjectIds []int64
  899. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 2).Find(&VideoProjectIds).Error
  900. if err != nil {
  901. return err
  902. }
  903. //fmt.Println("VideoProjectIds:", VideoProjectIds)
  904. // 任务id 对 项目id 的map
  905. taskIdToProjectIdMap := map[int]int64{}
  906. // 项目id 对 定时任务id 的map
  907. projectIdToAutoTaskIdMap := map[int64]int{}
  908. // 项目id 对 违约定时任务id 的map
  909. projectIdToAutoDefaultIdMap := map[int64]int{}
  910. // taskId 对 稿费形式的 map
  911. taskIdToFeeFormMap := make(map[int]int)
  912. var videoTaskIds []int
  913. for _, VideoProjectId := range VideoProjectIds {
  914. db2 := GetReadDB(context.Background())
  915. autoIds := gorm_model.ProjectInfo{}
  916. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", VideoProjectId).First(&autoIds)
  917. projectIdToAutoTaskIdMap[VideoProjectId] = int(autoIds.AutoTaskID)
  918. projectIdToAutoDefaultIdMap[VideoProjectId] = int(autoIds.AutoDefaultID)
  919. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  920. db1 := GetReadDB(context.Background())
  921. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", VideoProjectId, 0).Find(&videoTaskInfos)
  922. for _, videoTaskInfo := range videoTaskInfos {
  923. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  924. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  925. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  926. }
  927. }
  928. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  929. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  930. //fmt.Println("videoTaskIds:", videoTaskIds)
  931. // 首次提交初稿的任务记录id
  932. var videoTaskNeedModIds []int
  933. // 已提交初稿的任务记录id
  934. var submitVideoTaskNeedModIds []int
  935. for _, videoTaskId := range videoTaskIds {
  936. var taskInfo gorm_model.YoungeeTaskInfo
  937. db3 := GetReadDB(context.Background())
  938. // 保存所有初稿上传状态为 1 且任务状态为10 的任务记录的id
  939. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  940. if taskInfo.SketchStatus == 1 && taskInfo.TaskStage == 10 {
  941. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  942. }
  943. // 保存所有初稿上传状态为 3 且任务状态为10 的任务记录的id
  944. if taskInfo.SketchStatus == 3 && taskInfo.TaskStage == 10 {
  945. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  946. }
  947. }
  948. //fmt.Println("videoTaskNeedModIds:", videoTaskNeedModIds)
  949. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  950. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  951. db2 := GetReadDB(context.Background())
  952. var DraftDefaultInMv int32
  953. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&DraftDefaultInMv)
  954. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  955. var sketchDefaultRate int
  956. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  957. db6 := GetReadDB(context.Background())
  958. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  959. } else {
  960. db6 := GetReadDB(context.Background())
  961. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  962. }
  963. db1 := GetReadDB(context.Background())
  964. var taskScriptInfo gorm_model.YounggeeScriptInfo
  965. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ? AND is_ok = ?", videoTaskNeedModId, 1).Find(&taskScriptInfo)
  966. if taskScriptInfo.AutoSketchBreakAt.IsZero() {
  967. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoSketchBreakAt: taskScriptInfo.AutoAgreeAt.Add(dd)})
  968. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  969. if taskScriptInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  970. db4 := GetReadDB(context.Background())
  971. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  972. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  973. if err1 != nil {
  974. return err1
  975. }
  976. db3 := GetReadDB(context.Background())
  977. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  978. if err2 != nil {
  979. return err2
  980. }
  981. fmt.Println("创建时已违约则创建视频类型的初稿违约记录")
  982. }
  983. } else {
  984. if taskScriptInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  985. db4 := GetReadDB(context.Background())
  986. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  987. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  988. if err1 != nil {
  989. return err1
  990. }
  991. db3 := GetReadDB(context.Background())
  992. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  993. if err2 != nil {
  994. return err2
  995. }
  996. fmt.Println("已创建视频类型的初稿违约记录")
  997. }
  998. }
  999. }
  1000. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  1001. db2 := GetReadDB(context.Background())
  1002. var DraftDefaultInMv int32
  1003. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&DraftDefaultInMv)
  1004. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  1005. var sketchDefaultRate int
  1006. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  1007. db6 := GetReadDB(context.Background())
  1008. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  1009. } else {
  1010. db6 := GetReadDB(context.Background())
  1011. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  1012. }
  1013. db1 := GetReadDB(context.Background())
  1014. var taskSketchInfo gorm_model.YounggeeSketchInfo
  1015. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  1016. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  1017. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  1018. if err4 != nil {
  1019. return err4
  1020. }
  1021. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  1022. if taskSketchInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1023. db4 := GetReadDB(context.Background())
  1024. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1025. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1026. if err1 != nil {
  1027. return err1
  1028. }
  1029. db3 := GetReadDB(context.Background())
  1030. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  1031. if err2 != nil {
  1032. return err2
  1033. }
  1034. fmt.Println("创建时已违约则创建已提交初稿的视频类型的初稿违约记录")
  1035. }
  1036. } else {
  1037. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  1038. db4 := GetReadDB(context.Background())
  1039. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1040. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1041. if err1 != nil {
  1042. return err1
  1043. }
  1044. db3 := GetReadDB(context.Background())
  1045. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate}).Error
  1046. if err2 != nil {
  1047. return err2
  1048. }
  1049. fmt.Println("创建已提交初稿的视频类型的初稿违约记录")
  1050. }
  1051. }
  1052. }
  1053. return nil
  1054. }
  1055. func GetAutoScriptDefaultTask() error {
  1056. db := GetReadDB(context.Background())
  1057. var VideoProjectIds []int64
  1058. var specialProjectIds []int
  1059. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? AND project_form != 4", 1, 2).Find(&VideoProjectIds).Error
  1060. if err != nil {
  1061. return err
  1062. }
  1063. db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? AND project_form = 4", 1, 2).Find(&specialProjectIds)
  1064. //fmt.Println("VideoProjectIds:", VideoProjectIds)
  1065. //fmt.Println("specialProjectIds:", specialProjectIds)
  1066. // 任务id 对 项目id 的map
  1067. taskIdToProjectIdMap := map[int]int64{}
  1068. // 项目id 对 定时任务id 的map
  1069. projectIdToAutoTaskIdMap := map[int64]int{}
  1070. // 项目id 对 违约定时任务id 的map
  1071. projectIdToAutoDefaultIdMap := map[int64]int{}
  1072. // taskId 对 稿费形式的 map
  1073. taskIdToFeeFormMap := make(map[int]int)
  1074. // 素材微原创的任务id
  1075. var specialVideoTaskIds []int
  1076. // 需要判断是否违约的素材微原创的任务id
  1077. var specialTaskNeedModIds []int
  1078. // 首次提交脚本的任务记录id
  1079. var videoTaskNeedModIds []int
  1080. // 已提交脚本的任务记录id
  1081. var submitVideoTaskNeedModIds []int
  1082. for _, specialProjectId := range specialProjectIds {
  1083. db2 := GetReadDB(context.Background())
  1084. var projectInfo gorm_model.ProjectInfo
  1085. db2.Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", specialProjectId).First(&projectInfo)
  1086. projectIdToAutoTaskIdMap[int64(specialProjectId)] = int(projectInfo.AutoTaskID)
  1087. projectIdToAutoDefaultIdMap[int64(specialProjectId)] = int(projectInfo.AutoDefaultID)
  1088. var ScriptDefault int32
  1089. dbStart := GetReadDB(context.Background())
  1090. dbStart.Model(gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectInfo.AutoTaskID).First(&ScriptDefault)
  1091. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1092. t3 := projectInfo.PayAt.Add(dd)
  1093. if !projectInfo.PayAt.IsZero() && projectInfo.AutoScriptBreakAt.IsZero() {
  1094. db1 := GetReadDB(context.Background())
  1095. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).Updates(&gorm_model.ProjectInfo{AutoScriptBreakAt: &t3})
  1096. fmt.Println("已添加素材微原创的脚本违约自动处理时间")
  1097. }
  1098. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1099. db1 := GetReadDB(context.Background())
  1100. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", specialProjectId, 0).Find(&videoTaskInfos)
  1101. for _, videoTaskInfo := range videoTaskInfos {
  1102. taskIdToProjectIdMap[videoTaskInfo.TaskId] = int64(specialProjectId)
  1103. specialVideoTaskIds = append(specialVideoTaskIds, videoTaskInfo.TaskId)
  1104. specialTaskNeedMod := gorm_model.YoungeeTaskInfo{}
  1105. db3 := GetReadDB(context.Background())
  1106. // 保存所有脚本上传状态为 1 且任务状态为8 的任务记录的id
  1107. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskInfo.TaskId).First(&specialTaskNeedMod)
  1108. if specialTaskNeedMod.ScriptStatus == 1 && specialTaskNeedMod.TaskStage == 8 {
  1109. specialTaskNeedModIds = append(specialTaskNeedModIds, videoTaskInfo.TaskId)
  1110. }
  1111. // 保存所有脚本上传状态为 3 且任务状态为8 的任务记录的id
  1112. if specialTaskNeedMod.ScriptStatus == 3 && specialTaskNeedMod.TaskStage == 8 {
  1113. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskInfo.TaskId)
  1114. }
  1115. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  1116. }
  1117. }
  1118. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  1119. //fmt.Println("specialTaskNeedModIds:", specialTaskNeedModIds)
  1120. for _, specialTaskNeedModId := range specialTaskNeedModIds {
  1121. projectNeedMod := gorm_model.ProjectInfo{}
  1122. db1 := GetReadDB(context.Background())
  1123. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", taskIdToProjectIdMap[specialTaskNeedModId]).First(&projectNeedMod)
  1124. if projectNeedMod.AutoScriptBreakAt.Sub(time.Now()) <= 0 && !projectNeedMod.AutoScriptBreakAt.IsZero() {
  1125. var scriptDefaultRate int
  1126. if taskIdToFeeFormMap[specialTaskNeedModId] == 1 {
  1127. db6 := GetReadDB(context.Background())
  1128. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[specialTaskNeedModId]]).Find(&scriptDefaultRate)
  1129. } else {
  1130. db6 := GetReadDB(context.Background())
  1131. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[specialTaskNeedModId]]).Find(&scriptDefaultRate)
  1132. }
  1133. db2 := GetReadDB(context.Background())
  1134. err1 := db2.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1135. TaskID: specialTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[specialTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1136. if err1 != nil {
  1137. return err1
  1138. }
  1139. db3 := GetReadDB(context.Background())
  1140. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", specialTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate})
  1141. fmt.Println("已创建素材微原创的任务脚本违约记录")
  1142. }
  1143. }
  1144. var videoTaskIds []int
  1145. for _, VideoProjectId := range VideoProjectIds {
  1146. db2 := GetReadDB(context.Background())
  1147. var autoTaskId int
  1148. db2.Select("auto_task_id").Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", VideoProjectId).First(&autoTaskId)
  1149. projectIdToAutoTaskIdMap[VideoProjectId] = autoTaskId
  1150. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1151. db1 := GetReadDB(context.Background())
  1152. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", VideoProjectId, 0).Find(&videoTaskInfos)
  1153. for _, videoTaskInfo := range videoTaskInfos {
  1154. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  1155. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  1156. }
  1157. }
  1158. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1159. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  1160. //fmt.Println("videoTaskIds:", videoTaskIds)
  1161. for _, videoTaskId := range videoTaskIds {
  1162. var taskInfo gorm_model.YoungeeTaskInfo
  1163. db3 := GetReadDB(context.Background())
  1164. // 保存所有脚本上传状态为 1 且任务状态为8 的任务记录的id
  1165. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  1166. if taskInfo.ScriptStatus == 1 && taskInfo.TaskStage == 8 {
  1167. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  1168. }
  1169. // 保存所有脚本上传状态为 3 且任务状态为8 的任务记录的id
  1170. if taskInfo.ScriptStatus == 3 && taskInfo.TaskStage == 8 {
  1171. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  1172. }
  1173. }
  1174. //fmt.Println("videoTaskNeedModIds:", videoTaskNeedModIds)
  1175. //fmt.Println("submitVideoTaskNeedModIds:", submitVideoTaskNeedModIds)
  1176. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  1177. db2 := GetReadDB(context.Background())
  1178. var ScriptDefault int32
  1179. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&ScriptDefault)
  1180. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1181. db1 := GetReadDB(context.Background())
  1182. var taskLogisticInfo gorm_model.YoungeeTaskLogistics
  1183. db1.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskLogisticInfo)
  1184. var scriptDefaultRate int
  1185. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  1186. db6 := GetReadDB(context.Background())
  1187. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1188. } else {
  1189. db6 := GetReadDB(context.Background())
  1190. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1191. }
  1192. if taskLogisticInfo.AutoScriptBreakAt.IsZero() {
  1193. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{AutoScriptBreakAt: taskLogisticInfo.AutoSignAt.Add(dd)})
  1194. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1195. if taskLogisticInfo.AutoSignAt.Add(dd).Sub(time.Now()) <= 0 {
  1196. db4 := GetReadDB(context.Background())
  1197. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1198. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1199. if err1 != nil {
  1200. return err1
  1201. }
  1202. db3 := GetReadDB(context.Background())
  1203. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1204. if err2 != nil {
  1205. return err2
  1206. }
  1207. fmt.Println("创建时已违约则创建视频类型的脚本违约记录")
  1208. }
  1209. } else {
  1210. if taskLogisticInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1211. db4 := GetReadDB(context.Background())
  1212. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1213. TaskID: videoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[videoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1214. if err1 != nil {
  1215. return err1
  1216. }
  1217. db3 := GetReadDB(context.Background())
  1218. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1219. if err2 != nil {
  1220. return err2
  1221. }
  1222. fmt.Println("已创建视频类型的脚本违约记录")
  1223. }
  1224. }
  1225. }
  1226. // 根据最近一次拒绝时间判断当前是否违约
  1227. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  1228. db2 := GetReadDB(context.Background())
  1229. var ScriptDefault int32
  1230. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&ScriptDefault)
  1231. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1232. db1 := GetReadDB(context.Background())
  1233. var taskScriptInfo gorm_model.YounggeeScriptInfo
  1234. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskScriptInfo)
  1235. var scriptDefaultRate int
  1236. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  1237. db6 := GetReadDB(context.Background())
  1238. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1239. } else {
  1240. db6 := GetReadDB(context.Background())
  1241. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1242. }
  1243. if taskScriptInfo.AutoScriptBreakAt.IsZero() {
  1244. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoScriptBreakAt: taskScriptInfo.RejectAt.Add(dd)}).Error
  1245. if err4 != nil {
  1246. return err4
  1247. }
  1248. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1249. if taskScriptInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1250. db4 := GetReadDB(context.Background())
  1251. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1252. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1253. if err1 != nil {
  1254. return err1
  1255. }
  1256. db3 := GetReadDB(context.Background())
  1257. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1258. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1259. if err2 != nil {
  1260. return err2
  1261. }
  1262. fmt.Println("创建时已违约则创建已提交脚本的视频类型的脚本违约记录")
  1263. }
  1264. } else {
  1265. if taskScriptInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1266. db4 := GetReadDB(context.Background())
  1267. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1268. TaskID: submitVideoTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitVideoTaskNeedModId]), BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1269. if err1 != nil {
  1270. return err1
  1271. }
  1272. db3 := GetReadDB(context.Background())
  1273. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1274. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate}).Error
  1275. if err2 != nil {
  1276. return err2
  1277. }
  1278. fmt.Println("创建已提交脚本的视频类型的脚本违约记录")
  1279. }
  1280. }
  1281. }
  1282. return nil
  1283. }
  1284. func GetAutoLinkBreachTask() error {
  1285. db := GetReadDB(context.Background())
  1286. var projectIds []int64
  1287. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1288. if err != nil {
  1289. return err
  1290. }
  1291. //fmt.Println("projectIds:", projectIds)
  1292. // 任务id 对 项目id 的map
  1293. taskIdToProjectIdMap := map[int]int64{}
  1294. // 项目id 对 定时任务id 的map
  1295. projectIdToAutoTaskIdMap := map[int64]int{}
  1296. // 项目id 对 违约定时任务id 的map
  1297. projectIdToAutoDefaultIdMap := map[int64]int{}
  1298. var taskIds []int
  1299. // taskId 对 稿费形式的 map
  1300. taskIdToFeeFormMap := make(map[int]int)
  1301. for _, projectId := range projectIds {
  1302. db2 := GetReadDB(context.Background())
  1303. autoIds := gorm_model.ProjectInfo{}
  1304. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1305. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1306. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1307. var taskInfos []*gorm_model.YoungeeTaskInfo
  1308. db1 := GetReadDB(context.Background())
  1309. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  1310. for _, taskInfo := range taskInfos {
  1311. taskIdToProjectIdMap[taskInfo.TaskId] = projectId
  1312. taskIds = append(taskIds, taskInfo.TaskId)
  1313. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  1314. }
  1315. }
  1316. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1317. //fmt.Println("taskIds:", taskIds)
  1318. // 首次提交链接的任务记录id
  1319. var taskNeedModIds []int
  1320. // 已提交链接的任务记录id
  1321. var submitTaskNeedModIds []int
  1322. for _, taskId := range taskIds {
  1323. var taskInfo gorm_model.YoungeeTaskInfo
  1324. db3 := GetReadDB(context.Background())
  1325. // 保存所有链接上传状态为 1 且任务状态为12 的任务记录的id
  1326. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1327. if taskInfo.LinkStatus == 1 && taskInfo.TaskStage == 12 {
  1328. taskNeedModIds = append(taskNeedModIds, taskId)
  1329. }
  1330. // 保存所有链接上传状态为 3 且任务状态为12 的任务记录的id
  1331. if taskInfo.LinkStatus == 3 && taskInfo.TaskStage == 12 {
  1332. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1333. }
  1334. }
  1335. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  1336. //fmt.Println("submitTaskNeedModIds:", submitTaskNeedModIds)
  1337. for _, taskNeedModId := range taskNeedModIds {
  1338. db2 := GetReadDB(context.Background())
  1339. var linkBreach int32
  1340. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&linkBreach)
  1341. dd, _ := time.ParseDuration(conv.MustString(linkBreach, "") + "h")
  1342. var linkDefaultRate int
  1343. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1344. db6 := GetReadDB(context.Background())
  1345. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1346. } else {
  1347. db6 := GetReadDB(context.Background())
  1348. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1349. }
  1350. db1 := GetReadDB(context.Background())
  1351. var taskSketchInfo gorm_model.YounggeeSketchInfo
  1352. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskSketchInfo)
  1353. if taskSketchInfo.AutoLinkBreakAt.IsZero() {
  1354. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoLinkBreakAt: taskSketchInfo.AutoAgreeAt.Add(dd)})
  1355. fmt.Println("已添加链接违约自动处理时间")
  1356. if taskSketchInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1357. db4 := GetReadDB(context.Background())
  1358. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1359. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1360. if err1 != nil {
  1361. return err1
  1362. }
  1363. db3 := GetReadDB(context.Background())
  1364. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1365. if err2 != nil {
  1366. return err2
  1367. }
  1368. fmt.Println("创建时已违约则创建链接违约记录")
  1369. }
  1370. } else {
  1371. if taskSketchInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1372. db4 := GetReadDB(context.Background())
  1373. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1374. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1375. if err1 != nil {
  1376. return err1
  1377. }
  1378. db3 := GetReadDB(context.Background())
  1379. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1380. if err2 != nil {
  1381. return err2
  1382. }
  1383. fmt.Println("已创建链接违约记录")
  1384. }
  1385. }
  1386. }
  1387. // 根据最近一次拒绝时间判断当前是否违约
  1388. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1389. db2 := GetReadDB(context.Background())
  1390. var LinkBreach int32
  1391. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1392. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1393. var linkDefaultRate int
  1394. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1395. db6 := GetReadDB(context.Background())
  1396. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1397. } else {
  1398. db6 := GetReadDB(context.Background())
  1399. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1400. }
  1401. db1 := GetReadDB(context.Background())
  1402. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1403. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskLinkInfo)
  1404. if taskLinkInfo.AutoLinkBreakAt.IsZero() {
  1405. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoLinkBreakAt: taskLinkInfo.RejectAt.Add(dd)}).Error
  1406. if err4 != nil {
  1407. return err4
  1408. }
  1409. fmt.Println("已添加链接违约自动处理时间")
  1410. if taskLinkInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1411. db4 := GetReadDB(context.Background())
  1412. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1413. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1414. if err1 != nil {
  1415. return err1
  1416. }
  1417. db3 := GetReadDB(context.Background())
  1418. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1419. if err2 != nil {
  1420. return err2
  1421. }
  1422. fmt.Println("创建时已违约则创建已提交链接的链接违约记录")
  1423. }
  1424. } else {
  1425. if taskLinkInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1426. db4 := GetReadDB(context.Background())
  1427. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1428. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1429. if err1 != nil {
  1430. return err1
  1431. }
  1432. db3 := GetReadDB(context.Background())
  1433. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate}).Error
  1434. if err2 != nil {
  1435. return err2
  1436. }
  1437. fmt.Println("创建已提交链接的链接违约记录")
  1438. }
  1439. }
  1440. }
  1441. return nil
  1442. }
  1443. func GetAutoCaseCloseDefaultTask() error {
  1444. db := GetReadDB(context.Background())
  1445. var projectIds []int64
  1446. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1447. if err != nil {
  1448. return err
  1449. }
  1450. //fmt.Println("projectIds:", projectIds)
  1451. // 任务id 对 项目id 的map
  1452. taskIdToProjectIdMap := map[int]int64{}
  1453. // 项目id 对 定时任务id 的map
  1454. projectIdToAutoTaskIdMap := map[int64]int{}
  1455. // 项目id 对 违约定时任务id 的map
  1456. projectIdToAutoDefaultIdMap := map[int64]int{}
  1457. var taskIds []int
  1458. for _, projectId := range projectIds {
  1459. db2 := GetReadDB(context.Background())
  1460. autoIds := gorm_model.ProjectInfo{}
  1461. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1462. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1463. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1464. var taskInfos []int32
  1465. db1 := GetReadDB(context.Background())
  1466. db1.Select("task_id").Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND cur_default_type = ? ", projectId, 0).Find(&taskInfos)
  1467. for _, taskInfo := range taskInfos {
  1468. taskIdToProjectIdMap[int(taskInfo)] = projectId
  1469. taskIds = append(taskIds, int(taskInfo))
  1470. }
  1471. }
  1472. //fmt.Println("projectIdToAutoTaskIdMap:", projectIdToAutoTaskIdMap)
  1473. //fmt.Println("projectIdToAutoDefaultIdMap:", projectIdToAutoDefaultIdMap)
  1474. //fmt.Println("taskIdToProjectIdMap:", taskIdToProjectIdMap)
  1475. //fmt.Println("taskIds:", taskIds)
  1476. // 首次提交链接的任务记录id
  1477. var taskNeedModIds []int
  1478. // 已提交链接的任务记录id
  1479. var submitTaskNeedModIds []int
  1480. // taskId 对 稿费形式的 map
  1481. taskIdToFeeFormMap := make(map[int]int)
  1482. for _, taskId := range taskIds {
  1483. var taskInfo gorm_model.YoungeeTaskInfo
  1484. db3 := GetReadDB(context.Background())
  1485. // 保存所有数据上传状态为 1 且任务状态为14 的任务记录的id
  1486. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1487. if taskInfo.DataStatus == 1 && taskInfo.TaskStage == 14 {
  1488. taskNeedModIds = append(taskNeedModIds, taskId)
  1489. }
  1490. // 保存所有数据上传状态为 3 且任务状态为14 的任务记录的id
  1491. if taskInfo.DataStatus == 3 && taskInfo.TaskStage == 14 {
  1492. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1493. }
  1494. taskIdToFeeFormMap[taskId] = taskInfo.FeeForm
  1495. }
  1496. //fmt.Println("taskIdToFeeFormMap:", taskIdToFeeFormMap)
  1497. //fmt.Println("taskNeedModIds:", taskNeedModIds)
  1498. //fmt.Println("submitTaskNeedModIds:", submitTaskNeedModIds)
  1499. for _, taskNeedModId := range taskNeedModIds {
  1500. db2 := GetReadDB(context.Background())
  1501. var CaseCloseDefault int32
  1502. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&CaseCloseDefault)
  1503. dd, _ := time.ParseDuration(conv.MustString(CaseCloseDefault, "") + "h")
  1504. var dataDefaultRate int
  1505. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1506. db6 := GetReadDB(context.Background())
  1507. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1508. } else {
  1509. db6 := GetReadDB(context.Background())
  1510. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1511. }
  1512. //fmt.Println("dataDefaultRate:", dataDefaultRate)
  1513. db1 := GetReadDB(context.Background())
  1514. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1515. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskLinkInfo)
  1516. if taskLinkInfo.AutoDataBreakAt.IsZero() {
  1517. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoDataBreakAt: taskLinkInfo.AutoAgreeAt.Add(dd)})
  1518. fmt.Println("已添加数据违约自动处理时间")
  1519. if taskLinkInfo.AutoAgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1520. db4 := GetReadDB(context.Background())
  1521. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1522. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1523. if err1 != nil {
  1524. return err1
  1525. }
  1526. db3 := GetReadDB(context.Background())
  1527. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1528. if err2 != nil {
  1529. return err2
  1530. }
  1531. fmt.Println("创建时已违约则创建数据违约记录")
  1532. }
  1533. } else {
  1534. if taskLinkInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1535. db4 := GetReadDB(context.Background())
  1536. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1537. TaskID: taskNeedModId, ProjectID: int(taskIdToProjectIdMap[taskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1538. if err1 != nil {
  1539. return err1
  1540. }
  1541. db3 := GetReadDB(context.Background())
  1542. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1543. if err2 != nil {
  1544. return err2
  1545. }
  1546. fmt.Println("已创建数据违约记录")
  1547. }
  1548. }
  1549. }
  1550. // 根据最近一次拒绝时间判断当前是否违约
  1551. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1552. db2 := GetReadDB(context.Background())
  1553. var LinkBreach int32
  1554. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1555. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1556. var dataDefaultRate int
  1557. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1558. db6 := GetReadDB(context.Background())
  1559. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1560. } else {
  1561. db6 := GetReadDB(context.Background())
  1562. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1563. }
  1564. db1 := GetReadDB(context.Background())
  1565. var taskDataInfo gorm_model.YounggeeDataInfo
  1566. db1.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskDataInfo)
  1567. if taskDataInfo.AutoDataBreakAt.IsZero() {
  1568. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeDataInfo{AutoDataBreakAt: taskDataInfo.RejectAt.Add(dd)}).Error
  1569. if err4 != nil {
  1570. return err4
  1571. }
  1572. fmt.Println("已添加数据违约自动处理时间")
  1573. if taskDataInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1574. db4 := GetReadDB(context.Background())
  1575. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1576. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1577. if err1 != nil {
  1578. return err1
  1579. }
  1580. db3 := GetReadDB(context.Background())
  1581. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1582. if err2 != nil {
  1583. return err2
  1584. }
  1585. fmt.Println("创建时已违约则创建已提交数据的数据违约记录")
  1586. }
  1587. } else {
  1588. if taskDataInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1589. db4 := GetReadDB(context.Background())
  1590. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1591. TaskID: submitTaskNeedModId, ProjectID: int(taskIdToProjectIdMap[submitTaskNeedModId]), BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1592. if err1 != nil {
  1593. return err1
  1594. }
  1595. db3 := GetReadDB(context.Background())
  1596. err2 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1597. CurDefaultType: 7, DataBreakRate: dataDefaultRate}).Error
  1598. if err2 != nil {
  1599. return err2
  1600. }
  1601. fmt.Println("创建已提交数据的数据违约记录")
  1602. }
  1603. }
  1604. }
  1605. return nil
  1606. }
  1607. func GetHours(ctx context.Context) (*http_model.Hours, error) {
  1608. db := GetReadDB(ctx)
  1609. infoAuto := gorm_model.InfoAutoTask{}
  1610. err := db.Model(gorm_model.InfoAutoTask{}).Last(&infoAuto).Error
  1611. if err != nil {
  1612. return nil, err
  1613. }
  1614. Hours := http_model.Hours{}
  1615. Hours.DraftDefaultInPic = infoAuto.DraftDefaultInPic
  1616. Hours.DraftDefaultInMv = infoAuto.DraftDefaultInMv
  1617. Hours.ScriptDefault = infoAuto.ScriptDefault
  1618. Hours.Invalid = infoAuto.Invalid
  1619. Hours.LinkBreach = infoAuto.LinkBreach
  1620. Hours.CaseClose = infoAuto.CaseClose
  1621. Hours.CaseCloseDefault = infoAuto.CaseCloseDefault
  1622. Hours.ReviewInMv = infoAuto.ReviewInMv
  1623. Hours.ReviewUnlimited = infoAuto.ReviewUnlimited
  1624. Hours.PostReview = infoAuto.Postreview
  1625. Hours.SignInVirtual = infoAuto.SignInVirtual
  1626. Hours.SignInOffline = infoAuto.SignInOffline
  1627. return &Hours, err
  1628. }
  1629. func GetPercents(ctx context.Context) (*http_model.Percents, error) {
  1630. db := GetReadDB(ctx)
  1631. percentsInfo := gorm_model.InfoAutoDefaultHandle{}
  1632. err := db.Model(gorm_model.InfoAutoDefaultHandle{}).Last(&percentsInfo).Error
  1633. if err != nil {
  1634. return nil, err
  1635. }
  1636. Percents := http_model.Percents{}
  1637. Percents.DataReplaceNotUpload = percentsInfo.DataReplaceNotUpload
  1638. Percents.DataReplaceTimeOut = percentsInfo.DataReplaceTimeOut
  1639. Percents.DataOtherTimeOut = percentsInfo.DataOtherTimeOut
  1640. Percents.DataOtherNotUpload = percentsInfo.DataOtherNotUpload
  1641. Percents.LinkReplaceTimeOut = percentsInfo.LinkReplaceTimeOut
  1642. Percents.LinkReplaceNotUpload = percentsInfo.LinkReplaceNotUpload
  1643. Percents.LinkOtherTimeOut = percentsInfo.LinkOtherTimeOut
  1644. Percents.LinkOtherNotUpload = percentsInfo.LinkOtherNotUpload
  1645. Percents.ScriptReplaceTimeOut = percentsInfo.ScriptReplaceTimeOut
  1646. Percents.ScriptReplaceNotUpload = percentsInfo.ScriptReplaceNotUpload
  1647. Percents.ScriptOtherNotUpload = percentsInfo.ScriptOtherNotUpload
  1648. Percents.ScriptOtherTimeOut = percentsInfo.ScriptOtherTimeOut
  1649. Percents.SketchReplaceTimeOut = percentsInfo.SketchReplaceTimeOut
  1650. Percents.SketchOtherNotUpload = percentsInfo.SketchOtherNotUpload
  1651. Percents.SketchOtherTimeOut = percentsInfo.SketchOtherTimeOut
  1652. Percents.SketchReplaceNotUpload = percentsInfo.SketchReplaceNotUpload
  1653. return &Percents, err
  1654. }