diff --git a/network/wsNetwork.go b/network/wsNetwork.go index 92da09f1a6..525968ebf4 100644 --- a/network/wsNetwork.go +++ b/network/wsNetwork.go @@ -39,12 +39,12 @@ import ( "github.com/algorand/go-deadlock" "github.com/algorand/websocket" "github.com/gorilla/mux" - "golang.org/x/net/netutil" "github.com/algorand/go-algorand/config" "github.com/algorand/go-algorand/crypto" "github.com/algorand/go-algorand/logging" "github.com/algorand/go-algorand/logging/telemetryspec" + "github.com/algorand/go-algorand/network/limitlistener" "github.com/algorand/go-algorand/protocol" tools_network "github.com/algorand/go-algorand/tools/network" "github.com/algorand/go-algorand/tools/network/dnssec" @@ -748,7 +748,8 @@ func (wn *WebsocketNetwork) Start() { return } // wrap the original listener with a limited connection listener - listener = netutil.LimitListener(listener, wn.config.IncomingConnectionsLimit) + listener = limitlistener.RejectingLimitListener( + listener, uint64(wn.config.IncomingConnectionsLimit), wn.log) // wrap the limited connection listener with a requests tracker listener wn.listener = wn.requestsTracker.Listener(listener) wn.log.Debugf("listening on %s", wn.listener.Addr().String())