Changeset 732 in code for trunk/upstream.go

Timestamp:

Dec 2, 2021, 9:53:43 AM (4 years ago)

Author:

contact

Message:

Add context to connectToUpstream

File:

• trunk/upstream.go (modified) (6 diffs)

trunk/upstream.go

@@ -124 +124 @@
 }
 
-func connectToUpstream(network *network) (*upstreamConn, error) {
+func connectToUpstream(ctx context.Context, network *network) (*upstreamConn, error) {
         logger := &prefixLogger{network.user.logger, fmt.Sprintf("upstream %q: ", network.GetName())}
 
@@ -144 +144 @@
                 }
 
-                dialer.LocalAddr, err = network.user.localTCPAddrForHost(host)
+                dialer.LocalAddr, err = network.user.localTCPAddrForHost(ctx, host)
                 if err != nil {
                         return nil, fmt.Errorf("failed to pick local IP for remote host %q: %v", host, err)
@@ -172 +172 @@
                 }
 
-                netConn, err = dialer.Dial("tcp", addr)
+                netConn, err = dialer.DialContext(ctx, "tcp", addr)
                 if err != nil {
                         return nil, fmt.Errorf("failed to dial %q: %v", addr, err)
@@ -189 +189 @@
                 }
 
-                dialer.LocalAddr, err = network.user.localTCPAddrForHost(host)
+                dialer.LocalAddr, err = network.user.localTCPAddrForHost(ctx, host)
                 if err != nil {
                         return nil, fmt.Errorf("failed to pick local IP for remote host %q: %v", host, err)
@@ -195 +195 @@
 
                 logger.Printf("connecting to plain-text server at address %q", addr)
-                netConn, err = dialer.Dial("tcp", addr)
+                netConn, err = dialer.DialContext(ctx, "tcp", addr)
                 if err != nil {
                         return nil, fmt.Errorf("failed to dial %q: %v", addr, err)
@@ -201 +201 @@
         case "irc+unix", "unix":
                 logger.Printf("connecting to Unix socket at path %q", u.Path)
-                netConn, err = dialer.Dial("unix", u.Path)
+                netConn, err = dialer.DialContext(ctx, "unix", u.Path)
                 if err != nil {
                         return nil, fmt.Errorf("failed to connect to Unix socket %q: %v", u.Path, err)