ConnectionTracker keeps track of opened connections
type ConnectionTracker struct {
// contains filtered or unexported fields
}
func NewConnectionTracker() *ConnectionTracker
NewConnectionTracker returns a connection tracker for use with NewDialerWithTracker
func (c *ConnectionTracker) CloseAll()
CloseAll forcibly closes all tracked connections.
Note: new connections may get created before CloseAll returns.
func (c *ConnectionTracker) Track(conn net.Conn) net.Conn
Track adds the connection to the list of tracked connections, and returns a wrapped copy of the connection that stops tracking the connection when it is closed.
DialFunc is a shorthand for signature of net.DialContext.
type DialFunc func(ctx context.Context, network, address string) (net.Conn, error)
Dialer opens connections through Dial and tracks them.
type Dialer struct {
*ConnectionTracker
// contains filtered or unexported fields
}
func NewDialer(dial DialFunc) *Dialer
NewDialer creates a new Dialer instance. Equivalent to NewDialerWithTracker(dial, nil).
func NewDialerWithTracker(dial DialFunc, tracker *ConnectionTracker) *Dialer
NewDialerWithTracker creates a new Dialer instance.
If dial is not nil, it will be used to create new underlying connections. Otherwise net.DialContext is used. If tracker is not nil, it will be used to track new underlying connections. Otherwise NewConnectionTracker() is used.
func (d *Dialer) Dial(network, address string) (net.Conn, error)
Dial creates a new tracked connection.
func (d *Dialer) DialContext(ctx context.Context, network, address string) (net.Conn, error)
DialContext creates a new tracked connection.