chore: use redis.UniversalClient instead of *redis.Client (#88)

libi · Jan 14, 2024 · 0ff369b · 0ff369b
1 parent 68982eb
commit 0ff369b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/driver/driver.go b/driver/driver.go
@@ -30,14 +30,14 @@ type DriverV2 interface {
 	withOption(opt Option) (err error)
 }
 
-func NewRedisDriver(redisClient *redis.Client) DriverV2 {
+func NewRedisDriver(redisClient redis.UniversalClient) DriverV2 {
 	return newRedisDriver(redisClient)
 }
 
 func NewEtcdDriver(etcdCli *clientv3.Client) DriverV2 {
 	return newEtcdDriver(etcdCli)
 }
 
-func NewRedisZSetDriver(redisClient *redis.Client) DriverV2 {
+func NewRedisZSetDriver(redisClient redis.UniversalClient) DriverV2 {
 	return newRedisZSetDriver(redisClient)
 }
diff --git a/driver/redisdriver.go b/driver/redisdriver.go
@@ -17,7 +17,7 @@ const (
 )
 
 type RedisDriver struct {
-	c           *redis.Client
+	c           redis.UniversalClient
 	serviceName string
 	nodeID      string
 	timeout     time.Duration
@@ -32,7 +32,7 @@ type RedisDriver struct {
 	sync.Mutex
 }
 
-func newRedisDriver(redisClient *redis.Client) *RedisDriver {
+func newRedisDriver(redisClient redis.UniversalClient) *RedisDriver {
 	rd := &RedisDriver{
 		c: redisClient,
 		logger: &dlog.StdLogger{

diff --git a/driver/rediszsetdriver.go b/driver/rediszsetdriver.go
@@ -13,7 +13,7 @@ import (
 )
 
 type RedisZSetDriver struct {
-	c           *redis.Client
+	c           redis.UniversalClient
 	serviceName string
 	nodeID      string
 	timeout     time.Duration
@@ -28,7 +28,7 @@ type RedisZSetDriver struct {
 	sync.Mutex
 }
 
-func newRedisZSetDriver(redisClient *redis.Client) *RedisZSetDriver {
+func newRedisZSetDriver(redisClient redis.UniversalClient) *RedisZSetDriver {
 	rd := &RedisZSetDriver{
 		c: redisClient,
 		logger: &dlog.StdLogger{