cow_cron.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627
  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. case model.AvgRegistrationDays:
  125. pastureIndicatorList = e.AvgRegistrationDays(pastureList)
  126. case model.AvgPregnancyDays:
  127. pastureIndicatorList = e.AvgPregnancyDays(pastureList, startTime, endTime)
  128. case model.AvgGestationalAge:
  129. pastureIndicatorList = e.AvgGestationalAge(pastureList)
  130. case model.Month17UnPregnancyRate:
  131. pastureIndicatorList = e.MonthUnPregnancyRate(pastureList, 17)
  132. case model.Month20UnPregnancyRate:
  133. pastureIndicatorList = e.MonthUnPregnancyRate(pastureList, 20)
  134. }
  135. for pastureId, value := range pastureIndicatorList {
  136. e.UpdatePastureIndicators(pastureId, indicatorsDetail, startTime, value)
  137. }
  138. }
  139. return nil
  140. }
  141. // UpdateCowInfo 牛只基本信息维护
  142. func (e *Entry) UpdateCowInfo() error {
  143. pastureList := e.FindPastureList()
  144. for _, pasture := range pastureList {
  145. e.UpdateCowInfoByPasture(pasture.Id)
  146. }
  147. e.CreateCrontabLog(UpdateCowInfo)
  148. return nil
  149. }
  150. func (e *Entry) UpdateCowInfoByPasture(pastureId int64) {
  151. if ok := e.IsExistCrontabLog(UpdateCowInfo); ok {
  152. return
  153. }
  154. cowList := make([]*model.Cow, 0)
  155. if err := e.DB.Model(new(model.Cow)).
  156. Where("pasture_id = ?", pastureId).
  157. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  158. Find(&cowList).Error; err != nil {
  159. zaplog.Error("crontab", zap.Any("UpdateCowInfoByPasture", err))
  160. return
  161. }
  162. dateTime := time.Now().Local().Format(model.LayoutMonth)
  163. for _, cow := range cowList {
  164. // 周活动量
  165. weeklyActive := e.UpdateCowWeeklyHigh(pastureId, cow)
  166. zaplog.Info("crontab", zap.Any("weeklyActive", weeklyActive))
  167. cow.EventUpdate(weeklyActive)
  168. if err := e.DB.Model(new(model.Cow)).
  169. Select("day_age", "calving_age", "pregnancy_age", "admission_age", "abortion_age", "cow_type", "weekly_active", "lactation_age", "dry_milk_age", "mating_age").
  170. Where("id = ?", cow.Id).
  171. Updates(cow).Error; err != nil {
  172. zaplog.Error("Crontab", zap.Any("UpdateCowDayAge", err))
  173. }
  174. // 每月怀孕数据
  175. cowPregnantList := make([]*model.CowPregnant, 0)
  176. if cow.IsPregnant == pasturePb.IsShow_Ok || cow.BreedStatus == pasturePb.BreedStatus_Pregnant {
  177. cowPregnantList = append(cowPregnantList, model.NewCowPregnant(pastureId, cow, dateTime))
  178. }
  179. if len(cowPregnantList) > 0 {
  180. if err := e.DB.Model(new(model.CowPregnant)).
  181. Where("pasture_id = ?", pastureId).
  182. Where("date_time = ?", dateTime).
  183. Delete(&model.CowPregnant{}).Error; err != nil {
  184. zaplog.Error("Crontab", zap.Any("UpdateCowPregnant", err))
  185. }
  186. if err := e.DB.Model(new(model.CowPregnant)).
  187. Create(&cowPregnantList).Error; err != nil {
  188. zaplog.Error("Crontab", zap.Any("UpdateCowPregnant", err))
  189. }
  190. }
  191. }
  192. }
  193. // ImmunizationPlan 免疫计划,生成工作单
  194. func (e *Entry) ImmunizationPlan() error {
  195. if ok := e.IsExistCrontabLog(ImmunizationPlan); ok {
  196. return nil
  197. }
  198. planList := make([]*model.ImmunizationPlan, 0)
  199. if err := e.DB.Model(new(model.ImmunizationPlan)).
  200. Where("is_show = ?", pasturePb.IsShow_Ok).
  201. Order("pasture_id").
  202. Find(&planList).Error; err != nil {
  203. return xerr.WithStack(err)
  204. }
  205. var todayCount int32 = 0
  206. nowTime := time.Now().Local()
  207. for _, plan := range planList {
  208. cowList := make([]*model.Cow, 0)
  209. pref := e.DB.Table(fmt.Sprintf("%s as a", new(model.Cow).TableName())).
  210. Select("a.*").
  211. Where("a.pasture_id = ?", plan.PastureId).
  212. Where("a.admission_status = ?", pasturePb.AdmissionStatus_Admission).
  213. 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())
  214. if plan.CowType > 0 {
  215. pref.Where("a.cow_type = ?", plan.CowType)
  216. }
  217. switch plan.Conditions {
  218. case pasturePb.ImmunizationConditions_Days_Age:
  219. pref.Where("a.day_age = ?", plan.Value)
  220. case pasturePb.ImmunizationConditions_Days_After_Delivery:
  221. pref.Where("a.calving_age = ?", plan.Value)
  222. case pasturePb.ImmunizationConditions_Days_Of_Pregnancy:
  223. pref.Where("a.pregnancy_age = ?", plan.Value).
  224. Where("a.is_pregnant = ?", pasturePb.IsShow_Ok)
  225. case pasturePb.ImmunizationConditions_Month:
  226. continue
  227. case pasturePb.ImmunizationConditions_Admission_Days:
  228. pref.Where("a.admission_age = ?", plan.Value)
  229. case pasturePb.ImmunizationConditions_Other_Vaccine_After:
  230. if plan.ImmunizationPlanId > 0 {
  231. pref.Joins("INNER JOIN event_immunization_plan as b ON b.plan_id = ? ", plan.ImmunizationPlanId).
  232. Where("b.cow_id = a.id").
  233. Where("DATE_ADD(b.reality_day, INTERVAL ? DAY) = ?", plan.Value, time.Now().Local().Format(model.LayoutDate2)).
  234. Where("b.status = ?", pasturePb.IsShow_Ok)
  235. }
  236. }
  237. if err := pref.Find(&cowList).Debug().Error; err != nil {
  238. return xerr.WithStack(err)
  239. }
  240. if len(cowList) <= 0 {
  241. continue
  242. }
  243. newImmunizationPlanCowList := model.NewCowImmunizationPlanList(cowList, plan)
  244. if err := e.DB.Model(new(model.EventImmunizationPlan)).Create(newImmunizationPlanCowList).Error; err != nil {
  245. zaplog.Error("ImmunizationPlan",
  246. zap.Any("err", err),
  247. zap.Any("plan", plan),
  248. zap.Any("cowList", cowList),
  249. zap.Any("newImmunizationPlanCowList", newImmunizationPlanCowList),
  250. )
  251. return xerr.WithStack(err)
  252. }
  253. todayCount = int32(len(newImmunizationPlanCowList))
  254. if todayCount > 0 {
  255. endDay := nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[pasturePb.CalendarType_Immunisation])
  256. e.CreatedCalendar(plan.PastureId, pasturePb.CalendarType_Immunisation, nowTime.Format(model.LayoutDate2), endDay.Format(model.LayoutDate2), todayCount)
  257. }
  258. e.CreateCrontabLog(ImmunizationPlan)
  259. }
  260. return nil
  261. }
  262. // SameTimePlan 同期计划,生成工作单
  263. func (e *Entry) SameTimePlan() error {
  264. if ok := e.IsExistCrontabLog(SameTimePlan); ok {
  265. return nil
  266. }
  267. sameTimeList := make([]*model.SameTime, 0)
  268. if err := e.DB.Model(new(model.SameTime)).
  269. Where("is_show = ?", pasturePb.IsShow_Ok).
  270. Order("pasture_id").
  271. Find(&sameTimeList).Error; err != nil {
  272. return xerr.WithStack(err)
  273. }
  274. if len(sameTimeList) < 0 {
  275. return nil
  276. }
  277. defer func() {
  278. e.CreateCrontabLog(SameTimePlan)
  279. }()
  280. currWeek := time.Now().Local().Weekday()
  281. for _, sameTime := range sameTimeList {
  282. if len(sameTime.WeekType) <= 0 {
  283. continue
  284. }
  285. wts := strings.Split(sameTime.WeekType, model.WeekTypeSeparator)
  286. if len(wts) <= 0 {
  287. continue
  288. }
  289. var info bool
  290. for _, wt := range wts {
  291. if time.Weekday(int(pasturePb.Week_Kind_value[wt])) == currWeek {
  292. info = true
  293. break
  294. }
  295. }
  296. if !info {
  297. continue
  298. }
  299. cowList := make([]*model.Cow, 0)
  300. pref := e.DB.Model(new(model.Cow)).
  301. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  302. Where("sex = ?", pasturePb.Genders_Female).
  303. Where("pasture_id = ?", sameTime.PastureId).
  304. Where("is_pregnant = ?", pasturePb.IsShow_No)
  305. switch sameTime.CowType {
  306. case pasturePb.SameTimeCowType_Breeding_Calf:
  307. pref.Where("calving_age BETWEEN ? AND ?", sameTime.PostpartumDaysStart, sameTime.PostpartumDaysEnd).
  308. Where("lact > ?", 0)
  309. case pasturePb.SameTimeCowType_Empty:
  310. pref.Where(
  311. e.DB.Where("breed_status = ?", pasturePb.BreedStatus_Empty).
  312. Or("breed_status = ?", pasturePb.BreedStatus_Abort),
  313. )
  314. default:
  315. continue
  316. }
  317. if err := pref.Where(`NOT EXISTS (SELECT 1 FROM event_cow_same_time WHERE event_cow_same_time.cow_id = cow.id
  318. AND event_cow_same_time.status = ?)`, pasturePb.IsShow_No).
  319. Find(&cowList).Error; err != nil {
  320. zaplog.Error("crontab", zap.Any("err", err), zap.Any("sameTime", sameTime))
  321. return xerr.WithStack(err)
  322. }
  323. if len(cowList) <= 0 {
  324. continue
  325. }
  326. if err := e.GenerateCalendarBySameTimePlan(cowList, sameTime); err != nil {
  327. zaplog.Error("crontab",
  328. zap.Any("GenerateCalendarBySameTimePlan", err),
  329. zap.Any("cowList", cowList),
  330. zap.Any("plan", sameTime),
  331. )
  332. continue
  333. }
  334. }
  335. return nil
  336. }
  337. // UpdateSameTime 更新每天同期数据
  338. func (e *Entry) UpdateSameTime() error {
  339. calendarTypeList := backend.CalendarTypeEnumList("")
  340. showDay := time.Now().Local().Format(model.LayoutDate2)
  341. for _, v := range calendarTypeList {
  342. count := int64(0)
  343. if err := e.DB.Model(new(model.EventCowSameTime)).
  344. Where("same_time_type = ?", v.Value).
  345. Where("status = ?", pasturePb.IsShow_No).
  346. Count(&count).Error; err != nil {
  347. zaplog.Error("crontab", zap.Any("UpdateSameTime", err), zap.Any("count", count))
  348. }
  349. if count >= 0 {
  350. continue
  351. }
  352. isExist := int64(0)
  353. if err := e.DB.Model(new(model.Calendar)).
  354. Where("calendar_type = ?", v.Value).
  355. Where("show_day = ?", showDay).
  356. Count(&isExist).Error; err != nil {
  357. continue
  358. }
  359. if isExist <= 0 {
  360. continue
  361. }
  362. if err := e.DB.Model(new(model.Calendar)).
  363. Where("calendar_type = ?", v.Value).
  364. Where("show_day = ?", showDay).
  365. Updates(map[string]interface{}{
  366. "count": count,
  367. }).Error; err != nil {
  368. continue
  369. }
  370. }
  371. return nil
  372. }
  373. // SystemBasicCrontab 基础配置计划任务
  374. func (e *Entry) SystemBasicCrontab() error {
  375. if ok := e.IsExistCrontabLog(SystemBasicCrontab); ok {
  376. return nil
  377. }
  378. defer func() {
  379. e.CreateCrontabLog(SystemBasicCrontab)
  380. }()
  381. systemBasicList := make([]*model.SystemBasic, 0)
  382. if err := e.DB.Model(new(model.SystemBasic)).
  383. Where("is_show = ?", pasturePb.IsShow_Ok).
  384. Where("name IN ?", []string{
  385. model.PregnantCheckForFirst,
  386. model.PregnantCheckForSecond,
  387. model.WeaningAge,
  388. model.PregnancyAge,
  389. model.DryMilkAge,
  390. }).Order("pasture_id").
  391. Find(&systemBasicList).Error; err != nil {
  392. zaplog.Error("crontab", zap.Any("PregnancyCheck", err))
  393. return xerr.WithStack(err)
  394. }
  395. currWeekValue := time.Now().Local().Weekday()
  396. for _, systemBasic := range systemBasicList {
  397. // 周执行
  398. if systemBasic.Name == model.PregnantCheckForFirst && systemBasic.WeekValue >= 0 &&
  399. time.Weekday(systemBasic.WeekValue) != currWeekValue {
  400. continue
  401. }
  402. cowList := make([]*model.Cow, 0)
  403. pref := e.DB.Model(new(model.Cow)).
  404. Where("admission_status = ? ", pasturePb.AdmissionStatus_Admission).
  405. Where("pasture_id = ?", systemBasic.PastureId)
  406. switch systemBasic.Name {
  407. case model.PregnantCheckForFirst: // 初检清单
  408. pref.Where("breed_status = ?", pasturePb.BreedStatus_Breeding).
  409. Where("last_mating_at > ?", 0).
  410. Where("DATE(FROM_UNIXTIME(last_mating_at)) BETWEEN DATE_SUB(CURDATE(), INTERVAL ? DAY) AND DATE_SUB(CURDATE(), INTERVAL ? DAY)", systemBasic.MaxValue, systemBasic.MinValue).
  411. Where(`NOT EXISTS (SELECT 1 FROM event_pregnant_check WHERE event_pregnant_check.cow_id = cow.id
  412. AND event_pregnant_check.status = ? AND event_pregnant_check.pregnant_check_name = ?)`, pasturePb.IsShow_No, model.PregnantCheckForFirst)
  413. case model.PregnantCheckForSecond: // 复检清单 过滤初检空怀的牛只
  414. pref.Where("breed_status IN (?)", []pasturePb.BreedStatus_Kind{pasturePb.BreedStatus_Pregnant}).
  415. Where("last_mating_at > ?", 0).
  416. Where("DATE(FROM_UNIXTIME(last_mating_at)) = DATE_SUB(CURDATE(), INTERVAL ? DAY)", systemBasic.MinValue).
  417. Where(`NOT EXISTS (SELECT 1 FROM event_pregnant_check WHERE event_pregnant_check.cow_id = cow.id
  418. AND event_pregnant_check.status = ? AND event_pregnant_check.pregnant_check_name = ? )`, pasturePb.IsShow_No, model.PregnantCheckForSecond)
  419. case model.WeaningAge: // 断奶清单
  420. pref.Where("day_age = ?", systemBasic.MinValue).
  421. Where("NOT EXISTS (SELECT 1 FROM event_weaning WHERE event_weaning.cow_id = cow.id AND event_weaning.status = ?)", pasturePb.IsShow_No)
  422. case model.PregnancyAge: // 产犊清单
  423. pref.Where("pregnancy_age BETWEEN ? AND ?", systemBasic.MinValue, systemBasic.MaxValue).
  424. Where("breed_status = ?", pasturePb.BreedStatus_Pregnant).
  425. Where("NOT EXISTS (SELECT 1 FROM event_calving WHERE event_calving.cow_id = cow.id AND event_calving.status = ?)", pasturePb.IsShow_No)
  426. case model.DryMilkAge: // 干奶清单
  427. pref.Where("pregnancy_age BETWEEN ? AND ?", systemBasic.MinValue, systemBasic.MaxValue).
  428. Where("breed_status = ?", pasturePb.BreedStatus_Pregnant).
  429. Where("is_pregnant = ?", pasturePb.IsShow_Ok).
  430. 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)
  431. default:
  432. continue
  433. }
  434. if err := pref.Find(&cowList).Error; err != nil {
  435. zaplog.Error("crontab", zap.Any("PregnancyCheck", err), zap.Any("cowList", cowList))
  436. continue
  437. }
  438. if len(cowList) <= 0 {
  439. continue
  440. }
  441. e.InitEventData(cowList, systemBasic)
  442. }
  443. return nil
  444. }
  445. func (e *Entry) DeleteOldOriginal() error {
  446. if err := e.DB.Model(new(model.NeckRingOriginal)).
  447. Where("active_date <= ?", time.Now().Local().AddDate(0, -1, 0).Format(model.LayoutDate2)).
  448. Delete(&model.NeckRingOriginal{}).Error; err != nil {
  449. zaplog.Error("crontab", zap.Any("DeleteOldOriginal", err))
  450. }
  451. if err := e.DB.Model(new(model.NeckActiveHabit)).
  452. Where("heat_date <= ?", time.Now().Local().AddDate(0, -1, 0).Format(model.LayoutDate2)).
  453. Delete(&model.NeckActiveHabit{}).Error; err != nil {
  454. zaplog.Error("crontab", zap.Any("DeleteNeckActiveHabit", err))
  455. }
  456. return nil
  457. }
  458. // UpdateDiseaseToCalendar 更新每天治疗中牛头数到日历表中
  459. func (e *Entry) UpdateDiseaseToCalendar() error {
  460. pastureList := e.FindPastureList()
  461. for _, pasture := range pastureList {
  462. // 更新每天治疗中牛头数到日历表中
  463. var count int64
  464. if err := e.DB.Model(new(model.EventCowDisease)).
  465. Where("pasture_id = ?", pasture.Id).
  466. Where("health_status IN (?)", []pasturePb.HealthStatus_Kind{pasturePb.HealthStatus_Disease, pasturePb.HealthStatus_Treatment}).
  467. Count(&count).Error; err != nil {
  468. zaplog.Error("crontab", zap.Any("UpdateDisease", err))
  469. }
  470. if count > 0 {
  471. calendarTypeName := backend.CalendarTypeMap()[pasturePb.CalendarType_Disease]
  472. startDay := time.Now().Local().Format(model.LayoutDate2)
  473. newCalendar := model.NewCalendar(pasture.Id, calendarTypeName, pasturePb.CalendarType_Disease, startDay, startDay, int32(count))
  474. if err := e.DB.Model(new(model.Calendar)).
  475. Create(newCalendar).Error; err != nil {
  476. zaplog.Error("crontab", zap.Any("UpdateDisease", err))
  477. }
  478. }
  479. }
  480. return nil
  481. }
  482. func (e *Entry) InitEventData(cowList []*model.Cow, systemBasic *model.SystemBasic) {
  483. calendarType := pasturePb.CalendarType_Invalid
  484. nowTime := time.Now().Local()
  485. startDay, endDay := "", ""
  486. switch systemBasic.Name {
  487. case model.PregnantCheckForFirst, model.PregnantCheckForSecond:
  488. penMap, _ := e.GetPenMapList(systemBasic.PastureId)
  489. calendarType = pasturePb.CalendarType_Pregnancy_Check
  490. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  491. eventPregnantCheckDataList := model.NewEventPregnantCheckList(systemBasic.PastureId, cowList, penMap, systemBasic.Name, startDay, endDay)
  492. if err := e.DB.Model(new(model.EventPregnantCheck)).Create(eventPregnantCheckDataList).Error; err != nil {
  493. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventPregnantCheckDataList", eventPregnantCheckDataList))
  494. return
  495. }
  496. case model.WeaningAge:
  497. calendarType = pasturePb.CalendarType_Weaning
  498. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  499. eventWeaningDataList := model.NewEventWeaningList(systemBasic.PastureId, cowList, startDay, endDay)
  500. if err := e.DB.Model(new(model.EventWeaning)).Create(eventWeaningDataList).Error; err != nil {
  501. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventWeaningDataList", eventWeaningDataList))
  502. return
  503. }
  504. case model.PregnancyAge:
  505. calendarType = pasturePb.CalendarType_Calving
  506. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  507. eventCalvingList := model.NewEventCalvingList(systemBasic.PastureId, cowList, startDay, endDay)
  508. if err := e.DB.Model(new(model.EventCalving)).Create(eventCalvingList).Error; err != nil {
  509. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventCalvingList", eventCalvingList))
  510. return
  511. }
  512. case model.DryMilkAge:
  513. calendarType = pasturePb.CalendarType_DryMilk
  514. startDay, endDay = nowTime.Format(model.LayoutDate2), nowTime.AddDate(0, 0, model.CalendarTypeEndDaysMap[calendarType]).Format(model.LayoutDate2)
  515. eventCalvingList := model.NewEventDryMilkList(systemBasic.PastureId, cowList, startDay, endDay)
  516. if err := e.DB.Model(new(model.EventDryMilk)).Create(eventCalvingList).Error; err != nil {
  517. zaplog.Error("crontab", zap.Any("InitEventData", err), zap.Any("eventCalvingList", eventCalvingList))
  518. return
  519. }
  520. }
  521. e.CreatedCalendar(systemBasic.PastureId, calendarType, startDay, endDay, int32(len(cowList)))
  522. }
  523. // UpdateCowWeeklyHigh 更新牛只周活动量
  524. func (e *Entry) UpdateCowWeeklyHigh(pastureId int64, cow *model.Cow) int32 {
  525. highData, err := e.GetSystemNeckRingConfigure(pastureId, model.High)
  526. if err != nil || highData == nil || highData.Value <= 0 {
  527. return 0
  528. }
  529. minWeeklyActive, er := e.GetSystemNeckRingConfigure(pastureId, model.MinWeeklyActive)
  530. if er != nil || minWeeklyActive == nil || minWeeklyActive.Value <= 0 {
  531. return 0
  532. }
  533. weeklyActiveModelList := make([]*model.WeeklyActiveModel, 0)
  534. startTime := time.Now().Local().AddDate(0, 0, -7).Format(model.LayoutDate2)
  535. endTime := time.Now().Local().AddDate(0, 0, -1).Format(model.LayoutDate2)
  536. 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)
  537. if err = e.DB.Model(new(model.NeckActiveHabit)).
  538. Select(selectStr).
  539. Where("pasture_id = ?", pastureId).
  540. Where("cow_id = ?", cow.Id).
  541. Where("heat_date BETWEEN ? AND ?", startTime, endTime).
  542. Where("high > ?", highData.Value).
  543. Having("nb > ?", 8).
  544. Order("high").
  545. Group("heat_date").
  546. Find(&weeklyActiveModelList).Error; err != nil {
  547. zaplog.Error("crontab", zap.Any("UpdateCowWeeklyHigh", err))
  548. return 0
  549. }
  550. if len(weeklyActiveModelList) < 3 {
  551. return 0
  552. }
  553. // 1. 先按High值排序
  554. sort.Slice(weeklyActiveModelList, func(i, j int) bool {
  555. return weeklyActiveModelList[i].High < weeklyActiveModelList[j].High
  556. })
  557. countHigh := len(weeklyActiveModelList)
  558. sumHigh := int32(0)
  559. minHigh := weeklyActiveModelList[0].High
  560. maxHigh := weeklyActiveModelList[len(weeklyActiveModelList)-1].High
  561. for _, v := range weeklyActiveModelList {
  562. sumHigh += v.High
  563. }
  564. diff := 0
  565. x := int32(0)
  566. if countHigh > 3 {
  567. diff = 3
  568. x = weeklyActiveModelList[1].High
  569. } else if countHigh == 3 {
  570. diff = 2
  571. } else {
  572. return 0
  573. }
  574. avgHigh := float64(sumHigh-minHigh-maxHigh-x) / float64(countHigh-diff)
  575. return int32(avgHigh)
  576. }