encrypt session key

middleware/auth/session.go

@@ -2,11 +2,11 @@ package auth
 
 type Session interface {
 	// StoreJwtToken store jwt token is redis, make it expired for feature
-	StoreJwtToken(key string, value string, timeout int64) error
+	StoreJwtToken(token string, uid int64, timeout int64) error
 
 	// IsExistsJwtToken judge whether token is invalid
-	IsExistsJwtToken(key string) bool
+	IsExistsJwtToken(token string) bool
 
 	// DeleteJwtToken delete jwt token
-	DeleteJwtToken(key string) bool
+	DeleteJwtToken(token string) bool
 }

middleware/auth/session_debug.go

@@ -6,7 +6,7 @@ func NewDebugSessionStore() *DebugSessionStore {
 	return &DebugSessionStore{}
 }
 
-func (*DebugSessionStore) StoreJwtToken(key string, value string, timeout int64) error {
+func (*DebugSessionStore) StoreJwtToken(key string, uid int64, timeout int64) error {
 	return nil
 }
 

middleware/auth/session_redis.go

@@ -3,6 +3,7 @@ package auth
 import (
 	"time"
 
+	"git.links123.net/links123.com/pkg/utils"
 	"github.com/go-redis/redis"
 )
 
@@ -22,14 +23,21 @@ func NewRedisSessionStore(address string, password string) *RedisSessionStore {
 	return session
 }
 
-func (rss *RedisSessionStore) StoreJwtToken(key string, value string, timeout int64) error {
-	return rss.client.Set(key, value, time.Duration(timeout) * time.Second).Err()
+func (rss *RedisSessionStore) StoreJwtToken(token string, uid int64, timeout int64) error {
+	k := utils.Md5(token)
+	v := map[string]int64{"user_id": uid, "created": time.Now().Unix()}
+
+	return rss.client.Set(k, v, time.Duration(timeout)*time.Second).Err()
 }
 
-func (rss *RedisSessionStore) IsExistsJwtToken(key string) bool {
-	return rss.client.Exists(key).Val() == 1
+func (rss *RedisSessionStore) IsExistsJwtToken(token string) bool {
+	k := utils.Md5(token)
+
+	return rss.client.Exists(k).Val() == 1
 }
 
-func (rss *RedisSessionStore) DeleteJwtToken(key string) bool {
-	return rss.client.Del(key).Val() == 1
+func (rss *RedisSessionStore) DeleteJwtToken(token string) bool {
+	k := utils.Md5(token)
+
+	return rss.client.Del(k).Val() == 1
 }

utils/encrypt.go

@@ -0,0 +1,14 @@
+package utils
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+)
+
+func Md5(src string) string {
+	h := md5.New()
+	h.Write([]byte(src))
+	d := h.Sum(nil)
+
+	return string(hex.EncodeToString(d))
+}