123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674 |
- package backend
- import (
- "context"
- "errors"
- "fmt"
- "kpt-pasture/model"
- "kpt-pasture/util"
- "net/http"
- "sort"
- "strings"
- "sync"
- "time"
- "gorm.io/gorm"
- "gitee.com/xuyiping_admin/pkg/xerr"
- pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
- )
- func (s *StoreEntry) Detail(ctx context.Context, req *pasturePb.SearchEventRequest) (*pasturePb.CowInfoResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- if req.EarNumber == "" && req.NeckRingNumber == "" {
- return nil, xerr.Custom("请输入牛号或项圈号")
- }
- cowInfo := &model.Cow{}
- pref := s.DB.Model(new(model.Cow)).
- Where("pasture_id = ?", userModel.AppPasture.Id)
- if req.EarNumber != "" {
- pref.Where("ear_number = ?", req.EarNumber)
- }
- if req.NeckRingNumber != "" {
- pref.Where("neck_ring_number = ?", req.NeckRingNumber)
- }
- if err = pref.Order("id desc").
- First(cowInfo).Error; err != nil {
- if errors.Is(err, gorm.ErrRecordNotFound) {
- return nil, xerr.Custom("该牛只未找到")
- } else {
- return nil, xerr.WithStack(err)
- }
- }
- cowTypeMap := s.CowTypeMap()
- breedStatusMap := s.CowBreedStatusMap()
- cowKindMap := s.CowKindMap()
- cowSourceMap := s.CowSourceMap()
- admissionStatusMap := s.AdmissionStatusMap()
- healthStatusMap := s.HealthStatusMap()
- purposeMap := s.PurposeMap()
- systemBasic, err := s.GetSystemBasicByName(ctx, userModel.AppPasture.Id, model.PregnancyAge)
- if err != nil {
- return nil, xerr.Custom("请在基础参数配置妊娠天数")
- }
- cowDetails := model.CowSlice([]*model.Cow{cowInfo}).ToPB(
- cowTypeMap, breedStatusMap, cowKindMap, cowSourceMap,
- admissionStatusMap, healthStatusMap, purposeMap, systemBasic.MinValue,
- )
- if len(cowDetails) != 1 {
- return nil, xerr.Custom("该牛只未找到")
- }
- data := cowDetails[0]
- return &pasturePb.CowInfoResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: data,
- }, nil
- }
- func (s *StoreEntry) List(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchCowListResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- systemBasic, err := s.GetSystemBasicByName(ctx, userModel.AppPasture.Id, model.PregnancyAge)
- if err != nil {
- return nil, xerr.Custom("请在基础参数配置妊娠天数")
- }
- cowList := make([]*model.Cow, 0)
- var count int64 = 0
- pref := s.DB.Model(new(model.Cow)).
- Where("pasture_id = ?", userModel.AppPasture.Id)
- if len(req.CowId) > 0 {
- cowIds := strings.Split(req.CowId, ",")
- pref.Where("id IN ?", cowIds)
- }
- if req.EarNumber != "" {
- pref.Where("ear_number like ?", fmt.Sprintf("%s%s%s", "%", req.EarNumber, "%"))
- }
- if req.Id > 0 {
- pref.Where("id = ?", req.Id)
- }
- if req.PenId > 0 {
- pref.Where("pen_id = ?", req.PenId)
- }
- if req.CowType > 0 {
- pref.Where("cow_type = ?", req.CowType)
- }
- if req.BreedStatus > 0 {
- pref.Where("breed_status = ?", req.BreedStatus)
- }
- if req.CowKind > 0 {
- pref.Where("cow_kind = ?", req.CowKind)
- }
- if req.Sex > 0 {
- pref.Where("sex = ?", req.Sex)
- }
- if req.Lact > 0 {
- pref.Where("lact = ?", req.Lact)
- }
- if req.CowSource > 0 {
- pref.Where("source_id = ?", req.CowSource)
- }
- if err = pref.Order("id desc").
- Count(&count).
- Limit(int(pagination.PageSize)).
- Offset(int(pagination.PageOffset)).
- Find(&cowList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- cowTypeMap := s.CowTypeMap()
- breedStatusMap := s.CowBreedStatusMap()
- cowKindMap := s.CowKindMap()
- cowSourceMap := s.CowSourceMap()
- admissionStatusMap := s.AdmissionStatusMap()
- healthStatusMap := s.HealthStatusMap()
- purposeMap := s.PurposeMap()
- return &pasturePb.SearchCowListResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: &pasturePb.SearchCowData{
- List: model.CowSlice(cowList).ToPB(
- cowTypeMap, breedStatusMap, cowKindMap, cowSourceMap,
- admissionStatusMap, healthStatusMap, purposeMap, systemBasic.MinValue,
- ),
- Total: int32(count),
- PageSize: pagination.PageSize,
- Page: pagination.Page,
- },
- }, nil
- }
- func (s *StoreEntry) EventList(ctx context.Context, req *pasturePb.SearchCowEventListRequest, pagination *pasturePb.PaginationModel) (*pasturePb.CowEventListResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- eventCowLogList := make([]*model.EventCowLog, 0)
- cowInfo, err := s.GetCowEventByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
- if err != nil {
- return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
- }
- eventCowLog := &model.EventCowLog{CowId: cowInfo.Id}
- pref := s.DB.Table(eventCowLog.TableName()).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("cow_id = ?", cowInfo.Id)
- if req.Lact >= 0 {
- pref.Where("lact = ?", req.Lact)
- }
- if req.EventCategoryKind > 0 {
- pref.Where("event_type = ?", req.EventCategoryKind)
- }
- if err = pref.Order("event_at DESC").
- Limit(int(pagination.PageSize)).
- Offset(int(pagination.PageOffset)).
- Find(&eventCowLogList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- eventCategoryMap := s.EventCategoryMap()
- return &pasturePb.CowEventListResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: &pasturePb.CowEventData{
- List: model.EventCowLogSlice(eventCowLogList).ToPB(eventCategoryMap),
- Total: int32(len(eventCowLogList)),
- PageSize: pagination.PageSize,
- Page: pagination.Page,
- },
- }, nil
- }
- func (s *StoreEntry) BehaviorCurve(ctx context.Context, req *pasturePb.CowBehaviorCurveRequest) (*model.CowBehaviorCurveResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- cowInfo, err := s.GetCowEventByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
- if err != nil {
- return nil, xerr.Customf("错误的牛只信息: %d", req.CowId)
- }
- nowTime := time.Now().Local()
- nowDayZero := util.TimeParseLocalUnix(nowTime.Format(model.LayoutDate2))
- endDataTime := nowTime.Format(model.LayoutDate2)
- startDataTime := nowTime.AddDate(0, 0, -30).Format(model.LayoutDate2)
- dayRange, err := util.GetDaysBetween(startDataTime, endDataTime)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- if len(dayRange) <= 0 {
- return nil, xerr.Customf("错误的日期范围")
- }
- // 行为曲线数据
- neckActiveHabitList := make([]*model.NeckActiveHabit, 0)
- if err = s.DB.Table(new(model.NeckActiveHabit).TableName()).
- Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("is_show = ?", pasturePb.IsShow_Ok).
- Where("cow_id > ?", 0).
- Where("heat_date IN (?)", dayRange).
- Order("heat_date, frameid").
- Find(&neckActiveHabitList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- data := model.NeckActiveHabitSlice(neckActiveHabitList).ToPB(req.CurveName)
- q1, q3 := s.CowIQR(cowInfo, req.CurveName, dayRange, data.DateTimeList)
- data.IQR3 = q3
- data.IQR1 = q1
- eventMapList := s.EventTypeMap()
- for k, v := range eventMapList {
- if k == pasturePb.EventType_Enter || k == pasturePb.EventType_Body_Score || k == pasturePb.EventType_Birth ||
- k == pasturePb.EventType_Weaning || k == pasturePb.EventType_Sale || k == pasturePb.EventType_Weight ||
- k == pasturePb.EventType_Castrated {
- continue
- }
- data.EventMap[k] = v
- }
- data.LowActivity = 62
- data.MiddleActivity = 80
- // 牛只事件列表
- eventLogList := make([]*model.EventCowLog, 0)
- eventLog := &model.EventCowLog{CowId: cowInfo.Id}
- if err = s.DB.Table(eventLog.TableName()).
- Where("cow_id = ?", cowInfo.Id).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("event_at BETWEEN ? AND ?", nowDayZero-(30*86400), nowDayZero+86400).
- Order("event_at").
- Find(&eventLogList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- for _, v := range eventLogList {
- if v.EventType == pasturePb.EventType_Enter || v.EventType == pasturePb.EventType_Body_Score ||
- v.EventType == pasturePb.EventType_Birth || v.EventType == pasturePb.EventType_Weaning ||
- v.EventType == pasturePb.EventType_Sale || v.EventType == pasturePb.EventType_Weight ||
- v.EventType == pasturePb.EventType_Castrated {
- continue
- }
- if v.EventAt <= 0 {
- continue
- }
- eventAt := time.Unix(v.EventAt, 0).Local()
- data.EventList = append(data.EventList, &pasturePb.CowEvent{
- EventTypeKind: v.EventType,
- EventTypeName: v.EventTypeName,
- EventDescription: v.EventDescription,
- Remarks: v.Remarks,
- EventAtFormat: fmt.Sprintf("%s 09", eventAt.Format(model.LayoutDate2)),
- })
- }
- // 发情数据
- estrusList := make([]*model.EventEstrus, 0)
- if err = s.DB.Table(new(model.EventEstrus).TableName()).
- Where("cow_id = ?", cowInfo.Id).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("reality_day BETWEEN ? AND ?", startDataTime, endDataTime).
- Order("reality_day").
- Find(&estrusList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- for _, v := range estrusList {
- if data.EstrusList[v.Level] == nil {
- data.EstrusList[v.Level] = make([]string, 0)
- }
- if v.RealityDay > 0 {
- // 格式化为到小时的字符串
- hourStr := time.Unix(v.RealityDay, 0).Local().Format(model.LayoutHour)
- data.EstrusList[v.Level] = append(data.EstrusList[v.Level], hourStr)
- }
- }
- return &model.CowBehaviorCurveResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: data,
- }, nil
- }
- func (s *StoreEntry) CowIQR(cowInfo *model.Cow, curveName string, dayRange []string, dateTimeList []string) ([]int32, []int32) {
- q1, q3 := make([]int32, 0), make([]int32, 0)
- if curveName == "" || curveName == "active" || curveName == "behavior" || len(dateTimeList) <= 0 {
- return q1, q3
- }
- penId := cowInfo.PenId
- cowList := make([]*model.Cow, 0)
- if err := s.DB.Table(new(model.Cow).TableName()).
- Where("pen_id = ?", penId).
- Where("pasture_id = ?", cowInfo.PastureId).
- Where("neck_ring_number != ?", "").
- Find(&cowList).Error; err != nil {
- return q1, q3
- }
- cowIds := make([]int64, 0)
- for _, cow := range cowList {
- cowIds = append(cowIds, cow.Id)
- }
- neckActiveHabitList, err := s.GetNeckActiveHabitsConcurrent(cowInfo, cowIds, dayRange)
- if err != nil {
- return q1, q3
- }
- neckActiveHabitMap := make(map[string][]*model.NeckActiveHabit)
- for _, habit := range neckActiveHabitList {
- activeTime := habit.ActiveTime[:13]
- neckActiveHabitMap[activeTime] = append(neckActiveHabitMap[activeTime], habit)
- }
- penOriginal := make(map[string][]int32)
- for _, dt := range dateTimeList {
- habits, ok := neckActiveHabitMap[dt]
- if !ok {
- penOriginal[dt] = append(penOriginal[dt], int32(0))
- continue
- }
- switch curveName {
- case "rumina":
- for _, habit := range habits {
- penOriginal[dt] = append(penOriginal[dt], habit.Rumina)
- }
- case "intake":
- for _, habit := range habits {
- penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
- }
- case "inactive":
- for _, habit := range habits {
- penOriginal[dt] = append(penOriginal[dt], habit.Inactive)
- }
- case "chew":
- for _, habit := range habits {
- penOriginal[dt] = append(penOriginal[dt], habit.Rumina+habit.Intake)
- }
- case "immobility":
- for _, habit := range habits {
- penOriginal[dt] = append(penOriginal[dt], 120-habit.Active)
- }
- }
- }
- if len(penOriginal) > 0 {
- for _, dataList := range penOriginal {
- dl := len(dataList)
- if dl < 4 {
- q1 = append(q1, int32(0))
- q3 = append(q3, int32(0))
- continue
- }
- sort.Slice(dataList, func(i, j int) bool {
- return dataList[i] < dataList[j]
- })
- v1 := (len(dataList) + 1) / 4
- // 防止下标溢出panic
- s1 := float64(dataList[v1-1]) + 0.25*float64(dataList[v1]-dataList[v1-1])
- v3 := v1 * 3
- s3 := float64(dataList[v3-1]) + 0.75*float64(dataList[v3]-dataList[v3-1])
- q1 = append(q1, int32(s1))
- q3 = append(q3, int32(s3))
- }
- }
- return q1, q3
- }
- func (s *StoreEntry) GetNeckActiveHabitsConcurrent(cowInfo *model.Cow, cowIds []int64, dayRange []string) ([]*model.NeckActiveHabit, error) {
- const (
- batchSize = 100
- workerCount = 5
- )
- resultsChan := make(chan []*model.NeckActiveHabit)
- errChan := make(chan error, 1) // 缓冲 1,避免 goroutine 阻塞
- doneChan := make(chan struct{})
- var (
- neckActiveHabitList []*model.NeckActiveHabit
- mu sync.Mutex
- )
- // 结果收集 goroutine
- go func() {
- defer close(doneChan)
- for batchResults := range resultsChan {
- mu.Lock()
- neckActiveHabitList = append(neckActiveHabitList, batchResults...)
- mu.Unlock()
- }
- }()
- sem := make(chan struct{}, workerCount)
- var wg sync.WaitGroup
- // 分发任务
- go func() {
- defer func() {
- wg.Wait() // 等待所有 goroutine 完成
- close(resultsChan) // 安全关闭
- }()
- for i := 0; i < len(cowIds); i += batchSize {
- end := i + batchSize
- if end > len(cowIds) {
- end = len(cowIds)
- }
- batch := cowIds[i:end]
- wg.Add(1)
- sem <- struct{}{} // 获取令牌(可能阻塞,但不会死锁)
- go func(batch []int64) {
- defer func() {
- <-sem // 释放令牌
- wg.Done()
- }()
- var batchResults []*model.NeckActiveHabit
- if err := s.DB.Table(new(model.NeckActiveHabit).TableName()).
- Select("rumina,intake,inactive,gasp,high,active,active_time").
- Where("pasture_id = ?", cowInfo.PastureId).
- Where("is_show = ?", pasturePb.IsShow_Ok).
- Where("cow_id IN (?)", batch).
- Where("heat_date BETWEEN ? AND ?", dayRange[0], dayRange[len(dayRange)-1]).
- Find(&batchResults).Error; err != nil {
- select {
- case errChan <- fmt.Errorf("batch query error: %v", err):
- default:
- }
- return
- }
- if len(batchResults) > 0 {
- resultsChan <- batchResults
- }
- }(batch)
- }
- }()
- // 等待结果
- select {
- case err := <-errChan:
- return nil, err
- case <-doneChan:
- return neckActiveHabitList, nil
- }
- }
- func (s *StoreEntry) CowGrowthCurve(ctx context.Context, req *pasturePb.CowGrowthCurveRequest) (*pasturePb.CowGrowthCurveResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
- if err != nil {
- return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
- }
- weightList := make([]*model.EventWeight, 0)
- if err = s.DB.Table(new(model.EventWeight).TableName()).
- Where("cow_id = ?", cowInfo.Id).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Order("weight_at").
- Find(&weightList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- eventCowLogList := make([]*model.EventCowLog, 0)
- eventCowLog := &model.EventCowLog{CowId: cowInfo.Id}
- if err = s.DB.Table(eventCowLog.TableName()).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("cow_id = ?", cowInfo.Id).
- Order("id desc").
- Find(&eventCowLogList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- return &pasturePb.CowGrowthCurveResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: model.EventWeightSlice(weightList).ToPB(cowInfo, eventCowLogList),
- }, nil
- }
- func (s *StoreEntry) CowLactCurve(ctx context.Context, req *pasturePb.CowLactCurveRequest) (*pasturePb.CowLactCurveResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- cowInfo, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
- if err != nil {
- return nil, xerr.Customf("错误的牛只信息: %d", req.CowId)
- }
- cowLactList := make([]*model.CowLact, 0)
- if err = s.DB.Table(new(model.CowLact).TableName()).
- Where("cow_id = ?", req.CowId).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Order("lact").
- Find(&cowLactList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- data := &pasturePb.CowLactCurveData{
- DateTime: make(map[int32]string),
- WeekAvgMilk: make([]float32, 0),
- DayMilk: make([]float32, 0),
- DHI: make([]float32, 0),
- MilkProductionTrend: make([]float32, 0),
- DayHigh: make([]int32, 0),
- DayRumina: make([]int32, 0),
- DayIntake: make([]int32, 0),
- DayInactive: make([]int32, 0),
- DayChew: make([]int32, 0),
- DayImmobility: make([]int32, 0),
- EstrusWarning: make(map[int32]int32),
- }
- if len(cowLactList) <= 0 {
- return &pasturePb.CowLactCurveResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: data,
- }, nil
- }
- starTime := ""
- endTime := ""
- cowLactMap := make(map[int32]*model.CowLact)
- for _, v := range cowLactList {
- cowLactMap[v.Lact] = v
- if v.Lact == req.Lact {
- starTime = v.StartTime
- }
- }
- if st, ok := cowLactMap[req.Lact+1]; ok {
- et, _ := util.TimeParseLocal(model.LayoutDate2, st.StartTime)
- endTime = et.AddDate(0, 0, -1).Format(model.LayoutDate2)
- } else {
- endTime = time.Now().Local().Format(model.LayoutDate2)
- }
- neckRingList := make([]*model.NeckActiveHabit, 0)
- if err = s.DB.Model(new(model.NeckActiveHabit)).
- Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("cow_id > ?", 0).
- Where("heat_date BETWEEN ? AND ?", starTime, endTime).
- Order("heat_date, frameid").Find(&neckRingList).Error; err != nil {
- return nil, xerr.WithStack(err)
- }
- return &pasturePb.CowLactCurveResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: data,
- }, nil
- }
- func (s *StoreEntry) BehaviorRate(ctx context.Context, req *pasturePb.CowBehaviorRateRequest) (*pasturePb.CowBehaviorRateResponse, error) {
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
- if err != nil {
- return nil, xerr.Customf("错误的牛只信息: %s", req.EarNumber)
- }
- if req.EndAt <= 0 || req.StartAt <= 0 || req.EndAt < req.StartAt {
- return nil, xerr.Customf("时间范围错误")
- }
- t1 := time.Unix(int64(req.StartAt), 0).Local().Format(model.LayoutDate2)
- t2 := time.Unix(int64(req.EndAt), 0).Local().Format(model.LayoutDate2)
- dataBetween, err := util.GetDaysBetween(t1, t2)
- if err != nil {
- return nil, xerr.WithStack(err)
- }
- neckActiveHabitList := make([]*model.NeckActiveHabit, 0)
- if err = s.DB.Model(new(model.NeckActiveHabit)).
- Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
- Where("neck_ring_number = ?", cowInfo.NeckRingNumber).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("cow_id > ?", 0).
- Where("heat_date BETWEEN ? AND ?", t1, t2).
- Order("heat_date").
- Group("heat_date").
- Find(&neckActiveHabitList).Error; err != nil {
- }
- cowList := make([]*model.Cow, 0)
- if err = s.DB.Model(new(model.Cow)).
- Where("neck_ring_number != ?", "").
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("pen_id = ?", cowInfo.PenId).
- Find(&cowList).Error; err != nil {
- }
- neckRingNumbers := make([]string, 0)
- for _, v := range cowList {
- neckRingNumbers = append(neckRingNumbers, v.NeckRingNumber)
- }
- groupNeckActiveHabitList := make([]*model.NeckActiveHabit, 0)
- if err = s.DB.Model(new(model.NeckActiveHabit)).
- Select("heat_date,SUM(rumina) as rumina,SUM(intake) as intake,SUM(inactive) as inactive,SUM(gasp) AS gasp").
- Where("neck_ring_number IN ?", neckRingNumbers).
- Where("pasture_id = ?", userModel.AppPasture.Id).
- Where("cow_id > ?", 0).
- Where("heat_date BETWEEN ? AND ?", t1, t2).
- Order("heat_date").
- Group("heat_date").
- Find(&groupNeckActiveHabitList).Error; err != nil {
- }
- return &pasturePb.CowBehaviorRateResponse{
- Code: http.StatusOK,
- Msg: "ok",
- Data: model.NeckActiveHabitSlice(neckActiveHabitList).ToPB2(dataBetween, groupNeckActiveHabitList),
- }, nil
- }
|