156 lines
3.2 KiB
Go
156 lines
3.2 KiB
Go
|
/**
|
|||
|
|
* package api
|
|||
|
|
* 作用：用户管理相关API处理器
|
|||
|
|
*/
|
|||
|
|
package api
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"math/rand"
|
|||
|
|
"strconv"
|
|||
|
|
"time"
|
|||
|
|
"xk-websocket-v2/internal/model"
|
|||
|
|
"xk-websocket-v2/internal/service"
|
|||
|
|
"xk-websocket-v2/internal/utils"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* GetMyInfoHandler
|
|||
|
|
* 功能：获取当前用户信息
|
|||
|
|
* 路径：GET /api/user/my-info
|
|||
|
|
* 需要：JWT认证
|
|||
|
|
*/
|
|||
|
|
func GetMyInfoHandler(c *gin.Context) {
|
|||
|
|
// 从Context获取用户ID（由JWT中间件注入）
|
|||
|
|
userID, exists := c.Get("user_id")
|
|||
|
|
if !exists {
|
|||
|
|
utils.Unauthorized(c, "未认证")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
user, err := service.UserSvc.GetUserByID(userID.(string))
|
|||
|
|
if err != nil {
|
|||
|
|
utils.NotFound(c, "用户不存在")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
utils.SuccessWithData(c, user, "获取成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* GetUserListHandler
|
|||
|
|
* 功能：获取用户列表
|
|||
|
|
* 路径：GET /api/user/list
|
|||
|
|
*/
|
|||
|
|
func GetUserListHandler(c *gin.Context) {
|
|||
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|||
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
|||
|
|
|
|||
|
|
if page < 1 {
|
|||
|
|
page = 1
|
|||
|
|
}
|
|||
|
|
if pageSize < 1 || pageSize > 100 {
|
|||
|
|
pageSize = 20
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
users, total, err := service.UserSvc.GetUserList(page, pageSize)
|
|||
|
|
if err != nil {
|
|||
|
|
utils.InternalError(c, "查询失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 确保返回空数组而不是null
|
|||
|
|
if users == nil {
|
|||
|
|
users = []model.User{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
utils.SuccessWithData(c, gin.H{
|
|||
|
|
"data": users,
|
|||
|
|
"total": total,
|
|||
|
|
"page": page,
|
|||
|
|
"size": pageSize,
|
|||
|
|
}, "获取成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* CreateUserHandler
|
|||
|
|
* 功能：创建用户（管理员）
|
|||
|
|
* 路径：POST /api/user/create
|
|||
|
|
*/
|
|||
|
|
func CreateUserHandler(c *gin.Context) {
|
|||
|
|
var user model.User
|
|||
|
|
if err := c.ShouldBindJSON(&user); err != nil {
|
|||
|
|
utils.BadRequest(c, "参数错误: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 生成用户ID
|
|||
|
|
user.ID = generateUserID()
|
|||
|
|
|
|||
|
|
if err := service.UserSvc.CreateUser(&user); err != nil {
|
|||
|
|
utils.BadRequest(c, "创建用户失败: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
user.Password = "" // 清除密码
|
|||
|
|
utils.SuccessWithData(c, user, "创建成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* UpdateUserHandler
|
|||
|
|
* 功能：更新用户信息
|
|||
|
|
* 路径：POST /api/user/update
|
|||
|
|
*/
|
|||
|
|
func UpdateUserHandler(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
ID string `json:"id" binding:"required"`
|
|||
|
|
Updates map[string]interface{} `json:"updates" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
utils.BadRequest(c, "参数错误: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := service.UserSvc.UpdateUser(req.ID, req.Updates); err != nil {
|
|||
|
|
utils.BadRequest(c, "更新失败: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
utils.Success(c, "更新成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* DeleteUserHandler
|
|||
|
|
* 功能：删除用户
|
|||
|
|
* 路径：POST /api/user/delete
|
|||
|
|
*/
|
|||
|
|
func DeleteUserHandler(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
ID string `json:"id" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
utils.BadRequest(c, "参数错误")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := service.UserSvc.DeleteUser(req.ID); err != nil {
|
|||
|
|
utils.BadRequest(c, "删除失败: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
utils.Success(c, "删除成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 辅助函数：生成用户ID
|
|||
|
|
func generateUserID() string {
|
|||
|
|
// 使用时间戳+随机数生成用户ID
|
|||
|
|
timestamp := time.Now().UnixNano()
|
|||
|
|
random := rand.Intn(1000000)
|
|||
|
|
return fmt.Sprintf("user_%d_%d", timestamp, random)
|
|||
|
|
}
|
|||
|
|
|