add redigo interface

go-redsync · Oct 31, 2023 · ab335fb · ab335fb
1 parent e1f61b4
commit ab335fb
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/redis/redigo/redigo.go b/redis/redigo/redigo.go
@@ -9,8 +9,13 @@ import (
 	"github.com/gomodule/redigo/redis"
 )
 
+type Pool interface {
+	Get() redis.Conn
+	GetContext(ctx context.Context) (redis.Conn, error)
+}
+
 type pool struct {
-	delegate *redis.Pool
+	delegate Pool
 }
 
 func (p *pool) Get(ctx context.Context) (redsyncredis.Conn, error) {
@@ -25,7 +30,7 @@ func (p *pool) Get(ctx context.Context) (redsyncredis.Conn, error) {
 }
 
 // NewPool returns a Redigo-based pool implementation.
-func NewPool(delegate *redis.Pool) redsyncredis.Pool {
+func NewPool(delegate Pool) redsyncredis.Pool {
 	return &pool{delegate}
 }