cow.go 21 KB

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