kpt-pasture/model/neck_ring_estrus_warning.go

package model

import (
	"kpt-pasture/util"
	"time"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
)

type NeckRingEstrusWarning struct {
	Id               int64                      `json:"id"`
	NeckRingEstrusId int64                      `json:"neckRingEstrusId"`
	PastureId        int64                      `json:"pastureId"`
	CowId            int64                      `json:"cowId"`
	EarNumber        string                     `json:"earNumber"`
	NeckRingNumber   string                     `json:"neckRingNumber"`
	FirstTime        string                     `json:"firstTime"`
	DateTime         string                     `json:"dateTime"`
	LastTime         string                     `json:"lastTime"`
	IsPeak           pasturePb.IsShow_Kind      `json:"isPeak"`
	WarningKind      pasturePb.Warning_Kind     `json:"warningKind"`
	Level            pasturePb.EstrusLevel_Kind `json:"level"`
	HighChange       string                     `json:"highChange"`
	IsShow           pasturePb.IsShow_Kind      `json:"isShow"`
	CreatedAt        int64                      `json:"createdAt"`
	UpdatedAt        int64                      `json:"updatedAt"`
}

func (n *NeckRingEstrusWarning) TableName() string {
	return "neck_ring_estrus_warning"
}

func NewNeckRingEstrusWarning(
	neckRingEstrusId, pastureId, cowId int64,
	earNumber, neckRingNumber, firstTime, dateTime, lastTime string,
	warningKind pasturePb.Warning_Kind,
	level pasturePb.EstrusLevel_Kind,
) *NeckRingEstrusWarning {
	return &NeckRingEstrusWarning{
		NeckRingEstrusId: neckRingEstrusId,
		PastureId:        pastureId,
		CowId:            cowId,
		EarNumber:        earNumber,
		NeckRingNumber:   neckRingNumber,
		FirstTime:        firstTime,
		DateTime:         dateTime,
		LastTime:         lastTime,
		WarningKind:      warningKind,
		Level:            level,
		IsShow:           pasturePb.IsShow_Ok,
	}
}

// CalculatePzHour 计算最佳配置时间
func (n *NeckRingEstrusWarning) CalculatePzHour(lact int32) time.Time {
	var pzHour time.Time
	dateTime, _ := util.TimeParseLocal(LayoutTime, n.DateTime)
	firstTime, _ := util.TimeParseLocal(LayoutTime, n.FirstTime)

	// 条件判断
	if n.IsPeak == pasturePb.IsShow_Ok || dateTime.Sub(firstTime).Hours() >= 8 {
		pzHour = dateTime.Add(8 * time.Hour) // v.datetime + INTERVAL 8 HOUR
	} else {
		pzHour = firstTime.Add(16 * time.Hour) // v.firsttime + INTERVAL 16 HOUR
	}
	// 胎次调整
	if lact >= 3 {
		pzHour = pzHour.Add(-1 * time.Hour) // 减去 1 小时
	}
	return pzHour
}

type NeckRingEstrusWarningSlice []*NeckRingEstrusWarning

func (n NeckRingEstrusWarningSlice) ToPB(
	cowMap map[int64]*Cow,
	eventLogMap map[int64]string,
	matingWindowPeriodKind pasturePb.MatingWindowPeriod_Kind,
	matingWindowPeriodKindList []pasturePb.MatingWindowPeriod_Kind,
) []*pasturePb.EstrusItem {
	res := make([]*pasturePb.EstrusItem, 0)
	nowTime := time.Now().Local()
	for _, v := range n {
		cow, ok := cowMap[v.CowId]
		if !ok {
			cow = &Cow{Id: v.CowId}
		}
		lastBreedEventDetails := ""
		desc, ok := eventLogMap[cow.Id]
		if ok {
			lastBreedEventDetails = desc
		}

		pzHour := v.CalculatePzHour(cow.Lact)
		estrusInterval := int32(0)
		if v.LastTime != "" {
			lastTime, _ := util.TimeParseLocal(LayoutTime, v.LastTime)
			diff := pzHour.Sub(lastTime)
			estrusInterval = int32(diff.Hours() / 24)
		}

		optimumMatingStartTime := pzHour.Add(-4 * time.Hour)
		optimumMatingEndTime := pzHour.Add(4 * time.Hour)

		found := false
		if len(matingWindowPeriodKindList) > 0 && len(matingWindowPeriodKindList) <= 2 {
			for _, periodKind := range matingWindowPeriodKindList {
				if isIPeriod(periodKind, nowTime, optimumMatingStartTime, optimumMatingEndTime) {
					found = true
					break
				}
			}
		}

		if matingWindowPeriodKind > pasturePb.MatingWindowPeriod_Invalid {
			found = isIPeriod(matingWindowPeriodKind, nowTime, optimumMatingStartTime, optimumMatingEndTime)
		}

		if found {
			continue
		}

		firstTimeParseLocal, _ := util.TimeParseLocal(LayoutTime, v.FirstTime)
		afterAlarmTimeForHours := int32(nowTime.Sub(firstTimeParseLocal).Hours())
		postPeakTimeForHours := int32(0)
		if v.IsPeak == pasturePb.IsShow_Ok {
			dateTimeParseLocal, _ := util.TimeParseLocal(LayoutTime, v.DateTime)
			postPeakTimeForHours = int32(nowTime.Sub(dateTimeParseLocal).Hours())
		}

		data := &pasturePb.EstrusItem{
			Id:                     int32(v.Id),
			CowId:                  int32(v.CowId),
			EarNumber:              v.EarNumber,
			PenId:                  cow.PenId,
			PenName:                cow.PenName,
			MatingTimes:            cow.MatingTimes,
			Lact:                   cow.Lact,
			CalvingAge:             cow.CalvingAge,
			AbortionAge:            cow.AbortionAge,
			OptimumMatingStartTime: optimumMatingStartTime.Format(LayoutTime),
			OptimumMatingEndTime:   optimumMatingEndTime.Format(LayoutTime),
			LastBreedEventDetails:  lastBreedEventDetails,
			Level:                  v.Level,
			EstrusInterval:         estrusInterval,
			BestMatingTime:         pzHour.Format(LayoutTime),
			EstrusStartTime:        pzHour.Add(-8 * time.Hour).Format(LayoutTime),
			EstrusEndTime:          pzHour.Add(8 * time.Hour).Format(LayoutTime),
			AfterAlarmTimeForHours: afterAlarmTimeForHours,
			PostPeakTimeForHours:   postPeakTimeForHours,
		}
		res = append(res, data)
	}
	return res
}

func isIPeriod(periodKind pasturePb.MatingWindowPeriod_Kind, nowTime, optimumMatingStartTime, optimumMatingEndTime time.Time) bool {
	switch periodKind {
	case pasturePb.MatingWindowPeriod_Front:
		return !nowTime.Before(optimumMatingStartTime)
	case pasturePb.MatingWindowPeriod_Middle:
		return !(nowTime.After(optimumMatingStartTime) && nowTime.Before(optimumMatingEndTime))
	case pasturePb.MatingWindowPeriod_Behind:
		return !(nowTime.After(optimumMatingEndTime))
	default:
		return false
	}
}

const (
	Nb1    = 12
	Nb2    = 8
	MinNb1 = 10
)

type EstrusWarning struct {
	NeckRingEstrusId int64  `json:"neckRingEstrusId"`
	HighChange       string `json:"highChange"`
	CowId            int64  `json:"cowId"`
	DateTime         string `json:"dateTime"`
	Nb1              int32  `json:"nb1"`
	Nb2              int32  `json:"nb2"`
}