brontide: Make dialer parameter in brontide.Dial non-optional

This commit is contained in:
MeshCollider authored and Olaoluwa Osuntokun committed 2018-02-05 17:37:46 -08:00
1 parent 69f207e23f
commit 945be73bca
2 files changed
+3 -9

No files matched your search

+2 -8
View File
@@ -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
} }
+1 -1
View File
@@ -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