pig-farm-controller/internal/app/controller/management/pig_batch_health_controller.go

package management

import (
	"errors"
	"strconv"

	"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
	"git.huangwc.com/pig/pig-farm-controller/internal/app/dto"
	"git.huangwc.com/pig/pig-farm-controller/internal/app/service"
	"github.com/gin-gonic/gin"
)

// RecordSickPigs godoc
// @Summary      记录新增病猪事件
// @Description  记录猪批次中新增病猪的数量、治疗地点和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordSickPigsRequest true "记录病猪请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-sick-pigs [post]
func (c *PigBatchController) RecordSickPigs(ctx *gin.Context) {
	const action = "记录新增病猪事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordSickPigsRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordSickPigs(operatorID, uint(batchID), req.PenID, req.Quantity, req.TreatmentLocation, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录新增病猪事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}

// RecordSickPigRecovery godoc
// @Summary      记录病猪康复事件
// @Description  记录猪批次中病猪康复的数量、治疗地点和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordSickPigRecoveryRequest true "记录病猪康复请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-sick-pig-recovery [post]
func (c *PigBatchController) RecordSickPigRecovery(ctx *gin.Context) {
	const action = "记录病猪康复事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordSickPigRecoveryRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordSickPigRecovery(operatorID, uint(batchID), req.PenID, req.Quantity, req.TreatmentLocation, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录病猪康复事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}

// RecordSickPigDeath godoc
// @Summary      记录病猪死亡事件
// @Description  记录猪批次中病猪死亡的数量、治疗地点和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordSickPigDeathRequest true "记录病猪死亡请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-sick-pig-death [post]
func (c *PigBatchController) RecordSickPigDeath(ctx *gin.Context) {
	const action = "记录病猪死亡事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordSickPigDeathRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordSickPigDeath(operatorID, uint(batchID), req.PenID, req.Quantity, req.TreatmentLocation, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录病猪死亡事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}

// RecordSickPigCull godoc
// @Summary      记录病猪淘汰事件
// @Description  记录猪批次中病猪淘汰的数量、治疗地点和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordSickPigCullRequest true "记录病猪淘汰请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-sick-pig-cull [post]
func (c *PigBatchController) RecordSickPigCull(ctx *gin.Context) {
	const action = "记录病猪淘汰事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordSickPigCullRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordSickPigCull(operatorID, uint(batchID), req.PenID, req.Quantity, req.TreatmentLocation, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录病猪淘汰事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}

// RecordDeath godoc
// @Summary      记录正常猪只死亡事件
// @Description  记录猪批次中正常猪只死亡的数量和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordDeathRequest true "记录正常猪只死亡请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-death [post]
func (c *PigBatchController) RecordDeath(ctx *gin.Context) {
	const action = "记录正常猪只死亡事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordDeathRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordDeath(operatorID, uint(batchID), req.PenID, req.Quantity, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录正常猪只死亡事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}

// RecordCull godoc
// @Summary      记录正常猪只淘汰事件
// @Description  记录猪批次中正常猪只淘汰的数量和发生时间
// @Tags         猪批次管理
// @Accept       json
// @Produce      json
// @Param        id path int true "猪批次ID"
// @Param        body body dto.RecordCullRequest true "记录正常猪只淘汰请求信息"
// @Success      200 {object} controller.Response "记录成功"
// @Router       /api/v1/pig-batches/{id}/record-cull [post]
func (c *PigBatchController) RecordCull(ctx *gin.Context) {
	const action = "记录正常猪只淘汰事件"
	batchID, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
	if err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪批次ID格式", action, "ID格式错误", ctx.Param("id"))
		return
	}

	var req dto.RecordCullRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", req)
		return
	}

	operatorID, err := controller.GetOperatorIDFromContext(ctx)

	err = c.service.RecordCull(operatorID, uint(batchID), req.PenID, req.Quantity, req.HappenedAt, req.Remarks)
	if err != nil {
		if errors.Is(err, service.ErrPigBatchNotFound) || errors.Is(err, service.ErrPenNotFound) {
			controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), batchID)
			return
		} else if errors.Is(err, service.ErrInvalidOperation) || errors.Is(err, service.ErrPigBatchNotActive) || errors.Is(err, service.ErrPenNotAssociatedWithBatch) || errors.Is(err, service.ErrPenNotEmpty) {
			controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), batchID)
			return
		}
		c.logger.Errorf("%s: 业务逻辑失败: %v", action, err)
		controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "记录正常猪只淘汰事件失败", action, err.Error(), batchID)
		return
	}

	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "记录成功", nil, action, "记录成功", batchID)
}