kpt-pasture/module/backend/event_check.go

package backend

import (
	"context"
	"errors"
	"kpt-pasture/model"
	"kpt-pasture/util"
	"time"

	"gorm.io/gorm"

	"go.uber.org/zap"

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"

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

// PregnantCheckBatchModel 批量孕检
type PregnantCheckBatchModel struct {
	EventPregnancyCheck *model.EventPregnantCheck
	Cow                 *model.Cow
	OperationUser       *model.SystemUser
	PregnantCheckAt     int32                              // 孕检日期
	PregnantCheckResult pasturePb.PregnantCheckResult_Kind // 孕检结果
	PregnantCheckMethod pasturePb.PregnantCheckMethod_Kind // 孕检方式
	Remarks             string
	LastMating          *model.EventMating
}

// MatingTimes 更新配次
type MatingTimes struct {
	Mt            int32
	CowId         int64
	EventMatingId int64
}

type AbortionCheckBatchModel struct {
	EventAbortion *model.EventAbortion
	Cow           *model.Cow
	OperationUser *model.SystemUser
	IsLact        pasturePb.IsShow_Kind
}

func (s *StoreEntry) EnterCheck(ctx context.Context, pastureId int64, req *pasturePb.EventEnterRequest) error {
	var count int64
	if err := s.DB.Model(new(model.Cow)).
		Where("ear_number = ?", req.EarNumber).
		Where("pasture_id = ?", pastureId).
		Count(&count).Error; err != nil {
		return xerr.WithStack(err)
	}
	if count > 0 {
		return xerr.Custom("该牛只已存在")
	}

	if req.BirthAt > req.EnterAt {
		return xerr.Custom("出生时间不能大于入场时间")
	}

	return nil
}

func (s *StoreEntry) MatingCreateCheck(ctx context.Context, pastureId int64, req *pasturePb.EventMatingBatch) ([]*model.EventMatingCheckBatchModel, error) {
	if len(req.Items) <= 0 {
		return nil, xerr.Custom("请选择相关牛只")
	}

	if len(req.Items) > 50 {
		return nil, xerr.Custom("最多只能选择50只牛只")
	}
	eventMatingCheckBatchModelList := make([]*model.EventMatingCheckBatchModel, 0)

	for _, v := range req.Items {
		cowInfo, err := s.GetCowInfoByEarNumber(ctx, pastureId, v.EarNumber)
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(v.OperationId))
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		frozenSemen := &model.FrozenSemen{}
		if err = s.DB.Where("bull_id = ?", v.FrozenSemenNumber).
			First(frozenSemen).Error; err != nil {
			return nil, xerr.Custom("未找到冻精信息")
		}

		if frozenSemen.Quantity < v.FrozenSemenCount {
			return nil, xerr.Custom("冻精数量不足")
		}

		if cowInfo.Sex != pasturePb.Genders_Female {
			return nil, xerr.Customf("牛只: %d,不是母牛", cowInfo.EarNumber)
		}

		if int64(v.MatingAt) < cowInfo.LastMatingAt {
			return nil, xerr.Customf("牛只: %s,最近一次配种时间: %d,不能小于本次配种时间: %d", cowInfo.EarNumber, cowInfo.LastMatingAt, v.MatingAt)
		}

		if int64(v.MatingAt) < cowInfo.LastPregnantCheckAt {
			return nil, xerr.Customf("牛只: %s,最近一次孕检时间: %d,不能小于本次配种时间: %d", cowInfo.EarNumber, cowInfo.LastPregnantCheckAt, v.MatingAt)
		}

		if int64(v.MatingAt) < cowInfo.LastAbortionAt {
			return nil, xerr.Customf("牛只: %s,最近一次流产时间: %d,不能小于本次配种时间: %d", cowInfo.EarNumber, cowInfo.LastAbortionAt, v.MatingAt)
		}

		if int64(v.MatingAt) < cowInfo.BirthAt {
			return nil, xerr.Customf("牛只: %s,出生时间: %d,不能小于本次配种时间: %d", cowInfo.EarNumber, cowInfo.BirthAt, v.MatingAt)
		}

		if cowInfo.BreedStatus == pasturePb.BreedStatus_Pregnant || cowInfo.BreedStatus == pasturePb.BreedStatus_No_Mating {
			return nil, xerr.Customf("牛只: %s,当前状态为: %s,不能进行配种", cowInfo.EarNumber, cowInfo.BreedStatus.String())
		}
		eventMatingCheckBatchModelList = append(eventMatingCheckBatchModelList, &model.EventMatingCheckBatchModel{
			Cow:              cowInfo,
			FrozenSemen:      frozenSemen,
			OperationUser:    operationUser,
			MatingAt:         int64(v.MatingAt),
			FrozenSemenCount: v.FrozenSemenCount,
			Remarks:          v.Remarks,
			ExposeEstrusType: v.ExposeEstrusType,
		})
	}

	return eventMatingCheckBatchModelList, nil
}

func (s *StoreEntry) PregnantCheckDataCheck(ctx context.Context, pastureId int64, req *pasturePb.EventPregnantCheckBatch) ([]*PregnantCheckBatchModel, error) {
	if len(req.Items) <= 0 {
		return nil, xerr.Custom("请选择相关牛只数据")
	}

	if len(req.Items) > 50 {
		return nil, xerr.Custom("一次性最多限制提交50牛数据")
	}

	pregnantCheckBatchModelList := make([]*PregnantCheckBatchModel, 0)
	cowInfo := &model.Cow{}
	var err error
	for _, item := range req.Items {
		cowInfo, err = s.GetCowInfoByEarNumber(ctx, pastureId, item.EarNumber)
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		if cowInfo.Sex != pasturePb.Genders_Female {
			return nil, xerr.Customf("牛只: %d,不是母牛", cowInfo.Id)
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(item.OperationId))
		if err != nil {
			zaplog.Error("PregnantCheckDataCheck", zap.Any("id", item.OperationId), zap.Any("error", err.Error()))
			return nil, xerr.Customf("获取操作人员信息失败")
		}

		// 过滤掉没有配种状态的牛只
		if cowInfo.BreedStatus != pasturePb.BreedStatus_Breeding && cowInfo.BreedStatus != pasturePb.BreedStatus_Pregnant {
			return nil, xerr.Customf("牛只: %s 未参加配种,不能进行孕检", cowInfo.EarNumber)
		}

		if cowInfo.LastMatingAt <= 0 {
			return nil, xerr.Customf("牛只: %s,最近一次配种数据异常", cowInfo.EarNumber)
		}

		itemEventPregnantCheck, err := s.FindEventPregnantCheckIsExIstByCowId(ctx, cowInfo)
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		if itemEventPregnantCheck.Id <= 0 {
			return nil, xerr.Customf("未发现该牛只: %s 孕检数据", cowInfo.EarNumber)
		}

		lastEventMating, err := s.FindLastEventMatingByCowId(ctx, pastureId, cowInfo.Id)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, xerr.Customf("未发现该牛只: %s 配种数据", cowInfo.EarNumber)
		}

		if lastEventMating == nil || lastEventMating.Status == pasturePb.IsShow_No {
			return nil, xerr.Customf("未发现该牛只: %s 配种数据", cowInfo.EarNumber)
		}

		pregnantCheckBatchModelList = append(pregnantCheckBatchModelList, &PregnantCheckBatchModel{
			Cow:                 cowInfo,
			OperationUser:       operationUser,
			PregnantCheckAt:     item.PregnantCheckAt,
			PregnantCheckMethod: item.PregnantCheckMethod,
			PregnantCheckResult: item.PregnantCheckResult,
			Remarks:             item.Remarks,
			EventPregnancyCheck: itemEventPregnantCheck,
			LastMating:          lastEventMating,
		})
	}

	return pregnantCheckBatchModelList, nil
}

func (s *StoreEntry) EstrusCheckDataCheck(ctx context.Context, userModel *model.UserModel, req *pasturePb.EventEstrusBatch) (*model.EventEstrusBatch, error) {
	pastureId := userModel.AppPasture.Id
	exposeEstrusType := pasturePb.ExposeEstrusType_Natural_Estrus
	if req.ExposeEstrusType == pasturePb.ExposeEstrusType_Neck_Ring {
		exposeEstrusType = pasturePb.ExposeEstrusType_Neck_Ring
	}
	unMatingReasonsMap := s.UnMatingReasonsMap()

	res := &model.EventEstrusBatch{
		ExposeEstrusType: exposeEstrusType,
		EventEstrusList:  make([]*model.EventEstrus, 0),
		EventMatingList:  make([]*model.EventMating, 0),
		EarNumbers:       make([]string, 0),
	}

	for _, item := range req.Items {
		cowInfo, err := s.GetCowInfoByEarNumber(ctx, pastureId, item.EarNumber)
		if err != nil {
			return nil, xerr.Custom("牛只信息不存在")
		}

		if cowInfo.Sex != pasturePb.Genders_Female {
			return nil, xerr.Custom("该牛只不是母牛")
		}

		if item.EstrusAt <= 0 {
			return nil, xerr.Custom("发情时间不能为空")
		}

		if int64(item.EstrusAt) <= cowInfo.BirthAt {
			return nil, xerr.Custom("发情时间不能小于出生时间")
		}

		if int64(item.EstrusAt) <= cowInfo.LastCalvingAt {
			return nil, xerr.Custom("发情时间不能小于产犊时间")
		}

		estrusAt := time.Unix(int64(item.EstrusAt), 0).Local().Format(model.LayoutDate2)
		estrusLocalStartTime := util.TimeParseLocalUnix(estrusAt)
		estrusLocalEndTime := util.TimeParseLocalEndUnix(estrusAt)
		isExists := s.FindEventEstrusByCowId(pastureId, cowInfo.Id, estrusLocalStartTime, estrusLocalEndTime)
		// 如果存在，并且当天已经提交过则跳过
		if isExists {
			return nil, xerr.Customf("该牛只:%s,今天已经提交过发情数据", item.EarNumber)
		}

		operationUser, _ := s.GetSystemUserById(ctx, int64(item.OperationId))
		item.OperationName = operationUser.Name
		newEventEstrus := model.NewEventEstrus(
			pastureId, cowInfo, exposeEstrusType, pasturePb.IsShow_Ok,
			item.IsMating, int64(item.EstrusAt), operationUser, userModel.SystemUser,
		)

		if item.IsMating == pasturePb.IsShow_Ok {
			isExists = s.FindEventMatingByCowId(pastureId, cowInfo.Id)
			if isExists {
				return nil, xerr.Customf("配种清单已经有该牛只数据 :%s,今天已经提交过配种数据", item.EarNumber)
			}
			newEventMating := model.NewEventMating(pastureId, cowInfo, time.Now().Local().Unix(), exposeEstrusType)
			res.EventMatingList = append(res.EventMatingList, newEventMating)
		} else {
			newEventEstrus.UnMatingReasonsKind = item.UnMatingReasonsKind
			newEventEstrus.UnMatingReasonsName = unMatingReasonsMap[item.UnMatingReasonsKind]
		}

		newEventEstrus.Remarks = item.Remarks
		res.EventEstrusList = append(res.EventEstrusList, newEventEstrus)
		res.EarNumbers = append(res.EarNumbers, item.EarNumber)
	}
	return res, nil
}

func (s *StoreEntry) AbortionEventDataCheck(ctx context.Context, userModel *model.UserModel, items []*pasturePb.EventAbortionItem) ([]*AbortionCheckBatchModel, error) {
	if len(items) <= 0 {
		return nil, xerr.Custom("请选择相关数据")
	}
	if len(items) > 50 {
		return nil, xerr.Custom("一次性最多限制提交50牛数据")
	}
	abortionCheckBatchModelList := make([]*AbortionCheckBatchModel, 0)
	for _, item := range items {
		cow, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, item.EarNumber)
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		if cow.Sex != pasturePb.Genders_Female {
			return nil, xerr.Customf("牛只: %s,不是母牛", cow.EarNumber)
		}

		if cow.IsPregnant != pasturePb.IsShow_Ok {
			return nil, xerr.Customf("牛只: %s,不是怀孕状态", cow.EarNumber)
		}

		if cow.BreedStatus != pasturePb.BreedStatus_Pregnant {
			return nil, xerr.Customf("牛只: %s,不是怀孕状态", cow.EarNumber)
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(item.OperationId))
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		item.OperationName = operationUser.Name
		item.AbortionReasonsName = s.AbortionReasonsMap()[item.AbortionReasons]
		newEventAbortion := model.NewEventAbortion(userModel.AppPasture.Id, cow, item, pasturePb.IsShow_No)
		abortionCheckBatchModelList = append(abortionCheckBatchModelList, &AbortionCheckBatchModel{
			EventAbortion: newEventAbortion,
			Cow:           cow,
			OperationUser: operationUser,
			IsLact:        item.IsLact,
		})
	}
	return abortionCheckBatchModelList, nil
}

func (s *StoreEntry) ForbiddenMatingCheck(ctx context.Context, userModel *model.UserModel, items []*pasturePb.ForbiddenMatingItem) ([]*model.EventForbiddenMatingItem, error) {
	res := make([]*model.EventForbiddenMatingItem, 0)
	forbiddenMatingReasonsMap := s.ForbiddenMatingReasonsMap()
	for _, v := range items {
		cowInfo, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, v.EarNumber)
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		if cowInfo.Sex != pasturePb.Genders_Female {
			return nil, xerr.Customf("牛只: %s,不是母牛", cowInfo.EarNumber)
		}

		if v.ForbiddenMatingAt < int32(cowInfo.BirthAt) {
			return nil, xerr.Customf("牛只: %s,禁止配种时间不能小于出生时间", cowInfo.EarNumber)
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(v.OperationId))
		if err != nil {
			return nil, xerr.WithStack(err)
		}

		res = append(res, &model.EventForbiddenMatingItem{
			Cow:                        cowInfo,
			ForbiddenMatingAt:          int64(v.ForbiddenMatingAt),
			ForbiddenMatingReasonsKind: v.ForbiddenMatingReasonsKind,
			ForbiddenMatingReasonsName: forbiddenMatingReasonsMap[v.ForbiddenMatingReasonsKind],
			Remarks:                    v.Remarks,
			OperationUser:              operationUser,
			MessageUser:                userModel.SystemUser,
		})
	}
	return res, nil
}

func (s *StoreEntry) DeathCheck(ctx context.Context, req *pasturePb.EventDeathBatch, userModel *model.UserModel) ([]*model.EventDeathModel, error) {
	newEventDeathList := make([]*model.EventDeathModel, 0)
	for _, item := range req.Items {
		cow, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, item.EarNumber)
		if err != nil {
			zaplog.Error("DeathBatch", zap.Any("item", item), zap.Any("err", err))
			return nil, xerr.Customf("获取牛只信息失败: %s", item.EarNumber)
		}

		if cow.BirthAt <= int64(item.DeathAt) {
			return nil, xerr.Customf("牛只: %s,死亡时间不能早于出生时间", cow.EarNumber)
		}

		lastEventLog, err := s.GetEventCowLog(ctx, userModel.AppPasture.Id, cow.Id)
		if err != nil {
			zaplog.Error("DeathBatch", zap.Any("item", item), zap.Any("err", err))
			return nil, xerr.Customf("获取牛只信息失败: %s", item.EarNumber)
		}

		if int64(item.DeathAt) < lastEventLog.EventAt {
			return nil, xerr.Customf("牛只: %s,死亡时间不能早于上一次事件时间", cow.EarNumber)
		}

		if name, ok := s.DeadReasonMap()[item.DeathReasonKind]; ok {
			item.DeathReasonName = name
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(item.OperationId))
		if err != nil {
			zaplog.Error("DeathBatch", zap.Any("item", item), zap.Any("err", err))
			return nil, xerr.Customf("获取操作人员信息失败: %d", item.OperationId)
		}

		if name, ok := s.CowDeathDestinationMap()[item.DeathDestinationKind]; ok {
			item.DeathDestinationName = name
		}

		newEventDeath := model.NewEventDeath(userModel.AppPasture.Id, cow, item, userModel.SystemUser, operationUser)
		eventDeathModel := &model.EventDeathModel{
			Cow:         cow,
			EventDeath:  newEventDeath,
			NeckRing:    nil,
			CalvingCalf: nil,
		}

		// 犊牛死亡更新
		if cow.DayAge <= model.CalfDefaultDayAge {
			calvingCalf, ok := s.IsExistCalvingCalf(userModel.AppPasture.Id, cow.Id)
			if ok && calvingCalf != nil {
				eventDeathModel.CalvingCalf = calvingCalf
			}
		}

		if neckRing, ok := s.NeckRingIsExist(userModel.AppPasture.Id, item.EarNumber); ok && neckRing != nil {
			eventDeathModel.NeckRing = neckRing
		}
		newEventDeathList = append(newEventDeathList, eventDeathModel)
	}

	return newEventDeathList, nil
}