event: 配种事件

go.mod

@@ -3,7 +3,7 @@ module kpt-pasture
 go 1.17
 
 require (
-	gitee.com/xuyiping_admin/go_proto v0.0.0-20240526140108-73f4ee04f1c5
+	gitee.com/xuyiping_admin/go_proto v0.0.0-20240528055802-b5a257a61f02
 	gitee.com/xuyiping_admin/pkg v0.0.0-20231218082641-aac597b8a015
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/getsentry/sentry-go v0.23.0

go.sum

@@ -82,6 +82,14 @@ gitee.com/xuyiping_admin/go_proto v0.0.0-20240526024805-eac3258223e5 h1:2iGgtLkY
 gitee.com/xuyiping_admin/go_proto v0.0.0-20240526024805-eac3258223e5/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
 gitee.com/xuyiping_admin/go_proto v0.0.0-20240526140108-73f4ee04f1c5 h1:ni/CWwJhAilmh95iOIFJOup5eWCsSkYOVHq18sO+AFY=
 gitee.com/xuyiping_admin/go_proto v0.0.0-20240526140108-73f4ee04f1c5/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528044746-8077903e39e3 h1:mOhqL1VT4bLaOKlpswHr1g1hV5Obh81ykT6ViSCy5Rk=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528044746-8077903e39e3/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528051116-6bb4d505f164 h1:K1M7Ap1ulo2As3m7LPaStFk8zUeGrHOa6gGhpaU6OTw=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528051116-6bb4d505f164/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528053721-d0592ac1e139 h1:qPsLiTnAFo8J+uAGTFEJOXaCnrfF7Joj05rvw+Pu7b4=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528053721-d0592ac1e139/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528055802-b5a257a61f02 h1:SzB3bZJ/qb/aKDUqRkufiJr9d26CeTygLW+SeCZ97Js=
+gitee.com/xuyiping_admin/go_proto v0.0.0-20240528055802-b5a257a61f02/go.mod h1:BKrFW6YLDectlQcQk3FYKBeXvjEiodAKJ5rq7O/QiPE=
 gitee.com/xuyiping_admin/pkg v0.0.0-20231218082641-aac597b8a015 h1:dfb5dRd57L2HKjdwLT93UFmPYFPOmEl56gtZmqcNnaE=
 gitee.com/xuyiping_admin/pkg v0.0.0-20231218082641-aac597b8a015/go.mod h1:Fk4GYI/v0IK3XFrm1Gn+VkgCz5Y7mfswD5hsTJYOG6A=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=

http/handler/event/event.go

@@ -282,3 +282,54 @@ func PregnantCheckEventCreate(c *gin.Context) {
 		Data: &operationPb.Success{Success: true},
 	})
 }
+
+func MatingEventList(c *gin.Context) {
+	var req pasturePb.SearchEventRequest
+	if err := ginutil.BindProto(c, &req); err != nil {
+		apierr.AbortBadRequest(c, http.StatusBadRequest, err)
+		return
+	}
+
+	pagination := &pasturePb.PaginationModel{
+		Page:       int32(c.GetInt(middleware.Page)),
+		PageSize:   int32(c.GetInt(middleware.PageSize)),
+		PageOffset: int32(c.GetInt(middleware.PageOffset)),
+	}
+
+	res, err := middleware.Dependency(c).StoreEventHub.OpsService.MatingList(c, &req, pagination)
+	if err != nil {
+		apierr.ClassifiedAbort(c, err)
+		return
+	}
+	ginutil.JSONResp(c, res)
+}
+
+func MatingCreate(c *gin.Context) {
+	var req pasturePb.MatingEventRequest
+	if err := ginutil.BindProto(c, &req); err != nil {
+		apierr.AbortBadRequest(c, http.StatusBadRequest, err)
+		return
+	}
+
+	if err := valid.ValidateStruct(&req,
+		valid.Field(&req.CowId, valid.Required),
+		valid.Field(&req.StaffMemberId, valid.Required),
+		valid.Field(&req.ExposeEstrusType, valid.Required),
+		valid.Field(&req.StaffMemberId, valid.Required),
+		valid.Field(&req.MatingAt, valid.Required),
+	); err != nil {
+		apierr.AbortBadRequest(c, http.StatusBadRequest, err)
+		return
+	}
+
+	if err := middleware.BackendOperation(c).OpsService.MatingCreate(c, &req); err != nil {
+		apierr.ClassifiedAbort(c, err)
+		return
+	}
+
+	ginutil.JSONResp(c, &operationPb.CommonOK{
+		Code: http.StatusOK,
+		Msg:  "ok",
+		Data: &operationPb.Success{Success: true},
+	})
+}

http/route/event_api.go

@@ -27,5 +27,8 @@ func EventAPI(opts ...func(engine *gin.Engine)) func(s *gin.Engine) {
 
 		eventRoute.POST("/pregnant/check/list", event.PregnantCheckEventList)
 		eventRoute.POST("/pregnant/check/create", event.PregnantCheckEventCreate)
+
+		eventRoute.POST("/mating/list", event.MatingEventList)
+		eventRoute.POST("/mating/create", event.MatingCreate)
 	}
 }

model/cow.go

@@ -95,10 +95,23 @@ func NewCalfCow(motherId, fatherId int64, calf *EventCalvingCalf) *Cow {
 	}
 }
 
+// GetDayAge 日龄
 func (c *Cow) GetDayAge() int32 {
 	return int32(math.Floor(float64(time.Now().Unix()-c.BirthAt) / 86400))
 }
 
+// GetDaysPregnant 在胎天数
 func (c *Cow) GetDaysPregnant() int32 {
-	return int32(math.Floor(float64(time.Now().Unix()-c.LastMatingAt) / 86400))
+	if c.Status == pasturePb.CowStatus_Breeding || c.Status == pasturePb.CowStatus_Pregnant {
+		return int32(math.Floor(float64(time.Now().Unix()-c.LastMatingAt) / 86400))
+	}
+	return 0
+}
+
+// GetLactationDays 泌乳天数
+func (c *Cow) GetLactationDays() int32 {
+	if c.Status == pasturePb.CowStatus_Calving {
+		return int32(math.Floor(float64(time.Now().Unix()-c.LastCalvingAt) / 86400))
+	}
+	return 0
 }

model/event_mating.go

@@ -0,0 +1,77 @@
+package model
+
+import pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
+
+type EventMating struct {
+	Id                int64                           `json:"id"`
+	CowId             int64                           `json:"cow_id"`
+	DayAge            int64                           `json:"day_age"`
+	Lact              int8                            `json:"lact"`
+	LactationDays     int32                           `json:"lactation_days"`
+	ExposeEstrusType  pasturePb.ExposeEstrusType_Kind `json:"expose_estrus_type"`
+	MatingAt          int64                           `json:"mating_at"`
+	BullId            int64                           `json:"bull_id"`
+	FrozenSemenNumber string                          `json:"frozen_semen_number"`
+	StallNumberId     int64                           `json:"stall_number_id"`
+	OperationId       int64                           `json:"operation_id"`
+	Remarks           string                          `json:"remarks"`
+	CreatedAt         int64                           `json:"created_at"`
+	UpdatedAt         int64                           `json:"updated_at"`
+}
+
+func (e *EventMating) TableName() string {
+	return "event_mating"
+}
+
+func NewEventMating(cow *Cow, currentUser *SystemUser, req *pasturePb.MatingEventRequest) *EventMating {
+	return &EventMating{
+		CowId:             cow.Id,
+		DayAge:            int64(cow.GetDayAge()),
+		Lact:              int8(cow.Lact),
+		LactationDays:     cow.GetLactationDays(),
+		ExposeEstrusType:  req.ExposeEstrusType,
+		MatingAt:          int64(req.MatingAt),
+		BullId:            int64(req.BullId),
+		FrozenSemenNumber: req.FrozenSemenNumber,
+		StallNumberId:     int64(req.StaffMemberId),
+		OperationId:       currentUser.Id,
+		Remarks:           req.Remarks,
+	}
+}
+
+type EventMatingSlice []*EventMating
+
+func (e EventMatingSlice) ToPB(user []*SystemUser, exposeEstrusTypeMap map[pasturePb.ExposeEstrusType_Kind]string) []*pasturePb.SearchMatingList {
+	res := make([]*pasturePb.SearchMatingList, len(e))
+	for i, v := range e {
+		operationName, staffMemberName := "", ""
+		for _, u := range user {
+			if v.OperationId == u.Id {
+				operationName = u.Name
+			}
+			if v.StallNumberId == u.Id {
+				staffMemberName = u.Name
+			}
+		}
+		res[i] = &pasturePb.SearchMatingList{
+			Id:                   int32(v.Id),
+			CowId:                int32(v.CowId),
+			DayAge:               int32(v.DayAge),
+			Lact:                 int32(v.Lact),
+			ExposeEstrusType:     v.ExposeEstrusType,
+			ExposeEstrusTypeName: exposeEstrusTypeMap[v.ExposeEstrusType],
+			MatingAt:             int32(v.MatingAt),
+			BullId:               int32(v.BullId),
+			FrozenSemenNumber:    v.FrozenSemenNumber,
+			LactationDays:        v.LactationDays,
+			StaffMemberId:        int32(v.StallNumberId),
+			StaffMemberName:      staffMemberName,
+			Remarks:              v.Remarks,
+			OperationId:          int32(v.OperationId),
+			OperationName:        operationName,
+			CreatedAt:            int32(v.CreatedAt),
+			UpdatedAt:            int32(v.UpdatedAt),
+		}
+	}
+	return res
+}

module/backend/config_data.go

@@ -492,17 +492,40 @@ func (s *StoreEntry) UnitEnumList() []*pasturePb.ConfigOptionsList {
 	return configOptions
 }
 
-func (s *StoreEntry) DrugUsageMaps(key pasturePb.DrugUsage_Kind) string {
-	res := ""
+func (s *StoreEntry) ExposeEstrusTypeEnumList() []*pasturePb.ConfigOptionsList {
+	configOptions := make([]*pasturePb.ConfigOptionsList, 0)
+	configOptions = append(configOptions,
+		&pasturePb.ConfigOptionsList{
+			Value:    int32(pasturePb.ExposeEstrusType_Neck_Ring),
+			Label:    "脖环揭发",
+			Disabled: true,
+		}, &pasturePb.ConfigOptionsList{
+			Value:    int32(pasturePb.ExposeEstrusType_Foot_Ring),
+			Label:    "脚环/计步器揭发",
+			Disabled: true,
+		}, &pasturePb.ConfigOptionsList{
+			Value:    int32(pasturePb.ExposeEstrusType_Manual_Observation),
+			Label:    "人工观察",
+			Disabled: true,
+		})
+	return configOptions
+}
+
+func (s *StoreEntry) DrugUsageMaps() map[pasturePb.DrugUsage_Kind]string {
+	res := make(map[pasturePb.DrugUsage_Kind]string)
 	for _, v := range s.DrugUsageEnumList() {
-		if v.Value != int32(key) {
-			continue
-		}
-		res = v.Label
+		res[pasturePb.DrugUsage_Kind(v.Value)] = v.Label
 	}
 	return res
 }
 
+func (s *StoreEntry) ExposeEstrusTypeMap() map[pasturePb.ExposeEstrusType_Kind]string {
+	res := make(map[pasturePb.ExposeEstrusType_Kind]string)
+	for _, v := range s.ExposeEstrusTypeEnumList() {
+		res[pasturePb.ExposeEstrusType_Kind(v.Value)] = v.Label
+	}
+	return res
+}
 func (s *StoreEntry) BarnTypeOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
 	return &pasturePb.ConfigOptionsListResponse{
 		Code:    http.StatusOK,
@@ -611,6 +634,8 @@ func (s *StoreEntry) SystemBaseConfigOptions(ctx context.Context, optionsName st
 		configOptions = s.PregnantCheckResultEnumList()
 	case "pregnantCheckMethod":
 		configOptions = s.PregnantCheckMethodEnumList()
+	case "exposeEstrusType":
+		configOptions = s.ExposeEstrusTypeEnumList()
 	}
 	return &pasturePb.ConfigOptionsListResponse{
 		Code:    http.StatusOK,

module/backend/event.go

@@ -354,6 +354,7 @@ func (s *StoreEntry) PregnantCheckCreate(ctx context.Context, req *pasturePb.Pre
 	cowList := strings.Split(req.CowId, ",")
 	eventPregnantCheckList := make([]*model.EventPregnantCheck, 0)
 	currentUser, _ := s.GetCurrentSystemUser(ctx)
+
 	for _, v := range cowList {
 		cowId, _ := strconv.ParseInt(v, 10, 64)
 		cow, err := s.GetCowInfo(ctx, cowId)
@@ -368,3 +369,56 @@ func (s *StoreEntry) PregnantCheckCreate(ctx context.Context, req *pasturePb.Pre
 	}
 	return nil
 }
+
+func (s *StoreEntry) MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.MatingEventResponse, error) {
+	matingList := make([]*model.EventMating, 0)
+	var count int64 = 0
+	pref := s.DB.Table(new(model.EventMating).TableName())
+	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(&matingList).Error; err != nil {
+		return nil, xerr.WithStack(err)
+	}
+
+	systemUserList, _ := s.SystemUserList(ctx)
+	exposeEstrusTypeMap := s.ExposeEstrusTypeMap()
+	return &pasturePb.MatingEventResponse{
+		Code:    http.StatusOK,
+		Message: "ok",
+		Data: &pasturePb.SearchMatingData{
+			List:     model.EventMatingSlice(matingList).ToPB(systemUserList, exposeEstrusTypeMap),
+			Total:    int32(count),
+			PageSize: pagination.PageSize,
+			Page:     pagination.Page,
+		},
+	}, nil
+}
+
+func (s *StoreEntry) MatingCreate(ctx context.Context, req *pasturePb.MatingEventRequest) error {
+	if len(req.CowId) <= 0 {
+		return xerr.Custom("请选择相关牛只")
+	}
+	cowList := strings.Split(req.CowId, ",")
+	matingList := make([]*model.EventMating, 0)
+	currentUser, _ := s.GetCurrentSystemUser(ctx)
+
+	for _, v := range cowList {
+		cowId, _ := strconv.ParseInt(v, 10, 64)
+		cow, err := s.GetCowInfo(ctx, cowId)
+		if err != nil {
+			return xerr.WithStack(err)
+		}
+		matingList = append(matingList, model.NewEventMating(cow, currentUser, req))
+	}
+
+	if err := s.DB.Create(matingList).Error; err != nil {
+		return xerr.WithStack(err)
+	}
+	return nil
+}

module/backend/interface.go

@@ -133,6 +133,8 @@ type EventService interface {
 	CalvingCreate(ctx context.Context, req *pasturePb.CalvingEventRequest) error
 	PregnantCheckList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.PregnantCheckEventResponse, error)
 	PregnantCheckCreate(ctx context.Context, req *pasturePb.PregnantCheckEventRequest) error
+	MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.MatingEventResponse, error)
+	MatingCreate(ctx context.Context, req *pasturePb.MatingEventRequest) error
 }
 
 //go:generate mockgen -destination mock/CowService.go -package kptservicemock kpt-pasture/module/backend CowService