cow_cron.go 21 KB

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