cow.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670
  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 = model.LowActivity
  222. data.MiddleActivity = model.MiddleActivity
  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.NeckRingEstrus, 0)
  255. if err = s.DB.Model(new(model.NeckRingEstrus)).
  256. Select("max(id) as id,active_level,active_time,is_peak").
  257. Where("cow_id = ?", cowInfo.Id).
  258. Where("pasture_id = ?", userModel.AppPasture.Id).
  259. Where("active_time BETWEEN ? AND ?", fmt.Sprintf("%s 00:00:00", startDataTime), fmt.Sprintf("%s 23:59:59", endDataTime)).
  260. Where("is_peak = ?", pasturePb.IsShow_Ok).
  261. Find(&estrusList).Error; err != nil {
  262. return nil, xerr.WithStack(err)
  263. }
  264. for _, v := range estrusList {
  265. if data.EstrusList[v.ActiveLevel] == nil {
  266. data.EstrusList[v.ActiveLevel] = make([]string, 0)
  267. }
  268. data.EstrusList[v.ActiveLevel] = append(data.EstrusList[v.ActiveLevel], strings.TrimSuffix(v.ActiveTime, ":00:00"))
  269. }
  270. return &model.CowBehaviorCurveResponse{
  271. Code: http.StatusOK,
  272. Msg: "ok",
  273. Data: data,
  274. }, nil
  275. }
  276. func (s *StoreEntry) CowIQR(cowInfo *model.Cow, curveName string, dayRange []string, dateTimeList []string) ([]int32, []int32) {
  277. q1, q3 := make([]int32, 0), make([]int32, 0)
  278. if curveName == "" || curveName == "active" || curveName == "behavior" || len(dateTimeList) <= 0 {
  279. return q1, q3
  280. }
  281. penId := cowInfo.PenId
  282. cowList := make([]*model.Cow, 0)
  283. if err := s.DB.Table(new(model.Cow).TableName()).
  284. Where("pen_id = ?", penId).
  285. Where("pasture_id = ?", cowInfo.PastureId).
  286. Where("neck_ring_number != ?", "").
  287. Find(&cowList).Error; err != nil {
  288. return q1, q3
  289. }
  290. cowIds := make([]int64, 0)
  291. for _, cow := range cowList {
  292. cowIds = append(cowIds, cow.Id)
  293. }
  294. neckActiveHabitList, err := s.GetNeckActiveHabitsConcurrent(cowInfo, cowIds, dayRange)
  295. if err != nil {
  296. return q1, q3
  297. }
  298. neckActiveHabitMap := make(map[string][]*model.NeckActiveHabit)
  299. for _, habit := range neckActiveHabitList {
  300. activeTime := habit.ActiveTime[:13]
  301. neckActiveHabitMap[activeTime] = append(neckActiveHabitMap[activeTime], habit)
  302. }
  303. penOriginal := make(map[string][]int32)
  304. for _, dt := range dateTimeList {
  305. habits, ok := neckActiveHabitMap[dt]
  306. if !ok {
  307. penOriginal[dt] = append(penOriginal[dt], int32(0))
  308. continue
  309. }
  310. switch curveName {
  311. case "rumina":
  312. for _, habit := range habits {
  313. penOriginal[dt] = append(penOriginal[dt], habit.Rumina)
  314. }
  315. case "intake":
  316. for _, habit := range habits {
  317. penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
  318. }
  319. case "inactive":
  320. for _, habit := range habits {
  321. penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
  322. }
  323. case "chew":
  324. for _, habit := range habits {
  325. penOriginal[dt] = append(penOriginal[dt], habit.Rumina+habit.Intake)
  326. }
  327. case "immobility":
  328. for _, habit := range habits {
  329. penOriginal[dt] = append(penOriginal[dt], 120-habit.Active)
  330. }
  331. }
  332. }
  333. if len(penOriginal) > 0 {
  334. for _, dataList := range penOriginal {
  335. dl := len(dataList)
  336. if dl < 4 {
  337. q1 = append(q1, int32(0))
  338. q3 = append(q3, int32(0))
  339. continue
  340. }
  341. sort.Slice(dataList, func(i, j int) bool {
  342. return dataList[i] < dataList[j]
  343. })
  344. v1 := (len(dataList) + 1) / 4
  345. // 防止下标溢出panic
  346. s1 := float64(dataList[v1-1]) + 0.25*float64(dataList[v1]-dataList[v1-1])
  347. v3 := v1 * 3
  348. s3 := float64(dataList[v3-1]) + 0.75*float64(dataList[v3]-dataList[v3-1])
  349. q1 = append(q1, int32(s1))
  350. q3 = append(q3, int32(s3))
  351. }
  352. }
  353. return q1, q3
  354. }
  355. func (s *StoreEntry) GetNeckActiveHabitsConcurrent(cowInfo *model.Cow, cowIds []int64, dayRange []string) ([]*model.NeckActiveHabit, error) {
  356. const (
  357. batchSize = 100
  358. workerCount = 5
  359. )
  360. resultsChan := make(chan []*model.NeckActiveHabit)
  361. errChan := make(chan error, 1) // 缓冲 1,避免 goroutine 阻塞
  362. doneChan := make(chan struct{})
  363. var (
  364. neckActiveHabitList []*model.NeckActiveHabit
  365. mu sync.Mutex
  366. )
  367. // 结果收集 goroutine
  368. go func() {
  369. defer close(doneChan)
  370. for batchResults := range resultsChan {
  371. mu.Lock()
  372. neckActiveHabitList = append(neckActiveHabitList, batchResults...)
  373. mu.Unlock()
  374. }
  375. }()
  376. sem := make(chan struct{}, workerCount)
  377. var wg sync.WaitGroup
  378. // 分发任务
  379. go func() {
  380. defer func() {
  381. wg.Wait() // 等待所有 goroutine 完成
  382. close(resultsChan) // 安全关闭
  383. }()
  384. for i := 0; i < len(cowIds); i += batchSize {
  385. end := i + batchSize
  386. if end > len(cowIds) {
  387. end = len(cowIds)
  388. }
  389. batch := cowIds[i:end]
  390. wg.Add(1)
  391. sem <- struct{}{} // 获取令牌(可能阻塞,但不会死锁)
  392. go func(batch []int64) {
  393. defer func() {
  394. <-sem // 释放令牌
  395. wg.Done()
  396. }()
  397. var batchResults []*model.NeckActiveHabit
  398. if err := s.DB.Table(new(model.NeckActiveHabit).TableName()).
  399. Select("rumina,intake,inactive,gasp,high,active,active_time").
  400. Where("pasture_id = ?", cowInfo.PastureId).
  401. Where("is_show = ?", pasturePb.IsShow_Ok).
  402. Where("cow_id IN (?)", batch).
  403. Where("heat_date BETWEEN ? AND ?", dayRange[0], dayRange[len(dayRange)-1]).
  404. Find(&batchResults).Error; err != nil {
  405. select {
  406. case errChan <- fmt.Errorf("batch query error: %v", err):
  407. default:
  408. }
  409. return
  410. }
  411. if len(batchResults) > 0 {
  412. resultsChan <- batchResults
  413. }
  414. }(batch)
  415. }
  416. }()
  417. // 等待结果
  418. select {
  419. case err := <-errChan:
  420. return nil, err
  421. case <-doneChan:
  422. return neckActiveHabitList, nil
  423. }
  424. }
  425. func (s *StoreEntry) CowGrowthCurve(ctx context.Context, req *pasturePb.CowGrowthCurveRequest) (*pasturePb.CowGrowthCurveResponse, error) {
  426. userModel, err := s.GetUserModel(ctx)
  427. if err != nil {
  428. return nil, xerr.WithStack(err)
  429. }
  430. cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  431. if err != nil {
  432. return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
  433. }
  434. weightList := make([]*model.EventWeight, 0)
  435. if err = s.DB.Table(new(model.EventWeight).TableName()).
  436. Where("cow_id = ?", cowInfo.Id).
  437. Where("pasture_id = ?", userModel.AppPasture.Id).
  438. Order("weight_at").
  439. Find(&weightList).Error; err != nil {
  440. return nil, xerr.WithStack(err)
  441. }
  442. eventCowLogList := make([]*model.EventCowLog, 0)
  443. eventCowLog := &model.EventCowLog{CowId: cowInfo.Id}
  444. if err = s.DB.Table(eventCowLog.TableName()).
  445. Where("pasture_id = ?", userModel.AppPasture.Id).
  446. Where("cow_id = ?", cowInfo.Id).
  447. Order("id desc").
  448. Find(&eventCowLogList).Error; err != nil {
  449. return nil, xerr.WithStack(err)
  450. }
  451. return &pasturePb.CowGrowthCurveResponse{
  452. Code: http.StatusOK,
  453. Msg: "ok",
  454. Data: model.EventWeightSlice(weightList).ToPB(cowInfo, eventCowLogList),
  455. }, nil
  456. }
  457. func (s *StoreEntry) CowLactCurve(ctx context.Context, req *pasturePb.CowLactCurveRequest) (*pasturePb.CowLactCurveResponse, error) {
  458. userModel, err := s.GetUserModel(ctx)
  459. if err != nil {
  460. return nil, xerr.WithStack(err)
  461. }
  462. cowInfo, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  463. if err != nil {
  464. return nil, xerr.Customf("错误的牛只信息: %d", req.CowId)
  465. }
  466. cowLactList := make([]*model.CowLact, 0)
  467. if err = s.DB.Table(new(model.CowLact).TableName()).
  468. Where("cow_id = ?", req.CowId).
  469. Where("pasture_id = ?", userModel.AppPasture.Id).
  470. Order("lact").
  471. Find(&cowLactList).Error; err != nil {
  472. return nil, xerr.WithStack(err)
  473. }
  474. data := &pasturePb.CowLactCurveData{
  475. DateTime: make(map[int32]string),
  476. WeekAvgMilk: make([]float32, 0),
  477. DayMilk: make([]float32, 0),
  478. DHI: make([]float32, 0),
  479. MilkProductionTrend: make([]float32, 0),
  480. DayHigh: make([]int32, 0),
  481. DayRumina: make([]int32, 0),
  482. DayIntake: make([]int32, 0),
  483. DayInactive: make([]int32, 0),
  484. DayChew: make([]int32, 0),
  485. DayImmobility: make([]int32, 0),
  486. EstrusWarning: make(map[int32]int32),
  487. }
  488. if len(cowLactList) <= 0 {
  489. return &pasturePb.CowLactCurveResponse{
  490. Code: http.StatusOK,
  491. Msg: "ok",
  492. Data: data,
  493. }, nil
  494. }
  495. starTime := ""
  496. endTime := ""
  497. cowLactMap := make(map[int32]*model.CowLact)
  498. for _, v := range cowLactList {
  499. cowLactMap[v.Lact] = v
  500. if v.Lact == req.Lact {
  501. starTime = v.StartTime
  502. }
  503. }
  504. if st, ok := cowLactMap[req.Lact+1]; ok {
  505. et, _ := util.TimeParseLocal(model.LayoutDate2, st.StartTime)
  506. endTime = et.AddDate(0, 0, -1).Format(model.LayoutDate2)
  507. } else {
  508. endTime = time.Now().Local().Format(model.LayoutDate2)
  509. }
  510. neckRingList := make([]*model.NeckActiveHabit, 0)
  511. if err = s.DB.Model(new(model.NeckActiveHabit)).
  512. Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
  513. Where("pasture_id = ?", userModel.AppPasture.Id).
  514. Where("cow_id > ?", 0).
  515. Where("heat_date BETWEEN ? AND ?", starTime, endTime).
  516. Order("heat_date, frameid").Find(&neckRingList).Error; err != nil {
  517. return nil, xerr.WithStack(err)
  518. }
  519. return &pasturePb.CowLactCurveResponse{
  520. Code: http.StatusOK,
  521. Msg: "ok",
  522. Data: data,
  523. }, nil
  524. }
  525. func (s *StoreEntry) BehaviorRate(ctx context.Context, req *pasturePb.CowBehaviorRateRequest) (*pasturePb.CowBehaviorRateResponse, error) {
  526. userModel, err := s.GetUserModel(ctx)
  527. if err != nil {
  528. return nil, xerr.WithStack(err)
  529. }
  530. cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  531. if err != nil {
  532. return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
  533. }
  534. if req.EndAt <= 0 || req.StartAt <= 0 || req.EndAt < req.StartAt {
  535. return nil, xerr.Customf("时间范围错误")
  536. }
  537. t1 := time.Unix(int64(req.StartAt), 0).Local().Format(model.LayoutDate2)
  538. t2 := time.Unix(int64(req.EndAt), 0).Local().Format(model.LayoutDate2)
  539. dataBetween, err := util.GetDaysBetween(t1, t2)
  540. if err != nil {
  541. return nil, xerr.WithStack(err)
  542. }
  543. neckActiveHabitList := make([]*model.NeckActiveHabit, 0)
  544. if err = s.DB.Model(new(model.NeckActiveHabit)).
  545. Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
  546. Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
  547. Where("pasture_id = ?", userModel.AppPasture.Id).
  548. Where("cow_id > ?", 0).
  549. Where("heat_date BETWEEN ? AND ?", t1, t2).
  550. Order("heat_date").
  551. Group("heat_date").
  552. Find(&neckActiveHabitList).Error; err != nil {
  553. }
  554. cowList := make([]*model.Cow, 0)
  555. if err = s.DB.Model(new(model.Cow)).
  556. Where("neck_ring_number != ?", "").
  557. Where("pasture_id = ?", userModel.AppPasture.Id).
  558. Where("pen_id = ?", cowInfo.PenId).
  559. Find(&cowList).Error; err != nil {
  560. }
  561. neckRingNumbers := make([]string, 0)
  562. for _, v := range cowList {
  563. neckRingNumbers = append(neckRingNumbers, v.NeckRingNumber)
  564. }
  565. groupNeckActiveHabitList := make([]*model.NeckActiveHabit, 0)
  566. if err = s.DB.Model(new(model.NeckActiveHabit)).
  567. Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
  568. Where("neck_ring_number IN ?", neckRingNumbers).
  569. Where("pasture_id = ?", userModel.AppPasture.Id).
  570. Where("cow_id > ?", 0).
  571. Where("heat_date BETWEEN ? AND ?", t1, t2).
  572. Order("heat_date").
  573. Group("heat_date").
  574. Find(&groupNeckActiveHabitList).Error; err != nil {
  575. }
  576. return &pasturePb.CowBehaviorRateResponse{
  577. Code: http.StatusOK,
  578. Msg: "ok",
  579. Data: model.NeckActiveHabitSlice(neckActiveHabitList).ToPB2(dataBetween, groupNeckActiveHabitList),
  580. }, nil
  581. }