cow.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673
  1. package backend
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "kpt-pasture/model"
  7. "kpt-pasture/util"
  8. "net/http"
  9. "sort"
  10. "strings"
  11. "sync"
  12. "time"
  13. "gorm.io/gorm"
  14. "gitee.com/xuyiping_admin/pkg/xerr"
  15. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  16. )
  17. func (s *StoreEntry) Detail(ctx context.Context, req *pasturePb.SearchEventRequest) (*pasturePb.CowInfoResponse, error) {
  18. userModel, err := s.GetUserModel(ctx)
  19. if err != nil {
  20. return nil, xerr.WithStack(err)
  21. }
  22. if req.EarNumber == "" && req.NeckRingNumber == "" {
  23. return nil, xerr.Custom("请输入牛号或项圈号")
  24. }
  25. cowInfo := &model.Cow{}
  26. pref := s.DB.Model(new(model.Cow)).
  27. Where("pasture_id = ?", userModel.AppPasture.Id)
  28. if req.EarNumber != "" {
  29. pref.Where("ear_number = ?", req.EarNumber)
  30. }
  31. if req.NeckRingNumber != "" {
  32. pref.Where("neck_ring_number = ?", req.NeckRingNumber)
  33. }
  34. if err = pref.First(cowInfo).Error; err != nil {
  35. if errors.Is(err, gorm.ErrRecordNotFound) {
  36. return nil, xerr.Custom("该牛只未找到")
  37. } else {
  38. return nil, xerr.WithStack(err)
  39. }
  40. }
  41. cowTypeMap := s.CowTypeMap()
  42. breedStatusMap := s.CowBreedStatusMap()
  43. cowKindMap := s.CowKindMap()
  44. cowSourceMap := s.CowSourceMap()
  45. admissionStatusMap := s.AdmissionStatusMap()
  46. healthStatusMap := s.HealthStatusMap()
  47. purposeMap := s.PurposeMap()
  48. systemBasic, err := s.GetSystemBasicByName(ctx, userModel.AppPasture.Id, model.PregnancyAge)
  49. if err != nil {
  50. return nil, xerr.Custom("请在基础参数配置妊娠天数")
  51. }
  52. cowDetails := model.CowSlice([]*model.Cow{cowInfo}).ToPB(
  53. cowTypeMap, breedStatusMap, cowKindMap, cowSourceMap,
  54. admissionStatusMap, healthStatusMap, purposeMap, systemBasic.MinValue,
  55. )
  56. if len(cowDetails) != 1 {
  57. return nil, xerr.Custom("该牛只未找到")
  58. }
  59. data := cowDetails[0]
  60. return &pasturePb.CowInfoResponse{
  61. Code: http.StatusOK,
  62. Msg: "ok",
  63. Data: data,
  64. }, nil
  65. }
  66. func (s *StoreEntry) List(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchCowListResponse, error) {
  67. userModel, err := s.GetUserModel(ctx)
  68. if err != nil {
  69. return nil, xerr.WithStack(err)
  70. }
  71. systemBasic, err := s.GetSystemBasicByName(ctx, userModel.AppPasture.Id, model.PregnancyAge)
  72. if err != nil {
  73. return nil, xerr.Custom("请在基础参数配置妊娠天数")
  74. }
  75. cowList := make([]*model.Cow, 0)
  76. var count int64 = 0
  77. pref := s.DB.Model(new(model.Cow)).
  78. Where("pasture_id = ?", userModel.AppPasture.Id)
  79. if len(req.CowId) > 0 {
  80. cowIds := strings.Split(req.CowId, ",")
  81. pref.Where("id IN ?", cowIds)
  82. }
  83. if req.EarNumber != "" {
  84. pref.Where("ear_number like ?", fmt.Sprintf("%s%s%s", "%", req.EarNumber, "%"))
  85. }
  86. if req.Id > 0 {
  87. pref.Where("id = ?", req.Id)
  88. }
  89. if req.PenId > 0 {
  90. pref.Where("pen_id = ?", req.PenId)
  91. }
  92. if req.CowType > 0 {
  93. pref.Where("cow_type = ?", req.CowType)
  94. }
  95. if req.BreedStatus > 0 {
  96. pref.Where("breed_status = ?", req.BreedStatus)
  97. }
  98. if req.CowKind > 0 {
  99. pref.Where("cow_kind = ?", req.CowKind)
  100. }
  101. if req.Sex > 0 {
  102. pref.Where("sex = ?", req.Sex)
  103. }
  104. if req.Lact > 0 {
  105. pref.Where("lact = ?", req.Lact)
  106. }
  107. if req.CowSource > 0 {
  108. pref.Where("source_id = ?", req.CowSource)
  109. }
  110. if err = pref.Order("id desc").
  111. Count(&count).
  112. Limit(int(pagination.PageSize)).
  113. Offset(int(pagination.PageOffset)).
  114. Find(&cowList).Error; err != nil {
  115. return nil, xerr.WithStack(err)
  116. }
  117. cowTypeMap := s.CowTypeMap()
  118. breedStatusMap := s.CowBreedStatusMap()
  119. cowKindMap := s.CowKindMap()
  120. cowSourceMap := s.CowSourceMap()
  121. admissionStatusMap := s.AdmissionStatusMap()
  122. healthStatusMap := s.HealthStatusMap()
  123. purposeMap := s.PurposeMap()
  124. return &pasturePb.SearchCowListResponse{
  125. Code: http.StatusOK,
  126. Msg: "ok",
  127. Data: &pasturePb.SearchCowData{
  128. List: model.CowSlice(cowList).ToPB(
  129. cowTypeMap, breedStatusMap, cowKindMap, cowSourceMap,
  130. admissionStatusMap, healthStatusMap, purposeMap, systemBasic.MinValue,
  131. ),
  132. Total: int32(count),
  133. PageSize: pagination.PageSize,
  134. Page: pagination.Page,
  135. },
  136. }, nil
  137. }
  138. func (s *StoreEntry) EventList(ctx context.Context, req *pasturePb.SearchCowEventListRequest, pagination *pasturePb.PaginationModel) (*pasturePb.CowEventListResponse, error) {
  139. userModel, err := s.GetUserModel(ctx)
  140. if err != nil {
  141. return nil, xerr.WithStack(err)
  142. }
  143. eventCowLogList := make([]*model.EventCowLog, 0)
  144. cowInfo, err := s.GetCowEventByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  145. if err != nil {
  146. return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
  147. }
  148. eventCowLog := &model.EventCowLog{CowId: cowInfo.Id}
  149. pref := s.DB.Table(eventCowLog.TableName()).
  150. Where("pasture_id = ?", userModel.AppPasture.Id).
  151. Where("cow_id = ?", cowInfo.Id)
  152. if req.Lact >= 0 {
  153. pref.Where("lact = ?", req.Lact)
  154. }
  155. if req.EventCategoryKind > 0 {
  156. pref.Where("event_type = ?", req.EventCategoryKind)
  157. }
  158. if err = pref.Order("event_at DESC").
  159. Limit(int(pagination.PageSize)).
  160. Offset(int(pagination.PageOffset)).
  161. Find(&eventCowLogList).Error; err != nil {
  162. return nil, xerr.WithStack(err)
  163. }
  164. eventCategoryMap := s.EventCategoryMap()
  165. return &pasturePb.CowEventListResponse{
  166. Code: http.StatusOK,
  167. Msg: "ok",
  168. Data: &pasturePb.CowEventData{
  169. List: model.EventCowLogSlice(eventCowLogList).ToPB(eventCategoryMap),
  170. Total: int32(len(eventCowLogList)),
  171. PageSize: pagination.PageSize,
  172. Page: pagination.Page,
  173. },
  174. }, nil
  175. }
  176. func (s *StoreEntry) BehaviorCurve(ctx context.Context, req *pasturePb.CowBehaviorCurveRequest) (*model.CowBehaviorCurveResponse, error) {
  177. userModel, err := s.GetUserModel(ctx)
  178. if err != nil {
  179. return nil, xerr.WithStack(err)
  180. }
  181. cowInfo, err := s.GetCowEventByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  182. if err != nil {
  183. return nil, xerr.Customf("错误的牛只信息: %d", req.CowId)
  184. }
  185. nowTime := time.Now().Local()
  186. nowDayZero := util.TimeParseLocalUnix(nowTime.Format(model.LayoutDate2))
  187. endDataTime := nowTime.Format(model.LayoutDate2)
  188. startDataTime := nowTime.AddDate(0, 0, -30).Format(model.LayoutDate2)
  189. dayRange, err := util.GetDaysBetween(startDataTime, endDataTime)
  190. if err != nil {
  191. return nil, xerr.WithStack(err)
  192. }
  193. if len(dayRange) <= 0 {
  194. return nil, xerr.Customf("错误的日期范围")
  195. }
  196. // 行为曲线数据
  197. neckActiveHabitList := make([]*model.NeckActiveHabit, 0)
  198. if err = s.DB.Table(new(model.NeckActiveHabit).TableName()).
  199. Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
  200. Where("pasture_id = ?", userModel.AppPasture.Id).
  201. Where("is_show = ?", pasturePb.IsShow_Ok).
  202. Where("cow_id > ?", 0).
  203. Where("heat_date IN (?)", dayRange).
  204. Order("heat_date, frameid").
  205. Find(&neckActiveHabitList).Error; err != nil {
  206. return nil, xerr.WithStack(err)
  207. }
  208. data := model.NeckActiveHabitSlice(neckActiveHabitList).ToPB(req.CurveName)
  209. q1, q3 := s.CowIQR(cowInfo, req.CurveName, dayRange, data.DateTimeList)
  210. data.IQR3 = q3
  211. data.IQR1 = q1
  212. eventMapList := s.EventTypeMap()
  213. for k, v := range eventMapList {
  214. if k == pasturePb.EventType_Enter || k == pasturePb.EventType_Body_Score || k == pasturePb.EventType_Birth ||
  215. k == pasturePb.EventType_Weaning || k == pasturePb.EventType_Sale || k == pasturePb.EventType_Weight ||
  216. k == pasturePb.EventType_Castrated {
  217. continue
  218. }
  219. data.EventMap[k] = v
  220. }
  221. data.LowActivity = 62
  222. data.MiddleActivity = 80
  223. // 牛只事件列表
  224. eventLogList := make([]*model.EventCowLog, 0)
  225. eventLog := &model.EventCowLog{CowId: cowInfo.Id}
  226. if err = s.DB.Table(eventLog.TableName()).
  227. Where("cow_id = ?", cowInfo.Id).
  228. Where("pasture_id = ?", userModel.AppPasture.Id).
  229. Where("event_at BETWEEN ? AND ?", nowDayZero-(30*86400), nowDayZero+86400).
  230. Order("event_at").
  231. Find(&eventLogList).Error; err != nil {
  232. return nil, xerr.WithStack(err)
  233. }
  234. for _, v := range eventLogList {
  235. if v.EventType == pasturePb.EventType_Enter || v.EventType == pasturePb.EventType_Body_Score ||
  236. v.EventType == pasturePb.EventType_Birth || v.EventType == pasturePb.EventType_Weaning ||
  237. v.EventType == pasturePb.EventType_Sale || v.EventType == pasturePb.EventType_Weight ||
  238. v.EventType == pasturePb.EventType_Castrated {
  239. continue
  240. }
  241. if v.EventAt <= 0 {
  242. continue
  243. }
  244. eventAt := time.Unix(v.EventAt, 0).Local()
  245. data.EventList = append(data.EventList, &pasturePb.CowEvent{
  246. EventTypeKind: v.EventType,
  247. EventTypeName: v.EventTypeName,
  248. EventDescription: v.EventDescription,
  249. Remarks: v.Remarks,
  250. EventAtFormat: fmt.Sprintf("%s 09", eventAt.Format(model.LayoutDate2)),
  251. })
  252. }
  253. // 发情数据
  254. estrusList := make([]*model.EventEstrus, 0)
  255. if err = s.DB.Table(new(model.EventEstrus).TableName()).
  256. Where("cow_id = ?", cowInfo.Id).
  257. Where("pasture_id = ?", userModel.AppPasture.Id).
  258. Where("reality_day BETWEEN ? AND ?", startDataTime, endDataTime).
  259. Order("reality_day").
  260. Find(&estrusList).Error; err != nil {
  261. return nil, xerr.WithStack(err)
  262. }
  263. for _, v := range estrusList {
  264. if data.EstrusList[v.Level] == nil {
  265. data.EstrusList[v.Level] = make([]string, 0)
  266. }
  267. if v.RealityDay > 0 {
  268. // 格式化为到小时的字符串
  269. hourStr := time.Unix(v.RealityDay, 0).Local().Format(model.LayoutHour)
  270. data.EstrusList[v.Level] = append(data.EstrusList[v.Level], hourStr)
  271. }
  272. }
  273. return &model.CowBehaviorCurveResponse{
  274. Code: http.StatusOK,
  275. Msg: "ok",
  276. Data: data,
  277. }, nil
  278. }
  279. func (s *StoreEntry) CowIQR(cowInfo *model.Cow, curveName string, dayRange []string, dateTimeList []string) ([]int32, []int32) {
  280. q1, q3 := make([]int32, 0), make([]int32, 0)
  281. if curveName == "" || curveName == "active" || curveName == "behavior" || len(dateTimeList) <= 0 {
  282. return q1, q3
  283. }
  284. penId := cowInfo.PenId
  285. cowList := make([]*model.Cow, 0)
  286. if err := s.DB.Table(new(model.Cow).TableName()).
  287. Where("pen_id = ?", penId).
  288. Where("pasture_id = ?", cowInfo.PastureId).
  289. Where("neck_ring_number != ?", "").
  290. Find(&cowList).Error; err != nil {
  291. return q1, q3
  292. }
  293. cowIds := make([]int64, 0)
  294. for _, cow := range cowList {
  295. cowIds = append(cowIds, cow.Id)
  296. }
  297. neckActiveHabitList, err := s.GetNeckActiveHabitsConcurrent(cowInfo, cowIds, dayRange)
  298. if err != nil {
  299. return q1, q3
  300. }
  301. neckActiveHabitMap := make(map[string][]*model.NeckActiveHabit)
  302. for _, habit := range neckActiveHabitList {
  303. activeTime := habit.ActiveTime[:13]
  304. neckActiveHabitMap[activeTime] = append(neckActiveHabitMap[activeTime], habit)
  305. }
  306. penOriginal := make(map[string][]int32)
  307. for _, dt := range dateTimeList {
  308. habits, ok := neckActiveHabitMap[dt]
  309. if !ok {
  310. penOriginal[dt] = append(penOriginal[dt], int32(0))
  311. continue
  312. }
  313. switch curveName {
  314. case "rumina":
  315. for _, habit := range habits {
  316. penOriginal[dt] = append(penOriginal[dt], habit.Rumina)
  317. }
  318. case "intake":
  319. for _, habit := range habits {
  320. penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
  321. }
  322. case "inactive":
  323. for _, habit := range habits {
  324. penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
  325. }
  326. case "chew":
  327. for _, habit := range habits {
  328. penOriginal[dt] = append(penOriginal[dt], habit.Rumina+habit.Intake)
  329. }
  330. case "immobility":
  331. for _, habit := range habits {
  332. penOriginal[dt] = append(penOriginal[dt], 120-habit.Active)
  333. }
  334. }
  335. }
  336. if len(penOriginal) > 0 {
  337. for _, dataList := range penOriginal {
  338. dl := len(dataList)
  339. if dl < 4 {
  340. q1 = append(q1, int32(0))
  341. q3 = append(q3, int32(0))
  342. continue
  343. }
  344. sort.Slice(dataList, func(i, j int) bool {
  345. return dataList[i] < dataList[j]
  346. })
  347. v1 := (len(dataList) + 1) / 4
  348. // 防止下标溢出panic
  349. s1 := float64(dataList[v1-1]) + 0.25*float64(dataList[v1]-dataList[v1-1])
  350. v3 := v1 * 3
  351. s3 := float64(dataList[v3-1]) + 0.75*float64(dataList[v3]-dataList[v3-1])
  352. q1 = append(q1, int32(s1))
  353. q3 = append(q3, int32(s3))
  354. }
  355. }
  356. return q1, q3
  357. }
  358. func (s *StoreEntry) GetNeckActiveHabitsConcurrent(cowInfo *model.Cow, cowIds []int64, dayRange []string) ([]*model.NeckActiveHabit, error) {
  359. const (
  360. batchSize = 100
  361. workerCount = 5
  362. )
  363. resultsChan := make(chan []*model.NeckActiveHabit)
  364. errChan := make(chan error, 1) // 缓冲 1,避免 goroutine 阻塞
  365. doneChan := make(chan struct{})
  366. var (
  367. neckActiveHabitList []*model.NeckActiveHabit
  368. mu sync.Mutex
  369. )
  370. // 结果收集 goroutine
  371. go func() {
  372. defer close(doneChan)
  373. for batchResults := range resultsChan {
  374. mu.Lock()
  375. neckActiveHabitList = append(neckActiveHabitList, batchResults...)
  376. mu.Unlock()
  377. }
  378. }()
  379. sem := make(chan struct{}, workerCount)
  380. var wg sync.WaitGroup
  381. // 分发任务
  382. go func() {
  383. defer func() {
  384. wg.Wait() // 等待所有 goroutine 完成
  385. close(resultsChan) // 安全关闭
  386. }()
  387. for i := 0; i < len(cowIds); i += batchSize {
  388. end := i + batchSize
  389. if end > len(cowIds) {
  390. end = len(cowIds)
  391. }
  392. batch := cowIds[i:end]
  393. wg.Add(1)
  394. sem <- struct{}{} // 获取令牌(可能阻塞,但不会死锁)
  395. go func(batch []int64) {
  396. defer func() {
  397. <-sem // 释放令牌
  398. wg.Done()
  399. }()
  400. var batchResults []*model.NeckActiveHabit
  401. if err := s.DB.Table(new(model.NeckActiveHabit).TableName()).
  402. Select("rumina,intake,inactive,gasp,high,active,active_time").
  403. Where("pasture_id = ?", cowInfo.PastureId).
  404. Where("is_show = ?", pasturePb.IsShow_Ok).
  405. Where("cow_id IN (?)", batch).
  406. Where("heat_date BETWEEN ? AND ?", dayRange[0], dayRange[len(dayRange)-1]).
  407. Find(&batchResults).Error; err != nil {
  408. select {
  409. case errChan <- fmt.Errorf("batch query error: %v", err):
  410. default:
  411. }
  412. return
  413. }
  414. if len(batchResults) > 0 {
  415. resultsChan <- batchResults
  416. }
  417. }(batch)
  418. }
  419. }()
  420. // 等待结果
  421. select {
  422. case err := <-errChan:
  423. return nil, err
  424. case <-doneChan:
  425. return neckActiveHabitList, nil
  426. }
  427. }
  428. func (s *StoreEntry) CowGrowthCurve(ctx context.Context, req *pasturePb.CowGrowthCurveRequest) (*pasturePb.CowGrowthCurveResponse, error) {
  429. userModel, err := s.GetUserModel(ctx)
  430. if err != nil {
  431. return nil, xerr.WithStack(err)
  432. }
  433. cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  434. if err != nil {
  435. return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
  436. }
  437. weightList := make([]*model.EventWeight, 0)
  438. if err = s.DB.Table(new(model.EventWeight).TableName()).
  439. Where("cow_id = ?", cowInfo.Id).
  440. Where("pasture_id = ?", userModel.AppPasture.Id).
  441. Order("weight_at").
  442. Find(&weightList).Error; err != nil {
  443. return nil, xerr.WithStack(err)
  444. }
  445. eventCowLogList := make([]*model.EventCowLog, 0)
  446. eventCowLog := &model.EventCowLog{CowId: cowInfo.Id}
  447. if err = s.DB.Table(eventCowLog.TableName()).
  448. Where("pasture_id = ?", userModel.AppPasture.Id).
  449. Where("cow_id = ?", cowInfo.Id).
  450. Order("id desc").
  451. Find(&eventCowLogList).Error; err != nil {
  452. return nil, xerr.WithStack(err)
  453. }
  454. return &pasturePb.CowGrowthCurveResponse{
  455. Code: http.StatusOK,
  456. Msg: "ok",
  457. Data: model.EventWeightSlice(weightList).ToPB(cowInfo, eventCowLogList),
  458. }, nil
  459. }
  460. func (s *StoreEntry) CowLactCurve(ctx context.Context, req *pasturePb.CowLactCurveRequest) (*pasturePb.CowLactCurveResponse, error) {
  461. userModel, err := s.GetUserModel(ctx)
  462. if err != nil {
  463. return nil, xerr.WithStack(err)
  464. }
  465. cowInfo, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  466. if err != nil {
  467. return nil, xerr.Customf("错误的牛只信息: %d", req.CowId)
  468. }
  469. cowLactList := make([]*model.CowLact, 0)
  470. if err = s.DB.Table(new(model.CowLact).TableName()).
  471. Where("cow_id = ?", req.CowId).
  472. Where("pasture_id = ?", userModel.AppPasture.Id).
  473. Order("lact").
  474. Find(&cowLactList).Error; err != nil {
  475. return nil, xerr.WithStack(err)
  476. }
  477. data := &pasturePb.CowLactCurveData{
  478. DateTime: make(map[int32]string),
  479. WeekAvgMilk: make([]float32, 0),
  480. DayMilk: make([]float32, 0),
  481. DHI: make([]float32, 0),
  482. MilkProductionTrend: make([]float32, 0),
  483. DayHigh: make([]int32, 0),
  484. DayRumina: make([]int32, 0),
  485. DayIntake: make([]int32, 0),
  486. DayInactive: make([]int32, 0),
  487. DayChew: make([]int32, 0),
  488. DayImmobility: make([]int32, 0),
  489. EstrusWarning: make(map[int32]int32),
  490. }
  491. if len(cowLactList) <= 0 {
  492. return &pasturePb.CowLactCurveResponse{
  493. Code: http.StatusOK,
  494. Msg: "ok",
  495. Data: data,
  496. }, nil
  497. }
  498. starTime := ""
  499. endTime := ""
  500. cowLactMap := make(map[int32]*model.CowLact)
  501. for _, v := range cowLactList {
  502. cowLactMap[v.Lact] = v
  503. if v.Lact == req.Lact {
  504. starTime = v.StartTime
  505. }
  506. }
  507. if st, ok := cowLactMap[req.Lact+1]; ok {
  508. et, _ := util.TimeParseLocal(model.LayoutDate2, st.StartTime)
  509. endTime = et.AddDate(0, 0, -1).Format(model.LayoutDate2)
  510. } else {
  511. endTime = time.Now().Local().Format(model.LayoutDate2)
  512. }
  513. neckRingList := make([]*model.NeckActiveHabit, 0)
  514. if err = s.DB.Model(new(model.NeckActiveHabit)).
  515. Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
  516. Where("pasture_id = ?", userModel.AppPasture.Id).
  517. Where("cow_id > ?", 0).
  518. Where("heat_date BETWEEN ? AND ?", starTime, endTime).
  519. Order("heat_date, frameid").Find(&neckRingList).Error; err != nil {
  520. return nil, xerr.WithStack(err)
  521. }
  522. return &pasturePb.CowLactCurveResponse{
  523. Code: http.StatusOK,
  524. Msg: "ok",
  525. Data: data,
  526. }, nil
  527. }
  528. func (s *StoreEntry) BehaviorRate(ctx context.Context, req *pasturePb.CowBehaviorRateRequest) (*pasturePb.CowBehaviorRateResponse, error) {
  529. userModel, err := s.GetUserModel(ctx)
  530. if err != nil {
  531. return nil, xerr.WithStack(err)
  532. }
  533. cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  534. if err != nil {
  535. return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
  536. }
  537. if req.EndAt <= 0 || req.StartAt <= 0 || req.EndAt < req.StartAt {
  538. return nil, xerr.Customf("时间范围错误")
  539. }
  540. t1 := time.Unix(int64(req.StartAt), 0).Local().Format(model.LayoutDate2)
  541. t2 := time.Unix(int64(req.EndAt), 0).Local().Format(model.LayoutDate2)
  542. dataBetween, err := util.GetDaysBetween(t1, t2)
  543. if err != nil {
  544. return nil, xerr.WithStack(err)
  545. }
  546. neckActiveHabitList := make([]*model.NeckActiveHabit, 0)
  547. if err = s.DB.Model(new(model.NeckActiveHabit)).
  548. Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
  549. Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
  550. Where("pasture_id = ?", userModel.AppPasture.Id).
  551. Where("cow_id > ?", 0).
  552. Where("heat_date BETWEEN ? AND ?", t1, t2).
  553. Order("heat_date").
  554. Group("heat_date").
  555. Find(&neckActiveHabitList).Error; err != nil {
  556. }
  557. cowList := make([]*model.Cow, 0)
  558. if err = s.DB.Model(new(model.Cow)).
  559. Where("neck_ring_number != ?", "").
  560. Where("pasture_id = ?", userModel.AppPasture.Id).
  561. Where("pen_id = ?", cowInfo.PenId).
  562. Find(&cowList).Error; err != nil {
  563. }
  564. neckRingNumbers := make([]string, 0)
  565. for _, v := range cowList {
  566. neckRingNumbers = append(neckRingNumbers, v.NeckRingNumber)
  567. }
  568. groupNeckActiveHabitList := make([]*model.NeckActiveHabit, 0)
  569. if err = s.DB.Model(new(model.NeckActiveHabit)).
  570. Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
  571. Where("neck_ring_number IN ?", neckRingNumbers).
  572. Where("pasture_id = ?", userModel.AppPasture.Id).
  573. Where("cow_id > ?", 0).
  574. Where("heat_date BETWEEN ? AND ?", t1, t2).
  575. Order("heat_date").
  576. Group("heat_date").
  577. Find(&groupNeckActiveHabitList).Error; err != nil {
  578. }
  579. return &pasturePb.CowBehaviorRateResponse{
  580. Code: http.StatusOK,
  581. Msg: "ok",
  582. Data: model.NeckActiveHabitSlice(neckActiveHabitList).ToPB2(dataBetween, groupNeckActiveHabitList),
  583. }, nil
  584. }