blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

token modify

Browse Source
This commit is contained in:
Gordon 2021-11-30 15:37:51 +08:00
parent aff733bbd9
commit b02830c891
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/utils/jwt_token_test.go
View File

@ -81,3 +81,9 @@ func Test_ParseToken(t *testing.T) {
assert.Equal(t, claims.UID, uid) assert.Equal(t, claims.UID, uid)
} }
} }
func Test_GetClaimFromToken(t *testing.T) {
token := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVSUQiOiJvcGVuSU0xMjM0NTYiLCJQbGF0Zm9ybSI6IiIsImV4cCI6MTYzODg0NjQ3NiwibmJmIjoxNjM4MjQxNjc2LCJpYXQiOjE2MzgyNDE2NzZ9.W8RZB7ec5ySFj-rGE2Aho2z32g3MprQMdCyPiQu_C2I"
c, err := token_verify.GetClaimFromToken(token)
assert.Nil(t, c)
assert.Nil(t, err)
}

8
pkg/common/token_verify/jwt_token.go
View File

@ -55,7 +55,7 @@ func secret() jwt.Keyfunc {
} }
} }
func getClaimFromToken(tokensString string) (*Claims, error) { func GetClaimFromToken(tokensString string) (*Claims, error) {
token, err := jwt.ParseWithClaims(tokensString, &Claims{}, secret()) token, err := jwt.ParseWithClaims(tokensString, &Claims{}, secret())
if err != nil { if err != nil {
if ve, ok := err.(*jwt.ValidationError); ok { if ve, ok := err.(*jwt.ValidationError); ok {
@ -82,7 +82,7 @@ func getClaimFromToken(tokensString string) (*Claims, error) {
func ParseToken(tokensString string) (claims *Claims, err error) { func ParseToken(tokensString string) (claims *Claims, err error) {
claims, err = getClaimFromToken(tokensString) claims, err = GetClaimFromToken(tokensString)
if err != nil { if err != nil {
log.NewError("", "token validate err", err.Error()) log.NewError("", "token validate err", err.Error())
return nil, err return nil, err
@ -112,7 +112,7 @@ func ParseToken(tokensString string) (claims *Claims, err error) {
return nil, &constant.ErrTokenUnknown return nil, &constant.ErrTokenUnknown
} }
} }
return nil, err return nil, &constant.ErrTokenUnknown
} }
//func MakeTheTokenInvalid(currentClaims *Claims, platformClass string) (bool, error) { //func MakeTheTokenInvalid(currentClaims *Claims, platformClass string) (bool, error) {
@ -132,7 +132,7 @@ func ParseToken(tokensString string) (claims *Claims, err error) {
//} //}
func ParseRedisInterfaceToken(redisToken interface{}) (*Claims, error) { func ParseRedisInterfaceToken(redisToken interface{}) (*Claims, error) {
return getClaimFromToken(string(redisToken.([]uint8))) return GetClaimFromToken(string(redisToken.([]uint8)))
} }
//Validation token, false means failure, true means successful verification //Validation token, false means failure, true means successful verification