添加close

Author: letsfire

interface.go

@@ -2,12 +2,14 @@ package redigo
 
 import (
 	"fmt"
+	"io"
 	"time"
 
 	"github.com/gomodule/redigo/redis"
 )
 
 type ModeInterface interface {
+	io.Closer
 	fmt.Stringer
 	GetConn() redis.Conn
 	NewConn() (redis.Conn, error)

mode/alone/alone.go

@@ -16,6 +16,10 @@ func (am *aloneMode) NewConn() (redis.Conn, error) {
 	return am.pool.Dial()
 }
 
+func (am *aloneMode) Close() error {
+	return am.pool.Close()
+}
+
 func (am *aloneMode) String() string { return "alone" }
 
 func New(optFuncs ...OptFunc) redigo.ModeInterface {

mode/cluster/cluster.go

@@ -19,6 +19,10 @@ func (cm *clusterMode) NewConn() (redis.Conn, error) {
 	return cm.rc.Dial()
 }
 
+func (cm *clusterMode) Close() error {
+	return cm.rc.Close()
+}
+
 func (cm *clusterMode) String() string { return "cluster" }
 
 func New(optFuncs ...OptFunc) redigo.ModeInterface {

mode/sentinel/sentinel.go

@@ -22,6 +22,10 @@ func (sm *sentinelMode) NewConn() (redis.Conn, error) {
 	return sm.pool.Dial()
 }
 
+func (sm *sentinelMode) Close() error {
+	return sm.pool.Close()
+}
+
 func (sm *sentinelMode) String() string { return "sentinel" }
 
 func New(optFuncs ...OptFunc) redigo.ModeInterface {

redigo.go

@@ -24,6 +24,10 @@ func (c *Client) Mode() string {
 	return c.mode.String()
 }
 
+func (c *Client) Close() error {
+	return c.mode.Close()
+}
+
 func (c *Client) Execute(fn ExecuteFunc) (res interface{}, err error) {
 	conn := c.mode.GetConn()
 	defer conn.Close()