1
0
mirror of https://github.com/redis/go-redis.git synced 2025-07-31 05:04:23 +03:00

Add OnConnect context

This commit is contained in:
Vladimir Mihailenco
2020-06-10 10:36:22 +03:00
parent e779df5ab1
commit ef82e3705c
7 changed files with 27 additions and 18 deletions

View File

@ -5,6 +5,7 @@ import (
"errors"
"fmt"
"math/rand"
"net"
"strconv"
"sync"
"sync/atomic"
@ -62,7 +63,8 @@ type RingOptions struct {
// Following options are copied from Options struct.
OnConnect func(*Conn) error
Dialer func(ctx context.Context, network, addr string) (net.Conn, error)
OnConnect func(ctx context.Context, cn *Conn) error
Username string
DB int
@ -115,6 +117,7 @@ func (opt *RingOptions) init() {
func (opt *RingOptions) clientOptions() *Options {
return &Options{
Dialer: opt.Dialer,
OnConnect: opt.OnConnect,
DB: opt.DB,