egg_im/internal/business/domain/user/repo/auth_cache.go

package repo

import (
	"egg-im/internal/business/domain/user/model"
	"egg-im/pkg/db"
	"egg-im/pkg/gerrors"
	"egg-im/pkg/util"
	"encoding/json"
	"strconv"

	"github.com/go-redis/redis"
)

const (
	AuthKey = "auth:"
)

type authCache struct{}

var AuthCache = new(authCache)

func (*authCache) Get(userId, deviceId int64) (*model.Device, error) {
	bytes, err := db.RedisCli.HGet(AuthKey+strconv.FormatInt(userId, 10), strconv.FormatInt(deviceId, 10)).Bytes()
	if err != nil && err != redis.Nil {
		return nil, gerrors.WrapError(err)
	}
	if err == redis.Nil {
		return nil, nil
	}

	var device model.Device
	err = json.Unmarshal(bytes, &device)
	if err != nil {
		return nil, gerrors.WrapError(err)
	}
	return &device, nil
}

func (*authCache) Set(userId, deviceId int64, device model.Device) error {
	bytes, err := json.Marshal(device)
	if err != nil {
		return gerrors.WrapError(err)
	}

	_, err = db.RedisCli.HSet(AuthKey+strconv.FormatInt(userId, 10), strconv.FormatInt(deviceId, 10), bytes).Result()
	if err != nil {
		return gerrors.WrapError(err)
	}
	return nil
}

func (*authCache) GetAll(userId int64) (map[int64]model.Device, error) {
	result, err := db.RedisCli.HGetAll(AuthKey + strconv.FormatInt(userId, 10)).Result()
	if err != nil {
		return nil, gerrors.WrapError(err)
	}

	var devices = make(map[int64]model.Device, len(result))

	for k, v := range result {
		deviceId, err := strconv.ParseInt(k, 10, 64)
		if err != nil {
			return nil, gerrors.WrapError(err)
		}

		var device model.Device
		err = json.Unmarshal(util.Str2bytes(v), &device)
		if err != nil {
			return nil, gerrors.WrapError(err)
		}
		devices[deviceId] = device
	}
	return devices, nil
}