Changeset 67 in code

Timestamp:

Feb 18, 2020, 4:26:17 PM (5 years ago)

Author:

contact

Message:

Enable TCP keep-alive on all connections

Location:

trunk

Files:

• server.go (modified) (2 diffs)
• upstream.go (modified) (1 diff)

trunk/server.go

@@ -6 +6 @@
         "net"
         "sync"
+        "time"
 
         "gopkg.in/irc.v3"
 )
+
+// TODO: make configurable
+var keepAlivePeriod = time.Minute
+
+func setKeepAlive(c net.Conn) error {
+        tcpConn, ok := c.(*net.TCPConn)
+        if !ok {
+                return fmt.Errorf("cannot enable keep-alive on a non-TCP connection")
+        }
+        if err := tcpConn.SetKeepAlive(true); err != nil {
+                return err
+        }
+        return tcpConn.SetKeepAlivePeriod(keepAlivePeriod)
+}
 
 type Logger interface {
@@ -178 +193 @@
                 }
 
+                setKeepAlive(netConn)
+
                 dc := newDownstreamConn(s, netConn)
                 go func() {

trunk/upstream.go

@@ -56 +56 @@
                 return nil, fmt.Errorf("failed to dial %q: %v", upstream.Addr, err)
         }
+
+        setKeepAlive(netConn)
 
         msgs := make(chan *irc.Message, 64)