operate.go 85 KB

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