Close connection on end game

1 files changed, 5 insertions, 1 deletions

diff --git a/pkg/p2p/network.go b/pkg/p2p/network.go
index 2ebec43..7ff0746 100644
--- a/pkg/p2p/network.go
+++ b/pkg/p2p/network.go
@@ -51,6 +51,7 @@ type TCPNetwork struct {
 	id          NetworkID
 	listener    net.Listener
 	connections map[NetworkID]net.Conn
+	isClosed    bool
 }
 
 // Initiliaze a new TCP network
@@ -129,6 +130,8 @@ func (n *TCPNetwork) startServer() error {
 		return err
 	}
 
+	n.isClosed = false
+
 	go n.listenLoop()
 
 	n.Logger.Sugar().Infof("server started on %s\n", n.ListenAddr)
@@ -141,6 +144,7 @@ func (n *TCPNetwork) listenLoop() error {
 	for {
 		conn, err := n.listener.Accept()
 		if errors.Is(err, net.ErrClosed) {
+			n.isClosed = true
 			n.Logger.Sugar().Errorf("connection is closed in such a way: %v\n", err)
 			return err
 		}
@@ -218,7 +222,7 @@ func (n *TCPNetwork) retryConnect(remoteID NetworkID, addr string) {
 		if err != nil {
 			n.Logger.Sugar().Errorf("failed to connect to %s: %v. Retrying in %v...", remoteID, err, n.RetryDelay)
 			time.Sleep(n.RetryDelay)
-			if n.RetryDelay < 30*time.Second {
+			if !n.isClosed && n.RetryDelay < 30*time.Second {
 				n.RetryDelay *= 2
 			} else {
 				n.Lock()