kpt-pasture/module/backend/event_base.go

package backend

import (
	"context"
	"fmt"
	"kpt-pasture/model"
	"net/http"
	"strconv"
	"strings"

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

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

func (s *StoreEntry) ParseCowIds(ctx context.Context, cowIds string) ([]*model.Cow, error) {
	if len(cowIds) == 0 {
		return nil, xerr.Custom("cow id is required")
	}
	cowIdStr := strings.Split(cowIds, ",")
	var cowIdInts = make([]int64, 0)
	for _, v := range cowIdStr {
		cowId, err := strconv.ParseInt(v, 10, 64)
		if err != nil {
			return nil, xerr.Customf("错误的牛号: %s", v)
		}
		cowIdInts = append(cowIdInts, cowId)
	}

	cowList, err := s.GetCowInfoByCowIds(ctx, cowIdInts)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	return cowList, nil
}

func (s *StoreEntry) EnterList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchEnterEventResponse, error) {
	eventEnterList := make([]*model.EventEnter, 0)
	var count int64 = 0
	pref := s.DB.Model(new(model.EventEnter))
	if len(req.CowId) > 0 {
		cowIds := strings.Split(req.CowId, ",")
		pref.Where("cow_id IN ?", cowIds)
	}

	if err := pref.Order("id desc").
		Count(&count).Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&eventEnterList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	penMap := s.PenMap(ctx)
	breedStatusMap := s.CowBreedStatusMap()
	cowSourceMap := s.CowSourceMap()
	cowTypeMap := s.CowTypeMap()
	cowKindMap := s.CowKindMap()

	return &pasturePb.SearchEnterEventResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchEnterEventData{
			List:     model.EventEnterSlice(eventEnterList).ToPB(penMap, breedStatusMap, cowSourceMap, cowTypeMap, cowKindMap),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) CreateEnter(ctx context.Context, req *pasturePb.EventEnterRequest) error {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil || currentUser.Id <= 0 {
		return xerr.Customf("当前用户信息错误")
	}
	req.MessengerId = int32(currentUser.Id)
	req.MessengerName = currentUser.Name
	if req.OperationId > 0 {
		systemUser, _ := s.GetSystemUserById(ctx, int64(req.OperationId))
		req.OperationName = systemUser.Name
	}

	newCowData := model.NewCow(req)
	newEventEnter := model.NewEventEnter(newCowData.Id, req)
	if err = s.DB.Transaction(func(tx *gorm.DB) error {
		if err = tx.Create(newCowData).Error; err != nil {
			return xerr.WithStack(err)
		}
		if err = tx.Create(newEventEnter).Error; err != nil {
			return xerr.WithStack(err)
		}
		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) GroupTransferList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchTransferGroupEventResponse, error) {
	eventGroupTransferList := make([]*pasturePb.EventTransferGroupData, 0)
	var count int64 = 0
	pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventTransferGroup).TableName())).
		Select(`a.id,a.cow_id,a.pen_in_id as transfer_in_pen_id,a.pen_out_id as transfer_out_pen_id,a.lact,a.remarks,a.transfer_reason_id,a.transfer_reason_name,
			a.transfer_date,a.created_at,a.operation_id,a.operation_name,b.name as transfer_in_pen_name,c.name as transfer_out_pen_name,
			f.lact,f.ear_number`).
		Joins(fmt.Sprintf("JOIN %s AS b ON a.pen_in_id = b.id", new(model.Pen).TableName())).
		Joins(fmt.Sprintf("JOIN %s AS c on a.pen_out_id = c.id", new(model.Pen).TableName())).
		Joins(fmt.Sprintf("JOIN %s AS f ON a.cow_id = f.id", new(model.Cow).TableName()))
	if len(req.CowId) > 0 {
		cowIds := strings.Split(req.CowId, ",")
		pref.Where("a.cow_id IN ?", cowIds)
	}

	if err := pref.Order("a.id desc").Group("a.id").
		Count(&count).Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&eventGroupTransferList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchTransferGroupEventResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchTransferGroupEventData{
			List:     eventGroupTransferList,
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) CreateGroupTransfer(ctx context.Context, req *pasturePb.TransferGroupEventRequest) (err error) {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil || currentUser.Id <= 0 {
		return xerr.Customf("当前用户信息错误")
	}

	newEventTransferGroupModelList := make([]*model.EventTransferGroupModel, 0)
	for _, v := range req.Body {
		cow, err := s.GetCowInfoByCowId(ctx, int64(v.CowId))
		if err != nil {
			return xerr.WithStack(err)
		}
		// 转去栏舍和当前栏舍相同，则不处理
		if cow.PenId == v.TransferInPenId {
			continue
		}
		operationUser, err := s.GetSystemUserById(ctx, int64(v.OperationId))
		if err != nil {
			return xerr.WithStack(err)
		}
		newEventTransferGroup := model.NewEventTransferGroup(cow, v, currentUser, operationUser)
		newEventTransferGroupModelList = append(newEventTransferGroupModelList, &model.EventTransferGroupModel{
			Cow:                cow,
			EventTransferGroup: newEventTransferGroup,
		})
	}
	if len(newEventTransferGroupModelList) <= 0 {
		return nil
	}

	defer func() {
		if err != nil {
			for _, etg := range newEventTransferGroupModelList {
				cowLogs := s.SubmitEventLog(ctx, etg.Cow, pasturePb.EventType_Transfer_Ben, pasturePb.ExposeEstrusType_Invalid, etg.EventTransferGroup)
				s.DB.Table(cowLogs.TableName()).Create(cowLogs)
			}
		}
	}()

	if err = s.DB.Transaction(func(tx *gorm.DB) error {
		for _, v := range newEventTransferGroupModelList {
			if err = tx.Create(v.EventTransferGroup).Error; err != nil {
				return xerr.WithStack(err)
			}
			if err = tx.Model(v.Cow).Update("pen_id", v.EventTransferGroup.PenInId).Error; err != nil {
				return xerr.WithStack(err)
			}
		}
		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}

	return nil
}

func (s *StoreEntry) BodyScoreList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchBodyScoreEventResponse, error) {
	bodyScoreList := make([]*pasturePb.BodyScoreList, 0)
	var count int64 = 0
	pref := s.DB.Model(new(model.EventBodyScore)).Select("*,score as body_score")
	if len(req.CowId) > 0 {
		cowIds := strings.Split(req.CowId, ",")
		pref.Where("cow_id IN ?", cowIds)
	}

	if err := pref.Order("id desc").
		Count(&count).Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&bodyScoreList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchBodyScoreEventResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchBodyScoreData{
			List:     bodyScoreList,
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) CreateBodyScore(ctx context.Context, req *pasturePb.BodyScoreEventRequest) error {
	if len(req.CowId) <= 0 {
		return xerr.Custom("请选择相关牛只")
	}
	currentSystemUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Customf("获取当前用户失败: %s", err.Error())
	}

	operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
	if err != nil {
		return xerr.WithStack(err)
	}
	req.OperationName = operationUser.Name
	bodyScourEvent := make([]*model.EventBodyScore, 0)
	cowList, err := s.ParseCowIds(ctx, req.CowId)
	if err != nil {
		return xerr.WithStack(err)
	}
	for _, cow := range cowList {
		bodyScourEvent = append(bodyScourEvent, model.NewEventBodyScore(cow, currentSystemUser, req))
	}
	if len(bodyScourEvent) <= 0 {
		return nil
	}
	return s.DB.Create(bodyScourEvent).Error
}

func (s *StoreEntry) WeightList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchWeightEventResponse, error) {
	weightList := make([]*pasturePb.SearchWeightList, 0)
	var count int64 = 0
	pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventWeight).TableName())).
		Select(`a.id,a.cow_id,a.ear_number,a.weight / 100 as weight,a.lact,a.day_age,a.weight_at,a.remarks,a.created_at,
		a.updated_at,a.message_id,a.operation_id,a.message_name,a.operation_name`)
	if len(req.CowId) > 0 {
		cowIds := strings.Split(req.CowId, ",")
		pref.Where("a.cow_id IN ?", cowIds)
	}

	if err := pref.Order("a.id desc").
		Count(&count).Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&weightList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchWeightEventResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchWeightData{
			List:     weightList,
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) WeightBatch(ctx context.Context, req *pasturePb.EventWeight) (err error) {
	if len(req.WeightItems) <= 0 {
		return xerr.Custom("请选择相关牛只")
	}

	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("当前登录用户失败,请退出重新登录")
	}

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

	defer func() {
		if err == nil {
			// 记录事件日志
			for _, item := range req.WeightItems {
				cow, _ := s.GetCowInfoByCowId(ctx, int64(item.CowId))
				cowLogs := s.SubmitEventLog(ctx, cow, pasturePb.EventType_Weight, pasturePb.ExposeEstrusType_Invalid, req)
				s.DB.Table(cowLogs.TableName()).Create(cowLogs)
			}
		}
	}()
	cow := &model.Cow{}
	for _, item := range req.WeightItems {
		cow, err = s.GetCowInfoByCowId(ctx, int64(item.CowId))
		if err != nil {
			return xerr.WithStack(err)
		}
		item.WeightAt = req.WeightAt

		// 更新牛只信息
		cow.EventWeightUpdate(int64(item.Weight), int64(req.WeightAt))
		if err = s.DB.Model(new(model.Cow)).
			Select("last_second_weight_at,last_second_weight,last_weight_at,current_weight").
			Where("id = ?", cow.Id).
			Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
			Updates(cow).Error; err != nil {
			return xerr.WithStack(err)
		}

		// 创建牛只的体重记录
		eventWeight := model.NewEventWeight(cow, currentUser, int32(item.Weight*1000), item.Height, req)
		if err = s.DB.Create(eventWeight).Error; err != nil {
			return xerr.WithStack(err)
		}
	}
	return err
}

func (s *StoreEntry) DepartureBatch(ctx context.Context, req *pasturePb.EventDepartureBatch) (err error) {
	if len(req.Item) <= 0 {
		return xerr.Custom("请选择相关牛只")
	}

	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("当前登录用户失败,请退出重新登录")
	}

	newEventDepartureList := make([]*model.EventDeparture, 0)
	cow := &model.Cow{}
	for _, item := range req.Item {
		cow, err = s.GetCowInfoByCowId(ctx, int64(item.CowId))
		if err != nil {
			zaplog.Error("DepartureBatch", zap.Any("item", item), zap.Any("err", err))
			return xerr.Customf("获取牛只信息失败: %d", item.CowId)
		}

		operationUser, err := s.GetSystemUserById(ctx, int64(item.OperationId))
		if err != nil {
			zaplog.Error("DepartureBatch", zap.Any("item", item), zap.Any("err", err))
			return xerr.Customf("获取操作人员信息失败: %d", item.OperationId)
		}
		reasonName := ""
		switch item.DepartureType {
		case pasturePb.DepartureType_Death:
			reasonName = s.DeadReasonMap()[pasturePb.DeadReason_Kind(item.DepartureReason)]
		case pasturePb.DepartureType_Out:
			reasonName = s.OutReasonMap()[pasturePb.OutReason_Kind(item.DepartureReason)]
		}
		newEventDepartureList = append(newEventDepartureList, model.NewEventDeparture(cow, item, reasonName, currentUser, operationUser))
	}

	if len(newEventDepartureList) <= 0 {
		return nil
	}

	if err = s.DB.Create(newEventDepartureList).Error; err != nil {
		return xerr.WithStack(err)
	}

	// 记录事件日志
	for _, item := range newEventDepartureList {
		eventType := pasturePb.EventType_Death
		if item.DepartureType == pasturePb.DepartureType_Out {
			eventType = pasturePb.EventType_Out
		}
		cowLogs := s.SubmitEventLog(ctx, cow, eventType, pasturePb.ExposeEstrusType_Invalid, item)
		s.DB.Table(cowLogs.TableName()).Create(cowLogs)
	}
	return nil
}