operate.go 85 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817
  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. err := db9.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectInfo.ProjectID).
  764. Updates(map[string]interface{}{"project_status": 10, "payment_amount": realPayment}).Error
  765. if err != nil {
  766. log.Println("DB GetAutoCaseCloseTask error in data:", err)
  767. return err
  768. }
  769. }
  770. }
  771. return nil
  772. }
  773. func GetAutoInvalidTask() error {
  774. db := GetReadDB(context.Background())
  775. var projectInfos []*gorm_model.ProjectInfo
  776. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND project_status = ? ", 1, 6).Find(&projectInfos).Error
  777. if err != nil {
  778. return err
  779. }
  780. var projectIds []string
  781. projectIdToAutoTaskIdMap := map[string]int{}
  782. for _, projectInfo := range projectInfos {
  783. projectIds = append(projectIds, projectInfo.ProjectID)
  784. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  785. }
  786. for _, projectId := range projectIds {
  787. // 获取 autoTaskId 及其对应的限制时间
  788. autoTaskId := projectIdToAutoTaskIdMap[projectId]
  789. dbStart := GetReadDB(context.Background())
  790. var Invalid int32
  791. dbStart.Model(gorm_model.InfoAutoTask{}).Select("invalid").Where("auto_task_id = ?", autoTaskId).First(&Invalid)
  792. //fmt.Println("t:", autoTaskInfo.Invalid)
  793. db1 := GetReadDB(context.Background())
  794. project := gorm_model.ProjectInfo{}
  795. db1.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&project)
  796. //if project.AutoFailAt.IsZero() {
  797. // dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  798. // t := project.RecruitDdl.Add(dd)
  799. // db2 := GetReadDB(context.Background())
  800. // db2.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  801. // fmt.Println("已添加失效自动处理时间")
  802. //}
  803. dd, _ := time.ParseDuration(conv.MustString(Invalid, "") + "h")
  804. project.RecruitDdl.Add(dd)
  805. t := project.RecruitDdl.Add(dd)
  806. //fmt.Println("已存在的失效时间是否跟要添加的相等", project.AutoFailAt.Equal(t))
  807. if project.AutoFailAt == nil || project.AutoFailAt.IsZero() {
  808. db3 := GetReadDB(context.Background())
  809. db3.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{AutoFailAt: &t})
  810. fmt.Println("已添加失效自动处理时间")
  811. }
  812. projectNeedMod := gorm_model.ProjectInfo{}
  813. db2 := GetReadDB(context.Background())
  814. db2 = db2.Where("project_id = ?", projectId).First(&projectNeedMod)
  815. fmt.Println("失效自动处理时间为:", projectNeedMod.AutoFailAt)
  816. if !projectNeedMod.AutoFailAt.IsZero() || projectNeedMod.AutoFailAt != nil {
  817. fmt.Printf("项目 %v 失效自动处理时间不为空\n", projectNeedMod.ProjectID)
  818. fmt.Println("距离失效自动处理时间还有:,项目当前状态为", projectNeedMod.AutoFailAt.Sub(time.Now()), projectNeedMod.ProjectStatus)
  819. if projectNeedMod.AutoFailAt.Sub(time.Now()) <= 0 && projectNeedMod.ProjectStatus < 8 {
  820. db4 := GetReadDB(context.Background())
  821. err1 := db4.Model(gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).Updates(&gorm_model.ProjectInfo{ProjectStatus: 8, FailReason: 1}).Error
  822. if err1 != nil {
  823. return err1
  824. }
  825. fmt.Println("已更新项目状态为超时未支付的失效状态")
  826. }
  827. }
  828. }
  829. return nil
  830. }
  831. // 图文-初稿超时违约判断
  832. func GetAutoDraftDefaultInPicTask() error {
  833. db := GetReadDB(context.Background())
  834. var projectInfos []*gorm_model.ProjectInfo
  835. err := db.Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 1).Find(&projectInfos).Error
  836. if err != nil {
  837. return err
  838. }
  839. var projectIds []string
  840. projectIdToAutoTaskIdMap := map[string]int{} // 项目id 对 定时任务id 的map
  841. projectIdToAutoDefaultIdMap := map[string]int{} // 项目id 对 违约扣款设置id 的map
  842. taskIdToFeeFormMap := make(map[string]int) // taskId 对 稿费形式的 map
  843. TaskIdToProjectId := make(map[string]string) // taskId 对 项目id的 map
  844. var taskNeedModIds []string // 首次提交初稿的任务记录id
  845. var submitTaskNeedModIds []string // 修改后提交初稿的任务记录id
  846. // 构造map及list
  847. for _, projectInfo := range projectInfos {
  848. projectIds = append(projectIds, projectInfo.ProjectID)
  849. projectIdToAutoTaskIdMap[projectInfo.ProjectID] = int(projectInfo.AutoTaskID)
  850. projectIdToAutoDefaultIdMap[projectInfo.ProjectID] = int(projectInfo.AutoDefaultID)
  851. }
  852. for _, projectId := range projectIds {
  853. db1 := GetReadDB(context.Background())
  854. var taskInfos []gorm_model.YoungeeTaskInfo
  855. db1.Model(gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ? AND task_stage = ? AND cur_default_type = ?", projectId, 2, 9, 0).Find(&taskInfos)
  856. for _, taskInfo := range taskInfos {
  857. TaskIdToProjectId[taskInfo.TaskId] = projectId
  858. taskNeedMod := gorm_model.YoungeeTaskInfo{}
  859. db2 := GetReadDB(context.Background())
  860. // 保存所有满足物流状态为3 且 初稿上传状态为 1或 3 的任务记录的id
  861. db2.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskInfo.TaskId).First(&taskNeedMod)
  862. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  863. if taskNeedMod.TaskStage == 9 && taskNeedMod.ScriptStatus == 1 {
  864. taskNeedModIds = append(taskNeedModIds, taskInfo.TaskId)
  865. }
  866. if taskNeedMod.TaskStage == 9 && taskNeedMod.ScriptStatus == 3 {
  867. submitTaskNeedModIds = append(submitTaskNeedModIds, taskInfo.TaskId)
  868. }
  869. }
  870. }
  871. // 判断应首次上传初稿的任务
  872. for _, taskNeedModId := range taskNeedModIds {
  873. // 获取 autoTaskId 及其对应的限制时间
  874. autoTaskId := projectIdToAutoTaskIdMap[TaskIdToProjectId[taskNeedModId]]
  875. dbStart := GetReadDB(context.Background())
  876. var DraftDefaultInPic int32
  877. dbStart.Model(gorm_model.InfoAutoTask{}).Select("draft_default_in_pic").Where("auto_task_id = ?", autoTaskId).First(&DraftDefaultInPic)
  878. db3 := GetReadDB(context.Background())
  879. taskLogisticNeedMod := gorm_model.YoungeeTaskLogistics{}
  880. db3.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", taskNeedModId).First(&taskLogisticNeedMod)
  881. // 查询违约扣款比例
  882. var sketchDefaultRate int
  883. if taskIdToFeeFormMap[taskNeedModId] == 1 { // 稿费形式为产品置换
  884. db6 := GetReadDB(context.Background())
  885. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[TaskIdToProjectId[taskNeedModId]]).Find(&sketchDefaultRate)
  886. } else { // 稿费形式为其他
  887. db6 := GetReadDB(context.Background())
  888. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[TaskIdToProjectId[taskNeedModId]]).Find(&sketchDefaultRate)
  889. }
  890. // 添加初稿违约自动处理时间
  891. if taskLogisticNeedMod.AutoSketchBreakAt == nil || taskLogisticNeedMod.AutoSketchBreakAt.IsZero() {
  892. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInPic, "") + "h")
  893. db4 := GetReadDB(context.Background())
  894. t := taskLogisticNeedMod.SignedTime.Add(dd)
  895. db4.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{
  896. AutoSketchBreakAt: &t,
  897. })
  898. taskLogisticNeedMod.AutoSketchBreakAt = &t
  899. fmt.Println("已添加图片初稿违约自动处理时间")
  900. }
  901. // 判断是否超时违约
  902. if taskLogisticNeedMod.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  903. // 超时违约处理
  904. db5 := GetReadDB(context.Background())
  905. err1 := db5.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  906. TaskID: taskNeedModId, ProjectID: TaskIdToProjectId[taskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  907. if err1 != nil {
  908. return err1
  909. }
  910. taskInfo := gorm_model.YoungeeTaskInfo{}
  911. dbt := GetReadDB(context.Background())
  912. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskLogisticNeedMod.TaskID).Find(&taskInfo).Error
  913. if err2 != nil {
  914. return err2
  915. }
  916. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.LinkBreakRate+taskInfo.DataBreakRate+sketchDefaultRate+taskInfo.ScriptBreakRate)/100)
  917. if settleAmount < 0 {
  918. settleAmount = float64(0.01)
  919. }
  920. db8 := GetReadDB(context.Background())
  921. db8.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskLogisticNeedMod.TaskID).Updates(
  922. &gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate, SettleAmount: settleAmount})
  923. fmt.Println("已创建图文类型的初稿违约记录")
  924. }
  925. }
  926. // 判断应修改后上传初稿的任务
  927. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  928. // 获取 autoTaskId 及其对应的限制时间
  929. db2 := GetReadDB(context.Background())
  930. var DraftDefaultInMv int32
  931. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[TaskIdToProjectId[submitTaskNeedModId]]).First(&DraftDefaultInMv)
  932. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  933. // 查询违约扣款比例
  934. var sketchDefaultRate int
  935. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  936. db6 := GetReadDB(context.Background())
  937. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[TaskIdToProjectId[submitTaskNeedModId]]).Find(&sketchDefaultRate)
  938. } else {
  939. db6 := GetReadDB(context.Background())
  940. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[TaskIdToProjectId[submitTaskNeedModId]]).Find(&sketchDefaultRate)
  941. }
  942. // 添加初稿违约自动处理时间
  943. db1 := GetReadDB(context.Background())
  944. var taskSketchInfo gorm_model.YounggeeSketchInfo
  945. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 1", submitTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  946. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  947. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  948. if err4 != nil {
  949. return err4
  950. }
  951. taskSketchInfo.AutoSketchBreakAt = taskSketchInfo.RejectAt.Add(dd)
  952. fmt.Println("已添加图文形式的初稿违约自动处理时间")
  953. }
  954. // 判断是否违约
  955. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  956. db4 := GetReadDB(context.Background())
  957. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  958. TaskID: submitTaskNeedModId, ProjectID: TaskIdToProjectId[submitTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  959. if err1 != nil {
  960. return err1
  961. }
  962. taskInfo := gorm_model.YoungeeTaskInfo{}
  963. dbt := GetReadDB(context.Background())
  964. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Find(&taskInfo).Error
  965. if err2 != nil {
  966. return err2
  967. }
  968. settleAmount := taskInfo.TaskReward * (1.0 - float64(sketchDefaultRate+taskInfo.DataBreakRate+taskInfo.LinkBreakRate+taskInfo.ScriptBreakRate)/100)
  969. if settleAmount < 0 {
  970. settleAmount = float64(0.01)
  971. }
  972. db3 := GetReadDB(context.Background())
  973. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(
  974. &gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate, SettleAmount: settleAmount}).Error
  975. if err2 != nil {
  976. return err2
  977. }
  978. fmt.Println("创建已提交初稿的图文类型的初稿违约记录")
  979. }
  980. }
  981. return nil
  982. }
  983. // 视频-初稿超时违约判断
  984. func GetAutoDraftDefaultInMvTask() error {
  985. db := GetReadDB(context.Background())
  986. var VideoProjectIds []string
  987. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ? ", 1, 2).Find(&VideoProjectIds).Error
  988. if err != nil {
  989. return err
  990. }
  991. // 任务id 对 项目id 的map
  992. taskIdToProjectIdMap := map[string]string{}
  993. // 项目id 对 定时任务id 的map
  994. projectIdToAutoTaskIdMap := map[string]int{}
  995. // 项目 id 对 违约定时任务 id 的 map
  996. projectIdToAutoDefaultIdMap := map[string]int{}
  997. // taskId 对 稿费形式的 map
  998. taskIdToFeeFormMap := make(map[string]int)
  999. var videoTaskIds []string
  1000. // 构造map及list
  1001. for _, VideoProjectId := range VideoProjectIds {
  1002. db2 := GetReadDB(context.Background())
  1003. autoIds := gorm_model.ProjectInfo{}
  1004. db2.Model(&gorm_model.ProjectInfo{}).Select([]string{"auto_task_id", "auto_default_id"}).Where("project_id = ?", VideoProjectId).First(&autoIds)
  1005. projectIdToAutoTaskIdMap[VideoProjectId] = int(autoIds.AutoTaskID)
  1006. projectIdToAutoDefaultIdMap[VideoProjectId] = int(autoIds.AutoDefaultID)
  1007. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1008. db1 := GetReadDB(context.Background())
  1009. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ? AND task_stage = ? AND cur_default_type = ? ", VideoProjectId, 2, 9, 0).Find(&videoTaskInfos)
  1010. for _, videoTaskInfo := range videoTaskInfos {
  1011. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  1012. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  1013. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  1014. }
  1015. }
  1016. // 首次提交初稿的任务记录id
  1017. var videoTaskNeedModIds []string
  1018. // 已提交初稿的任务记录id
  1019. var submitVideoTaskNeedModIds []string
  1020. for _, videoTaskId := range videoTaskIds {
  1021. var taskInfo gorm_model.YoungeeTaskInfo
  1022. db3 := GetReadDB(context.Background())
  1023. // 保存所有初稿上传状态为 1 且任务阶段为9 的任务记录的id
  1024. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  1025. if taskInfo.SketchStatus == 1 && taskInfo.TaskStage == 9 {
  1026. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  1027. }
  1028. // 保存所有初稿上传状态为 3 且任务阶段为9 的任务记录的id
  1029. if taskInfo.SketchStatus == 3 && taskInfo.TaskStage == 9 {
  1030. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  1031. }
  1032. }
  1033. // 判断应首次提交初稿的任务
  1034. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  1035. db2 := GetReadDB(context.Background())
  1036. var DraftDefaultInMv int32
  1037. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&DraftDefaultInMv)
  1038. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  1039. var sketchDefaultRate int
  1040. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  1041. db6 := GetReadDB(context.Background())
  1042. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  1043. } else {
  1044. db6 := GetReadDB(context.Background())
  1045. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&sketchDefaultRate)
  1046. }
  1047. db1 := GetReadDB(context.Background())
  1048. var taskScriptInfo gorm_model.YounggeeScriptInfo
  1049. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ? AND is_ok = ?", videoTaskNeedModId, 1).Find(&taskScriptInfo)
  1050. if taskScriptInfo.AutoSketchBreakAt.IsZero() {
  1051. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoSketchBreakAt: taskScriptInfo.AgreeAt.Add(dd)})
  1052. taskScriptInfo.AutoSketchBreakAt = taskScriptInfo.AgreeAt.Add(dd)
  1053. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  1054. }
  1055. if taskScriptInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  1056. db4 := GetReadDB(context.Background())
  1057. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1058. TaskID: videoTaskNeedModId, ProjectID: taskIdToProjectIdMap[videoTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1059. if err1 != nil {
  1060. return err1
  1061. }
  1062. taskInfo := gorm_model.YoungeeTaskInfo{}
  1063. dbt := GetReadDB(context.Background())
  1064. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskInfo).Error
  1065. if err2 != nil {
  1066. return err2
  1067. }
  1068. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.LinkBreakRate+taskInfo.DataBreakRate+sketchDefaultRate+taskInfo.ScriptBreakRate)/100)
  1069. if settleAmount < 0 {
  1070. settleAmount = 0.01
  1071. }
  1072. db3 := GetReadDB(context.Background())
  1073. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(
  1074. &gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate, SettleAmount: settleAmount}).Error
  1075. if err2 != nil {
  1076. return err2
  1077. }
  1078. fmt.Println("已创建视频类型的初稿违约记录")
  1079. }
  1080. }
  1081. // 判断应修改后上传初稿的任务
  1082. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  1083. db2 := GetReadDB(context.Background())
  1084. var DraftDefaultInMv int32
  1085. db2.Model(&gorm_model.InfoAutoTask{}).Select("draft_default_in_mv").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&DraftDefaultInMv)
  1086. dd, _ := time.ParseDuration(conv.MustString(DraftDefaultInMv, "") + "h")
  1087. var sketchDefaultRate int
  1088. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  1089. db6 := GetReadDB(context.Background())
  1090. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  1091. } else {
  1092. db6 := GetReadDB(context.Background())
  1093. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&sketchDefaultRate)
  1094. }
  1095. db1 := GetReadDB(context.Background())
  1096. var taskSketchInfo gorm_model.YounggeeSketchInfo
  1097. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? and is_review = 1", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskSketchInfo)
  1098. if taskSketchInfo.AutoSketchBreakAt.IsZero() {
  1099. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoSketchBreakAt: taskSketchInfo.RejectAt.Add(dd)}).Error
  1100. if err4 != nil {
  1101. return err4
  1102. }
  1103. taskSketchInfo.AutoSketchBreakAt = taskSketchInfo.RejectAt.Add(dd)
  1104. fmt.Println("已添加视频形式的初稿违约自动处理时间")
  1105. }
  1106. if taskSketchInfo.AutoSketchBreakAt.Sub(time.Now()) <= 0 {
  1107. db4 := GetReadDB(context.Background())
  1108. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1109. TaskID: submitVideoTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitVideoTaskNeedModId], BreakType: 2, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1110. if err1 != nil {
  1111. return err1
  1112. }
  1113. taskInfo := gorm_model.YoungeeTaskInfo{}
  1114. dbt := GetReadDB(context.Background())
  1115. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Find(&taskInfo).Error
  1116. if err2 != nil {
  1117. return err2
  1118. }
  1119. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.LinkBreakRate+taskInfo.DataBreakRate+sketchDefaultRate+taskInfo.ScriptBreakRate)/100)
  1120. if settleAmount < 0 {
  1121. settleAmount = 0.01
  1122. }
  1123. db3 := GetReadDB(context.Background())
  1124. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(
  1125. &gorm_model.YoungeeTaskInfo{CurDefaultType: 3, SketchBreakRate: sketchDefaultRate, SettleAmount: settleAmount}).Error
  1126. if err2 != nil {
  1127. return err2
  1128. }
  1129. fmt.Println("创建已提交初稿的视频类型的初稿违约记录")
  1130. }
  1131. }
  1132. return nil
  1133. }
  1134. func GetAutoScriptDefaultTask() error {
  1135. db := GetReadDB(context.Background())
  1136. var VideoProjectIds []string
  1137. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ? AND content_type = ?", 1, 2).Find(&VideoProjectIds).Error
  1138. if err != nil {
  1139. return err
  1140. }
  1141. // 任务id 对 项目id 的map
  1142. taskIdToProjectIdMap := map[string]string{}
  1143. // 项目id 对 定时任务id 的map
  1144. projectIdToAutoTaskIdMap := map[string]int{}
  1145. // 项目id 对 违约定时任务id 的map
  1146. projectIdToAutoDefaultIdMap := map[string]int{}
  1147. // taskId 对 稿费形式的 map
  1148. taskIdToFeeFormMap := make(map[string]int)
  1149. // 首次提交脚本的任务记录id
  1150. var videoTaskNeedModIds []string
  1151. // 已提交脚本的任务记录id
  1152. var submitVideoTaskNeedModIds []string
  1153. var videoTaskIds []string
  1154. // 构造map及list
  1155. for _, VideoProjectId := range VideoProjectIds {
  1156. db2 := GetReadDB(context.Background())
  1157. var autoTaskId int
  1158. db2.Model(&gorm_model.ProjectInfo{}).Select("auto_task_id").Where("project_id = ?", VideoProjectId).First(&autoTaskId)
  1159. projectIdToAutoTaskIdMap[VideoProjectId] = autoTaskId
  1160. var videoTaskInfos []*gorm_model.YoungeeTaskInfo
  1161. db1 := GetReadDB(context.Background())
  1162. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ? AND task_stage = ? AND cur_default_type = ? ", VideoProjectId, 2, 7, 0).Find(&videoTaskInfos)
  1163. for _, videoTaskInfo := range videoTaskInfos {
  1164. taskIdToProjectIdMap[videoTaskInfo.TaskId] = VideoProjectId
  1165. taskIdToFeeFormMap[videoTaskInfo.TaskId] = videoTaskInfo.FeeForm
  1166. videoTaskIds = append(videoTaskIds, videoTaskInfo.TaskId)
  1167. }
  1168. }
  1169. for _, videoTaskId := range videoTaskIds {
  1170. var taskInfo gorm_model.YoungeeTaskInfo
  1171. db3 := GetReadDB(context.Background())
  1172. // 保存所有脚本上传状态为 1 且任务状态为8 的任务记录的id
  1173. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskId).First(&taskInfo)
  1174. if taskInfo.ScriptStatus == 1 && taskInfo.TaskStage == 7 {
  1175. videoTaskNeedModIds = append(videoTaskNeedModIds, videoTaskId)
  1176. }
  1177. // 保存所有脚本上传状态为 3 且任务状态为8 的任务记录的id
  1178. if taskInfo.ScriptStatus == 3 && taskInfo.TaskStage == 7 {
  1179. submitVideoTaskNeedModIds = append(submitVideoTaskNeedModIds, videoTaskId)
  1180. }
  1181. }
  1182. // 判断应首次上传脚本的任务
  1183. for _, videoTaskNeedModId := range videoTaskNeedModIds {
  1184. db2 := GetReadDB(context.Background())
  1185. var ScriptDefault int32
  1186. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).First(&ScriptDefault)
  1187. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1188. db1 := GetReadDB(context.Background())
  1189. var taskLogisticInfo gorm_model.YoungeeTaskLogistics
  1190. db1.Model(gorm_model.YoungeeTaskLogistics{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskLogisticInfo)
  1191. var scriptDefaultRate int
  1192. if taskIdToFeeFormMap[videoTaskNeedModId] == 1 {
  1193. db6 := GetReadDB(context.Background())
  1194. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1195. } else {
  1196. db6 := GetReadDB(context.Background())
  1197. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[videoTaskNeedModId]]).Find(&scriptDefaultRate)
  1198. }
  1199. if taskLogisticInfo.AutoScriptBreakAt == nil || taskLogisticInfo.AutoScriptBreakAt.IsZero() {
  1200. t := taskLogisticInfo.SignedTime.Add(dd)
  1201. db1.Where("task_id = ?", videoTaskNeedModId).Updates(&gorm_model.YoungeeTaskLogistics{AutoScriptBreakAt: &t})
  1202. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1203. if taskLogisticInfo.SignedTime.Add(dd).Sub(time.Now()) <= 0 {
  1204. db4 := GetReadDB(context.Background())
  1205. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1206. TaskID: videoTaskNeedModId, ProjectID: taskIdToProjectIdMap[videoTaskNeedModId], BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1207. if err1 != nil {
  1208. return err1
  1209. }
  1210. taskInfo := gorm_model.YoungeeTaskInfo{}
  1211. dbt := GetReadDB(context.Background())
  1212. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskInfo).Error
  1213. if err2 != nil {
  1214. return err2
  1215. }
  1216. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+taskInfo.DataBreakRate+taskInfo.LinkBreakRate+scriptDefaultRate)/100)
  1217. if settleAmount < 0 {
  1218. settleAmount = float64(0.01)
  1219. }
  1220. db3 := GetReadDB(context.Background())
  1221. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(
  1222. &gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate, SettleAmount: settleAmount}).Error
  1223. if err2 != nil {
  1224. return err2
  1225. }
  1226. fmt.Println("创建时已违约则创建视频类型的脚本违约记录")
  1227. }
  1228. } else {
  1229. if taskLogisticInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1230. db4 := GetReadDB(context.Background())
  1231. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1232. TaskID: videoTaskNeedModId, ProjectID: taskIdToProjectIdMap[videoTaskNeedModId], BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1233. if err1 != nil {
  1234. return err1
  1235. }
  1236. taskInfo := gorm_model.YoungeeTaskInfo{}
  1237. dbt := GetReadDB(context.Background())
  1238. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Find(&taskInfo).Error
  1239. if err2 != nil {
  1240. return err2
  1241. }
  1242. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+taskInfo.DataBreakRate+taskInfo.LinkBreakRate+scriptDefaultRate)/100)
  1243. if settleAmount < 0 {
  1244. settleAmount = 0.01
  1245. }
  1246. db3 := GetReadDB(context.Background())
  1247. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", videoTaskNeedModId).Updates(
  1248. &gorm_model.YoungeeTaskInfo{CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate, SettleAmount: settleAmount}).Error
  1249. if err2 != nil {
  1250. return err2
  1251. }
  1252. fmt.Println("已创建视频类型的脚本违约记录")
  1253. }
  1254. }
  1255. }
  1256. // 根据最近一次拒绝时间判断当前是否违约
  1257. for _, submitVideoTaskNeedModId := range submitVideoTaskNeedModIds {
  1258. db2 := GetReadDB(context.Background())
  1259. var ScriptDefault int32
  1260. db2.Model(&gorm_model.InfoAutoTask{}).Select("script_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).First(&ScriptDefault)
  1261. dd, _ := time.ParseDuration(conv.MustString(ScriptDefault, "") + "h")
  1262. db1 := GetReadDB(context.Background())
  1263. var taskScriptInfo gorm_model.YounggeeScriptInfo
  1264. db1.Model(gorm_model.YounggeeScriptInfo{}).Where("task_id = ? and is_review = 1", submitVideoTaskNeedModId).Order("reject_at desc").First(&taskScriptInfo)
  1265. var scriptDefaultRate int
  1266. if taskIdToFeeFormMap[submitVideoTaskNeedModId] == 1 {
  1267. db6 := GetReadDB(context.Background())
  1268. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1269. } else {
  1270. db6 := GetReadDB(context.Background())
  1271. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("script_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitVideoTaskNeedModId]]).Find(&scriptDefaultRate)
  1272. }
  1273. if taskScriptInfo.AutoScriptBreakAt.IsZero() {
  1274. err4 := db1.Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YounggeeScriptInfo{AutoScriptBreakAt: taskScriptInfo.RejectAt.Add(dd)}).Error
  1275. if err4 != nil {
  1276. return err4
  1277. }
  1278. fmt.Println("已添加视频形式的脚本违约自动处理时间")
  1279. if taskScriptInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1280. db4 := GetReadDB(context.Background())
  1281. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1282. TaskID: submitVideoTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitVideoTaskNeedModId], BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1283. if err1 != nil {
  1284. return err1
  1285. }
  1286. taskInfo := gorm_model.YoungeeTaskInfo{}
  1287. dbt := GetReadDB(context.Background())
  1288. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Find(&taskInfo).Error
  1289. if err2 != nil {
  1290. return err2
  1291. }
  1292. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+taskInfo.DataBreakRate+taskInfo.LinkBreakRate+scriptDefaultRate)/100)
  1293. if settleAmount < 0 {
  1294. settleAmount = 0.01
  1295. }
  1296. db3 := GetReadDB(context.Background())
  1297. err3 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1298. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate, SettleAmount: settleAmount}).Error
  1299. if err3 != nil {
  1300. return err3
  1301. }
  1302. fmt.Println("创建时已违约则创建已提交脚本的视频类型的脚本违约记录")
  1303. }
  1304. } else {
  1305. if taskScriptInfo.AutoScriptBreakAt.Sub(time.Now()) <= 0 {
  1306. db4 := GetReadDB(context.Background())
  1307. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1308. TaskID: submitVideoTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitVideoTaskNeedModId], BreakType: 1, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1309. if err1 != nil {
  1310. return err1
  1311. }
  1312. taskInfo := gorm_model.YoungeeTaskInfo{}
  1313. dbt := GetReadDB(context.Background())
  1314. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Find(&taskInfo).Error
  1315. if err2 != nil {
  1316. return err2
  1317. }
  1318. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+taskInfo.DataBreakRate+taskInfo.LinkBreakRate+scriptDefaultRate)/100)
  1319. if settleAmount < 0 {
  1320. settleAmount = float64(0.01)
  1321. }
  1322. db3 := GetReadDB(context.Background())
  1323. err3 := db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitVideoTaskNeedModId).Updates(&gorm_model.YoungeeTaskInfo{
  1324. CurDefaultType: 1, ScriptBreakRate: scriptDefaultRate, SettleAmount: settleAmount}).Error
  1325. if err3 != nil {
  1326. return err3
  1327. }
  1328. fmt.Println("创建已提交脚本的视频类型的脚本违约记录")
  1329. }
  1330. }
  1331. }
  1332. return nil
  1333. }
  1334. func GetAutoLinkBreachTask() error {
  1335. db := GetReadDB(context.Background())
  1336. var projectIds []string
  1337. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1338. if err != nil {
  1339. return err
  1340. }
  1341. // 任务id 对 项目id 的map
  1342. taskIdToProjectIdMap := map[string]string{}
  1343. // 项目id 对 定时任务id 的map
  1344. projectIdToAutoTaskIdMap := map[string]int{}
  1345. // 项目id 对 违约定时任务id 的map
  1346. projectIdToAutoDefaultIdMap := map[string]int{}
  1347. var taskIds []string
  1348. // taskId 对 稿费形式的 map
  1349. taskIdToFeeFormMap := make(map[string]int)
  1350. for _, projectId := range projectIds {
  1351. db2 := GetReadDB(context.Background())
  1352. autoIds := gorm_model.ProjectInfo{}
  1353. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1354. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1355. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1356. var taskInfos []*gorm_model.YoungeeTaskInfo
  1357. db1 := GetReadDB(context.Background())
  1358. db1.Model(&gorm_model.YoungeeTaskInfo{}).Where("project_id = ? AND task_status = ? AND task_stage = ? AND cur_default_type = ? ", projectId, 2, 11, 0).Find(&taskInfos)
  1359. for _, taskInfo := range taskInfos {
  1360. taskIdToProjectIdMap[taskInfo.TaskId] = projectId
  1361. taskIds = append(taskIds, taskInfo.TaskId)
  1362. taskIdToFeeFormMap[taskInfo.TaskId] = taskInfo.FeeForm
  1363. }
  1364. }
  1365. // 首次提交链接的任务记录id
  1366. var taskNeedModIds []string
  1367. // 已提交链接的任务记录id
  1368. var submitTaskNeedModIds []string
  1369. for _, taskId := range taskIds {
  1370. var taskInfo gorm_model.YoungeeTaskInfo
  1371. db3 := GetReadDB(context.Background())
  1372. // 保存所有链接上传状态为 1 且任务状态为12 的任务记录的id
  1373. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1374. if taskInfo.LinkStatus == 1 && taskInfo.TaskStage == 11 {
  1375. taskNeedModIds = append(taskNeedModIds, taskId)
  1376. }
  1377. // 保存所有链接上传状态为 3 且任务状态为12 的任务记录的id
  1378. if taskInfo.LinkStatus == 3 && taskInfo.TaskStage == 11 {
  1379. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1380. }
  1381. }
  1382. for _, taskNeedModId := range taskNeedModIds {
  1383. db2 := GetReadDB(context.Background())
  1384. var linkBreach int32
  1385. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&linkBreach)
  1386. dd, _ := time.ParseDuration(conv.MustString(linkBreach, "") + "h")
  1387. var linkDefaultRate int
  1388. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1389. db6 := GetReadDB(context.Background())
  1390. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1391. } else {
  1392. db6 := GetReadDB(context.Background())
  1393. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&linkDefaultRate)
  1394. }
  1395. db1 := GetReadDB(context.Background())
  1396. var taskSketchInfo gorm_model.YounggeeSketchInfo
  1397. db1.Model(gorm_model.YounggeeSketchInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskSketchInfo)
  1398. if taskSketchInfo.AutoLinkBreakAt.IsZero() {
  1399. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeSketchInfo{AutoLinkBreakAt: taskSketchInfo.AgreeAt.Add(dd)})
  1400. fmt.Println("已添加链接违约自动处理时间")
  1401. if taskSketchInfo.AgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1402. db4 := GetReadDB(context.Background())
  1403. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1404. TaskID: taskNeedModId, ProjectID: taskIdToProjectIdMap[taskNeedModId], BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1405. if err1 != nil {
  1406. return err1
  1407. }
  1408. taskInfo := gorm_model.YoungeeTaskInfo{}
  1409. dbt := GetReadDB(context.Background())
  1410. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Find(&taskInfo).Error
  1411. if err2 != nil {
  1412. return err2
  1413. }
  1414. settleAmount := taskInfo.TaskReward * (1.0 - float64(linkDefaultRate+taskInfo.DataBreakRate+taskInfo.SketchBreakRate+taskInfo.ScriptBreakRate)/100)
  1415. if settleAmount < 0 {
  1416. settleAmount = 0.01
  1417. }
  1418. db3 := GetReadDB(context.Background())
  1419. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(
  1420. &gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate, SettleAmount: settleAmount}).Error
  1421. if err2 != nil {
  1422. return err2
  1423. }
  1424. fmt.Println("创建时已违约则创建链接违约记录")
  1425. }
  1426. } else {
  1427. if taskSketchInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1428. db4 := GetReadDB(context.Background())
  1429. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1430. TaskID: taskNeedModId, ProjectID: taskIdToProjectIdMap[taskNeedModId], BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1431. if err1 != nil {
  1432. return err1
  1433. }
  1434. taskInfo := gorm_model.YoungeeTaskInfo{}
  1435. dbt := GetReadDB(context.Background())
  1436. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Find(&taskInfo).Error
  1437. if err2 != nil {
  1438. return err2
  1439. }
  1440. settleAmount := taskInfo.TaskReward * (1.0 - float64(linkDefaultRate+taskInfo.DataBreakRate+taskInfo.SketchBreakRate+taskInfo.ScriptBreakRate)/100)
  1441. if settleAmount < 0 {
  1442. settleAmount = 0.01
  1443. }
  1444. db3 := GetReadDB(context.Background())
  1445. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(
  1446. &gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate, SettleAmount: settleAmount}).Error
  1447. if err2 != nil {
  1448. return err2
  1449. }
  1450. fmt.Println("已创建链接违约记录")
  1451. }
  1452. }
  1453. }
  1454. // 根据最近一次拒绝时间判断当前是否违约
  1455. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1456. db2 := GetReadDB(context.Background())
  1457. var LinkBreach int32
  1458. db2.Model(&gorm_model.InfoAutoTask{}).Select("link_breach").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1459. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1460. var linkDefaultRate int
  1461. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1462. db6 := GetReadDB(context.Background())
  1463. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1464. } else {
  1465. db6 := GetReadDB(context.Background())
  1466. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("link_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&linkDefaultRate)
  1467. }
  1468. db1 := GetReadDB(context.Background())
  1469. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1470. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ?", submitTaskNeedModId).Order("reject_at desc").First(&taskLinkInfo)
  1471. if taskLinkInfo.AutoLinkBreakAt.IsZero() {
  1472. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoLinkBreakAt: taskLinkInfo.RejectAt.Add(dd)}).Error
  1473. if err4 != nil {
  1474. return err4
  1475. }
  1476. fmt.Println("已添加链接违约自动处理时间")
  1477. if taskLinkInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1478. db4 := GetReadDB(context.Background())
  1479. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1480. TaskID: submitTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitTaskNeedModId], BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1481. if err1 != nil {
  1482. return err1
  1483. }
  1484. taskInfo := gorm_model.YoungeeTaskInfo{}
  1485. dbt := GetReadDB(context.Background())
  1486. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Find(&taskInfo).Error
  1487. if err2 != nil {
  1488. return err2
  1489. }
  1490. settleAmount := taskInfo.TaskReward * (1.0 - float64(linkDefaultRate+taskInfo.DataBreakRate+taskInfo.SketchBreakRate+taskInfo.ScriptBreakRate)/100)
  1491. if settleAmount < 0 {
  1492. settleAmount = 0.01
  1493. }
  1494. db3 := GetReadDB(context.Background())
  1495. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(
  1496. &gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate, SettleAmount: settleAmount}).Error
  1497. if err2 != nil {
  1498. return err2
  1499. }
  1500. fmt.Println("创建时已违约则创建已提交链接的链接违约记录")
  1501. }
  1502. } else {
  1503. if taskLinkInfo.AutoLinkBreakAt.Sub(time.Now()) <= 0 {
  1504. db4 := GetReadDB(context.Background())
  1505. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1506. TaskID: submitTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitTaskNeedModId], BreakType: 3, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1507. if err1 != nil {
  1508. return err1
  1509. }
  1510. taskInfo := gorm_model.YoungeeTaskInfo{}
  1511. dbt := GetReadDB(context.Background())
  1512. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Find(&taskInfo).Error
  1513. if err2 != nil {
  1514. return err2
  1515. }
  1516. settleAmount := taskInfo.TaskReward * (1.0 - float64(linkDefaultRate+taskInfo.DataBreakRate+taskInfo.SketchBreakRate+taskInfo.ScriptBreakRate)/100)
  1517. if settleAmount < 0 {
  1518. settleAmount = float64(0.01)
  1519. }
  1520. db3 := GetReadDB(context.Background())
  1521. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(
  1522. &gorm_model.YoungeeTaskInfo{CurDefaultType: 5, LinkBreakRate: linkDefaultRate, SettleAmount: settleAmount}).Error
  1523. if err2 != nil {
  1524. return err2
  1525. }
  1526. fmt.Println("创建已提交链接的链接违约记录")
  1527. }
  1528. }
  1529. }
  1530. return nil
  1531. }
  1532. func GetAutoCaseCloseDefaultTask() error {
  1533. db := GetReadDB(context.Background())
  1534. var projectIds []string
  1535. err := db.Select("project_id").Model(gorm_model.ProjectInfo{}).Where("project_type = ?", 1).Find(&projectIds).Error
  1536. if err != nil {
  1537. return err
  1538. }
  1539. // 任务id 对 项目id 的map
  1540. taskIdToProjectIdMap := map[string]string{}
  1541. // 项目id 对 定时任务id 的map
  1542. projectIdToAutoTaskIdMap := map[string]int{}
  1543. // 项目id 对 违约定时任务id 的map
  1544. projectIdToAutoDefaultIdMap := map[string]int{}
  1545. var taskIds []string
  1546. for _, projectId := range projectIds {
  1547. db2 := GetReadDB(context.Background())
  1548. autoIds := gorm_model.ProjectInfo{}
  1549. db2.Select([]string{"auto_task_id", "auto_default_id"}).Model(&gorm_model.ProjectInfo{}).Where("project_id = ?", projectId).First(&autoIds)
  1550. projectIdToAutoTaskIdMap[projectId] = int(autoIds.AutoTaskID)
  1551. projectIdToAutoDefaultIdMap[projectId] = int(autoIds.AutoDefaultID)
  1552. var taskInfos []string
  1553. db1 := GetReadDB(context.Background())
  1554. 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)
  1555. for _, taskInfo := range taskInfos {
  1556. taskIdToProjectIdMap[taskInfo] = projectId
  1557. taskIds = append(taskIds, taskInfo)
  1558. }
  1559. }
  1560. // 首次提交链接的任务记录id
  1561. var taskNeedModIds []string
  1562. // 已提交链接的任务记录id
  1563. var submitTaskNeedModIds []string
  1564. // taskId 对 稿费形式的 map
  1565. taskIdToFeeFormMap := make(map[string]int)
  1566. for _, taskId := range taskIds {
  1567. var taskInfo gorm_model.YoungeeTaskInfo
  1568. db3 := GetReadDB(context.Background())
  1569. // 保存所有数据上传状态为 1 且任务状态为14 的任务记录的id
  1570. db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).First(&taskInfo)
  1571. if taskInfo.DataStatus == 1 && taskInfo.TaskStage == 13 {
  1572. taskNeedModIds = append(taskNeedModIds, taskId)
  1573. }
  1574. // 保存所有数据上传状态为 3 且任务状态为14 的任务记录的id
  1575. if taskInfo.DataStatus == 3 && taskInfo.TaskStage == 13 {
  1576. submitTaskNeedModIds = append(submitTaskNeedModIds, taskId)
  1577. }
  1578. taskIdToFeeFormMap[taskId] = taskInfo.FeeForm
  1579. }
  1580. for _, taskNeedModId := range taskNeedModIds {
  1581. db2 := GetReadDB(context.Background())
  1582. var CaseCloseDefault int32
  1583. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[taskNeedModId]]).First(&CaseCloseDefault)
  1584. dd, _ := time.ParseDuration(conv.MustString(CaseCloseDefault, "") + "h")
  1585. var dataDefaultRate int
  1586. if taskIdToFeeFormMap[taskNeedModId] == 1 {
  1587. db6 := GetReadDB(context.Background())
  1588. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1589. } else {
  1590. db6 := GetReadDB(context.Background())
  1591. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[taskNeedModId]]).Find(&dataDefaultRate)
  1592. }
  1593. //fmt.Println("dataDefaultRate:", dataDefaultRate)
  1594. db1 := GetReadDB(context.Background())
  1595. var taskLinkInfo gorm_model.YounggeeLinkInfo
  1596. db1.Model(gorm_model.YounggeeLinkInfo{}).Where("task_id = ? AND is_ok = ?", taskNeedModId, 1).Find(&taskLinkInfo)
  1597. if taskLinkInfo.AutoDataBreakAt.IsZero() {
  1598. db1.Where("task_id = ?", taskNeedModId).Updates(&gorm_model.YounggeeLinkInfo{AutoDataBreakAt: taskLinkInfo.AgreeAt.Add(dd)})
  1599. fmt.Println("已添加数据违约自动处理时间")
  1600. if taskLinkInfo.AgreeAt.Add(dd).Sub(time.Now()) <= 0 {
  1601. db4 := GetReadDB(context.Background())
  1602. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1603. TaskID: taskNeedModId, ProjectID: taskIdToProjectIdMap[taskNeedModId], BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1604. if err1 != nil {
  1605. return err1
  1606. }
  1607. taskInfo := gorm_model.YoungeeTaskInfo{}
  1608. dbt := GetReadDB(context.Background())
  1609. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Find(&taskInfo).Error
  1610. if err2 != nil {
  1611. return err2
  1612. }
  1613. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+dataDefaultRate+taskInfo.LinkBreakRate+taskInfo.ScriptBreakRate)/100)
  1614. if settleAmount < 0 {
  1615. settleAmount = 0.01
  1616. }
  1617. db3 := GetReadDB(context.Background())
  1618. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(
  1619. &gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate, SettleAmount: settleAmount}).Error
  1620. if err2 != nil {
  1621. return err2
  1622. }
  1623. fmt.Println("创建时已违约则创建数据违约记录")
  1624. }
  1625. } else {
  1626. if taskLinkInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1627. db4 := GetReadDB(context.Background())
  1628. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1629. TaskID: taskNeedModId, ProjectID: taskIdToProjectIdMap[taskNeedModId], BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1630. if err1 != nil {
  1631. return err1
  1632. }
  1633. taskInfo := gorm_model.YoungeeTaskInfo{}
  1634. dbt := GetReadDB(context.Background())
  1635. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Find(&taskInfo).Error
  1636. if err2 != nil {
  1637. return err2
  1638. }
  1639. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+dataDefaultRate+taskInfo.LinkBreakRate+taskInfo.ScriptBreakRate)/100)
  1640. if settleAmount < 0 {
  1641. settleAmount = float64(0.01)
  1642. }
  1643. fmt.Println("settleAmount: ", settleAmount)
  1644. db3 := GetReadDB(context.Background())
  1645. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", taskNeedModId).Updates(
  1646. &gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate, SettleAmount: settleAmount}).Error
  1647. if err2 != nil {
  1648. return err2
  1649. }
  1650. fmt.Println("已创建数据违约记录")
  1651. }
  1652. }
  1653. }
  1654. // 根据最近一次拒绝时间判断当前是否违约
  1655. for _, submitTaskNeedModId := range submitTaskNeedModIds {
  1656. db2 := GetReadDB(context.Background())
  1657. var LinkBreach int32
  1658. db2.Model(&gorm_model.InfoAutoTask{}).Select("case_close_default").Where("auto_task_id = ?", projectIdToAutoTaskIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).First(&LinkBreach)
  1659. dd, _ := time.ParseDuration(conv.MustString(LinkBreach, "") + "h")
  1660. var dataDefaultRate int
  1661. if taskIdToFeeFormMap[submitTaskNeedModId] == 1 {
  1662. db6 := GetReadDB(context.Background())
  1663. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_replace_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1664. } else {
  1665. db6 := GetReadDB(context.Background())
  1666. db6.Model(gorm_model.InfoAutoDefaultHandle{}).Select("data_other_time_out").Where("auto_default_id = ?", projectIdToAutoDefaultIdMap[taskIdToProjectIdMap[submitTaskNeedModId]]).Find(&dataDefaultRate)
  1667. }
  1668. db1 := GetReadDB(context.Background())
  1669. var taskDataInfo gorm_model.YounggeeDataInfo
  1670. db1.Model(gorm_model.YounggeeDataInfo{}).Where("task_id = ? and is_review = 1", submitTaskNeedModId).Order("reject_at desc").First(&taskDataInfo)
  1671. if taskDataInfo.AutoDataBreakAt.IsZero() {
  1672. err4 := db1.Where("task_id = ?", submitTaskNeedModId).Updates(&gorm_model.YounggeeDataInfo{AutoDataBreakAt: taskDataInfo.RejectAt.Add(dd)}).Error
  1673. if err4 != nil {
  1674. return err4
  1675. }
  1676. fmt.Println("已添加数据违约自动处理时间")
  1677. if taskDataInfo.RejectAt.Add(dd).Sub(time.Now()) <= 0 {
  1678. db4 := GetReadDB(context.Background())
  1679. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1680. TaskID: submitTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitTaskNeedModId], BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1681. if err1 != nil {
  1682. return err1
  1683. }
  1684. taskInfo := gorm_model.YoungeeTaskInfo{}
  1685. dbt := GetReadDB(context.Background())
  1686. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Find(&taskInfo).Error
  1687. if err2 != nil {
  1688. return err2
  1689. }
  1690. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+dataDefaultRate+taskInfo.LinkBreakRate+taskInfo.ScriptBreakRate)/100)
  1691. if settleAmount < 0 {
  1692. settleAmount = 0.01
  1693. }
  1694. db3 := GetReadDB(context.Background())
  1695. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(
  1696. &gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate, SettleAmount: settleAmount}).Error
  1697. if err2 != nil {
  1698. return err2
  1699. }
  1700. fmt.Println("创建时已违约则创建已提交数据的数据违约记录")
  1701. }
  1702. } else {
  1703. if taskDataInfo.AutoDataBreakAt.Sub(time.Now()) <= 0 {
  1704. db4 := GetReadDB(context.Background())
  1705. err1 := db4.Model(gorm_model.YoungeeContractInfo{}).Create(&gorm_model.YoungeeContractInfo{
  1706. TaskID: submitTaskNeedModId, ProjectID: taskIdToProjectIdMap[submitTaskNeedModId], BreakType: 4, BreakAt: time.Now(), DefaultStatus: 1}).Error
  1707. if err1 != nil {
  1708. return err1
  1709. }
  1710. taskInfo := gorm_model.YoungeeTaskInfo{}
  1711. dbt := GetReadDB(context.Background())
  1712. err2 := dbt.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Find(&taskInfo).Error
  1713. if err2 != nil {
  1714. return err2
  1715. }
  1716. settleAmount := taskInfo.TaskReward * (1.0 - float64(taskInfo.SketchBreakRate+dataDefaultRate+taskInfo.LinkBreakRate+taskInfo.ScriptBreakRate)/100)
  1717. if settleAmount < 0 {
  1718. settleAmount = 0.01
  1719. }
  1720. db3 := GetReadDB(context.Background())
  1721. err2 = db3.Model(gorm_model.YoungeeTaskInfo{}).Where("task_id = ?", submitTaskNeedModId).Updates(
  1722. &gorm_model.YoungeeTaskInfo{CurDefaultType: 7, DataBreakRate: dataDefaultRate, SettleAmount: settleAmount}).Error
  1723. if err2 != nil {
  1724. return err2
  1725. }
  1726. fmt.Println("创建已提交数据的数据违约记录")
  1727. }
  1728. }
  1729. }
  1730. return nil
  1731. }
  1732. func GetHours(ctx context.Context) (*http_model.Hours, error) {
  1733. db := GetReadDB(ctx)
  1734. infoAuto := gorm_model.InfoAutoTask{}
  1735. err := db.Model(gorm_model.InfoAutoTask{}).Last(&infoAuto).Error
  1736. if err != nil {
  1737. return nil, err
  1738. }
  1739. Hours := http_model.Hours{}
  1740. Hours.DraftDefaultInPic = infoAuto.DraftDefaultInPic
  1741. Hours.DraftDefaultInMv = infoAuto.DraftDefaultInMv
  1742. Hours.ScriptDefault = infoAuto.ScriptDefault
  1743. Hours.Invalid = infoAuto.Invalid
  1744. Hours.LinkBreach = infoAuto.LinkBreach
  1745. Hours.CaseClose = infoAuto.CaseClose
  1746. Hours.CaseCloseDefault = infoAuto.CaseCloseDefault
  1747. Hours.ReviewInMv = infoAuto.ReviewInMv
  1748. Hours.ReviewUnlimited = infoAuto.ReviewUnlimited
  1749. Hours.PostReview = infoAuto.Postreview
  1750. Hours.SignInVirtual = infoAuto.SignInVirtual
  1751. Hours.SignInOffline = infoAuto.SignInOffline
  1752. return &Hours, err
  1753. }
  1754. func GetPercents(ctx context.Context) (*http_model.Percents, error) {
  1755. db := GetReadDB(ctx)
  1756. percentsInfo := gorm_model.InfoAutoDefaultHandle{}
  1757. err := db.Model(gorm_model.InfoAutoDefaultHandle{}).Last(&percentsInfo).Error
  1758. if err != nil {
  1759. return nil, err
  1760. }
  1761. Percents := http_model.Percents{}
  1762. Percents.DataReplaceNotUpload = percentsInfo.DataReplaceNotUpload
  1763. Percents.DataReplaceTimeOut = percentsInfo.DataReplaceTimeOut
  1764. Percents.DataOtherTimeOut = percentsInfo.DataOtherTimeOut
  1765. Percents.DataOtherNotUpload = percentsInfo.DataOtherNotUpload
  1766. Percents.LinkReplaceTimeOut = percentsInfo.LinkReplaceTimeOut
  1767. Percents.LinkReplaceNotUpload = percentsInfo.LinkReplaceNotUpload
  1768. Percents.LinkOtherTimeOut = percentsInfo.LinkOtherTimeOut
  1769. Percents.LinkOtherNotUpload = percentsInfo.LinkOtherNotUpload
  1770. Percents.ScriptReplaceTimeOut = percentsInfo.ScriptReplaceTimeOut
  1771. Percents.ScriptReplaceNotUpload = percentsInfo.ScriptReplaceNotUpload
  1772. Percents.ScriptOtherNotUpload = percentsInfo.ScriptOtherNotUpload
  1773. Percents.ScriptOtherTimeOut = percentsInfo.ScriptOtherTimeOut
  1774. Percents.SketchReplaceTimeOut = percentsInfo.SketchReplaceTimeOut
  1775. Percents.SketchOtherNotUpload = percentsInfo.SketchOtherNotUpload
  1776. Percents.SketchOtherTimeOut = percentsInfo.SketchOtherTimeOut
  1777. Percents.SketchReplaceNotUpload = percentsInfo.SketchReplaceNotUpload
  1778. return &Percents, err
  1779. }