90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package diagram
|
|
|
|
import (
|
|
"context"
|
|
|
|
locker "modelRT/distributedlock"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
// "github.com/go-redis/redis"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// TODO 统一 storageClient与 rwLocker 中使用的 redis 版本
|
|
// RedisHash defines the encapsulation struct of redis hash type
|
|
type RedisHash struct {
|
|
ctx context.Context
|
|
rwLocker *locker.RedissionRWLocker
|
|
storageClient *redis.Client
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// SetRedisHashByMap define func of set redis hash by map struct
|
|
func (rh *RedisHash) SetRedisHashByMap(hashKey string, fields map[string]interface{}) error {
|
|
err := rh.rwLocker.WLock()
|
|
if err != nil {
|
|
rh.logger.Error("lock wLock by hashKey failed", zap.String("hashKey", hashKey), zap.Error(err))
|
|
return err
|
|
}
|
|
defer rh.rwLocker.UnWLock()
|
|
|
|
err = rh.storageClient.HSet(rh.ctx, hashKey, fields).Err()
|
|
if err != nil {
|
|
rh.logger.Error("set hash by map failed", zap.String("hashKey", hashKey), zap.Any("fields", fields), zap.Error(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SetRedisHashByKV define func of set redis hash by kv struct
|
|
func (rh *RedisHash) SetRedisHashByKV(hashKey string, field string, value interface{}) error {
|
|
err := rh.rwLocker.WLock()
|
|
if err != nil {
|
|
rh.logger.Error("lock wLock by hashKey failed", zap.String("hashKey", hashKey), zap.Error(err))
|
|
return err
|
|
}
|
|
defer rh.rwLocker.UnWLock()
|
|
|
|
err = rh.storageClient.HSet(rh.ctx, hashKey, field, value).Err()
|
|
if err != nil {
|
|
rh.logger.Error("set hash by kv failed", zap.String("hashKey", hashKey), zap.String("field", field), zap.Any("value", value), zap.Error(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// HGet define func of get specified field value from redis hash by key and field name
|
|
func (rh *RedisHash) HGet(hashKey string, field string) (string, error) {
|
|
err := rh.rwLocker.RLock()
|
|
if err != nil {
|
|
rh.logger.Error("lock rLock by hashKey failed", zap.String("hashKey", hashKey), zap.Error(err))
|
|
return "", err
|
|
}
|
|
defer rh.rwLocker.UnRLock()
|
|
|
|
result, err := rh.storageClient.HGet(rh.ctx, hashKey, field).Result()
|
|
if err != nil {
|
|
rh.logger.Error("set hash by kv failed", zap.String("hashKey", hashKey), zap.String("field", field), zap.Error(err))
|
|
return "", err
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// HGetAll define func of get all filelds from redis hash by key
|
|
func (rh *RedisHash) HGetAll(hashKey string) (map[string]string, error) {
|
|
err := rh.rwLocker.RLock()
|
|
if err != nil {
|
|
rh.logger.Error("lock rLock by hashKey failed", zap.String("hashKey", hashKey), zap.Error(err))
|
|
return nil, err
|
|
}
|
|
defer rh.rwLocker.UnRLock()
|
|
|
|
result, err := rh.storageClient.HGetAll(rh.ctx, hashKey).Result()
|
|
if err != nil {
|
|
rh.logger.Error("get all hash field by hash key failed", zap.String("hashKey", hashKey), zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return result, nil
|
|
}
|