1. 增加新类型中继设备
2. 优化代码
This commit is contained in:
@@ -6,6 +6,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
||||||
|
"git.huangwc.com/pig/pig-farm-controller/internal/controller"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
||||||
@@ -28,80 +29,130 @@ func NewController(deviceControlRepo repository.DeviceControlRepo, deviceRepo re
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// ControlRequest 设备控制请求结构体
|
// SwitchRequest 设备控制请求结构体
|
||||||
type ControlRequest struct {
|
type SwitchRequest struct {
|
||||||
ParentID *uint `json:"parent_id"` // 区域主控ID
|
ParentID *uint `json:"parent_id"` // 区域主控ID
|
||||||
DeviceType string `json:"device_type" binding:"required,oneof=fan water_curtain"`
|
DeviceType string `json:"device_type" binding:"required,oneof=fan water_curtain"`
|
||||||
DeviceID string `json:"device_id" binding:"required"`
|
DeviceID string `json:"device_id" binding:"required"`
|
||||||
Action string `json:"action" binding:"required,oneof=on off"`
|
Action string `json:"action" binding:"required,oneof=on off"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SwitchResponseData 设备控制响应数据结构体
|
||||||
|
type SwitchResponseData struct {
|
||||||
|
DeviceType string `json:"device_type"`
|
||||||
|
DeviceID string `json:"device_id"`
|
||||||
|
Action string `json:"action"`
|
||||||
|
}
|
||||||
|
|
||||||
// Switch 设备控制接口
|
// Switch 设备控制接口
|
||||||
func (c *Controller) Switch(ctx *gin.Context) {
|
func (c *Controller) Switch(ctx *gin.Context) {
|
||||||
// 从上下文中获取用户信息
|
// 从上下文中获取用户信息
|
||||||
userValue, exists := ctx.Get("user")
|
userValue, exists := ctx.Get("user")
|
||||||
if !exists {
|
if !exists {
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "无法获取用户信息")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
user, ok := userValue.(*middleware.AuthUser)
|
user, ok := userValue.(*middleware.AuthUser)
|
||||||
if !ok {
|
if !ok {
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户信息格式错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var req ControlRequest
|
var req SwitchRequest
|
||||||
if err := ctx.ShouldBindJSON(&req); err != nil {
|
if err := ctx.ShouldBindJSON(&req); err != nil {
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "请求参数错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取区域主控设备信息(如果提供了ParentID)
|
|
||||||
var location string
|
|
||||||
if req.ParentID != nil {
|
|
||||||
parentDevice, err := c.deviceRepo.FindByID(*req.ParentID)
|
|
||||||
if err != nil {
|
|
||||||
c.logger.Error("查找区域主控设备失败: " + err.Error())
|
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无效的区域主控ID"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
location = parentDevice.Name
|
|
||||||
} else {
|
|
||||||
location = "未知区域"
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: 实际的设备控制逻辑
|
// TODO: 实际的设备控制逻辑
|
||||||
// 这里暂时用TODO代替具体逻辑
|
// 这里暂时用TODO代替具体逻辑
|
||||||
|
|
||||||
// 创建设备控制记录
|
// 创建设备控制记录
|
||||||
control := &model.DeviceControl{
|
if err := c.createDeviceControlRecord(
|
||||||
UserID: user.ID,
|
user.ID,
|
||||||
Location: location,
|
req.DeviceID,
|
||||||
DeviceType: model.DeviceType(req.DeviceType),
|
req.DeviceType,
|
||||||
DeviceID: req.DeviceID,
|
req.Action,
|
||||||
Action: req.Action,
|
"success",
|
||||||
Status: "success", // 假设控制成功
|
"设备控制成功",
|
||||||
Result: "设备控制成功",
|
); err != nil {
|
||||||
}
|
|
||||||
|
|
||||||
if err := c.deviceControlRepo.Create(control); err != nil {
|
|
||||||
c.logger.Error("创建设备控制记录失败: " + err.Error())
|
c.logger.Error("创建设备控制记录失败: " + err.Error())
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "设备控制失败"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "设备控制失败")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
data := SwitchResponseData{
|
||||||
"message": "设备控制成功",
|
DeviceType: req.DeviceType,
|
||||||
"data": map[string]interface{}{
|
DeviceID: req.DeviceID,
|
||||||
"id": control.ID,
|
Action: req.Action,
|
||||||
"location": control.Location,
|
}
|
||||||
"device_type": control.DeviceType,
|
|
||||||
"device_id": control.DeviceID,
|
controller.SendSuccessResponse(ctx, "设备控制成功", data)
|
||||||
"action": control.Action,
|
}
|
||||||
"status": control.Status,
|
|
||||||
"result": control.Result,
|
// createDeviceControlRecord 创建设备控制记录
|
||||||
"created_at": control.CreatedAt,
|
func (c *Controller) createDeviceControlRecord(userID uint, deviceID, deviceType, action, status, result string) error {
|
||||||
},
|
// 获取设备信息
|
||||||
})
|
device, err := c.deviceRepo.FindByIDString(deviceID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 构建位置信息
|
||||||
|
var location string
|
||||||
|
switch device.Type {
|
||||||
|
case model.DeviceTypeRelay:
|
||||||
|
// 如果设备本身就是中继设备
|
||||||
|
location = device.Name
|
||||||
|
case model.DeviceTypePigPenController, model.DeviceTypeFeedMillController:
|
||||||
|
// 如果设备本身就是区域主控设备
|
||||||
|
if device.ParentID != nil {
|
||||||
|
// 获取中继设备
|
||||||
|
relayDevice, err := c.deviceRepo.FindByID(*device.ParentID)
|
||||||
|
if err != nil {
|
||||||
|
location = device.Name
|
||||||
|
} else {
|
||||||
|
location = relayDevice.Name + " -> " + device.Name
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
location = device.Name
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
// 如果是普通设备(风机、水帘等)
|
||||||
|
if device.ParentID != nil {
|
||||||
|
// 获取区域主控设备
|
||||||
|
parentDevice, err := c.deviceRepo.FindByID(*device.ParentID)
|
||||||
|
if err != nil {
|
||||||
|
location = "未知区域"
|
||||||
|
} else {
|
||||||
|
// 检查区域主控设备是否有上级设备(中继设备)
|
||||||
|
if parentDevice.ParentID != nil {
|
||||||
|
// 获取中继设备
|
||||||
|
relayDevice, err := c.deviceRepo.FindByID(*parentDevice.ParentID)
|
||||||
|
if err != nil {
|
||||||
|
location = parentDevice.Name
|
||||||
|
} else {
|
||||||
|
location = relayDevice.Name + " -> " + parentDevice.Name
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
location = parentDevice.Name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
location = "未知区域"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
control := &model.DeviceControl{
|
||||||
|
UserID: userID,
|
||||||
|
Location: location,
|
||||||
|
DeviceType: model.DeviceType(deviceType),
|
||||||
|
DeviceID: deviceID,
|
||||||
|
Action: action,
|
||||||
|
Status: status,
|
||||||
|
Result: result,
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.deviceControlRepo.Create(control)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
||||||
|
"git.huangwc.com/pig/pig-farm-controller/internal/controller"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
||||||
@@ -41,24 +42,24 @@ func (c *Controller) Create(ctx *gin.Context) {
|
|||||||
// 从上下文中获取用户信息
|
// 从上下文中获取用户信息
|
||||||
userValue, exists := ctx.Get("user")
|
userValue, exists := ctx.Get("user")
|
||||||
if !exists {
|
if !exists {
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "无法获取用户信息")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
user, ok := userValue.(*middleware.AuthUser)
|
user, ok := userValue.(*middleware.AuthUser)
|
||||||
if !ok {
|
if !ok {
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户信息格式错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var req CreateRequest
|
var req CreateRequest
|
||||||
if err := ctx.ShouldBindJSON(&req); err != nil {
|
if err := ctx.ShouldBindJSON(&req); err != nil {
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "请求参数错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 创建操作历史记录
|
// 创建操作历史记录
|
||||||
history := &model.OperationHistory{
|
operation := &model.OperationHistory{
|
||||||
UserID: user.ID,
|
UserID: user.ID,
|
||||||
Action: req.Action,
|
Action: req.Action,
|
||||||
Target: req.Target,
|
Target: req.Target,
|
||||||
@@ -67,30 +68,13 @@ func (c *Controller) Create(ctx *gin.Context) {
|
|||||||
Result: req.Result,
|
Result: req.Result,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := c.operationHistoryRepo.Create(history); err != nil {
|
if err := c.operationHistoryRepo.Create(operation); err != nil {
|
||||||
c.logger.Error("创建操作历史记录失败: " + err.Error())
|
c.logger.Error("创建操作历史记录失败: " + err.Error())
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "创建操作历史记录失败"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "创建操作历史记录失败")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
controller.SendSuccessResponse(ctx, "操作历史记录创建成功", nil)
|
||||||
"message": "操作历史记录创建成功",
|
|
||||||
"data": map[string]interface{}{
|
|
||||||
"id": history.ID,
|
|
||||||
"action": history.Action,
|
|
||||||
"target": history.Target,
|
|
||||||
"parameters": history.Parameters,
|
|
||||||
"status": history.Status,
|
|
||||||
"result": history.Result,
|
|
||||||
"created_at": history.CreatedAt,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// ListByUserRequest 按用户查询操作历史请求结构体
|
|
||||||
type ListByUserRequest struct {
|
|
||||||
Page int `form:"page" binding:"required"`
|
|
||||||
Limit int `form:"limit" binding:"required"`
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ListByUser 获取当前用户的所有操作历史记录
|
// ListByUser 获取当前用户的所有操作历史记录
|
||||||
@@ -98,115 +82,77 @@ func (c *Controller) ListByUser(ctx *gin.Context) {
|
|||||||
// 从上下文中获取用户信息
|
// 从上下文中获取用户信息
|
||||||
userValue, exists := ctx.Get("user")
|
userValue, exists := ctx.Get("user")
|
||||||
if !exists {
|
if !exists {
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "无法获取用户信息")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
user, ok := userValue.(*middleware.AuthUser)
|
user, ok := userValue.(*middleware.AuthUser)
|
||||||
if !ok {
|
if !ok {
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户信息格式错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 解析查询参数
|
// 获取分页参数
|
||||||
var req ListByUserRequest
|
page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
|
||||||
if err := ctx.ShouldBindQuery(&req); err != nil {
|
limit, _ := strconv.Atoi(ctx.DefaultQuery("limit", "10"))
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "查询参数错误"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if req.Page <= 0 {
|
// 确保分页参数有效
|
||||||
req.Page = 1
|
if page < 1 {
|
||||||
|
page = 1
|
||||||
}
|
}
|
||||||
|
if limit < 1 || limit > 100 {
|
||||||
if req.Limit <= 0 || req.Limit > 100 {
|
limit = 10
|
||||||
req.Limit = 10
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 计算偏移量
|
// 计算偏移量
|
||||||
offset := (req.Page - 1) * req.Limit
|
offset := (page - 1) * limit
|
||||||
|
|
||||||
// 查询用户操作历史记录
|
// 查询操作历史记录
|
||||||
histories, err := c.operationHistoryRepo.FindByUserID(user.ID)
|
operations, err := c.operationHistoryRepo.ListByUserID(user.ID, offset, limit)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Error("查询操作历史记录失败: " + err.Error())
|
c.logger.Error("获取操作历史记录失败: " + err.Error())
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "查询操作历史记录失败"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "获取操作历史记录失败")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 分页处理
|
controller.SendSuccessResponse(ctx, "获取操作历史记录成功", operations)
|
||||||
start := offset
|
|
||||||
end := start + req.Limit
|
|
||||||
if start > len(histories) {
|
|
||||||
start = len(histories)
|
|
||||||
}
|
|
||||||
if end > len(histories) {
|
|
||||||
end = len(histories)
|
|
||||||
}
|
|
||||||
|
|
||||||
pagedHistories := histories[start:end]
|
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
|
||||||
"message": "查询成功",
|
|
||||||
"data": map[string]interface{}{
|
|
||||||
"histories": pagedHistories,
|
|
||||||
"page": req.Page,
|
|
||||||
"limit": req.Limit,
|
|
||||||
"total": len(histories),
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRequest 获取单个操作历史记录请求结构体
|
|
||||||
type GetRequest struct {
|
|
||||||
ID string `uri:"id" binding:"required"`
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get 获取单个操作历史记录
|
// Get 获取单个操作历史记录
|
||||||
func (c *Controller) Get(ctx *gin.Context) {
|
func (c *Controller) Get(ctx *gin.Context) {
|
||||||
|
// 获取操作历史记录ID
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "无效的操作历史记录ID")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查询操作历史记录
|
||||||
|
operation, err := c.operationHistoryRepo.FindByID(uint(id))
|
||||||
|
if err != nil {
|
||||||
|
c.logger.Error("获取操作历史记录失败: " + err.Error())
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "获取操作历史记录失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// 从上下文中获取用户信息
|
// 从上下文中获取用户信息
|
||||||
userValue, exists := ctx.Get("user")
|
userValue, exists := ctx.Get("user")
|
||||||
if !exists {
|
if !exists {
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "无法获取用户信息")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
user, ok := userValue.(*middleware.AuthUser)
|
user, ok := userValue.(*middleware.AuthUser)
|
||||||
if !ok {
|
if !ok {
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户信息格式错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 解析路径参数
|
// 检查操作历史记录是否属于当前用户
|
||||||
var req GetRequest
|
if operation.UserID != user.ID {
|
||||||
if err := ctx.ShouldBindUri(&req); err != nil {
|
controller.SendErrorResponse(ctx, http.StatusForbidden, "无权访问该操作历史记录")
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "路径参数错误"})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 将ID转换为整数
|
controller.SendSuccessResponse(ctx, "获取操作历史记录成功", operation)
|
||||||
id, err := strconv.ParseUint(req.ID, 10, 32)
|
|
||||||
if err != nil {
|
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "ID格式错误"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查询操作历史记录
|
|
||||||
history, err := c.operationHistoryRepo.FindByID(uint(id))
|
|
||||||
if err != nil {
|
|
||||||
c.logger.Error("查询操作历史记录失败: " + err.Error())
|
|
||||||
ctx.JSON(http.StatusNotFound, gin.H{"error": "操作历史记录不存在"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// 检查是否是当前用户的记录
|
|
||||||
if history.UserID != user.ID {
|
|
||||||
ctx.JSON(http.StatusForbidden, gin.H{"error": "无权访问该记录"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
|
||||||
"message": "查询成功",
|
|
||||||
"data": history,
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|||||||
41
internal/controller/response.go
Normal file
41
internal/controller/response.go
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
// Package controller 提供控制器层的公共功能
|
||||||
|
// 包含公共响应结构体和处理函数等
|
||||||
|
package controller
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
// APIResponse 通用API响应结构体
|
||||||
|
type APIResponse struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
Data interface{} `json:"data,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// SuccessResponseData 成功响应数据结构体
|
||||||
|
type SuccessResponseData struct {
|
||||||
|
Message string `json:"message"`
|
||||||
|
Data interface{} `json:"data,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// sendSuccessResponse 发送成功响应的公共函数
|
||||||
|
func SendSuccessResponse(ctx *gin.Context, message string, data interface{}) {
|
||||||
|
response := APIResponse{
|
||||||
|
Code: http.StatusOK,
|
||||||
|
Message: message,
|
||||||
|
Data: data,
|
||||||
|
}
|
||||||
|
ctx.JSON(http.StatusOK, response)
|
||||||
|
}
|
||||||
|
|
||||||
|
// sendErrorResponse 发送错误响应的公共函数
|
||||||
|
func SendErrorResponse(ctx *gin.Context, code int, message string) {
|
||||||
|
response := APIResponse{
|
||||||
|
Code: code,
|
||||||
|
Message: message,
|
||||||
|
}
|
||||||
|
ctx.JSON(code, response)
|
||||||
|
}
|
||||||
@@ -6,6 +6,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
|
||||||
|
"git.huangwc.com/pig/pig-farm-controller/internal/controller"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
@@ -26,101 +27,104 @@ func NewController(userRepo repository.UserRepo) *Controller {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// RegisterRequest 注册请求结构体
|
// RegisterRequest 用户注册请求结构体
|
||||||
type RegisterRequest struct {
|
type RegisterRequest struct {
|
||||||
Username string `json:"username" binding:"required"`
|
Username string `json:"username" binding:"required"`
|
||||||
Password string `json:"password" binding:"required"`
|
Password string `json:"password" binding:"required"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// RegisterResponse 注册响应结构体
|
// LoginRequest 用户登录请求结构体
|
||||||
type RegisterResponse struct {
|
|
||||||
ID uint `json:"id"`
|
|
||||||
Username string `json:"username"`
|
|
||||||
CreatedAt string `json:"created_at"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Register 用户注册
|
|
||||||
func (c *Controller) Register(ctx *gin.Context) {
|
|
||||||
var req RegisterRequest
|
|
||||||
if err := ctx.ShouldBindJSON(&req); err != nil {
|
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
user, err := c.userRepo.CreateUser(req.Username, req.Password)
|
|
||||||
if err != nil {
|
|
||||||
c.logger.Error("创建用户失败: " + err.Error())
|
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "创建用户失败"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
response := RegisterResponse{
|
|
||||||
ID: user.ID,
|
|
||||||
Username: user.Username,
|
|
||||||
CreatedAt: user.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
|
||||||
"message": "用户创建成功",
|
|
||||||
"user": response,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// LoginRequest 登录请求结构体
|
|
||||||
type LoginRequest struct {
|
type LoginRequest struct {
|
||||||
Username string `json:"username" binding:"required"`
|
Username string `json:"username" binding:"required"`
|
||||||
Password string `json:"password" binding:"required"`
|
Password string `json:"password" binding:"required"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// LoginResponse 登录响应结构体
|
// RegisterResponseData 用户注册响应数据
|
||||||
type LoginResponse struct {
|
type RegisterResponseData struct {
|
||||||
|
ID uint `json:"id"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoginResponseData 用户登录响应数据
|
||||||
|
type LoginResponseData struct {
|
||||||
ID uint `json:"id"`
|
ID uint `json:"id"`
|
||||||
Username string `json:"username"`
|
Username string `json:"username"`
|
||||||
Token string `json:"token"`
|
Token string `json:"token"`
|
||||||
CreatedAt string `json:"created_at"`
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Login 用户登录
|
// Register 用户注册接口
|
||||||
|
func (c *Controller) Register(ctx *gin.Context) {
|
||||||
|
var req RegisterRequest
|
||||||
|
if err := ctx.ShouldBindJSON(&req); err != nil {
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "请求参数错误")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查用户名是否已存在
|
||||||
|
_, err := c.userRepo.FindByUsername(req.Username)
|
||||||
|
if err == nil {
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "用户名已存在")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 对密码进行哈希处理
|
||||||
|
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(req.Password), bcrypt.DefaultCost)
|
||||||
|
if err != nil {
|
||||||
|
c.logger.Error("密码哈希处理失败: " + err.Error())
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户注册失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建用户
|
||||||
|
user, err := c.userRepo.CreateUser(req.Username, string(hashedPassword))
|
||||||
|
if err != nil {
|
||||||
|
c.logger.Error("创建用户失败: " + err.Error())
|
||||||
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "用户注册失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
data := RegisterResponseData{
|
||||||
|
ID: user.ID,
|
||||||
|
Username: user.Username,
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.SendSuccessResponse(ctx, "用户注册成功", data)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Login 用户登录接口
|
||||||
func (c *Controller) Login(ctx *gin.Context) {
|
func (c *Controller) Login(ctx *gin.Context) {
|
||||||
var req LoginRequest
|
var req LoginRequest
|
||||||
if err := ctx.ShouldBindJSON(&req); err != nil {
|
if err := ctx.ShouldBindJSON(&req); err != nil {
|
||||||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
|
controller.SendErrorResponse(ctx, http.StatusBadRequest, "请求参数错误")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 查找用户
|
// 查找用户
|
||||||
user, err := c.userRepo.FindByUsername(req.Username)
|
user, err := c.userRepo.FindByUsername(req.Username)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Error("查找用户失败: " + err.Error())
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "用户名或密码错误")
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "用户名或密码错误"})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 验证密码
|
// 验证密码
|
||||||
err = bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(req.Password))
|
if err := bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(req.Password)); err != nil {
|
||||||
if err != nil {
|
controller.SendErrorResponse(ctx, http.StatusUnauthorized, "用户名或密码错误")
|
||||||
ctx.JSON(http.StatusUnauthorized, gin.H{"error": "用户名或密码错误"})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 生成JWT访问令牌
|
// 生成JWT token
|
||||||
authMiddleware := middleware.NewAuthMiddleware(c.userRepo)
|
token, err := middleware.NewAuthMiddleware(c.userRepo).GenerateToken(user.ID, user.Username)
|
||||||
token, err := authMiddleware.GenerateToken(user.ID, user.Username)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Error("生成令牌失败: " + err.Error())
|
c.logger.Error("生成JWT token失败: " + err.Error())
|
||||||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "登录失败"})
|
controller.SendErrorResponse(ctx, http.StatusInternalServerError, "登录失败")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
response := LoginResponse{
|
data := LoginResponseData{
|
||||||
ID: user.ID,
|
ID: user.ID,
|
||||||
Username: user.Username,
|
Username: user.Username,
|
||||||
Token: token,
|
Token: token,
|
||||||
CreatedAt: user.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx.JSON(http.StatusOK, gin.H{
|
controller.SendSuccessResponse(ctx, "登录成功", data)
|
||||||
"message": "登录成功",
|
|
||||||
"user": response,
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,6 +23,9 @@ const (
|
|||||||
|
|
||||||
// DeviceTypeFeedMillController 做料车间主控
|
// DeviceTypeFeedMillController 做料车间主控
|
||||||
DeviceTypeFeedMillController DeviceType = "feed_mill_controller"
|
DeviceTypeFeedMillController DeviceType = "feed_mill_controller"
|
||||||
|
|
||||||
|
// DeviceTypeRelay 中继设备
|
||||||
|
DeviceTypeRelay DeviceType = "relay"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Device 代表设备信息
|
// Device 代表设备信息
|
||||||
|
|||||||
@@ -3,6 +3,8 @@
|
|||||||
package repository
|
package repository
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"strconv"
|
||||||
|
|
||||||
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
"git.huangwc.com/pig/pig-farm-controller/internal/model"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
@@ -15,6 +17,9 @@ type DeviceRepo interface {
|
|||||||
// FindByID 根据ID查找设备
|
// FindByID 根据ID查找设备
|
||||||
FindByID(id uint) (*model.Device, error)
|
FindByID(id uint) (*model.Device, error)
|
||||||
|
|
||||||
|
// FindByIDString 根据ID字符串查找设备
|
||||||
|
FindByIDString(id string) (*model.Device, error)
|
||||||
|
|
||||||
// FindByParentID 根据上级设备ID查找设备
|
// FindByParentID 根据上级设备ID查找设备
|
||||||
FindByParentID(parentID uint) ([]*model.Device, error)
|
FindByParentID(parentID uint) ([]*model.Device, error)
|
||||||
|
|
||||||
@@ -83,6 +88,21 @@ func (r *deviceRepo) FindByID(id uint) (*model.Device, error) {
|
|||||||
return &device, nil
|
return &device, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// FindByIDString 根据ID字符串查找设备
|
||||||
|
func (r *deviceRepo) FindByIDString(id string) (*model.Device, error) {
|
||||||
|
deviceID, err := strconv.ParseUint(id, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var device model.Device
|
||||||
|
result := r.db.First(&device, deviceID)
|
||||||
|
if result.Error != nil {
|
||||||
|
return nil, result.Error
|
||||||
|
}
|
||||||
|
return &device, nil
|
||||||
|
}
|
||||||
|
|
||||||
// FindByParentID 根据上级设备ID查找设备
|
// FindByParentID 根据上级设备ID查找设备
|
||||||
func (r *deviceRepo) FindByParentID(parentID uint) ([]*model.Device, error) {
|
func (r *deviceRepo) FindByParentID(parentID uint) ([]*model.Device, error) {
|
||||||
var devices []*model.Device
|
var devices []*model.Device
|
||||||
|
|||||||
@@ -20,6 +20,9 @@ type OperationHistoryRepo interface {
|
|||||||
|
|
||||||
// List 获取操作历史记录列表(分页)
|
// List 获取操作历史记录列表(分页)
|
||||||
List(offset, limit int) ([]*model.OperationHistory, error)
|
List(offset, limit int) ([]*model.OperationHistory, error)
|
||||||
|
|
||||||
|
// ListByUserID 根据用户ID获取操作历史记录列表(分页)
|
||||||
|
ListByUserID(userID uint, offset, limit int) ([]*model.OperationHistory, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// operationHistoryRepo 操作历史仓库实现
|
// operationHistoryRepo 操作历史仓库实现
|
||||||
@@ -69,3 +72,13 @@ func (r *operationHistoryRepo) List(offset, limit int) ([]*model.OperationHistor
|
|||||||
}
|
}
|
||||||
return histories, nil
|
return histories, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ListByUserID 根据用户ID获取操作历史记录列表(分页)
|
||||||
|
func (r *operationHistoryRepo) ListByUserID(userID uint, offset, limit int) ([]*model.OperationHistory, error) {
|
||||||
|
var histories []*model.OperationHistory
|
||||||
|
result := r.db.Where("user_id = ?", userID).Offset(offset).Limit(limit).Order("created_at DESC").Find(&histories)
|
||||||
|
if result.Error != nil {
|
||||||
|
return nil, result.Error
|
||||||
|
}
|
||||||
|
return histories, nil
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user