kpt-pasture/module/crontab/cow_indicators_breed.go

package crontab

import (
	"fmt"
	"kpt-pasture/model"

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

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"
	"go.uber.org/zap"
)

// FindCalvingNumber 产犊事件总数
func (e *Entry) FindCalvingNumber(pastureList []*model.AppPastureList, startTime, endTime int64) map[int64]string {
	res := make(map[int64]string)
	for _, pasture := range pastureList {
		var count int64
		if err := e.DB.Model(new(model.EventSale)).
			Where("pasture_id = ?", pasture.Id).
			Where("status = ?", pasturePb.IsShow_Ok).
			Where("reality_day BETWEEN ? AND ?", startTime, endTime).
			Count(&count).Error; err != nil {
			zaplog.Error("FindCalvingNumber", zap.Any("pasture_id", pasture.Id), zap.Any("err", err))
		}
		res[pasture.Id] = fmt.Sprintf("%d", count)
	}
	return res
}

// LactationCow 泌乳牛头数 干奶牛头数 后备牛头数
func (e *Entry) LactationCow(pastureList []*model.AppPastureList, milkKind pasturePb.CowMilk_Kind) map[int64]string {
	res := make(map[int64]string)
	for _, pasture := range pastureList {
		var count int64
		if err := e.DB.Model(new(model.Cow)).
			Where("pasture_id = ?", pasture.Id).
			Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
			Where("milk_kind = ?", milkKind).
			Count(&count).Error; err != nil {
			zaplog.Error("LactationCow", zap.Any("pasture_id", pasture.Id), zap.Any("err", err))
		}
		res[pasture.Id] = fmt.Sprintf("%d", count)
	}
	return res
}