fix(auth): streamline auth err proccess

Author: ndyakov

redis.go

@@ -285,21 +285,23 @@ func (c *baseClient) _getConn(ctx context.Context) (*pool.Conn, error) {
 return cn, nil
 }
 
-func (c *baseClient) newReAuthCredentialsListener(ctx context.Context, cn *pool.Conn) auth.CredentialsListener {
-connPool := pool.NewSingleConnPool(c.connPool, cn)
-// hooksMixin are intentionally empty here
-conn := newConn(c.opt, connPool, nil)
+func (c *baseClient) newReAuthCredentialsListener(ctx context.Context, poolCn *pool.Conn) auth.CredentialsListener {
 ctx = c.context(ctx)
 return auth.NewReAuthCredentialsListener(
-c.reAuthConnection(ctx, conn),
-c.onAuthenticationErr(ctx, conn),
+c.reAuthConnection(ctx, poolCn),
+c.onAuthenticationErr(ctx, poolCn),
 )
 }
 
-func (c *baseClient) reAuthConnection(ctx context.Context, cn *Conn) func(credentials auth.Credentials) error {
+func (c *baseClient) reAuthConnection(ctx context.Context, poolCn *pool.Conn) func(credentials auth.Credentials) error {
 return func(credentials auth.Credentials) error {
 var err error
 username, password := credentials.BasicAuth()
+
+connPool := pool.NewSingleConnPool(c.connPool, poolCn)
+// hooksMixin are intentionally empty here
+cn := newConn(c.opt, connPool, nil)
+
 if username != "" {
 err = cn.AuthACL(ctx, username, password).Err()
 } else {
@@ -308,22 +310,13 @@ func (c *baseClient) reAuthConnection(ctx context.Context, cn *Conn) func(creden
 return err
 }
 }
-func (c *baseClient) onAuthenticationErr(ctx context.Context, cn *Conn) func(err error) {
+func (c *baseClient) onAuthenticationErr(ctx context.Context, poolCn *pool.Conn) func(err error) {
 return func(err error) {
-// since the connection pool of the *Conn will actually return us the underlying pool.Conn,
-// we can get it from the *Conn and remove it from the clients pool.
 if err != nil {
 if isBadConn(err, false, c.opt.Addr) {
-poolCn, getErr := cn.connPool.Get(ctx)
-if getErr == nil {
-c.connPool.Remove(ctx, poolCn, err)
-} else {
-// if we can't get the pool connection, we can only close the connection
-if err := cn.Close(); err != nil {
-log.Printf("failed to close connection: %v", err)
-}
-}
+c.connPool.CloseConn(poolCn)
 }
+internal.Logger.Printf(ctx, "redis: re-authentication failed: %v", err)
 }
 }
 }