brontide: Make dialer parameter in brontide.Dial non-optional
This commit is contained in:
parent
69f207e23f
commit
945be73bca
@ -33,17 +33,11 @@ var _ net.Conn = (*Conn)(nil)
|
|||||||
// public key. In the case of a handshake failure, the connection is closed and
|
// public key. In the case of a handshake failure, the connection is closed and
|
||||||
// a non-nil error is returned.
|
// a non-nil error is returned.
|
||||||
func Dial(localPriv *btcec.PrivateKey, netAddr *lnwire.NetAddress,
|
func Dial(localPriv *btcec.PrivateKey, netAddr *lnwire.NetAddress,
|
||||||
dialer ...func(string, string) (net.Conn, error)) (*Conn, error) {
|
dialer func(string, string) (net.Conn, error)) (*Conn, error) {
|
||||||
ipAddr := netAddr.Address.String()
|
ipAddr := netAddr.Address.String()
|
||||||
var conn net.Conn
|
var conn net.Conn
|
||||||
var err error
|
var err error
|
||||||
if dialer == nil {
|
conn, err = dialer("tcp", ipAddr)
|
||||||
// A dial function WAS NOT passed in.
|
|
||||||
conn, err = net.Dial("tcp", ipAddr)
|
|
||||||
} else {
|
|
||||||
// A dial function WAS passed in so we use it instead.
|
|
||||||
conn, err = dialer[0]("tcp", ipAddr)
|
|
||||||
}
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -47,7 +47,7 @@ func establishTestConnection() (net.Conn, net.Conn, func(), error) {
|
|||||||
conErrChan := make(chan error, 1)
|
conErrChan := make(chan error, 1)
|
||||||
connChan := make(chan net.Conn, 1)
|
connChan := make(chan net.Conn, 1)
|
||||||
go func() {
|
go func() {
|
||||||
conn, err := Dial(remotePriv, netAddr)
|
conn, err := Dial(remotePriv, netAddr, net.Dial)
|
||||||
|
|
||||||
conErrChan <- err
|
conErrChan <- err
|
||||||
connChan <- conn
|
connChan <- conn
|
||||||
|
Loading…
Reference in New Issue
Block a user