1. 增加新类型中继设备

2. 优化代码
This commit is contained in:
2025-09-08 12:27:54 +08:00
parent bab44e7e3a
commit c7e2aaee89
7 changed files with 285 additions and 207 deletions

View File

@@ -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)
} }

View File

@@ -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,
})
} }

View 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)
}

View File

@@ -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,
})
} }

View File

@@ -23,6 +23,9 @@ const (
// DeviceTypeFeedMillController 做料车间主控 // DeviceTypeFeedMillController 做料车间主控
DeviceTypeFeedMillController DeviceType = "feed_mill_controller" DeviceTypeFeedMillController DeviceType = "feed_mill_controller"
// DeviceTypeRelay 中继设备
DeviceTypeRelay DeviceType = "relay"
) )
// Device 代表设备信息 // Device 代表设备信息

View File

@@ -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

View File

@@ -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
}