cow_cron.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617
  1. package crontab
  2. import (
  3. "context"
  4. "fmt"
  5. "kpt-pasture/model"
  6. "kpt-pasture/module/backend"
  7. "kpt-pasture/util"
  8. "sort"
  9. "strings"
  10. "time"
  11. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  12. "gitee.com/xuyiping_admin/pkg/logger/zaplog"
  13. "gitee.com/xuyiping_admin/pkg/xerr"
  14. "go.uber.org/zap"
  15. )
  16. const (
  17. UpdateCowInfo = "UpdateCowInfo"
  18. ImmunizationPlan = "ImmunizationPlan"
  19. SameTimePlan = "SameTimePlan"
  20. WorkOrderMaster = "WorkOrderMaster"
  21. SystemBasicCrontab = "SystemBasicCrontab"
  22. NeckRingOriginal = "NeckRingOriginal"
  23. )
  24. // GenerateAsynqWorkOrder 异步生成工作单
  25. func (e *Entry) GenerateAsynqWorkOrder() error {
  26. if ok := e.IsExistCrontabLog(WorkOrderMaster); ok {
  27. return nil
  28. }
  29. defer func() {
  30. e.CreateCrontabLog(WorkOrderMaster)
  31. }()
  32. workOrderList := make([]*model.WorkOrderMaster, 0)
  33. if err := e.DB.Where("is_show = ?", pasturePb.IsShow_Ok).Find(&workOrderList).Error; err != nil {
  34. return err
  35. }
  36. for _, workOrder := range workOrderList {
  37. timeUnix, err := util.ConvertParseLocalUnix(workOrder.ExecTime)
  38. if timeUnix <= 0 || err != nil {
  39. zaplog.Error("crontab", zap.Any("GenerateWorkOrder", err), zap.Any("execTime", workOrder.ExecTime))
  40. continue
  41. }
  42. nowTime := time.Now().Local().Unix()
  43. if timeUnix < nowTime {
  44. continue
  45. }
  46. execTime := timeUnix - nowTime
  47. task := model.NewTaskWorkOrderPayload(workOrder.Id, time.Duration(execTime)*time.Second)
  48. if _, err = e.AsynqClient.CtxEnqueue(context.Background(), task); err != nil {
  49. zaplog.Error("PushMessage CtxEnqueue", zap.Any("Err", err))
  50. }
  51. }
  52. return nil
  53. }
  54. // Indicators 月度指标维护
  55. func (e *Entry) Indicators() error {
  56. indicatorsDetailsList := make([]*model.IndicatorsDetails, 0)
  57. if err := e.DB.Model(new(model.IndicatorsDetails)).
  58. Find(&indicatorsDetailsList).Error; err != nil {
  59. return err
  60. }
  61. pastureList := e.FindPastureList()
  62. startTime, endTime := util.GetMonthStartAndEndTimestamp()
  63. for _, indicatorsDetail := range indicatorsDetailsList {
  64. pastureIndicatorList := map[int64]string{}
  65. switch indicatorsDetail.Kind {
  66. case model.AllCow:
  67. pastureIndicatorList = e.FindPastureAllCow(pastureList)
  68. case model.AdultCow:
  69. pastureIndicatorList = e.FindPastureAdultCow(pastureList)
  70. case model.AvgCalvingInterval:
  71. pastureIndicatorList = e.FindAvgCalvingInterval(pastureList, startTime, endTime)
  72. case model.OutputNumber:
  73. pastureIndicatorList = e.FindOutputNumber(pastureList, startTime, endTime)
  74. case model.InputNumber:
  75. pastureIndicatorList = e.FindInputNumber(pastureList, startTime, endTime)
  76. case model.SalesVolume:
  77. pastureIndicatorList = e.FindSalesVolume(pastureList, startTime, endTime)
  78. case model.CalvingNumber:
  79. pastureIndicatorList = e.FindCalvingNumber(pastureList, startTime, endTime)
  80. case model.AdultAbortionRate:
  81. pastureIndicatorList = e.FindAdultAbortionRate(pastureList, "adult", startTime, endTime)
  82. case model.YouthAbortionRate:
  83. pastureIndicatorList = e.FindAdultAbortionRate(pastureList, "youth", startTime, endTime)
  84. case model.AllDieNumber:
  85. pastureIndicatorList = e.FindDeathNumber(pastureList, startTime, endTime, false)
  86. case model.DiseaseNumber:
  87. pastureIndicatorList = e.FindDiseaseNumber(pastureList, startTime, endTime)
  88. case model.CureNumber:
  89. pastureIndicatorList = e.FindCureNumber(pastureList, startTime, endTime)
  90. case model.OutNumber:
  91. pastureIndicatorList = e.FindDepartureNumber(pastureList, pasturePb.SalesType_Out, startTime, endTime)
  92. case model.CalfDieNumber:
  93. pastureIndicatorList = e.FindDeathNumber(pastureList, startTime, endTime, true)
  94. case model.LactationCow:
  95. pastureIndicatorList = e.LactationCow(pastureList, pasturePb.CowMilk_Lactation)
  96. case model.DryMilkCow:
  97. pastureIndicatorList = e.LactationCow(pastureList, pasturePb.CowMilk_Dry_Milk)
  98. case model.ReserveCow:
  99. pastureIndicatorList = e.LactationCow(pastureList, pasturePb.CowMilk_Reserve)
  100. case model.FirstBornSurvivalRate:
  101. pastureIndicatorList = e.FirstBornSurvivalRate(pastureList, startTime, endTime)
  102. case model.FirstBornDeathRate:
  103. pastureIndicatorList = e.FirstBornDeathRate(pastureList, startTime, endTime)
  104. case model.MultiparitySurvivalRate:
  105. pastureIndicatorList = e.MultiparitySurvivalRate(pastureList, startTime, endTime)
  106. case model.MultiparityDeathRate:
  107. pastureIndicatorList = e.MultiparityDeathRate(pastureList, startTime, endTime)
  108. case model.AvgAgeFirstMate:
  109. pastureIndicatorList = e.AvgAgeFirstMate(pastureList, startTime, endTime)
  110. case model.YouthPregnantRate:
  111. pastureIndicatorList = e.CowPregnantRate(pastureList, indicatorsDetail.Kind)
  112. case model.MultiparityPregnantRate:
  113. pastureIndicatorList = e.CowPregnantRate(pastureList, indicatorsDetail.Kind)
  114. case model.MultipartyFirstCheckRate:
  115. pastureIndicatorList = e.PregnantCheckRate(pastureList, startTime, endTime, true, model.PregnantCheckForFirst)
  116. case model.MultipartySecondCheckRate:
  117. pastureIndicatorList = e.PregnantCheckRate(pastureList, startTime, endTime, true, model.PregnantCheckForSecond)
  118. case model.YouthFirstCheckRate:
  119. pastureIndicatorList = e.PregnantCheckRate(pastureList, startTime, endTime, false, model.PregnantCheckForFirst)
  120. case model.YouthSecondCheckRate:
  121. pastureIndicatorList = e.PregnantCheckRate(pastureList, startTime, endTime, false, model.PregnantCheckForSecond)
  122. case model.ForbiddenCowNumber:
  123. pastureIndicatorList = e.ForbiddenCowNumber(pastureList, startTime, endTime)
  124. }
  125. for pastureId, value := range pastureIndicatorList {
  126. e.UpdatePastureIndicators(pastureId, indicatorsDetail, startTime, value)
  127. }
  128. }
  129. return nil
  130. }
  131. // UpdateCowInfo 牛只基本信息维护
  132. func (e *Entry) UpdateCowInfo() error {
  133. pastureList := e.FindPastureList()
  134. for _, pasture := range pastureList {
  135. e.UpdateCowInfoByPasture(pasture.Id)
  136. }
  137. e.CreateCrontabLog(UpdateCowInfo)
  138. return nil
  139. }
  140. func (e *Entry) UpdateCowInfoByPasture(pastureId int64) {
  141. if ok := e.IsExistCrontabLog(UpdateCowInfo); ok {
  142. return
  143. }
  144. cowList := make([]*model.Cow, 0)
  145. if err := e.DB.Model(new(model.Cow)).
  146. Where("pasture_id = ?", pastureId).
  147. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  148. Find(&cowList).Error; err != nil {
  149. zaplog.Error("crontab", zap.Any("UpdateCowInfoByPasture", err))
  150. return
  151. }
  152. dateTime := time.Now().Local().Format(model.LayoutMonth)
  153. for _, cow := range cowList {
  154. // 周活动量
  155. weeklyActive := e.UpdateCowWeeklyHigh(pastureId, cow)
  156. zaplog.Info("crontab", zap.Any("weeklyActive", weeklyActive))
  157. cow.EventUpdate(weeklyActive)
  158. if err := e.DB.Model(new(model.Cow)).
  159. Select("day_age", "calving_age", "pregnancy_age", "admission_age", "abortion_age", "cow_type", "weekly_active", "lactation_age", "dry_milk_age", "mating_age").
  160. Where("id = ?", cow.Id).
  161. Updates(cow).Error; err != nil {
  162. zaplog.Error("Crontab", zap.Any("UpdateCowDayAge", err))
  163. }
  164. // 每月怀孕数据
  165. cowPregnantList := make([]*model.CowPregnant, 0)
  166. if cow.IsPregnant == pasturePb.IsShow_Ok || cow.BreedStatus == pasturePb.BreedStatus_Pregnant {
  167. cowPregnantList = append(cowPregnantList, model.NewCowPregnant(pastureId, cow, dateTime))
  168. }
  169. if len(cowPregnantList) > 0 {
  170. if err := e.DB.Model(new(model.CowPregnant)).
  171. Where("pasture_id = ?", pastureId).
  172. Where("date_time = ?", dateTime).
  173. Delete(&model.CowPregnant{}).Error; err != nil {
  174. zaplog.Error("Crontab", zap.Any("UpdateCowPregnant", err))
  175. }
  176. if err := e.DB.Model(new(model.CowPregnant)).
  177. Create(&cowPregnantList).Error; err != nil {
  178. zaplog.Error("Crontab", zap.Any("UpdateCowPregnant", err))
  179. }
  180. }
  181. }
  182. }
  183. // ImmunizationPlan 免疫计划,生成工作单
  184. func (e *Entry) ImmunizationPlan() error {
  185. if ok := e.IsExistCrontabLog(ImmunizationPlan); ok {
  186. return nil
  187. }
  188. planList := make([]*model.ImmunizationPlan, 0)
  189. if err := e.DB.Model(new(model.ImmunizationPlan)).
  190. Where("is_show = ?", pasturePb.IsShow_Ok).
  191. Order("pasture_id").
  192. Find(&planList).Error; err != nil {
  193. return xerr.WithStack(err)
  194. }
  195. var todayCount int32 = 0
  196. nowTime := time.Now().Local()
  197. for _, plan := range planList {
  198. cowList := make([]*model.Cow, 0)
  199. pref := e.DB.Table(fmt.Sprintf("%s as a", new(model.Cow).TableName())).
  200. Select("a.*").
  201. Where("a.pasture_id = ?", plan.PastureId).
  202. Where("a.admission_status = ?", pasturePb.AdmissionStatus_Admission).
  203. Where("NOT EXISTS ( select 1 from event_immunization_plan b where b.pen_id = a.id and b.status = ? and b.plan_day > ?)", pasturePb.IsShow_No, nowTime.Unix())
  204. if plan.CowType > 0 {
  205. pref.Where("a.cow_type = ?", plan.CowType)
  206. }
  207. switch plan.Conditions {
  208. case pasturePb.ImmunizationConditions_Days_Age:
  209. pref.Where("a.day_age = ?", plan.Value)
  210. case pasturePb.ImmunizationConditions_Days_After_Delivery:
  211. pref.Where("a.calving_age = ?", plan.Value)
  212. case pasturePb.ImmunizationConditions_Days_Of_Pregnancy:
  213. pref.Where("a.pregnancy_age = ?", plan.Value).
  214. Where("a.is_pregnant = ?", pasturePb.IsShow_Ok)
  215. case pasturePb.ImmunizationConditions_Month:
  216. continue
  217. case pasturePb.ImmunizationConditions_Admission_Days:
  218. pref.Where("a.admission_age = ?", plan.Value)
  219. case pasturePb.ImmunizationConditions_Other_Vaccine_After:
  220. if plan.ImmunizationPlanId > 0 {
  221. pref.Joins("INNER JOIN event_immunization_plan as b ON b.plan_id = ? ", plan.ImmunizationPlanId).
  222. Where("b.cow_id = a.id").
  223. Where("DATE_ADD(b.reality_day, INTERVAL ? DAY) = ?", plan.Value, time.Now().Local().Format(model.LayoutDate2)).
  224. Where("b.status = ?", pasturePb.IsShow_Ok)
  225. }
  226. }
  227. if err := pref.Find(&cowList).Debug().Error; err != nil {
  228. return xerr.WithStack(err)
  229. }
  230. if len(cowList) <= 0 {
  231. continue
  232. }
  233. newImmunizationPlanCowList := model.NewCowImmunizationPlanList(cowList, plan)
  234. if err := e.DB.Model(new(model.EventImmunizationPlan)).Create(newImmunizationPlanCowList).Error; err != nil {
  235. zaplog.Error("ImmunizationPlan",
  236. zap.Any("err", err),
  237. zap.Any("plan", plan),
  238. zap.Any("cowList", cowList),
  239. zap.Any("newImmunizationPlanCowList", newImmunizationPlanCowList),
  240. )
  241. return xerr.WithStack(err)
  242. }
  243. todayCount = int32(len(newImmunizationPlanCowList))
  244. if todayCount > 0 {
  245. endDay := nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[pasturePb.CalendarType_Immunisation])
  246. e.CreatedCalendar(plan.PastureId, pasturePb.CalendarType_Immunisation, nowTime.Format(model.LayoutDate2), endDay.Format(model.LayoutDate2), todayCount)
  247. }
  248. e.CreateCrontabLog(ImmunizationPlan)
  249. }
  250. return nil
  251. }
  252. // SameTimePlan 同期计划,生成工作单
  253. func (e *Entry) SameTimePlan() error {
  254. if ok := e.IsExistCrontabLog(SameTimePlan); ok {
  255. return nil
  256. }
  257. sameTimeList := make([]*model.SameTime, 0)
  258. if err := e.DB.Model(new(model.SameTime)).
  259. Where("is_show = ?", pasturePb.IsShow_Ok).
  260. Order("pasture_id").
  261. Find(&sameTimeList).Error; err != nil {
  262. return xerr.WithStack(err)
  263. }
  264. if len(sameTimeList) < 0 {
  265. return nil
  266. }
  267. defer func() {
  268. e.CreateCrontabLog(SameTimePlan)
  269. }()
  270. currWeek := time.Now().Local().Weekday()
  271. for _, sameTime := range sameTimeList {
  272. if len(sameTime.WeekType) <= 0 {
  273. continue
  274. }
  275. wts := strings.Split(sameTime.WeekType, model.WeekTypeSeparator)
  276. if len(wts) <= 0 {
  277. continue
  278. }
  279. var info bool
  280. for _, wt := range wts {
  281. if time.Weekday(int(pasturePb.Week_Kind_value[wt])) == currWeek {
  282. info = true
  283. break
  284. }
  285. }
  286. if !info {
  287. continue
  288. }
  289. cowList := make([]*model.Cow, 0)
  290. pref := e.DB.Model(new(model.Cow)).
  291. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  292. Where("sex = ?", pasturePb.Genders_Female).
  293. Where("pasture_id = ?", sameTime.PastureId).
  294. Where("is_pregnant = ?", pasturePb.IsShow_No)
  295. switch sameTime.CowType {
  296. case pasturePb.SameTimeCowType_Breeding_Calf:
  297. pref.Where("calving_age BETWEEN ? AND ?", sameTime.PostpartumDaysStart, sameTime.PostpartumDaysEnd).
  298. Where("lact > ?", 0)
  299. case pasturePb.SameTimeCowType_Empty:
  300. pref.Where(
  301. e.DB.Where("breed_status = ?", pasturePb.BreedStatus_Empty).
  302. Or("breed_status = ?", pasturePb.BreedStatus_Abort),
  303. )
  304. default:
  305. continue
  306. }
  307. if err := pref.Where(`NOT EXISTS (SELECT 1 FROM event_cow_same_time WHERE event_cow_same_time.cow_id = cow.id
  308. AND event_cow_same_time.status = ?)`, pasturePb.IsShow_No).
  309. Find(&cowList).Error; err != nil {
  310. zaplog.Error("crontab", zap.Any("err", err), zap.Any("sameTime", sameTime))
  311. return xerr.WithStack(err)
  312. }
  313. if len(cowList) <= 0 {
  314. continue
  315. }
  316. if err := e.GenerateCalendarBySameTimePlan(cowList, sameTime); err != nil {
  317. zaplog.Error("crontab",
  318. zap.Any("GenerateCalendarBySameTimePlan", err),
  319. zap.Any("cowList", cowList),
  320. zap.Any("plan", sameTime),
  321. )
  322. continue
  323. }
  324. }
  325. return nil
  326. }
  327. // UpdateSameTime 更新每天同期数据
  328. func (e *Entry) UpdateSameTime() error {
  329. calendarTypeList := backend.CalendarTypeEnumList("")
  330. showDay := time.Now().Local().Format(model.LayoutDate2)
  331. for _, v := range calendarTypeList {
  332. count := int64(0)
  333. if err := e.DB.Model(new(model.EventCowSameTime)).
  334. Where("same_time_type = ?", v.Value).
  335. Where("status = ?", pasturePb.IsShow_No).
  336. Count(&count).Error; err != nil {
  337. zaplog.Error("crontab", zap.Any("UpdateSameTime", err), zap.Any("count", count))
  338. }
  339. if count >= 0 {
  340. continue
  341. }
  342. isExist := int64(0)
  343. if err := e.DB.Model(new(model.Calendar)).
  344. Where("calendar_type = ?", v.Value).
  345. Where("show_day = ?", showDay).
  346. Count(&isExist).Error; err != nil {
  347. continue
  348. }
  349. if isExist <= 0 {
  350. continue
  351. }
  352. if err := e.DB.Model(new(model.Calendar)).
  353. Where("calendar_type = ?", v.Value).
  354. Where("show_day = ?", showDay).
  355. Updates(map[string]interface{}{
  356. "count": count,
  357. }).Error; err != nil {
  358. continue
  359. }
  360. }
  361. return nil
  362. }
  363. // SystemBasicCrontab 基础配置计划任务
  364. func (e *Entry) SystemBasicCrontab() error {
  365. if ok := e.IsExistCrontabLog(SystemBasicCrontab); ok {
  366. return nil
  367. }
  368. defer func() {
  369. e.CreateCrontabLog(SystemBasicCrontab)
  370. }()
  371. systemBasicList := make([]*model.SystemBasic, 0)
  372. if err := e.DB.Model(new(model.SystemBasic)).
  373. Where("is_show = ?", pasturePb.IsShow_Ok).
  374. Where("name IN ?", []string{
  375. model.PregnantCheckForFirst,
  376. model.PregnantCheckForSecond,
  377. model.WeaningAge,
  378. model.PregnancyAge,
  379. model.DryMilkAge,
  380. }).Order("pasture_id").
  381. Find(&systemBasicList).Error; err != nil {
  382. zaplog.Error("crontab", zap.Any("PregnancyCheck", err))
  383. return xerr.WithStack(err)
  384. }
  385. currWeekValue := time.Now().Local().Weekday()
  386. for _, systemBasic := range systemBasicList {
  387. // 周执行
  388. if systemBasic.Name == model.PregnantCheckForFirst && systemBasic.WeekValue >= 0 &&
  389. time.Weekday(systemBasic.WeekValue) != currWeekValue {
  390. continue
  391. }
  392. cowList := make([]*model.Cow, 0)
  393. pref := e.DB.Model(new(model.Cow)).
  394. Where("admission_status = ? ", pasturePb.AdmissionStatus_Admission).
  395. Where("pasture_id = ?", systemBasic.PastureId)
  396. switch systemBasic.Name {
  397. case model.PregnantCheckForFirst: // 初检清单
  398. pref.Where("breed_status = ?", pasturePb.BreedStatus_Breeding).
  399. Where("last_mating_at > ?", 0).
  400. Where("DATE(FROM_UNIXTIME(last_mating_at)) BETWEEN DATE_SUB(CURDATE(), INTERVAL ? DAY) AND DATE_SUB(CURDATE(), INTERVAL ? DAY)", systemBasic.MaxValue, systemBasic.MinValue).
  401. Where(`NOT EXISTS (SELECT 1 FROM event_pregnant_check WHERE event_pregnant_check.cow_id = cow.id
  402. AND event_pregnant_check.status = ? AND event_pregnant_check.pregnant_check_name = ?)`, pasturePb.IsShow_No, model.PregnantCheckForFirst)
  403. case model.PregnantCheckForSecond: // 复检清单 过滤初检空怀的牛只
  404. pref.Where("breed_status IN (?)", []pasturePb.BreedStatus_Kind{pasturePb.BreedStatus_Pregnant}).
  405. Where("last_mating_at > ?", 0).
  406. Where("DATE(FROM_UNIXTIME(last_mating_at)) = DATE_SUB(CURDATE(), INTERVAL ? DAY)", systemBasic.MinValue).
  407. Where(`NOT EXISTS (SELECT 1 FROM event_pregnant_check WHERE event_pregnant_check.cow_id = cow.id
  408. AND event_pregnant_check.status = ? AND event_pregnant_check.pregnant_check_name = ? )`, pasturePb.IsShow_No, model.PregnantCheckForSecond)
  409. case model.WeaningAge: // 断奶清单
  410. pref.Where("day_age = ?", systemBasic.MinValue).
  411. Where("NOT EXISTS (SELECT 1 FROM event_weaning WHERE event_weaning.cow_id = cow.id AND event_weaning.status = ?)", pasturePb.IsShow_No)
  412. case model.PregnancyAge: // 产犊清单
  413. pref.Where("pregnancy_age BETWEEN ? AND ?", systemBasic.MinValue, systemBasic.MaxValue).
  414. Where("breed_status = ?", pasturePb.BreedStatus_Pregnant).
  415. Where("NOT EXISTS (SELECT 1 FROM event_calving WHERE event_calving.cow_id = cow.id AND event_calving.status = ?)", pasturePb.IsShow_No)
  416. case model.DryMilkAge: // 干奶清单
  417. pref.Where("pregnancy_age BETWEEN ? AND ?", systemBasic.MinValue, systemBasic.MaxValue).
  418. Where("breed_status = ?", pasturePb.BreedStatus_Pregnant).
  419. Where("is_pregnant = ?", pasturePb.IsShow_Ok).
  420. Where("NOT EXISTS (SELECT 1 FROM event_dry_milk WHERE event_dry_milk.cow_id = cow.id AND event_dry_milk.status = ?)", pasturePb.IsShow_No)
  421. default:
  422. continue
  423. }
  424. if err := pref.Find(&cowList).Error; err != nil {
  425. zaplog.Error("crontab", zap.Any("PregnancyCheck", err), zap.Any("cowList", cowList))
  426. continue
  427. }
  428. if len(cowList) <= 0 {
  429. continue
  430. }
  431. e.InitEventData(cowList, systemBasic)
  432. }
  433. return nil
  434. }
  435. func (e *Entry) DeleteOldOriginal() error {
  436. if err := e.DB.Model(new(model.NeckRingOriginal)).
  437. Where("active_date <= ?", time.Now().Local().AddDate(0, -1, 0).Format(model.LayoutDate2)).
  438. Delete(&model.NeckRingOriginal{}).Error; err != nil {
  439. zaplog.Error("crontab", zap.Any("DeleteOldOriginal", err))
  440. }
  441. if err := e.DB.Model(new(model.NeckActiveHabit)).
  442. Where("heat_date <= ?", time.Now().Local().AddDate(0, -1, 0).Format(model.LayoutDate2)).
  443. Delete(&model.NeckActiveHabit{}).Error; err != nil {
  444. zaplog.Error("crontab", zap.Any("DeleteNeckActiveHabit", err))
  445. }
  446. return nil
  447. }
  448. // UpdateDiseaseToCalendar 更新每天治疗中牛头数到日历表中
  449. func (e *Entry) UpdateDiseaseToCalendar() error {
  450. pastureList := e.FindPastureList()
  451. for _, pasture := range pastureList {
  452. // 更新每天治疗中牛头数到日历表中
  453. var count int64
  454. if err := e.DB.Model(new(model.EventCowDisease)).
  455. Where("pasture_id = ?", pasture.Id).
  456. Where("health_status IN (?)", []pasturePb.HealthStatus_Kind{pasturePb.HealthStatus_Disease, pasturePb.HealthStatus_Treatment}).
  457. Count(&count).Error; err != nil {
  458. zaplog.Error("crontab", zap.Any("UpdateDisease", err))
  459. }
  460. if count > 0 {
  461. calendarTypeName := backend.CalendarTypeMap()[pasturePb.CalendarType_Disease]
  462. startDay := time.Now().Local().Format(model.LayoutDate2)
  463. newCalendar := model.NewCalendar(pasture.Id, calendarTypeName, pasturePb.CalendarType_Disease, startDay, startDay, int32(count))
  464. if err := e.DB.Model(new(model.Calendar)).
  465. Create(newCalendar).Error; err != nil {
  466. zaplog.Error("crontab", zap.Any("UpdateDisease", err))
  467. }
  468. }
  469. }
  470. return nil
  471. }
  472. func (e *Entry) InitEventData(cowList []*model.Cow, systemBasic *model.SystemBasic) {
  473. calendarType := pasturePb.CalendarType_Invalid
  474. nowTime := time.Now().Local()
  475. startDay, endDay := "", ""
  476. switch systemBasic.Name {
  477. case model.PregnantCheckForFirst, model.PregnantCheckForSecond:
  478. penMap, _ := e.GetPenMapList(systemBasic.PastureId)
  479. calendarType = pasturePb.CalendarType_Pregnancy_Check
  480. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  481. eventPregnantCheckDataList := model.NewEventPregnantCheckList(systemBasic.PastureId, cowList, penMap, systemBasic.Name, startDay, endDay)
  482. if err := e.DB.Model(new(model.EventPregnantCheck)).Create(eventPregnantCheckDataList).Error; err != nil {
  483. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventPregnantCheckDataList", eventPregnantCheckDataList))
  484. return
  485. }
  486. case model.WeaningAge:
  487. calendarType = pasturePb.CalendarType_Weaning
  488. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  489. eventWeaningDataList := model.NewEventWeaningList(systemBasic.PastureId, cowList, startDay, endDay)
  490. if err := e.DB.Model(new(model.EventWeaning)).Create(eventWeaningDataList).Error; err != nil {
  491. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventWeaningDataList", eventWeaningDataList))
  492. return
  493. }
  494. case model.PregnancyAge:
  495. calendarType = pasturePb.CalendarType_Calving
  496. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  497. eventCalvingList := model.NewEventCalvingList(systemBasic.PastureId, cowList, startDay, endDay)
  498. if err := e.DB.Model(new(model.EventCalving)).Create(eventCalvingList).Error; err != nil {
  499. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventCalvingList", eventCalvingList))
  500. return
  501. }
  502. case model.DryMilkAge:
  503. calendarType = pasturePb.CalendarType_DryMilk
  504. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  505. eventCalvingList := model.NewEventDryMilkList(systemBasic.PastureId, cowList, startDay, endDay)
  506. if err := e.DB.Model(new(model.EventDryMilk)).Create(eventCalvingList).Error; err != nil {
  507. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventCalvingList", eventCalvingList))
  508. return
  509. }
  510. }
  511. e.CreatedCalendar(systemBasic.PastureId, calendarType, startDay, endDay, int32(len(cowList)))
  512. }
  513. // UpdateCowWeeklyHigh 更新牛只周活动量
  514. func (e *Entry) UpdateCowWeeklyHigh(pastureId int64, cow *model.Cow) int32 {
  515. highData, err := e.GetSystemNeckRingConfigure(pastureId, model.High)
  516. if err != nil || highData == nil || highData.Value <= 0 {
  517. return 0
  518. }
  519. minWeeklyActive, er := e.GetSystemNeckRingConfigure(pastureId, model.MinWeeklyActive)
  520. if er != nil || minWeeklyActive == nil || minWeeklyActive.Value <= 0 {
  521. return 0
  522. }
  523. weeklyActiveModelList := make([]*model.WeeklyActiveModel, 0)
  524. startTime := time.Now().Local().AddDate(0, 0, -7).Format(model.LayoutDate2)
  525. endTime := time.Now().Local().AddDate(0, 0, -1).Format(model.LayoutDate2)
  526. selectStr := fmt.Sprintf(`cow_id,heat_date,count(1) AS nb,IF(ROUND(AVG(high))>%d, ROUND(AVG(high)), %d) AS high`, minWeeklyActive.Value, minWeeklyActive.Value)
  527. if err = e.DB.Model(new(model.NeckActiveHabit)).
  528. Select(selectStr).
  529. Where("pasture_id = ?", pastureId).
  530. Where("cow_id = ?", cow.Id).
  531. Where("heat_date BETWEEN ? AND ?", startTime, endTime).
  532. Where("high > ?", highData.Value).
  533. Having("nb > ?", 8).
  534. Order("high").
  535. Group("heat_date").
  536. Find(&weeklyActiveModelList).Error; err != nil {
  537. zaplog.Error("crontab", zap.Any("UpdateCowWeeklyHigh", err))
  538. return 0
  539. }
  540. if len(weeklyActiveModelList) < 3 {
  541. return 0
  542. }
  543. // 1. 先按High值排序
  544. sort.Slice(weeklyActiveModelList, func(i, j int) bool {
  545. return weeklyActiveModelList[i].High < weeklyActiveModelList[j].High
  546. })
  547. countHigh := len(weeklyActiveModelList)
  548. sumHigh := int32(0)
  549. minHigh := weeklyActiveModelList[0].High
  550. maxHigh := weeklyActiveModelList[len(weeklyActiveModelList)-1].High
  551. for _, v := range weeklyActiveModelList {
  552. sumHigh += v.High
  553. }
  554. diff := 0
  555. x := int32(0)
  556. if countHigh > 3 {
  557. diff = 3
  558. x = weeklyActiveModelList[1].High
  559. } else if countHigh == 3 {
  560. diff = 2
  561. } else {
  562. return 0
  563. }
  564. avgHigh := float64(sumHigh-minHigh-maxHigh-x) / float64(countHigh-diff)
  565. return int32(avgHigh)
  566. }