kpt-pasture/model/event_weight.go

package model

import (
	"time"

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

type EventWeight struct {
	ID            int64  `json:"id"`
	PastureId     int64  `json:"pastureId"`
	CowId         int64  `json:"cowId"`
	EarNumber     string `json:"earNumber"`
	DayAge        int32  `json:"dayAge"`
	LactationDay  int64  `json:"lactationDay"`
	PenId         int32  `json:"penId"`
	Lact          int32  `json:"lact"`
	Weight        int32  `json:"weight"`
	Height        int32  `json:"height"`
	WeightAt      int64  `json:"weightAt"`
	Remarks       string `json:"remarks"`
	OperationId   int32  `json:"operationId"`
	OperationName string `json:"operationName"`
	MessageId     int64  `json:"messageId"`
	MessageName   string `json:"messageName"`
	CreatedAt     int64  `json:"created_at"`
	UpdatedAt     int64  `json:"updated_at"`
}

func (c *EventWeight) TableName() string {
	return "event_weight"
}

func NewEventWeight(cow *Cow, currentUser *SystemUser, weight, height int32, req *pasturePb.EventWeight) *EventWeight {
	return &EventWeight{
		PastureId:     currentUser.PastureId,
		CowId:         cow.Id,
		EarNumber:     cow.EarNumber,
		Weight:        weight,
		Height:        height,
		Lact:          cow.Lact,
		DayAge:        cow.GetDayAge(),
		WeightAt:      int64(req.WeightAt),
		Remarks:       req.Remarks,
		MessageId:     currentUser.Id,
		MessageName:   currentUser.Name,
		OperationId:   req.OperationId,
		OperationName: req.OperationName,
	}
}

type EventWeightSlice []*EventWeight

func (e EventWeightSlice) ToPB() []*pasturePb.CowGrowthCurveData {
	res := make([]*pasturePb.CowGrowthCurveData, len(e))
	for i, v := range e {
		weightAtFormat := ""
		if v.WeightAt > 0 {
			weightAtFormat = time.Unix(v.WeightAt, 0).Format(LayoutTime)
		}

		res[i] = &pasturePb.CowGrowthCurveData{
			Weight:          float32(v.Weight) / 1000,
			WeightAtFormat:  weightAtFormat,
			AvgWeight:       0, // 平均体重
			DayAddWeight:    0, // 日增重
			AvgDayAddWeight: 0, // 平均日增重
			MonthAddWeight:  0, // 月增重
		}
	}
}