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)
 		}
 	}
 }