Go语言——用户登录信息保存redis并使用

Go语言——用户登录信息保存redis并使用

获取指定前缀的所有Keys

核心代码如下

// 获取指定前缀所有key
res, err := d.redis.cache.Do(ctx, "keys", "kafka*")
if err != nil {
    logging.Errorf("获取指定前缀keys失败！！%s", err)
}

// redis所有key转为列表
ret := make([]string, 10)
for _, item := range res.([]interface{}) {
    if keyStr, ok := item.([]byte); ok {
        ret = append(ret, string(keyStr))
    }
}

用户信息插入Redis

核心代码如下

func (r *RedisCache) InsertRedisUserTimeOut(ctx context.Context, data *model.UserInfo) error {
    if data == nil {
        return errors.New("param error")
    }
    key := getUserInfoRedisKey(data.UID)

    var ub []byte
    ub, marshalErr := json.Marshal(data)
    if marshalErr != nil {
        logging.Errorf("[InsertUser] json.Marshal failed !! %s", marshalErr)
        return marshalErr
    }

    bol, err := r.cache.Set(ctx, key, ub)
  if !bol {
  logging.Errorf("[InsertUser] r.cache.set failed!! %s", err)
  }
  return err
}

struct如下

// struct如下
type UserInfo struct {
    UID      int64  `json:"uid"`
    Nick     string `json:"nick"`
    Gender   int    `json:"gender"`
    Age      int    `json:"age"`
    Birthday string `json:"birthday"`
    Portrait string `json:"portrait"`
    SelfDesc string `json:"self_desc"`

    // 兼容老版协议
    ID          int64  `json:"id"`
    Name        string `json:"name"`
    AvatarURL   string `json:"avatar_url"`
    Description string `json:"description"`
}

获取用户信息

核心代码如下

/**
 * @Description: 获取所有用户信息
 * @receiver d
 * @param ctx
 * @return []model.UserInfo
 * @return error
 */
func (d *Dao) GetUserInfoAll(ctx context.Context) ([]model.UserInfo, error) {

    // 获取指定前缀所有key
    res, err := d.redis.cache.Do(ctx, "keys", "kafka:user_info*")
    if err != nil {
        logging.Errorf("获取指定前缀keys失败！！%s", err)
    }

    // redis所有key转为列表
    ret := make([]interface{}, 0)
    for _, item := range res.([]interface{}) {
        if keyStr, ok := item.([]byte); ok {
            ret = append(ret, keyStr)
        }
    }

    userInfoList := make([]model.UserInfo, 0)
    // 获取用户列表
    mgetArr, mgetErr := d.redis.cache.MGet(ctx, ret...)
    if mgetErr != nil {
        logging.Errorf("[GetUserInfoAll] d.redis.cache.MGet failed!! %s", mgetErr)
        return userInfoList, nil
    }

    for i := range mgetArr {
        bytes := mgetArr[i]
        if bytes != nil {
            var user model.UserInfo
            json.Unmarshal(bytes, &user)
            userInfoList = append(userInfoList, user)
        }
    }

    return userInfoList, nil
}