Merge remote-tracking branch 'origin/errcode' into errcode

pkg/common/config/config.go

@@ -665,15 +665,6 @@ func initConfig(config interface{}, configName, configPath string) {
 func InitConfig(configPath string) {
 	initConfig(&Config, "config.yaml", configPath)
 	initConfig(&UsualConfig, "usualConfig.yaml", configPath)
-	if Config.Etcd.UserName == "" {
-		Config.Etcd.UserName = UsualConfig.Etcd.UserName
-	}
-	if Config.Etcd.Password == "" {
-		Config.Etcd.Password = UsualConfig.Etcd.Password
-	}
-	if Config.Etcd.Secret == "" {
-		Config.Etcd.Secret = UsualConfig.Etcd.Secret
-	}
 
 	if Config.Mysql.DBUserName == "" {
 		Config.Mysql.DBUserName = UsualConfig.Mysql.DBUserName

pkg/common/db/cache/redis.go

@@ -20,10 +20,10 @@ import (
 )
 
 const (
-	userIncrSeq             = "REDIS_USER_INCR_SEQ:" // user incr seq
-	appleDeviceToken        = "DEVICE_TOKEN"
-	userMinSeq              = "REDIS_USER_MIN_SEQ:"
-	uidPidToken             = "UID_PID_TOKEN_STATUS:"
+	userIncrSeq      = "REDIS_USER_INCR_SEQ:" // user incr seq
+	appleDeviceToken = "DEVICE_TOKEN"
+	userMinSeq       = "REDIS_USER_MIN_SEQ:"
+
 	getuiToken              = "GETUI_TOKEN"
 	getuiTaskID             = "GETUI_TASK_ID"
 	messageCache            = "MESSAGE_CACHE:"
@@ -94,33 +94,33 @@ func NewRedisClient(rdb redis.UniversalClient) *RedisClient {
 	return &RedisClient{rdb: rdb}
 }
 
-//Perform seq auto-increment operation of user messages
+// Perform seq auto-increment operation of user messages
 func (r *RedisClient) IncrUserSeq(uid string) (uint64, error) {
 	key := userIncrSeq + uid
 	seq, err := r.rdb.Incr(context.Background(), key).Result()
 	return uint64(seq), err
 }
 
-//Get the largest Seq
+// Get the largest Seq
 func (r *RedisClient) GetUserMaxSeq(uid string) (uint64, error) {
 	key := userIncrSeq + uid
 	seq, err := r.rdb.Get(context.Background(), key).Result()
 	return uint64(utils.StringToInt(seq)), err
 }
 
-//set the largest Seq
+// set the largest Seq
 func (r *RedisClient) SetUserMaxSeq(uid string, maxSeq uint64) error {
 	key := userIncrSeq + uid
 	return r.rdb.Set(context.Background(), key, maxSeq, 0).Err()
 }
 
-//Set the user's minimum seq
+// Set the user's minimum seq
 func (r *RedisClient) SetUserMinSeq(uid string, minSeq uint32) (err error) {
 	key := userMinSeq + uid
 	return r.rdb.Set(context.Background(), key, minSeq, 0).Err()
 }
 
-//Get the smallest Seq
+// Get the smallest Seq
 func (r *RedisClient) GetUserMinSeq(uid string) (uint64, error) {
 	key := userMinSeq + uid
 	seq, err := r.rdb.Get(context.Background(), key).Result()
@@ -159,7 +159,7 @@ func (r *RedisClient) SetGroupMinSeq(groupID string, minSeq uint32) error {
 	return r.rdb.Set(context.Background(), key, minSeq, 0).Err()
 }
 
-//Store userid and platform class to redis
+// Store userid and platform class to redis
 func (r *RedisClient) AddTokenFlag(userID string, platformID int, token string, flag int) error {
 	key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
 	log2.NewDebug("", "add token key is ", key)

pkg/common/db/cache/token.go

@@ -0,0 +1,75 @@
+package cache
+
+import (
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/tokenverify"
+	"Open_IM/pkg/utils"
+	"context"
+	go_redis "github.com/go-redis/redis/v8"
+	"github.com/golang-jwt/jwt/v4"
+)
+
+const (
+	uidPidToken = "UID_PID_TOKEN_STATUS:"
+)
+
+type Token interface {
+	//结果为空 不返回错误
+	GetTokensWithoutError(ctx context.Context, userID, platform string) (map[string]int, error)
+	//创建token
+	CreateToken(ctx context.Context, userID string, platformID int) (string, error)
+}
+
+type TokenRedis struct {
+	RedisClient  *RedisClient
+	AccessSecret string
+	AccessExpire int64
+}
+
+func NewTokenRedis(redisClient *RedisClient, accessSecret string, accessExpire int64) *TokenRedis {
+	return &TokenRedis{redisClient, accessSecret, accessExpire}
+}
+
+// 结果为空 不返回错误
+func (t *TokenRedis) GetTokensWithoutError(ctx context.Context, userID, platform string) (map[string]int, error) {
+	key := uidPidToken + userID + ":" + platform
+	m, err := t.RedisClient.GetClient().HGetAll(context.Background(), key).Result()
+	if err != nil && err == go_redis.Nil {
+		return nil, nil
+	}
+	mm := make(map[string]int)
+	for k, v := range m {
+		mm[k] = utils.StringToInt(v)
+	}
+	return mm, utils.Wrap(err, "")
+}
+
+// 创建token
+func (t *TokenRedis) CreateToken(ctx context.Context, userID string, platform string) (string, error) {
+	tokens, err := t.GetTokensWithoutError(ctx, userID, platform)
+	if err != nil {
+		return "", err
+	}
+	var deleteTokenKey []string
+	for k, v := range tokens {
+		_, err = tokenverify.GetClaimFromToken(k)
+		if err != nil || v != constant.NormalToken {
+			deleteTokenKey = append(deleteTokenKey, k)
+		}
+	}
+	if len(deleteTokenKey) != 0 {
+		key := uidPidToken + userID + ":" + platform
+		err := t.RedisClient.GetClient().HDel(context.Background(), key, deleteTokenKey...).Err()
+		if err != nil {
+			return "", err
+		}
+	}
+	claims := tokenverify.BuildClaims(userID, platform, t.AccessExpire)
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	tokenString, err := token.SignedString([]byte(t.AccessSecret))
+	if err != nil {
+		return "", utils.Wrap(err, "")
+	}
+	key := uidPidToken + userID + ":" + platform
+	return "", utils.Wrap(t.RedisClient.GetClient().HSet(context.Background(), key, tokenString, constant.NormalToken).Err(), "")
+}

pkg/common/db/controller/auth.go

@@ -1,9 +1,34 @@
 package controller
 
-import "context"
+import (
+	"Open_IM/pkg/common/db/cache"
+	"context"
+	"github.com/go-redis/redis/v8"
+)
 
 type AuthInterface interface {
-	GetTokens(ctx context.Context, userID, platform string) (map[string]int, error)
-	DeleteToken(ctx context.Context, userID, platform string) error
-	CreateToken(ctx context.Context, userID string, platformID int, ttl int64) (string, error)
+	//结果为空 不返回错误
+	GetTokensWithoutError(ctx context.Context, userID, platform string) (map[string]int, error)
+
+	//创建token
+	CreateToken(ctx context.Context, userID string, platform string) (string, error)
+}
+
+type AuthController struct {
+	database *cache.TokenRedis
+}
+
+func NewAuthController(rdb redis.UniversalClient, accessSecret string, accessExpire int64) *AuthController {
+	cache.NewRedisClient(rdb)
+	return &AuthController{database: cache.NewTokenRedis(cache.NewRedisClient(rdb), accessSecret, accessExpire)}
+}
+
+// 结果为空 不返回错误
+func (a *AuthController) GetTokensWithoutError(ctx context.Context, userID, platform string) (map[string]int, error) {
+	return a.database.GetTokensWithoutError(ctx, userID, platform)
+}
+
+// 创建token
+func (a *AuthController) CreateToken(ctx context.Context, userID string, platform string) (string, error) {
+	return a.database.CreateToken(ctx, userID, platform)
 }