Compare commits

..

6 Commits

7 changed files with 878 additions and 261 deletions

View File

@@ -0,0 +1,186 @@
package plan
import (
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
"gorm.io/datatypes"
)
// PlanToResponse 将Plan模型转换为PlanResponse
func PlanToResponse(plan *models.Plan) *PlanResponse {
if plan == nil {
return nil
}
response := &PlanResponse{
ID: plan.ID,
Name: plan.Name,
Description: plan.Description,
ExecutionType: plan.ExecutionType,
CronExpression: plan.CronExpression,
ContentType: plan.ContentType,
}
// 转换子计划
if plan.ContentType == models.PlanContentTypeSubPlans {
response.SubPlans = make([]SubPlanResponse, len(plan.SubPlans))
for i, subPlan := range plan.SubPlans {
response.SubPlans[i] = SubPlanToResponse(&subPlan)
}
}
// 转换任务
if plan.ContentType == models.PlanContentTypeTasks {
response.Tasks = make([]TaskResponse, len(plan.Tasks))
for i, task := range plan.Tasks {
response.Tasks[i] = TaskToResponse(&task)
}
}
return response
}
// PlanFromCreateRequest 将CreatePlanRequest转换为Plan模型并进行业务规则验证
func PlanFromCreateRequest(req *CreatePlanRequest) (*models.Plan, error) {
if req == nil {
return nil, nil
}
plan := &models.Plan{
Name: req.Name,
Description: req.Description,
ExecutionType: req.ExecutionType,
CronExpression: req.CronExpression,
ContentType: req.ContentType,
}
// 处理子计划 (通过ID引用)
if req.ContentType == models.PlanContentTypeSubPlans && req.SubPlanIDs != nil {
plan.SubPlans = make([]models.SubPlan, len(req.SubPlanIDs))
for i, childPlanID := range req.SubPlanIDs {
plan.SubPlans[i] = models.SubPlan{
ChildPlanID: childPlanID,
ExecutionOrder: i, // 默认执行顺序, ReorderSteps会再次确认
}
}
}
// 处理任务
if req.ContentType == models.PlanContentTypeTasks && req.Tasks != nil {
plan.Tasks = make([]models.Task, len(req.Tasks))
for i, taskReq := range req.Tasks {
// 使用来自请求的ExecutionOrder
plan.Tasks[i] = TaskFromRequest(&taskReq)
}
}
// 1. 首先,执行重复性验证
if err := plan.ValidateExecutionOrder(); err != nil {
// 如果检测到重复,立即返回错误
return nil, err
}
// 2. 然后,调用方法来修复顺序断层
plan.ReorderSteps()
return plan, nil
}
// PlanFromUpdateRequest 将UpdatePlanRequest转换为Plan模型并进行业务规则验证
func PlanFromUpdateRequest(req *UpdatePlanRequest) (*models.Plan, error) {
if req == nil {
return nil, nil
}
plan := &models.Plan{
Name: req.Name,
Description: req.Description,
ExecutionType: req.ExecutionType,
CronExpression: req.CronExpression,
ContentType: req.ContentType,
}
// 处理子计划 (通过ID引用)
if req.ContentType == models.PlanContentTypeSubPlans && req.SubPlanIDs != nil {
plan.SubPlans = make([]models.SubPlan, len(req.SubPlanIDs))
for i, childPlanID := range req.SubPlanIDs {
plan.SubPlans[i] = models.SubPlan{
ChildPlanID: childPlanID,
ExecutionOrder: i, // 默认执行顺序, ReorderSteps会再次确认
}
}
}
// 处理任务
if req.ContentType == models.PlanContentTypeTasks && req.Tasks != nil {
plan.Tasks = make([]models.Task, len(req.Tasks))
for i, taskReq := range req.Tasks {
// 使用来自请求的ExecutionOrder
plan.Tasks[i] = TaskFromRequest(&taskReq)
}
}
// 1. 首先,执行重复性验证
if err := plan.ValidateExecutionOrder(); err != nil {
// 如果检测到重复,立即返回错误
return nil, err
}
// 2. 然后,调用方法来修复顺序断层
plan.ReorderSteps()
return plan, nil
}
// SubPlanToResponse 将SubPlan模型转换为SubPlanResponse
func SubPlanToResponse(subPlan *models.SubPlan) SubPlanResponse {
if subPlan == nil {
return SubPlanResponse{}
}
response := SubPlanResponse{
ID: subPlan.ID,
ParentPlanID: subPlan.ParentPlanID,
ChildPlanID: subPlan.ChildPlanID,
ExecutionOrder: subPlan.ExecutionOrder,
}
// 如果有完整的子计划数据,也进行转换
if subPlan.ChildPlan != nil {
response.ChildPlan = PlanToResponse(subPlan.ChildPlan)
}
return response
}
// TaskToResponse 将Task模型转换为TaskResponse
func TaskToResponse(task *models.Task) TaskResponse {
if task == nil {
return TaskResponse{}
}
return TaskResponse{
ID: task.ID,
PlanID: task.PlanID,
Name: task.Name,
Description: task.Description,
ExecutionOrder: task.ExecutionOrder,
Type: task.Type,
Parameters: controller.Properties(task.Parameters),
}
}
// TaskFromRequest 将TaskRequest转换为Task模型
func TaskFromRequest(req *TaskRequest) models.Task {
if req == nil {
return models.Task{}
}
return models.Task{
Name: req.Name,
Description: req.Description,
ExecutionOrder: req.ExecutionOrder,
Type: req.Type,
Parameters: datatypes.JSON(req.Parameters),
}
}

View File

@@ -0,0 +1,459 @@
package plan_test
import (
"testing"
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller/plan"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
"github.com/stretchr/testify/assert"
"gorm.io/datatypes"
"gorm.io/gorm"
)
func TestPlanToResponse(t *testing.T) {
t.Run("nil plan", func(t *testing.T) {
response := plan.PlanToResponse(nil)
assert.Nil(t, response)
})
t.Run("basic plan without associations", func(t *testing.T) {
planModel := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "Test Plan",
Description: "A test plan",
ExecutionType: models.PlanExecutionTypeAutomatic,
CronExpression: "0 0 * * *",
ContentType: models.PlanContentTypeTasks,
}
response := plan.PlanToResponse(planModel)
assert.NotNil(t, response)
assert.Equal(t, uint(1), response.ID)
assert.Equal(t, "Test Plan", response.Name)
assert.Equal(t, "A test plan", response.Description)
assert.Equal(t, models.PlanExecutionTypeAutomatic, response.ExecutionType)
assert.Equal(t, "0 0 * * *", response.CronExpression)
assert.Equal(t, models.PlanContentTypeTasks, response.ContentType)
assert.Empty(t, response.SubPlans)
assert.Empty(t, response.Tasks)
})
t.Run("plan with sub plans", func(t *testing.T) {
childPlan := &models.Plan{
Model: gorm.Model{ID: 2},
Name: "Child Plan",
ContentType: models.PlanContentTypeTasks,
}
planModel := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "Parent Plan",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{
Model: gorm.Model{ID: 10},
ParentPlanID: 1,
ChildPlanID: 2,
ExecutionOrder: 1,
ChildPlan: childPlan,
},
},
}
response := plan.PlanToResponse(planModel)
assert.NotNil(t, response)
assert.Equal(t, uint(1), response.ID)
assert.Equal(t, "Parent Plan", response.Name)
assert.Equal(t, models.PlanContentTypeSubPlans, response.ContentType)
assert.Len(t, response.SubPlans, 1)
assert.Empty(t, response.Tasks)
subPlanResp := response.SubPlans[0]
assert.Equal(t, uint(10), subPlanResp.ID)
assert.Equal(t, uint(1), subPlanResp.ParentPlanID)
assert.Equal(t, uint(2), subPlanResp.ChildPlanID)
assert.Equal(t, 1, subPlanResp.ExecutionOrder)
assert.NotNil(t, subPlanResp.ChildPlan)
assert.Equal(t, "Child Plan", subPlanResp.ChildPlan.Name)
})
t.Run("plan with tasks", func(t *testing.T) {
params := datatypes.JSON([]byte(`{"device_id": 1, "value": 25}`))
planModel := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "Task Plan",
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{
Model: gorm.Model{ID: 10},
PlanID: 1,
Name: "Task 1",
Description: "First task",
ExecutionOrder: 1,
Type: models.TaskTypeWaiting,
Parameters: params,
},
},
}
response := plan.PlanToResponse(planModel)
assert.NotNil(t, response)
assert.Equal(t, uint(1), response.ID)
assert.Equal(t, "Task Plan", response.Name)
assert.Equal(t, models.PlanContentTypeTasks, response.ContentType)
assert.Len(t, response.Tasks, 1)
assert.Empty(t, response.SubPlans)
taskResp := response.Tasks[0]
assert.Equal(t, uint(10), taskResp.ID)
assert.Equal(t, uint(1), taskResp.PlanID)
assert.Equal(t, "Task 1", taskResp.Name)
assert.Equal(t, "First task", taskResp.Description)
assert.Equal(t, 1, taskResp.ExecutionOrder)
assert.Equal(t, models.TaskTypeWaiting, taskResp.Type)
assert.Equal(t, controller.Properties(params), taskResp.Parameters)
})
}
func TestPlanFromCreateRequest(t *testing.T) {
t.Run("nil request", func(t *testing.T) {
planModel, err := plan.PlanFromCreateRequest(nil)
assert.NoError(t, err)
assert.Nil(t, planModel)
})
t.Run("basic plan without associations", func(t *testing.T) {
req := &plan.CreatePlanRequest{
Name: "Test Plan",
Description: "A test plan",
ExecutionType: models.PlanExecutionTypeAutomatic,
CronExpression: "0 0 * * *",
ContentType: models.PlanContentTypeTasks,
}
planModel, err := plan.PlanFromCreateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Test Plan", planModel.Name)
assert.Equal(t, "A test plan", planModel.Description)
assert.Equal(t, models.PlanExecutionTypeAutomatic, planModel.ExecutionType)
assert.Equal(t, "0 0 * * *", planModel.CronExpression)
assert.Equal(t, models.PlanContentTypeTasks, planModel.ContentType)
assert.Empty(t, planModel.SubPlans)
assert.Empty(t, planModel.Tasks)
})
t.Run("plan with sub plan IDs", func(t *testing.T) {
req := &plan.CreatePlanRequest{
Name: "Parent Plan",
ContentType: models.PlanContentTypeSubPlans,
SubPlanIDs: []uint{2, 3},
}
planModel, err := plan.PlanFromCreateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Parent Plan", planModel.Name)
assert.Equal(t, models.PlanContentTypeSubPlans, planModel.ContentType)
assert.Len(t, planModel.SubPlans, 2)
assert.Empty(t, planModel.Tasks)
assert.Equal(t, uint(2), planModel.SubPlans[0].ChildPlanID)
assert.Equal(t, 1, planModel.SubPlans[0].ExecutionOrder)
assert.Equal(t, uint(3), planModel.SubPlans[1].ChildPlanID)
assert.Equal(t, 2, planModel.SubPlans[1].ExecutionOrder)
})
t.Run("plan with tasks", func(t *testing.T) {
params := controller.Properties([]byte(`{"device_id": 1, "value": 25}`))
req := &plan.CreatePlanRequest{
Name: "Task Plan",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{
Name: "Task 1",
Description: "First task",
ExecutionOrder: 1,
Type: models.TaskTypeWaiting,
Parameters: params,
},
},
}
planModel, err := plan.PlanFromCreateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Task Plan", planModel.Name)
assert.Equal(t, models.PlanContentTypeTasks, planModel.ContentType)
assert.Len(t, planModel.Tasks, 1)
assert.Empty(t, planModel.SubPlans)
task := planModel.Tasks[0]
assert.Equal(t, "Task 1", task.Name)
assert.Equal(t, "First task", task.Description)
assert.Equal(t, 1, task.ExecutionOrder)
assert.Equal(t, models.TaskTypeWaiting, task.Type)
assert.Equal(t, datatypes.JSON(params), task.Parameters)
})
t.Run("plan with tasks with gapped execution order", func(t *testing.T) {
req := &plan.CreatePlanRequest{
Name: "Task Plan with Gaps",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{Name: "Task 3", ExecutionOrder: 5},
{Name: "Task 1", ExecutionOrder: 2},
},
}
planModel, err := plan.PlanFromCreateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Len(t, planModel.Tasks, 2)
// After ReorderSteps, tasks are sorted by their original ExecutionOrder and then re-numbered.
assert.Equal(t, "Task 1", planModel.Tasks[0].Name)
assert.Equal(t, 1, planModel.Tasks[0].ExecutionOrder)
assert.Equal(t, "Task 3", planModel.Tasks[1].Name)
assert.Equal(t, 2, planModel.Tasks[1].ExecutionOrder)
})
t.Run("plan with duplicate task execution order", func(t *testing.T) {
req := &plan.CreatePlanRequest{
Name: "Invalid Plan",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{Name: "Task 1", ExecutionOrder: 1},
{Name: "Task 2", ExecutionOrder: 1}, // Duplicate order
},
}
planModel, err := plan.PlanFromCreateRequest(req)
assert.Error(t, err)
assert.Contains(t, err.Error(), "任务执行顺序重复")
assert.Nil(t, planModel)
})
}
func TestPlanFromUpdateRequest(t *testing.T) {
t.Run("nil request", func(t *testing.T) {
planModel, err := plan.PlanFromUpdateRequest(nil)
assert.NoError(t, err)
assert.Nil(t, planModel)
})
t.Run("basic plan without associations", func(t *testing.T) {
req := &plan.UpdatePlanRequest{
Name: "Updated Plan",
Description: "An updated plan",
ExecutionType: models.PlanExecutionTypeManual,
CronExpression: "0 30 * * *",
ContentType: models.PlanContentTypeTasks,
}
planModel, err := plan.PlanFromUpdateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Updated Plan", planModel.Name)
assert.Equal(t, "An updated plan", planModel.Description)
assert.Equal(t, models.PlanExecutionTypeManual, planModel.ExecutionType)
assert.Equal(t, "0 30 * * *", planModel.CronExpression)
assert.Equal(t, models.PlanContentTypeTasks, planModel.ContentType)
assert.Empty(t, planModel.SubPlans)
assert.Empty(t, planModel.Tasks)
})
t.Run("plan with sub plan IDs", func(t *testing.T) {
req := &plan.UpdatePlanRequest{
Name: "Updated Parent Plan",
ContentType: models.PlanContentTypeSubPlans,
SubPlanIDs: []uint{2, 3},
}
planModel, err := plan.PlanFromUpdateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Updated Parent Plan", planModel.Name)
assert.Equal(t, models.PlanContentTypeSubPlans, planModel.ContentType)
assert.Len(t, planModel.SubPlans, 2)
assert.Empty(t, planModel.Tasks)
assert.Equal(t, uint(2), planModel.SubPlans[0].ChildPlanID)
assert.Equal(t, 1, planModel.SubPlans[0].ExecutionOrder)
assert.Equal(t, uint(3), planModel.SubPlans[1].ChildPlanID)
assert.Equal(t, 2, planModel.SubPlans[1].ExecutionOrder)
})
t.Run("plan with tasks", func(t *testing.T) {
params := controller.Properties([]byte(`{"device_id": 1, "value": 25}`))
req := &plan.UpdatePlanRequest{
Name: "Updated Task Plan",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{
Name: "Task 1",
Description: "First task",
ExecutionOrder: 1,
Type: models.TaskTypeWaiting,
Parameters: params,
},
},
}
planModel, err := plan.PlanFromUpdateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Equal(t, "Updated Task Plan", planModel.Name)
assert.Equal(t, models.PlanContentTypeTasks, planModel.ContentType)
assert.Len(t, planModel.Tasks, 1)
assert.Empty(t, planModel.SubPlans)
task := planModel.Tasks[0]
assert.Equal(t, "Task 1", task.Name)
assert.Equal(t, 1, task.ExecutionOrder)
assert.Equal(t, datatypes.JSON(params), task.Parameters)
})
t.Run("plan with duplicate task execution order", func(t *testing.T) {
req := &plan.UpdatePlanRequest{
Name: "Invalid Updated Plan",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{Name: "Task 1", ExecutionOrder: 1},
{Name: "Task 2", ExecutionOrder: 1}, // Duplicate order
},
}
planModel, err := plan.PlanFromUpdateRequest(req)
assert.Error(t, err)
assert.Contains(t, err.Error(), "任务执行顺序重复")
assert.Nil(t, planModel)
})
t.Run("plan with tasks with gapped execution order", func(t *testing.T) {
req := &plan.UpdatePlanRequest{
Name: "Updated Task Plan with Gaps",
ContentType: models.PlanContentTypeTasks,
Tasks: []plan.TaskRequest{
{Name: "Task 3", ExecutionOrder: 5},
{Name: "Task 1", ExecutionOrder: 2},
},
}
planModel, err := plan.PlanFromUpdateRequest(req)
assert.NoError(t, err)
assert.NotNil(t, planModel)
assert.Len(t, planModel.Tasks, 2)
// After ReorderSteps, tasks are sorted by their original ExecutionOrder and then re-numbered.
assert.Equal(t, "Task 1", planModel.Tasks[0].Name)
assert.Equal(t, 1, planModel.Tasks[0].ExecutionOrder)
assert.Equal(t, "Task 3", planModel.Tasks[1].Name)
assert.Equal(t, 2, planModel.Tasks[1].ExecutionOrder)
})
}
func TestSubPlanToResponse(t *testing.T) {
t.Run("nil sub plan", func(t *testing.T) {
response := plan.SubPlanToResponse(nil)
assert.Equal(t, plan.SubPlanResponse{}, response)
})
t.Run("sub plan without child plan", func(t *testing.T) {
subPlan := &models.SubPlan{
Model: gorm.Model{ID: 10},
ParentPlanID: 1,
ChildPlanID: 2,
ExecutionOrder: 1,
}
response := plan.SubPlanToResponse(subPlan)
assert.Equal(t, uint(10), response.ID)
assert.Equal(t, uint(1), response.ParentPlanID)
assert.Equal(t, uint(2), response.ChildPlanID)
assert.Equal(t, 1, response.ExecutionOrder)
assert.Nil(t, response.ChildPlan)
})
t.Run("sub plan with child plan", func(t *testing.T) {
childPlan := &models.Plan{
Model: gorm.Model{ID: 2},
Name: "Child Plan",
}
subPlan := &models.SubPlan{
Model: gorm.Model{ID: 10},
ParentPlanID: 1,
ChildPlanID: 2,
ExecutionOrder: 1,
ChildPlan: childPlan,
}
response := plan.SubPlanToResponse(subPlan)
assert.Equal(t, uint(10), response.ID)
assert.Equal(t, uint(1), response.ParentPlanID)
assert.Equal(t, uint(2), response.ChildPlanID)
assert.Equal(t, 1, response.ExecutionOrder)
assert.NotNil(t, response.ChildPlan)
assert.Equal(t, "Child Plan", response.ChildPlan.Name)
})
}
func TestTaskToResponse(t *testing.T) {
t.Run("nil task", func(t *testing.T) {
response := plan.TaskToResponse(nil)
assert.Equal(t, plan.TaskResponse{}, response)
})
t.Run("task with parameters", func(t *testing.T) {
params := datatypes.JSON([]byte(`{"device_id": 1, "value": 25}`))
task := &models.Task{
Model: gorm.Model{ID: 10},
PlanID: 1,
Name: "Test Task",
Description: "A test task",
ExecutionOrder: 1,
Type: models.TaskTypeWaiting,
Parameters: params,
}
response := plan.TaskToResponse(task)
assert.Equal(t, uint(10), response.ID)
assert.Equal(t, uint(1), response.PlanID)
assert.Equal(t, "Test Task", response.Name)
assert.Equal(t, "A test task", response.Description)
assert.Equal(t, 1, response.ExecutionOrder)
assert.Equal(t, models.TaskTypeWaiting, response.Type)
assert.Equal(t, controller.Properties(params), response.Parameters)
})
}
func TestTaskFromRequest(t *testing.T) {
t.Run("nil request", func(t *testing.T) {
task := plan.TaskFromRequest(nil)
assert.Equal(t, models.Task{}, task)
})
t.Run("task with parameters", func(t *testing.T) {
params := controller.Properties([]byte(`{"device_id": 1, "value": 25}`))
req := &plan.TaskRequest{
Name: "Test Task",
Description: "A test task",
ExecutionOrder: 1,
Type: models.TaskTypeWaiting,
Parameters: params,
}
task := plan.TaskFromRequest(req)
assert.Equal(t, "Test Task", task.Name)
assert.Equal(t, "A test task", task.Description)
assert.Equal(t, 1, task.ExecutionOrder)
assert.Equal(t, models.TaskTypeWaiting, task.Type)
assert.Equal(t, datatypes.JSON(params), task.Parameters)
})
}

View File

@@ -3,6 +3,7 @@ package plan
import (
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
"github.com/gin-gonic/gin"
)
@@ -11,19 +12,25 @@ import (
// CreatePlanRequest 定义创建计划请求的结构体
type CreatePlanRequest struct {
Name string `json:"name" binding:"required" example:"猪舍温度控制计划"`
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
// 更多计划基本信息字段
// SubPlans 或 Tasks 可以在这里嵌套定义,以支持一次性创建
Name string `json:"name" binding:"required" example:"猪舍温度控制计划"`
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
ExecutionType models.PlanExecutionType `json:"execution_type" binding:"required" example:"automatic"`
CronExpression string `json:"cron_expression" example:"0 0 6 * * *"`
ContentType models.PlanContentType `json:"content_type" binding:"required" example:"tasks"`
SubPlanIDs []uint `json:"sub_plan_ids,omitempty"`
Tasks []TaskRequest `json:"tasks,omitempty"`
}
// PlanResponse 定义计划详情响应的结构体
type PlanResponse struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"猪舍温度控制计划"`
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
// 更多计划基本信息字段
// SubPlans 或 Tasks 也可以在这里返回
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"猪舍温度控制计划"`
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
ExecutionType models.PlanExecutionType `json:"execution_type" example:"automatic"`
CronExpression string `json:"cron_expression" example:"0 0 6 * * *"`
ContentType models.PlanContentType `json:"content_type" example:"tasks"`
SubPlans []SubPlanResponse `json:"sub_plans,omitempty"`
Tasks []TaskResponse `json:"tasks,omitempty"`
}
// ListPlansResponse 定义获取计划列表响应的结构体
@@ -34,9 +41,42 @@ type ListPlansResponse struct {
// UpdatePlanRequest 定义更新计划请求的结构体
type UpdatePlanRequest struct {
Name string `json:"name" example:"猪舍温度控制计划V2"`
Description string `json:"description" example:"更新后的描述"`
// 更多可更新字段
Name string `json:"name" example:"猪舍温度控制计划V2"`
Description string `json:"description" example:"更新后的描述"`
ExecutionType models.PlanExecutionType `json:"execution_type" example:"automatic"`
CronExpression string `json:"cron_expression" example:"0 0 6 * * *"`
ContentType models.PlanContentType `json:"content_type" example:"tasks"`
SubPlanIDs []uint `json:"sub_plan_ids,omitempty"`
Tasks []TaskRequest `json:"tasks,omitempty"`
}
// SubPlanResponse 定义子计划响应结构体
type SubPlanResponse struct {
ID uint `json:"id" example:"1"`
ParentPlanID uint `json:"parent_plan_id" example:"1"`
ChildPlanID uint `json:"child_plan_id" example:"2"`
ExecutionOrder int `json:"execution_order" example:"1"`
ChildPlan *PlanResponse `json:"child_plan,omitempty"`
}
// TaskRequest 定义任务请求结构体
type TaskRequest struct {
Name string `json:"name" example:"打开风扇"`
Description string `json:"description" example:"打开1号风扇"`
ExecutionOrder int `json:"execution_order" example:"1"`
Type models.TaskType `json:"type" example:"waiting"`
Parameters controller.Properties `json:"parameters,omitempty"`
}
// TaskResponse 定义任务响应结构体
type TaskResponse struct {
ID uint `json:"id" example:"1"`
PlanID uint `json:"plan_id" example:"1"`
Name string `json:"name" example:"打开风扇"`
Description string `json:"description" example:"打开1号风扇"`
ExecutionOrder int `json:"execution_order" example:"1"`
Type models.TaskType `json:"type" example:"waiting"`
Parameters controller.Properties `json:"parameters,omitempty"`
}
// --- Controller 定义 ---

View File

@@ -2,6 +2,7 @@ package models
import (
"fmt"
"sort"
"gorm.io/datatypes"
"gorm.io/gorm"
@@ -79,6 +80,42 @@ func (p Plan) ValidateExecutionOrder() error {
return nil
}
// ReorderSteps 重新排序计划中的步骤(任务或子计划),使其 ExecutionOrder 从 1 开始且连续。
// 这个方法假设重复的顺序已经被其他方法验证过它只负责修复断层和不从1开始的序列。
func (p *Plan) ReorderSteps() {
switch p.ContentType {
case PlanContentTypeTasks:
if len(p.Tasks) == 0 {
return
}
// 1. 按当前的 ExecutionOrder 对任务进行排序
sort.Slice(p.Tasks, func(i, j int) bool {
return p.Tasks[i].ExecutionOrder < p.Tasks[j].ExecutionOrder
})
// 2. 重新分配连续的 ExecutionOrder
for i := range p.Tasks {
p.Tasks[i].ExecutionOrder = i + 1
}
case PlanContentTypeSubPlans:
if len(p.SubPlans) == 0 {
return
}
// 1. 按当前的 ExecutionOrder 对子计划进行排序
sort.Slice(p.SubPlans, func(i, j int) bool {
return p.SubPlans[i].ExecutionOrder < p.SubPlans[j].ExecutionOrder
})
// 2. 重新分配连续的 ExecutionOrder
for i := range p.SubPlans {
p.SubPlans[i].ExecutionOrder = i + 1
}
}
}
// SubPlan 代表作为另一个计划一部分的子计划,具有执行顺序
type SubPlan struct {
gorm.Model

View File

@@ -1,22 +1,25 @@
package models_test
import (
"fmt"
"sort"
"testing"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
"github.com/stretchr/testify/assert"
)
func TestPlan_ValidateExecutionOrder(t *testing.T) {
tests := []struct {
name string
plan models.Plan
expectedError string // 期望的错误信息如果为nil则表示不期望错误
}{
func TestPlan_ReorderSteps(t *testing.T) {
type testCase struct {
name string
initialPlan *models.Plan
expectedOrders []int
}
testCases := []testCase{
// --- Test Cases for Tasks ---
{
name: "任务类型-无重复执行顺序",
plan: models.Plan{
name: "Tasks: 完美顺序",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: 1},
@@ -24,31 +27,78 @@ func TestPlan_ValidateExecutionOrder(t *testing.T) {
{ExecutionOrder: 3},
},
},
expectedError: "",
expectedOrders: []int{1, 2, 3},
},
{
name: "任务类型-重复执行顺序",
plan: models.Plan{
name: "Tasks: 有间断",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: 1},
{ExecutionOrder: 2},
{ExecutionOrder: 1}, // 重复
{ExecutionOrder: 3},
{ExecutionOrder: 5},
},
},
expectedError: fmt.Sprintf("任务执行顺序重复: %d", 1),
expectedOrders: []int{1, 2, 3},
},
{
name: "任务类型-空任务列表",
plan: models.Plan{
name: "Tasks: 从0开始",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: 0},
{ExecutionOrder: 1},
{ExecutionOrder: 2},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "Tasks: 完全无序",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: 8},
{ExecutionOrder: 2},
{ExecutionOrder: 4},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "Tasks: 包含负数",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: -5},
{ExecutionOrder: 10},
{ExecutionOrder: 2},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "Tasks: 空切片",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{},
},
expectedError: "",
expectedOrders: []int{},
},
{
name: "子计划类型-无重复执行顺序",
plan: models.Plan{
name: "Tasks: 单个元素",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeTasks,
Tasks: []models.Task{
{ExecutionOrder: 100},
},
},
expectedOrders: []int{1},
},
// --- Test Cases for SubPlans ---
{
name: "SubPlans: 完美顺序",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: 1},
@@ -56,48 +106,97 @@ func TestPlan_ValidateExecutionOrder(t *testing.T) {
{ExecutionOrder: 3},
},
},
expectedError: "",
expectedOrders: []int{1, 2, 3},
},
{
name: "子计划类型-重复执行顺序",
plan: models.Plan{
name: "SubPlans: 有间断",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: 1},
{ExecutionOrder: 2},
{ExecutionOrder: 1}, // 重复
{ExecutionOrder: 3},
{ExecutionOrder: 5},
},
},
expectedError: fmt.Sprintf("子计划执行顺序重复: %d", 1),
expectedOrders: []int{1, 2, 3},
},
{
name: "子计划类型-空子计划列表",
plan: models.Plan{
name: "SubPlans: 从0开始",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: 0},
{ExecutionOrder: 1},
{ExecutionOrder: 2},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "SubPlans: 完全无序",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: 8},
{ExecutionOrder: 2},
{ExecutionOrder: 4},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "SubPlans: 包含负数",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: -5},
{ExecutionOrder: 10},
{ExecutionOrder: 2},
},
},
expectedOrders: []int{1, 2, 3},
},
{
name: "SubPlans: 空切片",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{},
},
expectedError: "",
expectedOrders: []int{},
},
{
name: "未知内容类型",
plan: models.Plan{
ContentType: "UNKNOWN_TYPE", // 未知类型
Tasks: []models.Task{{ExecutionOrder: 1}},
name: "SubPlans: 单个元素",
initialPlan: &models.Plan{
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ExecutionOrder: 100},
},
},
expectedError: "", // 对于未知类型ValidateExecutionOrder 不会返回错误,因为它只处理已知类型
expectedOrders: []int{1},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
err := tt.plan.ValidateExecutionOrder()
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
// 调用被测试的方法
tc.initialPlan.ReorderSteps()
if tt.expectedError != "" {
assert.Error(t, err)
assert.Contains(t, err.Error(), tt.expectedError)
} else {
assert.NoError(t, err)
// 提取并验证最终的顺序
finalOrders := make([]int, 0)
if tc.initialPlan.ContentType == models.PlanContentTypeTasks {
for _, task := range tc.initialPlan.Tasks {
finalOrders = append(finalOrders, task.ExecutionOrder)
}
} else if tc.initialPlan.ContentType == models.PlanContentTypeSubPlans {
for _, subPlan := range tc.initialPlan.SubPlans {
finalOrders = append(finalOrders, subPlan.ExecutionOrder)
}
}
// 对 finalOrders 进行排序,以确保比较的一致性,因为 ReorderSteps 后的顺序是固定的
sort.Ints(finalOrders)
assert.Equal(t, tc.expectedOrders, finalOrders, "The final execution orders should be a continuous sequence starting from 1.")
})
}
}

View File

@@ -137,10 +137,10 @@ func (r *gormPlanRepository) CreatePlan(plan *models.Plan) error {
if plan.ContentType == models.PlanContentTypeSubPlans {
childIDsToValidate := make(map[uint]bool)
for _, subPlanLink := range plan.SubPlans {
if subPlanLink.ChildPlan == nil || subPlanLink.ChildPlan.ID == 0 {
if subPlanLink.ChildPlanID == 0 {
return ErrSubPlanIDIsZeroOnCreate
}
childIDsToValidate[subPlanLink.ChildPlan.ID] = true
childIDsToValidate[subPlanLink.ChildPlanID] = true
}
var ids []uint
@@ -346,10 +346,6 @@ func (r *gormPlanRepository) reconcileSubPlans(tx *gorm.DB, plan *models.Plan) e
return err
}
}
// 递归协调子计划节点
if err := r.reconcilePlanNode(tx, link.ChildPlan); err != nil {
return err
}
}
var linksToDelete []uint

View File

@@ -836,206 +836,6 @@ func TestUpdatePlan_Reconciliation(t *testing.T) {
assert.Equal(t, int64(1), taskCount, "新任务应被创建")
},
},
{
name: "递归更新-深层节点的变更",
setupDB: func(db *gorm.DB) uint {
db.Create(&models.Plan{Model: gorm.Model{ID: 1}, Name: "A", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 2}, Name: "B", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 3}, Name: "C", ContentType: models.PlanContentTypeTasks})
db.Create(&models.SubPlan{Model: gorm.Model{ID: 101}, ParentPlanID: 1, ChildPlanID: 2, ExecutionOrder: 1})
db.Create(&models.SubPlan{Model: gorm.Model{ID: 102}, ParentPlanID: 2, ChildPlanID: 3, ExecutionOrder: 1})
return 1
},
buildInput: func(db *gorm.DB) *models.Plan {
planC := &models.Plan{Model: gorm.Model{ID: 3}, Name: "C Updated", ContentType: models.PlanContentTypeTasks}
planB := &models.Plan{
Model: gorm.Model{ID: 2},
Name: "B",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{Model: gorm.Model{ID: 102}, ParentPlanID: 2, ChildPlanID: 3, ChildPlan: planC, ExecutionOrder: 2}},
}
planA := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "A Updated",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{Model: gorm.Model{ID: 101}, ParentPlanID: 1, ChildPlanID: 2, ChildPlan: planB, ExecutionOrder: 1}},
}
return planA
},
verifyDB: func(t *testing.T, db *gorm.DB, rootPlanID uint) {
var finalA, finalC models.Plan
var finalLinkB models.SubPlan
db.First(&finalA, 1)
assert.Equal(t, "A Updated", finalA.Name)
db.First(&finalLinkB, 102)
assert.Equal(t, 2, finalLinkB.ExecutionOrder)
db.First(&finalC, 3)
assert.Equal(t, "C Updated", finalC.Name)
},
},
{
name: "递归更新-深层节点的类型转换(C从SubPlans变为Tasks)",
setupDB: func(db *gorm.DB) uint {
// 初始状态: A -> B -> C -> D
db.Create(&models.Plan{Model: gorm.Model{ID: 1}, Name: "A", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 2}, Name: "B", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 3}, Name: "C", ContentType: models.PlanContentTypeSubPlans}) // C的初始类型
db.Create(&models.Plan{Model: gorm.Model{ID: 4}, Name: "D"})
db.Create(&models.SubPlan{ParentPlanID: 1, ChildPlanID: 2})
db.Create(&models.SubPlan{ParentPlanID: 2, ChildPlanID: 3})
db.Create(&models.SubPlan{Model: gorm.Model{ID: 99}, ParentPlanID: 3, ChildPlanID: 4}) // C->D 的关联
return 1
},
buildInput: func(db *gorm.DB) *models.Plan {
// 更新操作: C的类型变为Tasks并增加一个新Task
planC := &models.Plan{
Model: gorm.Model{ID: 3},
Name: "C",
ContentType: models.PlanContentTypeTasks, // 类型变更
Tasks: []models.Task{{Name: "C's New Task"}},
}
planB := &models.Plan{
Model: gorm.Model{ID: 2},
Name: "B",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 3, ChildPlan: planC}},
}
planA := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "A",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 2, ChildPlan: planB}},
}
return planA
},
verifyDB: func(t *testing.T, db *gorm.DB, rootPlanID uint) {
// 1. 验证 C->D 的旧关联已被删除
var linkCount int64
db.Model(&models.SubPlan{}).Where("id = ?", 99).Count(&linkCount)
assert.Equal(t, int64(0), linkCount, "C的旧子计划关联应被清理")
// 2. 验证 C 的新Task已被创建
var task models.Task
err := db.Where("plan_id = ?", 3).First(&task).Error
assert.NoError(t, err, "C的新任务应该被创建")
assert.Equal(t, "C's New Task", task.Name)
// 3. 验证 D 计划本身依然存在
var planDCount int64
db.Model(&models.Plan{}).Where("id = ?", 4).Count(&planDCount)
assert.Equal(t, int64(1), planDCount, "计划D本身不应被删除")
},
}, {
name: "递归更新-中间层分支替换(A->B变为A->D)",
setupDB: func(db *gorm.DB) uint {
// 初始状态: A -> B -> C
db.Create(&models.Plan{Model: gorm.Model{ID: 1}, Name: "A", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 2}, Name: "B", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 3}, Name: "C"})
db.Create(&models.SubPlan{Model: gorm.Model{ID: 101}, ParentPlanID: 1, ChildPlanID: 2}) // A->B
db.Create(&models.SubPlan{ParentPlanID: 2, ChildPlanID: 3}) // B->C
// 准备用于替换的分支: D -> E
db.Create(&models.Plan{Model: gorm.Model{ID: 4}, Name: "D", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 5}, Name: "E"})
db.Create(&models.SubPlan{ParentPlanID: 4, ChildPlanID: 5}) // D->E
return 1
},
buildInput: func(db *gorm.DB) *models.Plan {
// 更新操作: A的子计划从 B 替换为 D
planE := &models.Plan{Model: gorm.Model{ID: 5}, Name: "E Updated"} // 同时更新深层节点
planD := &models.Plan{
Model: gorm.Model{ID: 4},
Name: "D",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 5, ChildPlan: planE}},
}
planA := &models.Plan{
Model: gorm.Model{ID: 1},
Name: "A",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 4, ChildPlan: planD}}, // 新关联 A->D
}
return planA
},
verifyDB: func(t *testing.T, db *gorm.DB, rootPlanID uint) {
// 1. 验证 A->B 的旧关联已被删除
var linkCount int64
db.Model(&models.SubPlan{}).Where("id = ?", 101).Count(&linkCount)
assert.Equal(t, int64(0), linkCount, "A->B 的旧关联应被删除")
// 2. 验证 A->D 的新关联已创建
var newLink models.SubPlan
err := db.Where("parent_plan_id = ? AND child_plan_id = ?", 1, 4).First(&newLink).Error
assert.NoError(t, err, "A->D 的新关联应被创建")
// 3. 验证 B, C, D, E 计划本身都依然存在
var planCount int64
db.Model(&models.Plan{}).Where("id IN ?", []uint{2, 3, 4, 5}).Count(&planCount)
assert.Equal(t, int64(4), planCount, "所有被引用或解引用的计划本身都不应被删除")
// 4. 验证对新分支深层节点的递归更新已生效
var finalE models.Plan
db.First(&finalE, 5)
assert.Equal(t, "E Updated", finalE.Name)
},
},
{
name: "递归更新-菱形依赖下的冲突更新",
setupDB: func(db *gorm.DB) uint {
// 初始状态: A -> B -> D, A -> C -> D
db.Create(&models.Plan{Model: gorm.Model{ID: 1}, Name: "A", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 2}, Name: "B", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 3}, Name: "C", ContentType: models.PlanContentTypeSubPlans})
db.Create(&models.Plan{Model: gorm.Model{ID: 4}, Name: "D (Original)"}) // D的初始名字
// 创建关联
db.Create(&models.SubPlan{ParentPlanID: 1, ChildPlanID: 2, ExecutionOrder: 1}) // A->B
db.Create(&models.SubPlan{ParentPlanID: 1, ChildPlanID: 3, ExecutionOrder: 2}) // A->C
db.Create(&models.SubPlan{ParentPlanID: 2, ChildPlanID: 4}) // B->D
db.Create(&models.SubPlan{ParentPlanID: 3, ChildPlanID: 4}) // C->D
return 1
},
buildInput: func(db *gorm.DB) *models.Plan {
// 在一次调用中,通过不同路径对 D 进行不同的更新
planD_fromB := &models.Plan{Model: gorm.Model{ID: 4}, Name: "D (Updated from B)"}
planD_fromC := &models.Plan{Model: gorm.Model{ID: 4}, Name: "D (Updated from C)"}
planB := &models.Plan{
Model: gorm.Model{ID: 2},
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 4, ChildPlan: planD_fromB}},
}
planC := &models.Plan{
Model: gorm.Model{ID: 3},
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{{ChildPlanID: 4, ChildPlan: planD_fromC}},
}
planA := &models.Plan{
Model: gorm.Model{ID: 1},
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ChildPlanID: 2, ChildPlan: planB, ExecutionOrder: 1},
{ChildPlanID: 3, ChildPlan: planC, ExecutionOrder: 2}, // C 在 B 之后
},
}
return planA
},
verifyDB: func(t *testing.T, db *gorm.DB, rootPlanID uint) {
// 验证D 的最终名字应该符合“最后一次更新获胜”的原则
// 由于 planC 在 planB 之后被处理D 的名字应该是 "D (Updated from C)"
var finalD models.Plan
db.First(&finalD, 4)
assert.Equal(t, "D (Updated from C)", finalD.Name, "共享下游节点的更新应以后一次执行为准")
// 确保所有关联依然存在
var linkCount int64
db.Model(&models.SubPlan{}).Where("child_plan_id = ?", 4).Count(&linkCount)
assert.Equal(t, int64(2), linkCount, "D 的两个上游关联都应继续存在")
},
},
}
for _, tc := range testCases {
@@ -1134,7 +934,7 @@ func TestPlanRepository_Create(t *testing.T) {
Name: "父计划",
ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{
{ChildPlanID: 1, ExecutionOrder: 1, ChildPlan: &models.Plan{Model: gorm.Model{ID: 1}}}, // 关联已存在的子计划1
{ChildPlanID: 1, ExecutionOrder: 1}, // 关联已存在的子计划1
{ChildPlanID: 2, ExecutionOrder: 2, ChildPlan: &models.Plan{Model: gorm.Model{ID: 2}}}, // 关联已存在的子计划2
},
},