处理分享2

This commit is contained in:
2024-09-29 16:55:00 +08:00
parent 6431aa890b
commit c7daf18830
12 changed files with 302 additions and 96 deletions
+8 -8
View File
@@ -9,14 +9,6 @@ import (
// Auth Auth认证
func Auth() gin.HandlerFunc {
return func(c *gin.Context) {
// 获取用户id
userId := c.GetInt64("UserId")
if userId == 0 {
responses.Fail(c)
c.Abort()
return
}
// 获取客户端
client := c.GetInt("Client")
if client == 0 || (client != 1 && client != 2) {
@@ -32,6 +24,14 @@ func Auth() gin.HandlerFunc {
// 后台
if client == 2 {
// 获取用户id
userId := c.GetInt64("UserId")
if userId == 0 {
responses.Fail(c)
c.Abort()
return
}
adminUserDao := dao.AdminUserDao{}
adminUser, err := adminUserDao.GetAdminUserFirstById(userId)
if err != nil || adminUser == nil {
+31 -10
View File
@@ -53,20 +53,41 @@ func Jwt() gin.HandlerFunc {
return
}
// 转换类型
userId, err := strconv.ParseInt(t.UserId, 10, 64)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"message": "token错误",
"code": consts.TokenError,
"data": "",
})
var userId int64
var qaId int64
c.Abort()
return
// 用户id
if t.UserId != "" {
userId, err = strconv.ParseInt(t.UserId, 10, 64)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"message": "token错误",
"code": consts.TokenError,
"data": "",
})
c.Abort()
return
}
}
// 知识问答id
if t.QaId != "" {
qaId, err = strconv.ParseInt(t.QaId, 10, 64)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"message": "token错误",
"code": consts.TokenError,
"data": "",
})
c.Abort()
return
}
}
c.Set("UserId", userId) // 用户id
c.Set("QaId", qaId) // 知识问答id
c.Set("Client", t.Client) // 客户端(1:前台 2:后台)
c.Next()
}