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

package management

import (
	"context"
	"errors"
	"fmt"
	"strconv"

	"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
	"git.huangwc.com/pig/pig-farm-controller/internal/app/service"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"

	"github.com/labstack/echo/v4"
)

// mapAndSendError 统一映射服务层错误并发送响应。
// 这个函数将服务层返回的错误转换为控制器层应返回的HTTP状态码和审计信息。
func mapAndSendError(reqContext context.Context, c *PigBatchController, ctx echo.Context, action string, err error, id uint) error {
	if errors.Is(err, service.ErrPigBatchNotFound) ||
		errors.Is(err, service.ErrPenNotFound) ||
		errors.Is(err, service.ErrPenNotAssociatedWithBatch) {
		return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), action, err.Error(), id)
	} else if errors.Is(err, service.ErrInvalidOperation) ||
		errors.Is(err, service.ErrPigBatchActive) ||
		errors.Is(err, service.ErrPigBatchNotActive) ||
		errors.Is(err, service.ErrPenOccupiedByOtherBatch) ||
		errors.Is(err, service.ErrPenStatusInvalidForAllocation) ||
		errors.Is(err, service.ErrPenNotEmpty) {
		return controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), action, err.Error(), id)
	} else {
		logs.GetLogger(reqContext).Errorf("操作[%s]业务逻辑失败: %v", action, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, fmt.Sprintf("操作失败: %v", err), action, err.Error(), id)
	}
}

// idExtractorFunc 定义了一个函数类型，用于从echo.Context中提取主ID。
type idExtractorFunc func(ctx echo.Context) (uint, error)

// extractOperatorAndPrimaryID 封装了从echo.Context中提取操作员ID和主ID的通用逻辑。
// 它负责处理ID提取过程中的错误，并发送相应的HTTP响应。
//
// 参数:
//
//	c: *PigBatchController - 控制器实例，用于访问其日志。
//	ctx: echo.Context - Echo上下文。
//	action: string - 当前操作的描述，用于日志和审计。
//	idExtractor: idExtractorFunc - 可选函数，用于从ctx中提取主ID。如果为nil，则尝试从":id"路径参数中提取。
//
// 返回值:
//
//	operatorID: uint - 提取到的操作员ID。
//	primaryID: uint - 提取到的主ID。
//	err: error - 如果ID提取失败或发送错误响应，则返回错误。
func extractOperatorAndPrimaryID(
	c *PigBatchController,
	ctx echo.Context,
	action string,
	idExtractor idExtractorFunc,
) (operatorID uint, primaryID uint, err error) {
	// 1. 获取操作员ID
	operatorID, err = controller.GetOperatorIDFromContext(ctx)
	if err != nil {
		return 0, 0, controller.SendErrorWithAudit(ctx, controller.CodeUnauthorized, "未授权", action, "无法获取操作员ID", nil)
	}

	// 2. 提取主ID
	if idExtractor != nil {
		primaryID, err = idExtractor(ctx)
		if err != nil {
			return 0, 0, controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的ID格式", action, "ID格式错误", err.Error())
		}
	} else { // 默认从 ":id" 路径参数提取
		idParam := ctx.Param("id")
		if idParam == "" { // 有些端点可能没有 "id" 参数，例如列表或创建操作
			// 如果没有ID参数且没有自定义提取器，primaryID保持为0，这对于某些操作是可接受的
		} else {
			parsedID, err := strconv.ParseUint(idParam, 10, 32)
			if err != nil {
				return 0, 0, controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的ID格式", action, "ID格式错误", idParam)
			}
			primaryID = uint(parsedID)
		}
	}

	return operatorID, primaryID, nil
}

// handleAPIRequest 封装了控制器中处理带有请求体和路径参数的API请求的通用逻辑。
// 它负责请求体绑定、操作员ID获取、服务层调用、错误映射和响应发送。
func handleAPIRequest[Req any](
	reqContext context.Context,
	c *PigBatchController,
	ctx echo.Context,
	action string,
	reqDTO Req,
	serviceExecutor func(ctx echo.Context, operatorID uint, primaryID uint, req Req) error,
	successMsg string,
	idExtractor idExtractorFunc,
) error {
	// 1. 绑定请求体
	if err := ctx.Bind(&reqDTO); err != nil {
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体", action, "请求体绑定失败", reqDTO)
	}

	// 2. 提取操作员ID和主ID
	operatorID, primaryID, err := extractOperatorAndPrimaryID(c, ctx, action, idExtractor)
	if err != nil {
		return err // 错误已在 extractOperatorAndPrimaryID 中处理
	}

	// 3. 执行服务层逻辑
	err = serviceExecutor(ctx, operatorID, primaryID, reqDTO)
	if err != nil {
		return mapAndSendError(reqContext, c, ctx, action, err, primaryID)
	}

	// 4. 发送成功响应
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, successMsg, nil, action, successMsg, primaryID)
}

// handleNoBodyAPIRequest 封装了处理不带请求体，但有路径参数和操作员ID的API请求的通用逻辑。
func handleNoBodyAPIRequest(
	reqContext context.Context,
	c *PigBatchController,
	ctx echo.Context,
	action string,
	serviceExecutor func(ctx echo.Context, operatorID uint, primaryID uint) error,
	successMsg string,
	idExtractor idExtractorFunc,
) error {
	// 1. 提取操作员ID和主ID
	operatorID, primaryID, err := extractOperatorAndPrimaryID(c, ctx, action, idExtractor)
	if err != nil {
		return err // 错误已在 extractOperatorAndPrimaryID 中处理
	}

	// 2. 执行服务层逻辑
	err = serviceExecutor(ctx, operatorID, primaryID)
	if err != nil {
		return mapAndSendError(reqContext, c, ctx, action, err, primaryID)
	}

	// 3. 发送成功响应
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, successMsg, nil, action, successMsg, primaryID)
}

// handleAPIRequestWithResponse 封装了控制器中处理带有请求体、路径参数并返回响应DTO的API请求的通用逻辑。
func handleAPIRequestWithResponse[Req any, Resp any](
	reqContext context.Context,
	c *PigBatchController,
	ctx echo.Context,
	action string,
	reqDTO Req,
	serviceExecutor func(ctx echo.Context, operatorID uint, primaryID uint, req Req) (Resp, error), // serviceExecutor现在返回Resp
	successMsg string,
	idExtractor idExtractorFunc,
) error {
	// 1. 绑定请求体
	if err := ctx.Bind(&reqDTO); err != nil {
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, fmt.Sprintf("无效的请求体: %v", err), action, fmt.Sprintf("请求体绑定失败: %v", err), reqDTO)
	}

	// 2. 提取操作员ID和主ID
	operatorID, primaryID, err := extractOperatorAndPrimaryID(c, ctx, action, idExtractor)
	if err != nil {
		return err // 错误已在 extractOperatorAndPrimaryID 中处理
	}

	// 3. 执行服务层逻辑
	respDTO, err := serviceExecutor(ctx, operatorID, primaryID, reqDTO)
	if err != nil {
		return mapAndSendError(reqContext, c, ctx, action, err, primaryID)
	}

	// 4. 发送成功响应
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, successMsg, respDTO, action, successMsg, primaryID)
}

// handleNoBodyAPIRequestWithResponse 封装了处理不带请求体，但有路径参数和操作员ID，并返回响应DTO的API请求的通用逻辑。
func handleNoBodyAPIRequestWithResponse[Resp any](
	reqContext context.Context,
	c *PigBatchController,
	ctx echo.Context,
	action string,
	serviceExecutor func(ctx echo.Context, operatorID uint, primaryID uint) (Resp, error), // serviceExecutor现在返回Resp
	successMsg string,
	idExtractor idExtractorFunc,
) error {
	// 1. 提取操作员ID和主ID
	operatorID, primaryID, err := extractOperatorAndPrimaryID(c, ctx, action, idExtractor)
	if err != nil {
		return err // 错误已在 extractOperatorAndPrimaryID 中处理
	}

	// 2. 执行服务层逻辑
	respDTO, err := serviceExecutor(ctx, operatorID, primaryID)
	if err != nil {
		return mapAndSendError(reqContext, c, ctx, action, err, primaryID)
	}

	// 3. 发送成功响应
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, successMsg, respDTO, action, successMsg, primaryID)
}

// handleQueryAPIRequestWithResponse 封装了处理带有查询参数并返回响应DTO的API请求的通用逻辑。
func handleQueryAPIRequestWithResponse[Query any, Resp any](
	reqContext context.Context,
	c *PigBatchController,
	ctx echo.Context,
	action string,
	queryDTO Query,
	serviceExecutor func(ctx echo.Context, operatorID uint, query Query) (Resp, error), // serviceExecutor现在接收queryDTO
	successMsg string,
) error {
	// 1. 绑定查询参数
	if err := ctx.Bind(&queryDTO); err != nil {
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的查询参数", action, "查询参数绑定失败", queryDTO)
	}

	// 2. 获取操作员ID
	operatorID, err := controller.GetOperatorIDFromContext(ctx)
	if err != nil {
		return controller.SendErrorWithAudit(ctx, controller.CodeUnauthorized, "未授权", action, "无法获取操作员ID", nil)
	}

	// 3. 执行服务层逻辑
	respDTO, err := serviceExecutor(ctx, operatorID, queryDTO)
	if err != nil {
		// 对于列表查询，通常没有primaryID，所以传递0
		return mapAndSendError(reqContext, c, ctx, action, err, 0)
	}

	// 4. 发送成功响应
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, successMsg, respDTO, action, successMsg, nil)
}