operate.go 84 KB

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