285 lines
14 KiB
Go
285 lines
14 KiB
Go
package plan
|
||
|
||
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"
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/plan"
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||
"github.com/labstack/echo/v4"
|
||
)
|
||
|
||
// --- 控制器定义 ---
|
||
|
||
// Controller 定义了计划相关的控制器
|
||
type Controller struct {
|
||
logger *logs.Logger
|
||
planService service.PlanService
|
||
}
|
||
|
||
// NewController 创建一个新的 Controller 实例
|
||
func NewController(logger *logs.Logger, planService service.PlanService) *Controller {
|
||
return &Controller{
|
||
logger: logger,
|
||
planService: planService,
|
||
}
|
||
}
|
||
|
||
// --- 接口方法实现 ---
|
||
|
||
// CreatePlan godoc
|
||
// @Summary 创建计划
|
||
// @Description 创建一个新的计划，包括其基本信息和所有关联的子计划/任务。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param plan body dto.CreatePlanRequest true "计划信息"
|
||
// @Success 200 {object} controller.Response{data=dto.PlanResponse} "业务码为201代表创建成功"
|
||
// @Router /api/v1/plans [post]
|
||
func (c *Controller) CreatePlan(ctx echo.Context) error {
|
||
var req dto.CreatePlanRequest
|
||
const actionType = "创建计划"
|
||
if err := ctx.Bind(&req); err != nil {
|
||
c.logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体: "+err.Error(), actionType, "请求体绑定失败", req)
|
||
}
|
||
|
||
// 调用服务层创建计划
|
||
resp, err := c.planService.CreatePlan(&req)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层创建计划失败: %v", actionType, err)
|
||
// 根据服务层返回的错误类型，转换为相应的HTTP状态码
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划数据校验失败或关联计划不存在", req)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "创建计划失败: "+err.Error(), actionType, "服务层创建计划失败", req)
|
||
}
|
||
|
||
// 使用统一的成功响应函数
|
||
c.logger.Infof("%s: 计划创建成功, ID: %d", actionType, resp.ID)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeCreated, "计划创建成功", resp, actionType, "计划创建成功", resp)
|
||
}
|
||
|
||
// GetPlan godoc
|
||
// @Summary 获取计划详情
|
||
// @Description 根据计划ID获取单个计划的详细信息。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Produce json
|
||
// @Param id path int true "计划ID"
|
||
// @Success 200 {object} controller.Response{data=dto.PlanResponse} "业务码为200代表成功获取"
|
||
// @Router /api/v1/plans/{id} [get]
|
||
func (c *Controller) GetPlan(ctx echo.Context) error {
|
||
const actionType = "获取计划详情"
|
||
// 1. 从 URL 路径中获取 ID
|
||
idStr := ctx.Param("id")
|
||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 计划ID格式错误: %v, ID: %s", actionType, err, idStr)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的计划ID格式", actionType, "计划ID格式错误", idStr)
|
||
}
|
||
|
||
// 调用服务层获取计划详情
|
||
resp, err := c.planService.GetPlanByID(uint(id))
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层获取计划详情失败: %v, ID: %d", actionType, err, id)
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划不存在", id)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "获取计划详情失败: "+err.Error(), actionType, "服务层获取计划详情失败", id)
|
||
}
|
||
|
||
// 4. 发送成功响应
|
||
c.logger.Infof("%s: 获取计划详情成功, ID: %d", actionType, id)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取计划详情成功", resp, actionType, "获取计划详情成功", resp)
|
||
}
|
||
|
||
// ListPlans godoc
|
||
// @Summary 获取计划列表
|
||
// @Description 获取所有计划的列表，支持按类型过滤和分页
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Produce json
|
||
// @Param query query dto.ListPlansQuery false "查询参数"
|
||
// @Success 200 {object} controller.Response{data=dto.ListPlansResponse} "业务码为200代表成功获取列表"
|
||
// @Router /api/v1/plans [get]
|
||
func (c *Controller) ListPlans(ctx echo.Context) error {
|
||
const actionType = "获取计划列表"
|
||
var query dto.ListPlansQuery
|
||
if err := ctx.Bind(&query); err != nil {
|
||
c.logger.Errorf("%s: 查询参数绑定失败: %v", actionType, err)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的查询参数: "+err.Error(), actionType, "查询参数绑定失败", query)
|
||
}
|
||
|
||
// 调用服务层获取计划列表
|
||
resp, err := c.planService.ListPlans(&query)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层获取计划列表失败: %v", actionType, err)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "获取计划列表失败: "+err.Error(), actionType, "服务层获取计划列表失败", nil)
|
||
}
|
||
|
||
c.logger.Infof("%s: 获取计划列表成功, 数量: %d", actionType, len(resp.Plans))
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取计划列表成功", resp, actionType, "获取计划列表成功", resp)
|
||
}
|
||
|
||
// UpdatePlan godoc
|
||
// @Summary 更新计划
|
||
// @Description 根据计划ID更新计划的详细信息。系统计划不允许修改。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param id path int true "计划ID"
|
||
// @Param plan body dto.UpdatePlanRequest true "更新后的计划信息"
|
||
// @Success 200 {object} controller.Response{data=dto.PlanResponse} "业务码为200代表更新成功"
|
||
// @Router /api/v1/plans/{id} [put]
|
||
func (c *Controller) UpdatePlan(ctx echo.Context) error {
|
||
const actionType = "更新计划"
|
||
// 1. 从 URL 路径中获取 ID
|
||
idStr := ctx.Param("id")
|
||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 计划ID格式错误: %v, ID: %s", actionType, err, idStr)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的计划ID格式", actionType, "计划ID格式错误", idStr)
|
||
}
|
||
|
||
// 2. 绑定请求体
|
||
var req dto.UpdatePlanRequest
|
||
if err := ctx.Bind(&req); err != nil {
|
||
c.logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体: "+err.Error(), actionType, "请求体绑定失败", req)
|
||
}
|
||
|
||
// 调用服务层更新计划
|
||
resp, err := c.planService.UpdatePlan(uint(id), &req)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层更新计划失败: %v, ID: %d", actionType, err, id)
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划不存在", id)
|
||
} else if errors.Is(err, plan.ErrPlanCannotBeModified) { // 修改为 plan.ErrPlanCannotBeModified
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeForbidden, err.Error(), actionType, "系统计划不允许修改", id)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "更新计划失败: "+err.Error(), actionType, "服务层更新计划失败", req)
|
||
}
|
||
|
||
// 9. 发送成功响应
|
||
c.logger.Infof("%s: 计划更新成功, ID: %d", actionType, resp.ID)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "计划更新成功", resp, actionType, "计划更新成功", resp)
|
||
}
|
||
|
||
// DeletePlan godoc
|
||
// @Summary 删除计划
|
||
// @Description 根据计划ID删除计划。（软删除）系统计划不允许删除。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Produce json
|
||
// @Param id path int true "计划ID"
|
||
// @Success 200 {object} controller.Response "业务码为200代表删除成功"
|
||
// @Router /api/v1/plans/{id} [delete]
|
||
func (c *Controller) DeletePlan(ctx echo.Context) error {
|
||
const actionType = "删除计划"
|
||
// 1. 从 URL 路径中获取 ID
|
||
idStr := ctx.Param("id")
|
||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 计划ID格式错误: %v, ID: %s", actionType, err, idStr)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的计划ID格式", actionType, "计划ID格式错误", idStr)
|
||
}
|
||
|
||
// 调用服务层删除计划
|
||
err = c.planService.DeletePlan(uint(id))
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层删除计划失败: %v, ID: %d", actionType, err, id)
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划不存在", id)
|
||
} else if errors.Is(err, plan.ErrPlanCannotBeDeleted) { // 修改为 plan.ErrPlanCannotBeDeleted
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeForbidden, err.Error(), actionType, "系统计划不允许删除", id)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "删除计划失败: "+err.Error(), actionType, "服务层删除计划失败", id)
|
||
}
|
||
|
||
// 6. 发送成功响应
|
||
c.logger.Infof("%s: 计划删除成功, ID: %d", actionType, id)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "计划删除成功", nil, actionType, "计划删除成功", id)
|
||
}
|
||
|
||
// StartPlan godoc
|
||
// @Summary 启动计划
|
||
// @Description 根据计划ID启动一个计划的执行。系统计划不允许手动启动。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Produce json
|
||
// @Param id path int true "计划ID"
|
||
// @Success 200 {object} controller.Response "业务码为200代表成功启动计划"
|
||
// @Router /api/v1/plans/{id}/start [post]
|
||
func (c *Controller) StartPlan(ctx echo.Context) error {
|
||
const actionType = "启动计划"
|
||
// 1. 从 URL 路径中获取 ID
|
||
idStr := ctx.Param("id")
|
||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 计划ID格式错误: %v, ID: %s", actionType, err, idStr)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的计划ID格式", actionType, "计划ID格式错误", idStr)
|
||
}
|
||
|
||
// 调用服务层启动计划
|
||
err = c.planService.StartPlan(uint(id))
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层启动计划失败: %v, ID: %d", actionType, err, id)
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划不存在", id)
|
||
} else if errors.Is(err, plan.ErrPlanCannotBeStarted) { // 修改为 plan.ErrPlanCannotBeStarted
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeForbidden, err.Error(), actionType, "系统计划不允许手动启动", id)
|
||
} else if errors.Is(err, plan.ErrPlanAlreadyEnabled) { // 修改为 plan.ErrPlanAlreadyEnabled
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, err.Error(), actionType, "计划已处于启动状态", id)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "启动计划失败: "+err.Error(), actionType, "服务层启动计划失败", id)
|
||
}
|
||
|
||
// 6. 发送成功响应
|
||
c.logger.Infof("%s: 计划已成功启动, ID: %d", actionType, id)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "计划已成功启动", nil, actionType, "计划已成功启动", id)
|
||
}
|
||
|
||
// StopPlan godoc
|
||
// @Summary 停止计划
|
||
// @Description 根据计划ID停止一个正在执行的计划。系统计划不能被停止。
|
||
// @Tags 计划管理
|
||
// @Security BearerAuth
|
||
// @Produce json
|
||
// @Param id path int true "计划ID"
|
||
// @Success 200 {object} controller.Response "业务码为200代表成功停止计划"
|
||
// @Router /api/v1/plans/{id}/stop [post]
|
||
func (c *Controller) StopPlan(ctx echo.Context) error {
|
||
const actionType = "停止计划"
|
||
// 1. 从 URL 路径中获取 ID
|
||
idStr := ctx.Param("id")
|
||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 计划ID格式错误: %v, ID: %s", actionType, err, idStr)
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的计划ID格式", actionType, "计划ID格式错误", idStr)
|
||
}
|
||
|
||
// 调用服务层停止计划
|
||
err = c.planService.StopPlan(uint(id))
|
||
if err != nil {
|
||
c.logger.Errorf("%s: 服务层停止计划失败: %v, ID: %d", actionType, err, id)
|
||
if errors.Is(err, plan.ErrPlanNotFound) { // 修改为 plan.ErrPlanNotFound
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, err.Error(), actionType, "计划不存在", id)
|
||
} else if errors.Is(err, plan.ErrPlanCannotBeStopped) { // 修改为 plan.ErrPlanCannotBeStopped
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeForbidden, err.Error(), actionType, "系统计划不允许停止", id)
|
||
} else if errors.Is(err, plan.ErrPlanNotEnabled) { // 修改为 plan.ErrPlanNotEnabled
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, err.Error(), actionType, "计划未启用", id)
|
||
}
|
||
return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "停止计划失败: "+err.Error(), actionType, "服务层停止计划失败", id)
|
||
}
|
||
|
||
// 6. 发送成功响应
|
||
c.logger.Infof("%s: 计划已成功停止, ID: %d", actionType, id)
|
||
return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "计划已成功停止", nil, actionType, "计划已成功停止", id)
|
||
}
|