Compare commits
7 Commits
32e260ed73
...
01b11b6e42
| Author | SHA1 | Date | |
|---|---|---|---|
| 01b11b6e42 | |||
| cacb9a9b1d | |||
| 16bc3b08bc | |||
| d169f9b9d1 | |||
| fcea68c7c7 | |||
| 2b431377c1 | |||
| 986bdf15a6 |
266
docs/docs.go
266
docs/docs.go
@@ -162,6 +162,211 @@ const docTemplate = `{
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans": {
|
||||
"get": {
|
||||
"description": "获取所有计划的列表",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "获取计划列表",
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"post": {
|
||||
"description": "创建一个新的计划,包括其基本信息和所有关联的子计划/任务。",
|
||||
"consumes": [
|
||||
"application/json"
|
||||
],
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "创建计划",
|
||||
"parameters": [
|
||||
{
|
||||
"description": "计划信息",
|
||||
"name": "plan",
|
||||
"in": "body",
|
||||
"required": true,
|
||||
"schema": {
|
||||
"$ref": "#/definitions/plan.CreatePlanRequest"
|
||||
}
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}": {
|
||||
"get": {
|
||||
"description": "根据计划ID获取单个计划的详细信息。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "获取计划详情",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"put": {
|
||||
"description": "根据计划ID更新计划的详细信息。",
|
||||
"consumes": [
|
||||
"application/json"
|
||||
],
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "更新计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
},
|
||||
{
|
||||
"description": "更新后的计划信息",
|
||||
"name": "plan",
|
||||
"in": "body",
|
||||
"required": true,
|
||||
"schema": {
|
||||
"$ref": "#/definitions/plan.UpdatePlanRequest"
|
||||
}
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"delete": {
|
||||
"description": "根据计划ID删除计划。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "删除计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}/start": {
|
||||
"post": {
|
||||
"description": "根据计划ID启动一个计划的执行。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "启动计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}/stop": {
|
||||
"post": {
|
||||
"description": "根据计划ID停止一个正在执行的计划。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "停止计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/users": {
|
||||
"post": {
|
||||
"description": "根据用户名和密码创建一个新的系统用户。",
|
||||
@@ -345,6 +550,67 @@ const docTemplate = `{
|
||||
"DeviceTypeDevice"
|
||||
]
|
||||
},
|
||||
"plan.CreatePlanRequest": {
|
||||
"type": "object",
|
||||
"required": [
|
||||
"name"
|
||||
],
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "根据温度自动调节风扇和加热器"
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划"
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.ListPlansResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"plans": {
|
||||
"type": "array",
|
||||
"items": {
|
||||
"$ref": "#/definitions/plan.PlanResponse"
|
||||
}
|
||||
},
|
||||
"total": {
|
||||
"type": "integer",
|
||||
"example": 100
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.PlanResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "根据温度自动调节风扇和加热器"
|
||||
},
|
||||
"id": {
|
||||
"type": "integer",
|
||||
"example": 1
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划"
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.UpdatePlanRequest": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "更新后的描述"
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划V2"
|
||||
}
|
||||
}
|
||||
},
|
||||
"user.CreateUserRequest": {
|
||||
"type": "object",
|
||||
"required": [
|
||||
|
||||
@@ -151,6 +151,211 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans": {
|
||||
"get": {
|
||||
"description": "获取所有计划的列表",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "获取计划列表",
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"post": {
|
||||
"description": "创建一个新的计划,包括其基本信息和所有关联的子计划/任务。",
|
||||
"consumes": [
|
||||
"application/json"
|
||||
],
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "创建计划",
|
||||
"parameters": [
|
||||
{
|
||||
"description": "计划信息",
|
||||
"name": "plan",
|
||||
"in": "body",
|
||||
"required": true,
|
||||
"schema": {
|
||||
"$ref": "#/definitions/plan.CreatePlanRequest"
|
||||
}
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}": {
|
||||
"get": {
|
||||
"description": "根据计划ID获取单个计划的详细信息。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "获取计划详情",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"put": {
|
||||
"description": "根据计划ID更新计划的详细信息。",
|
||||
"consumes": [
|
||||
"application/json"
|
||||
],
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "更新计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
},
|
||||
{
|
||||
"description": "更新后的计划信息",
|
||||
"name": "plan",
|
||||
"in": "body",
|
||||
"required": true,
|
||||
"schema": {
|
||||
"$ref": "#/definitions/plan.UpdatePlanRequest"
|
||||
}
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"delete": {
|
||||
"description": "根据计划ID删除计划。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "删除计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}/start": {
|
||||
"post": {
|
||||
"description": "根据计划ID启动一个计划的执行。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "启动计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/plans/{id}/stop": {
|
||||
"post": {
|
||||
"description": "根据计划ID停止一个正在执行的计划。",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"计划管理"
|
||||
],
|
||||
"summary": "停止计划",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "integer",
|
||||
"description": "计划ID",
|
||||
"name": "id",
|
||||
"in": "path",
|
||||
"required": true
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "业务失败,具体错误码和信息见响应体(例如400, 404, 500)",
|
||||
"schema": {
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/users": {
|
||||
"post": {
|
||||
"description": "根据用户名和密码创建一个新的系统用户。",
|
||||
@@ -334,6 +539,67 @@
|
||||
"DeviceTypeDevice"
|
||||
]
|
||||
},
|
||||
"plan.CreatePlanRequest": {
|
||||
"type": "object",
|
||||
"required": [
|
||||
"name"
|
||||
],
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "根据温度自动调节风扇和加热器"
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划"
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.ListPlansResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"plans": {
|
||||
"type": "array",
|
||||
"items": {
|
||||
"$ref": "#/definitions/plan.PlanResponse"
|
||||
}
|
||||
},
|
||||
"total": {
|
||||
"type": "integer",
|
||||
"example": 100
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.PlanResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "根据温度自动调节风扇和加热器"
|
||||
},
|
||||
"id": {
|
||||
"type": "integer",
|
||||
"example": 1
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划"
|
||||
}
|
||||
}
|
||||
},
|
||||
"plan.UpdatePlanRequest": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"description": {
|
||||
"type": "string",
|
||||
"example": "更新后的描述"
|
||||
},
|
||||
"name": {
|
||||
"type": "string",
|
||||
"example": "猪舍温度控制计划V2"
|
||||
}
|
||||
}
|
||||
},
|
||||
"user.CreateUserRequest": {
|
||||
"type": "object",
|
||||
"required": [
|
||||
|
||||
@@ -79,6 +79,48 @@ definitions:
|
||||
x-enum-varnames:
|
||||
- DeviceTypeAreaController
|
||||
- DeviceTypeDevice
|
||||
plan.CreatePlanRequest:
|
||||
properties:
|
||||
description:
|
||||
example: 根据温度自动调节风扇和加热器
|
||||
type: string
|
||||
name:
|
||||
example: 猪舍温度控制计划
|
||||
type: string
|
||||
required:
|
||||
- name
|
||||
type: object
|
||||
plan.ListPlansResponse:
|
||||
properties:
|
||||
plans:
|
||||
items:
|
||||
$ref: '#/definitions/plan.PlanResponse'
|
||||
type: array
|
||||
total:
|
||||
example: 100
|
||||
type: integer
|
||||
type: object
|
||||
plan.PlanResponse:
|
||||
properties:
|
||||
description:
|
||||
example: 根据温度自动调节风扇和加热器
|
||||
type: string
|
||||
id:
|
||||
example: 1
|
||||
type: integer
|
||||
name:
|
||||
example: 猪舍温度控制计划
|
||||
type: string
|
||||
type: object
|
||||
plan.UpdatePlanRequest:
|
||||
properties:
|
||||
description:
|
||||
example: 更新后的描述
|
||||
type: string
|
||||
name:
|
||||
example: 猪舍温度控制计划V2
|
||||
type: string
|
||||
type: object
|
||||
user.CreateUserRequest:
|
||||
properties:
|
||||
password:
|
||||
@@ -225,6 +267,141 @@ paths:
|
||||
summary: 更新设备信息
|
||||
tags:
|
||||
- 设备管理
|
||||
/plans:
|
||||
get:
|
||||
description: 获取所有计划的列表
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 获取计划列表
|
||||
tags:
|
||||
- 计划管理
|
||||
post:
|
||||
consumes:
|
||||
- application/json
|
||||
description: 创建一个新的计划,包括其基本信息和所有关联的子计划/任务。
|
||||
parameters:
|
||||
- description: 计划信息
|
||||
in: body
|
||||
name: plan
|
||||
required: true
|
||||
schema:
|
||||
$ref: '#/definitions/plan.CreatePlanRequest'
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 创建计划
|
||||
tags:
|
||||
- 计划管理
|
||||
/plans/{id}:
|
||||
delete:
|
||||
description: 根据计划ID删除计划。
|
||||
parameters:
|
||||
- description: 计划ID
|
||||
in: path
|
||||
name: id
|
||||
required: true
|
||||
type: integer
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 404, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 删除计划
|
||||
tags:
|
||||
- 计划管理
|
||||
get:
|
||||
description: 根据计划ID获取单个计划的详细信息。
|
||||
parameters:
|
||||
- description: 计划ID
|
||||
in: path
|
||||
name: id
|
||||
required: true
|
||||
type: integer
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 404, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 获取计划详情
|
||||
tags:
|
||||
- 计划管理
|
||||
put:
|
||||
consumes:
|
||||
- application/json
|
||||
description: 根据计划ID更新计划的详细信息。
|
||||
parameters:
|
||||
- description: 计划ID
|
||||
in: path
|
||||
name: id
|
||||
required: true
|
||||
type: integer
|
||||
- description: 更新后的计划信息
|
||||
in: body
|
||||
name: plan
|
||||
required: true
|
||||
schema:
|
||||
$ref: '#/definitions/plan.UpdatePlanRequest'
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 404, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 更新计划
|
||||
tags:
|
||||
- 计划管理
|
||||
/plans/{id}/start:
|
||||
post:
|
||||
description: 根据计划ID启动一个计划的执行。
|
||||
parameters:
|
||||
- description: 计划ID
|
||||
in: path
|
||||
name: id
|
||||
required: true
|
||||
type: integer
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 404, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 启动计划
|
||||
tags:
|
||||
- 计划管理
|
||||
/plans/{id}/stop:
|
||||
post:
|
||||
description: 根据计划ID停止一个正在执行的计划。
|
||||
parameters:
|
||||
- description: 计划ID
|
||||
in: path
|
||||
name: id
|
||||
required: true
|
||||
type: integer
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: 业务失败,具体错误码和信息见响应体(例如400, 404, 500)
|
||||
schema:
|
||||
$ref: '#/definitions/controller.Response'
|
||||
summary: 停止计划
|
||||
tags:
|
||||
- 计划管理
|
||||
/users:
|
||||
post:
|
||||
consumes:
|
||||
|
||||
@@ -17,6 +17,7 @@ import (
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller/device"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller/plan"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/controller/user"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/service/token"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
|
||||
@@ -39,11 +40,12 @@ type API struct {
|
||||
config config.ServerConfig // API 服务器的配置,使用 infra/config 包中的 ServerConfig
|
||||
userController *user.Controller // 用户控制器实例
|
||||
deviceController *device.Controller // 设备控制器实例
|
||||
planController *plan.Controller // 计划控制器实例
|
||||
}
|
||||
|
||||
// NewAPI 创建并返回一个新的 API 实例
|
||||
// 负责初始化 Gin 引擎、设置全局中间件,并注入所有必要的依赖。
|
||||
func NewAPI(cfg config.ServerConfig, logger *logs.Logger, userRepo repository.UserRepository, deviceRepository repository.DeviceRepository, tokenService token.TokenService) *API {
|
||||
func NewAPI(cfg config.ServerConfig, logger *logs.Logger, userRepo repository.UserRepository, deviceRepository repository.DeviceRepository, planRepository repository.PlanRepository, tokenService token.TokenService) *API {
|
||||
// 设置 Gin 模式,例如 gin.ReleaseMode (生产模式) 或 gin.DebugMode (开发模式)
|
||||
// 从配置中获取 Gin 模式
|
||||
gin.SetMode(cfg.Mode)
|
||||
@@ -67,6 +69,8 @@ func NewAPI(cfg config.ServerConfig, logger *logs.Logger, userRepo repository.Us
|
||||
userController: user.NewController(userRepo, logger, tokenService),
|
||||
// 在 NewAPI 中初始化设备控制器,并将其作为 API 结构体的成员
|
||||
deviceController: device.NewController(deviceRepository, logger),
|
||||
// 在 NewAPI 中初始化计划控制器,并将其作为 API 结构体的成员
|
||||
planController: plan.NewController(logger, planRepository),
|
||||
}
|
||||
|
||||
api.setupRoutes() // 设置所有路由
|
||||
@@ -95,6 +99,18 @@ func (a *API) setupRoutes() {
|
||||
deviceGroup.PUT("/:id", a.deviceController.UpdateDevice)
|
||||
deviceGroup.DELETE("/:id", a.deviceController.DeleteDevice)
|
||||
}
|
||||
|
||||
// 计划相关路由组
|
||||
planGroup := v1.Group("/plans")
|
||||
{
|
||||
planGroup.POST("", a.planController.CreatePlan)
|
||||
planGroup.GET("", a.planController.ListPlans)
|
||||
planGroup.GET("/:id", a.planController.GetPlan)
|
||||
planGroup.PUT("/:id", a.planController.UpdatePlan)
|
||||
planGroup.DELETE("/:id", a.planController.DeletePlan)
|
||||
planGroup.POST("/:id/start", a.planController.StartPlan)
|
||||
planGroup.POST("/:id/stop", a.planController.StopPlan)
|
||||
}
|
||||
}
|
||||
|
||||
// 添加 Swagger UI 路由
|
||||
|
||||
168
internal/app/controller/plan/plan_controller.go
Normal file
168
internal/app/controller/plan/plan_controller.go
Normal file
@@ -0,0 +1,168 @@
|
||||
package plan
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"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/repository"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// --- 请求和响应 DTO 定义 ---
|
||||
|
||||
// CreatePlanRequest 定义创建计划请求的结构体
|
||||
type CreatePlanRequest struct {
|
||||
Name string `json:"name" binding:"required" example:"猪舍温度控制计划"`
|
||||
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
|
||||
// 更多计划基本信息字段
|
||||
// SubPlans 或 Tasks 可以在这里嵌套定义,以支持一次性创建
|
||||
}
|
||||
|
||||
// PlanResponse 定义计划详情响应的结构体
|
||||
type PlanResponse struct {
|
||||
ID uint `json:"id" example:"1"`
|
||||
Name string `json:"name" example:"猪舍温度控制计划"`
|
||||
Description string `json:"description" example:"根据温度自动调节风扇和加热器"`
|
||||
// 更多计划基本信息字段
|
||||
// SubPlans 或 Tasks 也可以在这里返回
|
||||
}
|
||||
|
||||
// ListPlansResponse 定义获取计划列表响应的结构体
|
||||
type ListPlansResponse struct {
|
||||
Plans []PlanResponse `json:"plans"`
|
||||
Total int `json:"total" example:"100"`
|
||||
}
|
||||
|
||||
// UpdatePlanRequest 定义更新计划请求的结构体
|
||||
type UpdatePlanRequest struct {
|
||||
Name string `json:"name" example:"猪舍温度控制计划V2"`
|
||||
Description string `json:"description" example:"更新后的描述"`
|
||||
// 更多可更新字段
|
||||
}
|
||||
|
||||
// --- Controller 定义 ---
|
||||
|
||||
// Controller 定义了计划相关的控制器
|
||||
type Controller struct {
|
||||
logger *logs.Logger
|
||||
planRepo repository.PlanRepository
|
||||
}
|
||||
|
||||
// NewController 创建一个新的 Controller 实例
|
||||
func NewController(logger *logs.Logger, planRepo repository.PlanRepository) *Controller {
|
||||
return &Controller{
|
||||
logger: logger,
|
||||
planRepo: planRepo,
|
||||
}
|
||||
}
|
||||
|
||||
// --- 接口方法实现 ---
|
||||
|
||||
// CreatePlan godoc
|
||||
// @Summary 创建计划
|
||||
// @Description 创建一个新的计划,包括其基本信息和所有关联的子计划/任务。
|
||||
// @Tags 计划管理
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Param plan body CreatePlanRequest true "计划信息"
|
||||
// @Success 200 {object} controller.Response{data=plan.PlanResponse} "业务码为201代表创建成功"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 500)"
|
||||
// @Router /plans [post]
|
||||
func (pc *Controller) CreatePlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来创建计划
|
||||
pc.logger.Infof("收到创建计划请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusCreated, "创建计划接口占位符", PlanResponse{ID: 0, Name: "占位计划"})
|
||||
}
|
||||
|
||||
// GetPlan godoc
|
||||
// @Summary 获取计划详情
|
||||
// @Description 根据计划ID获取单个计划的详细信息。
|
||||
// @Tags 计划管理
|
||||
// @Produce json
|
||||
// @Param id path int true "计划ID"
|
||||
// @Success 200 {object} controller.Response{data=plan.PlanResponse} "业务码为200代表成功获取"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 404, 500)"
|
||||
// @Router /plans/{id} [get]
|
||||
func (pc *Controller) GetPlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来获取计划
|
||||
pc.logger.Infof("收到获取计划 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "获取计划接口占位符", ListPlansResponse{Plans: []PlanResponse{}, Total: 0})
|
||||
|
||||
}
|
||||
|
||||
// ListPlans godoc
|
||||
// @Summary 获取计划列表
|
||||
// @Description 获取所有计划的列表
|
||||
// @Tags 计划管理
|
||||
// @Produce json
|
||||
// @Success 200 {object} controller.Response{data=plan.ListPlansResponse} "业务码为200代表成功获取列表"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 500)"
|
||||
// @Router /plans [get]
|
||||
func (pc *Controller) ListPlans(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来获取计划列表
|
||||
pc.logger.Infof("收到获取计划列表请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "获取计划列表接口占位符", ListPlansResponse{Plans: []PlanResponse{}, Total: 0})
|
||||
}
|
||||
|
||||
// UpdatePlan godoc
|
||||
// @Summary 更新计划
|
||||
// @Description 根据计划ID更新计划的详细信息。
|
||||
// @Tags 计划管理
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Param id path int true "计划ID"
|
||||
// @Param plan body UpdatePlanRequest true "更新后的计划信息"
|
||||
// @Success 200 {object} controller.Response{data=plan.PlanResponse} "业务码为200代表更新成功"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 404, 500)"
|
||||
// @Router /plans/{id} [put]
|
||||
func (pc *Controller) UpdatePlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来更新计划
|
||||
pc.logger.Infof("收到更新计划请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "更新计划接口占位符", PlanResponse{ID: 0, Name: "占位计划"})
|
||||
}
|
||||
|
||||
// DeletePlan godoc
|
||||
// @Summary 删除计划
|
||||
// @Description 根据计划ID删除计划。
|
||||
// @Tags 计划管理
|
||||
// @Produce json
|
||||
// @Param id path int true "计划ID"
|
||||
// @Success 200 {object} controller.Response "业务码为200代表删除成功"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 404, 500)"
|
||||
// @Router /plans/{id} [delete]
|
||||
func (pc *Controller) DeletePlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来删除计划
|
||||
pc.logger.Infof("收到删除计划请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "删除计划接口占位符", nil)
|
||||
}
|
||||
|
||||
// StartPlan godoc
|
||||
// @Summary 启动计划
|
||||
// @Description 根据计划ID启动一个计划的执行。
|
||||
// @Tags 计划管理
|
||||
// @Produce json
|
||||
// @Param id path int true "计划ID"
|
||||
// @Success 200 {object} controller.Response "业务码为200代表成功启动计划"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 404, 500)"
|
||||
// @Router /plans/{id}/start [post]
|
||||
func (pc *Controller) StartPlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来启动计划
|
||||
pc.logger.Infof("收到启动计划请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "启动计划接口占位符", nil)
|
||||
}
|
||||
|
||||
// StopPlan godoc
|
||||
// @Summary 停止计划
|
||||
// @Description 根据计划ID停止一个正在执行的计划。
|
||||
// @Tags 计划管理
|
||||
// @Produce json
|
||||
// @Param id path int true "计划ID"
|
||||
// @Success 200 {object} controller.Response "业务码为200代表成功停止计划"
|
||||
// @Failure 200 {object} controller.Response "业务失败,具体错误码和信息见响应体(例如400, 404, 500)"
|
||||
// @Router /plans/{id}/stop [post]
|
||||
func (pc *Controller) StopPlan(c *gin.Context) {
|
||||
// 占位符:此处应调用服务层或仓库层来停止计划
|
||||
pc.logger.Infof("收到停止计划请求 (占位符)")
|
||||
controller.SendResponse(c, http.StatusOK, "停止计划接口占位符", nil)
|
||||
}
|
||||
@@ -55,10 +55,13 @@ func NewApplication(configPath string) (*Application, error) {
|
||||
// 7. 初始化设备仓库
|
||||
deviceRepo := repository.NewGormDeviceRepository(storage.GetDB())
|
||||
|
||||
// 8. 初始化 API 服务器
|
||||
apiServer := api.NewAPI(cfg.Server, logger, userRepo, deviceRepo, tokenService)
|
||||
// 8. 初始化计划仓库
|
||||
planRepo := repository.NewGormPlanRepository(storage.GetDB())
|
||||
|
||||
// 9. 组装 Application 对象
|
||||
// 9. 初始化 API 服务器
|
||||
apiServer := api.NewAPI(cfg.Server, logger, userRepo, deviceRepo, planRepo, tokenService)
|
||||
|
||||
// 10. 组装 Application 对象
|
||||
app := &Application{
|
||||
Config: cfg,
|
||||
Logger: logger,
|
||||
@@ -121,15 +124,7 @@ func initStorage(cfg config.DatabaseConfig, logger *logs.Logger) (database.Stora
|
||||
}
|
||||
|
||||
// 执行数据库迁移
|
||||
// 这里需要添加所有需要自动迁移的模型
|
||||
var dbModels = []interface{}{
|
||||
&models.User{},
|
||||
&models.Device{},
|
||||
&models.Plan{},
|
||||
&models.SubPlan{},
|
||||
&models.Task{},
|
||||
}
|
||||
if err := storage.Migrate(dbModels...); err != nil {
|
||||
if err := storage.Migrate(models.GetAllModels()...); err != nil {
|
||||
return nil, fmt.Errorf("数据库迁移失败: %w", err)
|
||||
}
|
||||
|
||||
|
||||
13
internal/infra/models/models.go
Normal file
13
internal/infra/models/models.go
Normal file
@@ -0,0 +1,13 @@
|
||||
package models
|
||||
|
||||
// GetAllModels 返回一个包含所有数据库模型实例的切片。
|
||||
// 这个函数用于在数据库初始化时自动迁移所有的表结构。
|
||||
func GetAllModels() []interface{} {
|
||||
return []interface{}{
|
||||
&User{},
|
||||
&Device{},
|
||||
&Plan{},
|
||||
&SubPlan{},
|
||||
&Task{},
|
||||
}
|
||||
}
|
||||
@@ -58,9 +58,10 @@ func (Plan) TableName() string {
|
||||
type SubPlan struct {
|
||||
gorm.Model
|
||||
|
||||
ParentPlanID uint `gorm:"not null;index" json:"parent_plan_id"` // 父计划的ID
|
||||
ChildPlanID uint `gorm:"not null;index" json:"child_plan_id"` // 子计划的ID (它本身也是一个 Plan)
|
||||
Order int `gorm:"not null" json:"order"` // 在父计划中的执行顺序
|
||||
ParentPlanID uint `gorm:"not null;index" json:"parent_plan_id"` // 父计划的ID
|
||||
ChildPlanID uint `gorm:"not null;index" json:"child_plan_id"` // 子计划的ID (它本身也是一个 Plan)
|
||||
ExecutionOrder int `gorm:"not null" json:"execution_order"` // 在父计划中的执行顺序
|
||||
ChildPlan *Plan `gorm:"-" json:"child_plan"` // 完整子计划数据,仅内存中
|
||||
}
|
||||
|
||||
// TableName 自定义 GORM 使用的数据库表名
|
||||
@@ -72,12 +73,12 @@ func (SubPlan) TableName() string {
|
||||
type Task struct {
|
||||
gorm.Model
|
||||
|
||||
PlanID uint `gorm:"not null;index" json:"plan_id"` // 此任务所属计划的ID
|
||||
Name string `gorm:"not null" json:"name"`
|
||||
Description string `json:"description"`
|
||||
Order int `gorm:"not null" json:"order"` // 在计划中的执行顺序
|
||||
Type TaskType `gorm:"not null" json:"type"` // 任务的类型,对应 task 包中的具体动作
|
||||
Parameters datatypes.JSON `json:"parameters"` // 任务特定参数的JSON (例如: 设备ID, 值)
|
||||
PlanID uint `gorm:"not null;index" json:"plan_id"` // 此任务所属计划的ID
|
||||
Name string `gorm:"not null" json:"name"`
|
||||
Description string `json:"description"`
|
||||
ExecutionOrder int `gorm:"not null" json:"execution_order"` // 在计划中的执行顺序
|
||||
Type TaskType `gorm:"not null" json:"type"` // 任务的类型,对应 task 包中的具体动作
|
||||
Parameters datatypes.JSON `json:"parameters"` // 任务特定参数的JSON (例如: 设备ID, 值)
|
||||
}
|
||||
|
||||
// TableName 自定义 GORM 使用的数据库表名
|
||||
|
||||
@@ -12,12 +12,12 @@ import (
|
||||
|
||||
// setupTestDB 是一个共享的辅助函数,用于为集成测试创建一个干净的、内存中的 SQLite 数据库实例。
|
||||
func setupTestDB(t *testing.T) *gorm.DB {
|
||||
// "file::memory:?cache=shared" 是 GORM 连接内存 SQLite 的标准方式,确保在同一测试中的不同连接可以访问相同的数据。
|
||||
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
||||
// "file::memory:?cache=shared" 是 GORM 连接内存 SQLite 的标准方式,确保在同一测试中的不同连接可以访问相同的数据,而我们显然不需要这个
|
||||
db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{})
|
||||
assert.NoError(t, err, "连接内存数据库时发生错误")
|
||||
|
||||
// 自动迁移所有需要的表结构
|
||||
err = db.AutoMigrate(&models.User{}, &models.Device{}, &models.SubPlan{}, &models.Task{}, &models.Plan{})
|
||||
err = db.AutoMigrate(models.GetAllModels()...)
|
||||
assert.NoError(t, err, "数据库迁移时发生错误")
|
||||
|
||||
return db
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
@@ -12,6 +14,8 @@ type PlanRepository interface {
|
||||
ListBasicPlans() ([]models.Plan, error)
|
||||
// GetBasicPlanByID 根据ID获取计划的基本信息,不包含子计划和任务详情
|
||||
GetBasicPlanByID(id uint) (*models.Plan, error)
|
||||
// GetPlanByID 根据ID获取计划,包含子计划和任务详情
|
||||
GetPlanByID(id uint) (*models.Plan, error)
|
||||
}
|
||||
|
||||
// gormPlanRepository 是 PlanRepository 的 GORM 实现
|
||||
@@ -47,3 +51,48 @@ func (r *gormPlanRepository) GetBasicPlanByID(id uint) (*models.Plan, error) {
|
||||
}
|
||||
return &plan, nil
|
||||
}
|
||||
|
||||
// GetPlanByID 根据ID获取计划,包含子计划和任务详情
|
||||
func (r *gormPlanRepository) GetPlanByID(id uint) (*models.Plan, error) {
|
||||
var plan models.Plan
|
||||
|
||||
// 先获取基本计划信息
|
||||
result := r.db.First(&plan, id)
|
||||
if result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
|
||||
// 根据内容类型加载关联数据
|
||||
switch plan.ContentType {
|
||||
case models.PlanContentTypeSubPlans:
|
||||
// 加载子计划引用
|
||||
var subPlans []models.SubPlan
|
||||
result = r.db.Where("parent_plan_id = ?", plan.ID).Order("execution_order").Find(&subPlans)
|
||||
if result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
|
||||
// 递归加载每个子计划的完整信息
|
||||
for i := range subPlans {
|
||||
childPlan, err := r.GetPlanByID(subPlans[i].ChildPlanID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
subPlans[i].ChildPlan = childPlan
|
||||
}
|
||||
|
||||
plan.SubPlans = subPlans
|
||||
case models.PlanContentTypeTasks:
|
||||
// 加载任务
|
||||
result = r.db.Preload("Tasks", func(taskDB *gorm.DB) *gorm.DB {
|
||||
return taskDB.Order("execution_order")
|
||||
}).First(&plan, id)
|
||||
if result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
default:
|
||||
return nil, fmt.Errorf("未知的计划内容类型: %v; 计划ID: %v", plan.ContentType, plan.ID)
|
||||
}
|
||||
|
||||
return &plan, nil
|
||||
}
|
||||
|
||||
@@ -2,7 +2,9 @@ package repository_test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||||
@@ -134,3 +136,265 @@ func TestGetBasicPlanByID(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// TestGetPlanByID 测试 GetPlanByID 方法,确保它能根据ID正确返回计划的完整信息,包括关联数据。
|
||||
func TestGetPlanByID(t *testing.T) {
|
||||
type testCase struct {
|
||||
name string
|
||||
setupData func(db *gorm.DB) // 用于在测试前插入数据
|
||||
planID uint
|
||||
expectedPlan *models.Plan
|
||||
expectedError string
|
||||
}
|
||||
|
||||
testCases := []testCase{
|
||||
{
|
||||
name: "PlanNotFound",
|
||||
setupData: func(db *gorm.DB) {
|
||||
// 不插入任何数据
|
||||
},
|
||||
planID: 999,
|
||||
expectedPlan: nil,
|
||||
expectedError: "record not found",
|
||||
},
|
||||
{
|
||||
name: "PlanWithTasks",
|
||||
setupData: func(db *gorm.DB) {
|
||||
// 使用硬编码的ID创建计划,使测试可预测
|
||||
plan := models.Plan{
|
||||
Model: gorm.Model{ID: 1},
|
||||
Name: "Plan A",
|
||||
ContentType: models.PlanContentTypeTasks,
|
||||
}
|
||||
db.Create(&plan)
|
||||
// 创建任务,它们的ID将由数据库自动生成
|
||||
db.Create(&models.Task{PlanID: 1, Name: "Task 2", ExecutionOrder: 1})
|
||||
db.Create(&models.Task{PlanID: 1, Name: "Task 1", ExecutionOrder: 2})
|
||||
},
|
||||
planID: 1,
|
||||
expectedPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 1},
|
||||
Name: "Plan A",
|
||||
ContentType: models.PlanContentTypeTasks,
|
||||
Tasks: []models.Task{
|
||||
// 期望按 "order" 字段升序排序
|
||||
{PlanID: 1, Name: "Task 2", ExecutionOrder: 1},
|
||||
{PlanID: 1, Name: "Task 1", ExecutionOrder: 2},
|
||||
},
|
||||
},
|
||||
expectedError: "",
|
||||
},
|
||||
{
|
||||
name: "PlanWithMultiLevelSubPlans",
|
||||
setupData: func(db *gorm.DB) {
|
||||
// 创建一个三层结构的计划
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 20}, Name: "Grandparent Plan", ContentType: models.PlanContentTypeSubPlans})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 21}, Name: "Parent Plan", ContentType: models.PlanContentTypeSubPlans})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 22}, Name: "Child Plan With Tasks", ContentType: models.PlanContentTypeTasks})
|
||||
db.Create(&models.Task{PlanID: 22, Name: "Grandchild's Task", ExecutionOrder: 1})
|
||||
|
||||
// 创建关联关系
|
||||
db.Create(&models.SubPlan{ParentPlanID: 20, ChildPlanID: 21, ExecutionOrder: 1})
|
||||
db.Create(&models.SubPlan{ParentPlanID: 21, ChildPlanID: 22, ExecutionOrder: 1})
|
||||
},
|
||||
planID: 20,
|
||||
expectedPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 20},
|
||||
Name: "Grandparent Plan",
|
||||
ContentType: models.PlanContentTypeSubPlans,
|
||||
SubPlans: []models.SubPlan{
|
||||
{
|
||||
ParentPlanID: 20,
|
||||
ChildPlanID: 21,
|
||||
ExecutionOrder: 1,
|
||||
ChildPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 21},
|
||||
Name: "Parent Plan",
|
||||
ContentType: models.PlanContentTypeSubPlans,
|
||||
SubPlans: []models.SubPlan{
|
||||
{
|
||||
ParentPlanID: 21,
|
||||
ChildPlanID: 22,
|
||||
ExecutionOrder: 1,
|
||||
ChildPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 22},
|
||||
Name: "Child Plan With Tasks",
|
||||
ContentType: models.PlanContentTypeTasks,
|
||||
Tasks: []models.Task{
|
||||
{PlanID: 22, Name: "Grandchild's Task", ExecutionOrder: 1},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
expectedError: "",
|
||||
},
|
||||
{
|
||||
name: "UnknownContentType",
|
||||
setupData: func(db *gorm.DB) {
|
||||
db.Create(&models.Plan{
|
||||
Model: gorm.Model{ID: 30},
|
||||
Name: "Unknown Type Plan",
|
||||
ContentType: "INVALID_TYPE",
|
||||
})
|
||||
},
|
||||
planID: 30,
|
||||
expectedPlan: nil,
|
||||
expectedError: fmt.Sprintf("未知的计划内容类型: INVALID_TYPE; 计划ID: 30"),
|
||||
},
|
||||
// 新增场景:测试空的关联数据
|
||||
{
|
||||
name: "PlanWithTasksType_ButNoTasks",
|
||||
setupData: func(db *gorm.DB) {
|
||||
db.Create(&models.Plan{
|
||||
Model: gorm.Model{ID: 50},
|
||||
Name: "Plan with empty tasks",
|
||||
ContentType: models.PlanContentTypeTasks,
|
||||
})
|
||||
},
|
||||
planID: 50,
|
||||
expectedPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 50},
|
||||
Name: "Plan with empty tasks",
|
||||
ContentType: models.PlanContentTypeTasks,
|
||||
Tasks: []models.Task{}, // 期望一个空的切片
|
||||
},
|
||||
expectedError: "",
|
||||
},
|
||||
// 新增场景:测试复杂的同级排序
|
||||
{
|
||||
name: "PlanWithSubPlans_ComplexSorting",
|
||||
setupData: func(db *gorm.DB) {
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 60}, Name: "Main Plan For Sorting", ContentType: models.PlanContentTypeSubPlans})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 61}, Name: "SubPlan C", ContentType: models.PlanContentTypeTasks})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 62}, Name: "SubPlan A", ContentType: models.PlanContentTypeTasks})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 63}, Name: "SubPlan B", ContentType: models.PlanContentTypeTasks})
|
||||
// 故意打乱顺序插入
|
||||
db.Create(&models.SubPlan{ParentPlanID: 60, ChildPlanID: 61, ExecutionOrder: 3})
|
||||
db.Create(&models.SubPlan{ParentPlanID: 60, ChildPlanID: 62, ExecutionOrder: 1})
|
||||
db.Create(&models.SubPlan{ParentPlanID: 60, ChildPlanID: 63, ExecutionOrder: 2})
|
||||
},
|
||||
planID: 60,
|
||||
expectedPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 60},
|
||||
Name: "Main Plan For Sorting",
|
||||
ContentType: models.PlanContentTypeSubPlans,
|
||||
SubPlans: []models.SubPlan{
|
||||
{ParentPlanID: 60, ChildPlanID: 62, ExecutionOrder: 1, ChildPlan: &models.Plan{Model: gorm.Model{ID: 62}, Name: "SubPlan A", ContentType: models.PlanContentTypeTasks, Tasks: []models.Task{}}},
|
||||
{ParentPlanID: 60, ChildPlanID: 63, ExecutionOrder: 2, ChildPlan: &models.Plan{Model: gorm.Model{ID: 63}, Name: "SubPlan B", ContentType: models.PlanContentTypeTasks, Tasks: []models.Task{}}},
|
||||
{ParentPlanID: 60, ChildPlanID: 61, ExecutionOrder: 3, ChildPlan: &models.Plan{Model: gorm.Model{ID: 61}, Name: "SubPlan C", ContentType: models.PlanContentTypeTasks, Tasks: []models.Task{}}},
|
||||
},
|
||||
},
|
||||
expectedError: "",
|
||||
},
|
||||
// 新增场景:测试混合内容的子计划树
|
||||
{
|
||||
name: "PlanWithSubPlans_MixedContentTypes",
|
||||
setupData: func(db *gorm.DB) {
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 70}, Name: "Mixed Main Plan", ContentType: models.PlanContentTypeSubPlans})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 71}, Name: "Child with SubPlans", ContentType: models.PlanContentTypeSubPlans})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 72}, Name: "Grandchild with Tasks", ContentType: models.PlanContentTypeTasks})
|
||||
db.Create(&models.Task{PlanID: 72, Name: "Grandchild's Task", ExecutionOrder: 1})
|
||||
db.Create(&models.Plan{Model: gorm.Model{ID: 73}, Name: "Child with Tasks", ContentType: models.PlanContentTypeTasks})
|
||||
db.Create(&models.Task{PlanID: 73, Name: "Child's Task", ExecutionOrder: 1})
|
||||
|
||||
// 创建关联
|
||||
db.Create(&models.SubPlan{ParentPlanID: 70, ChildPlanID: 71, ExecutionOrder: 1}) // Main -> Child with SubPlans
|
||||
db.Create(&models.SubPlan{ParentPlanID: 70, ChildPlanID: 73, ExecutionOrder: 2}) // Main -> Child with Tasks
|
||||
db.Create(&models.SubPlan{ParentPlanID: 71, ChildPlanID: 72, ExecutionOrder: 1}) // Child with SubPlans -> Grandchild
|
||||
},
|
||||
planID: 70,
|
||||
expectedPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 70},
|
||||
Name: "Mixed Main Plan",
|
||||
ContentType: models.PlanContentTypeSubPlans,
|
||||
SubPlans: []models.SubPlan{
|
||||
{
|
||||
ParentPlanID: 70, ChildPlanID: 71, ExecutionOrder: 1,
|
||||
ChildPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 71}, Name: "Child with SubPlans", ContentType: models.PlanContentTypeSubPlans,
|
||||
SubPlans: []models.SubPlan{
|
||||
{ParentPlanID: 71, ChildPlanID: 72, ExecutionOrder: 1, ChildPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 72}, Name: "Grandchild with Tasks", ContentType: models.PlanContentTypeTasks,
|
||||
Tasks: []models.Task{{PlanID: 72, Name: "Grandchild's Task", ExecutionOrder: 1}},
|
||||
}},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
ParentPlanID: 70, ChildPlanID: 73, ExecutionOrder: 2,
|
||||
ChildPlan: &models.Plan{
|
||||
Model: gorm.Model{ID: 73}, Name: "Child with Tasks", ContentType: models.PlanContentTypeTasks,
|
||||
Tasks: []models.Task{{PlanID: 73, Name: "Child's Task", ExecutionOrder: 1}},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
expectedError: "",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range testCases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
// 使用 defer 确保数据库连接在测试结束后关闭
|
||||
sqlDB, _ := db.DB()
|
||||
defer sqlDB.Close()
|
||||
|
||||
tc.setupData(db)
|
||||
|
||||
repo := repository.NewGormPlanRepository(db)
|
||||
plan, err := repo.GetPlanByID(tc.planID)
|
||||
|
||||
if tc.expectedError != "" {
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), tc.expectedError)
|
||||
assert.Nil(t, plan)
|
||||
} else {
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, plan)
|
||||
|
||||
// 在比较之前,清理实际结果和期望结果中所有不确定的、由数据库自动生成的字段
|
||||
cleanPlanForComparison(plan)
|
||||
cleanPlanForComparison(tc.expectedPlan)
|
||||
|
||||
assert.Equal(t, tc.expectedPlan, plan)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// cleanPlanForComparison 递归地重置 Plan 对象及其关联对象中由数据库自动生成的字段(如ID和时间戳),
|
||||
// 以便在测试中断言它们与期望值相等。
|
||||
func cleanPlanForComparison(p *models.Plan) {
|
||||
if p == nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 重置 Plan 自身的时间戳
|
||||
p.CreatedAt = time.Time{}
|
||||
p.UpdatedAt = time.Time{}
|
||||
p.DeletedAt = gorm.DeletedAt{}
|
||||
|
||||
// 重置所有 Task 的自动生成字段
|
||||
for i := range p.Tasks {
|
||||
p.Tasks[i].ID = 0 // ID是自动生成的,必须重置为0才能与期望值匹配
|
||||
p.Tasks[i].CreatedAt = time.Time{}
|
||||
p.Tasks[i].UpdatedAt = time.Time{}
|
||||
p.Tasks[i].DeletedAt = gorm.DeletedAt{}
|
||||
}
|
||||
|
||||
// 重置所有 SubPlan 的自动生成字段,并递归清理子计划
|
||||
for i := range p.SubPlans {
|
||||
p.SubPlans[i].ID = 0 // SubPlan 连接记录的ID也是自动生成的
|
||||
p.SubPlans[i].CreatedAt = time.Time{}
|
||||
p.SubPlans[i].UpdatedAt = time.Time{}
|
||||
p.SubPlans[i].DeletedAt = gorm.DeletedAt{}
|
||||
|
||||
// 递归调用以清理嵌套的子计划
|
||||
cleanPlanForComparison(p.SubPlans[i].ChildPlan)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user