From 859c993a82d8795f8e0070d5176ca2aa9f85d797 Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Thu, 3 Feb 2022 10:36:08 +0100
Subject: [PATCH] Retry on temporary net.Listener failure

Instead of stopping to listen, retry on temporary failure. This
can happen when running out of FDs.
---
 server.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/server.go b/server.go
index ae01cfa..92768f3 100644
--- a/server.go
+++ b/server.go
@@ -220,6 +220,8 @@ func (ln *Listener) Start() error {
 	}
 	log.Printf("listening on %q", ln.Address)
 
+	ln.netLn = &retryListener{Listener: ln.netLn}
+
 	go func() {
 		if err := ln.serve(); err != nil {
 			log.Fatalf("listener %q: %v", ln.Address, err)
@@ -444,3 +446,30 @@ func sslTLV(state *tls.ConnectionState) (proxyproto.TLV, error) {
 
 	return pp2ssl.Marshal()
 }
+
+type retryListener struct {
+	net.Listener
+
+	delay time.Duration
+}
+
+func (ln *retryListener) Accept() (net.Conn, error) {
+	for {
+		conn, err := ln.Listener.Accept()
+		if ne, ok := err.(net.Error); ok && ne.Temporary() {
+			if ln.delay == 0 {
+				ln.delay = 5 * time.Millisecond
+			} else {
+				ln.delay *= 2
+			}
+			if max := 1 * time.Second; ln.delay > max {
+				ln.delay = max
+			}
+			log.Printf("listener %q: accept error (retrying in %v): %v", ln.Addr(), ln.delay, err)
+			time.Sleep(ln.delay)
+		} else {
+			ln.delay = 0
+			return conn, err
+		}
+	}
+}