奇淼（piexlmax
3 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
16 changed files with 286 additions and 115 deletions
-
5server/api/v1/system/sys_user.go
-
1server/config.yaml
-
11server/core/viper.go
-
3server/global/global.go
-
5server/go.mod
-
4server/initialize/timer.go
-
100server/middleware/jwt.go
-
52server/plugin/ws/utils/utils.go
-
83server/plugin/ws/ws.go
-
28server/service/system/jwt_black_list.go
-
71server/utils/jwt.go
-
2web/package.json
-
6web/src/core/gin-vue-admin.js
-
11web/src/core/global.js
-
18web/src/view/layout/aside/historyComponent/history.vue
-
1web/src/view/layout/index.vue
@ -0,0 +1,52 @@ |
|||
package utils |
|||
|
|||
import ( |
|||
"github.com/flipped-aurora/gin-vue-admin/server/global" |
|||
systemReq "github.com/flipped-aurora/gin-vue-admin/server/model/system/request" |
|||
"github.com/gin-gonic/gin" |
|||
uuid "github.com/satori/go.uuid" |
|||
) |
|||
|
|||
// 从Gin的Context中获取从jwt解析出来的用户ID
|
|||
func GetUserID(c *gin.Context) uint { |
|||
if claims, exists := c.Get("claims"); !exists { |
|||
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析出来的用户ID失败, 请检查路由是否使用jwt中间件!") |
|||
return 0 |
|||
} else { |
|||
waitUse := claims.(*systemReq.CustomClaims) |
|||
return waitUse.ID |
|||
} |
|||
} |
|||
|
|||
// 从Gin的Context中获取从jwt解析出来的用户UUID
|
|||
func GetUserUuid(c *gin.Context) uuid.UUID { |
|||
if claims, exists := c.Get("claims"); !exists { |
|||
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析出来的用户UUID失败, 请检查路由是否使用jwt中间件!") |
|||
return uuid.UUID{} |
|||
} else { |
|||
waitUse := claims.(*systemReq.CustomClaims) |
|||
return waitUse.UUID |
|||
} |
|||
} |
|||
|
|||
// 从Gin的Context中获取从jwt解析出来的用户角色id
|
|||
func GetUserAuthorityId(c *gin.Context) string { |
|||
if claims, exists := c.Get("claims"); !exists { |
|||
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析出来的用户UUID失败, 请检查路由是否使用jwt中间件!") |
|||
return "" |
|||
} else { |
|||
waitUse := claims.(*systemReq.CustomClaims) |
|||
return waitUse.AuthorityId |
|||
} |
|||
} |
|||
|
|||
// 从Gin的Context中获取从jwt解析出来的用户角色id
|
|||
func GetUserInfo(c *gin.Context) *systemReq.CustomClaims { |
|||
if claims, exists := c.Get("claims"); !exists { |
|||
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析出来的用户UUID失败, 请检查路由是否使用jwt中间件!") |
|||
return nil |
|||
} else { |
|||
waitUse := claims.(*systemReq.CustomClaims) |
|||
return waitUse |
|||
} |
|||
} |
|||
@ -0,0 +1,83 @@ |
|||
package ws |
|||
|
|||
import ( |
|||
"github.com/flipped-aurora/ws/core/biz" |
|||
"github.com/flipped-aurora/ws/core/data" |
|||
"github.com/gin-gonic/gin" |
|||
"go.uber.org/zap" |
|||
"nhooyr.io/websocket" |
|||
) |
|||
|
|||
type wsPlugin struct { |
|||
logger *zap.Logger // 日志输出对象
|
|||
manageBuf int64 // buffer
|
|||
registeredMsgHandler map[int32]func(biz.IMessage) bool // 消息处理
|
|||
checkMap map[string]biz.CheckFunc // 用户校验
|
|||
|
|||
admin biz.IManage |
|||
adminCase *biz.AdminCase |
|||
} |
|||
|
|||
func DefaultRegisteredMsgHandler(admin biz.IManage, logger *zap.Logger) map[int32]func(biz.IMessage) bool { |
|||
return map[int32]func(msg biz.IMessage) bool{ |
|||
1: func(msg biz.IMessage) bool { |
|||
// w.admin 里面找到注册客户端的方法
|
|||
client, ok := admin.FindClient(msg.GetTo()) |
|||
if !ok { |
|||
logger.Info("没有找到该用户") |
|||
return false |
|||
} |
|||
return client.SendMes(msg) |
|||
}, |
|||
} |
|||
} |
|||
|
|||
func DefaultCheckMap() map[string]biz.CheckFunc { |
|||
return map[string]biz.CheckFunc{ |
|||
"gva_ws": func(c interface{}) (string, bool) { |
|||
// 先断言是gin.content
|
|||
cc, ok := c.(*gin.Context) |
|||
if !ok { |
|||
return "", false |
|||
} |
|||
token := cc.Query("jwt") |
|||
// 可以携带jwt
|
|||
if len(token) == 0 { |
|||
return "", false |
|||
} |
|||
// 解析 jwt...
|
|||
|
|||
return token, true |
|||
}, |
|||
} |
|||
} |
|||
|
|||
func (w *wsPlugin) Register(g *gin.RouterGroup) { |
|||
// gva_ws 为身份校验函数
|
|||
g.GET("/ws", w.adminCase.HandlerWS("gva_ws", &websocket.AcceptOptions{ |
|||
InsecureSkipVerify: true, |
|||
})) |
|||
g.POST("/sendMsg", w.adminCase.SendMsg("gva_ws")) |
|||
|
|||
} |
|||
|
|||
func (w *wsPlugin) RouterPath() string { |
|||
return "gva_ws" |
|||
} |
|||
|
|||
func GenerateWs(logger *zap.Logger, manageBuf int64, checkMap map[string]biz.CheckFunc) *wsPlugin { |
|||
m := data.NewManage(manageBuf) |
|||
t := data.NewTopic() |
|||
h := data.NewHandle() |
|||
admin := data.NewAdmin(m, t, h, logger) |
|||
for s, checkFunc := range checkMap { |
|||
admin.AddCheckFunc(s, checkFunc) |
|||
} |
|||
registeredMsgHandler := DefaultRegisteredMsgHandler(admin, logger) |
|||
|
|||
for key, handler := range registeredMsgHandler { |
|||
admin.RegisteredMsgHandler(key, handler) |
|||
} |
|||
return &wsPlugin{logger: logger, manageBuf: manageBuf, |
|||
registeredMsgHandler: registeredMsgHandler, checkMap: checkMap, admin: admin, adminCase: biz.NewAdmin(admin)} |
|||
} |
|||
@ -0,0 +1,71 @@ |
|||
package utils |
|||
|
|||
import ( |
|||
"errors" |
|||
"github.com/dgrijalva/jwt-go" |
|||
"github.com/flipped-aurora/gin-vue-admin/server/global" |
|||
"github.com/flipped-aurora/gin-vue-admin/server/model/system/request" |
|||
) |
|||
|
|||
type JWT struct { |
|||
SigningKey []byte |
|||
} |
|||
|
|||
var ( |
|||
TokenExpired = errors.New("Token is expired") |
|||
TokenNotValidYet = errors.New("Token not active yet") |
|||
TokenMalformed = errors.New("That's not even a token") |
|||
TokenInvalid = errors.New("Couldn't handle this token:") |
|||
) |
|||
|
|||
func NewJWT() *JWT { |
|||
return &JWT{ |
|||
[]byte(global.GVA_CONFIG.JWT.SigningKey), |
|||
} |
|||
} |
|||
|
|||
// 创建一个token
|
|||
func (j *JWT) CreateToken(claims request.CustomClaims) (string, error) { |
|||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims) |
|||
return token.SignedString(j.SigningKey) |
|||
} |
|||
|
|||
// CreateTokenByOldToken 旧token 换新token 使用归并回源避免并发问题
|
|||
func (j *JWT) CreateTokenByOldToken(oldToken string, claims request.CustomClaims) (string, error) { |
|||
v, err, _ := global.GVA_Concurrency_Control.Do("JWT:"+oldToken, func() (interface{}, error) { |
|||
return j.CreateToken(claims) |
|||
}) |
|||
return v.(string), err |
|||
} |
|||
|
|||
// 解析 token
|
|||
func (j *JWT) ParseToken(tokenString string) (*request.CustomClaims, error) { |
|||
token, err := jwt.ParseWithClaims(tokenString, &request.CustomClaims{}, func(token *jwt.Token) (i interface{}, e error) { |
|||
return j.SigningKey, nil |
|||
}) |
|||
if err != nil { |
|||
if ve, ok := err.(*jwt.ValidationError); ok { |
|||
if ve.Errors&jwt.ValidationErrorMalformed != 0 { |
|||
return nil, TokenMalformed |
|||
} else if ve.Errors&jwt.ValidationErrorExpired != 0 { |
|||
// Token is expired
|
|||
return nil, TokenExpired |
|||
} else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 { |
|||
return nil, TokenNotValidYet |
|||
} else { |
|||
return nil, TokenInvalid |
|||
} |
|||
} |
|||
} |
|||
if token != nil { |
|||
if claims, ok := token.Claims.(*request.CustomClaims); ok && token.Valid { |
|||
return claims, nil |
|||
} |
|||
return nil, TokenInvalid |
|||
|
|||
} else { |
|||
return nil, TokenInvalid |
|||
|
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,11 @@ |
|||
import config from './config' |
|||
import { emitter } from '@/utils/bus.js' |
|||
|
|||
const closeThisPage = () => { |
|||
emitter.emit('closeThisPage') |
|||
} |
|||
|
|||
export const register = (app) => { |
|||
app.config.globalProperties.$GIN_VUE_ADMIN = config |
|||
app.config.globalProperties.$CloseThisPage = closeThisPage |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue