Close connection on end game

1 files changed, 14 insertions, 0 deletions

diff --git a/pkg/ui/multiplayer/multiplayer.go b/pkg/ui/multiplayer/multiplayer.go
index 0e390eb..ac3bc7a 100644
--- a/pkg/ui/multiplayer/multiplayer.go
+++ b/pkg/ui/multiplayer/multiplayer.go
@@ -3,6 +3,7 @@ package multiplayer
 import (
 	"time"
 
+	"github.com/boozec/rahanna/internal/logger"
 	"github.com/boozec/rahanna/pkg/p2p"
 	"go.uber.org/zap"
 )
@@ -48,3 +49,16 @@ func (n *GameNetwork) AddPeer(remoteID p2p.NetworkID, addr string) {
 func (n *GameNetwork) AddReceiveFunction(f p2p.NetworkMessageReceiveFunc) {
 	n.server.OnReceiveFn = f
 }
+
+func (n *GameNetwork) Close() error {
+	err := n.server.Close()
+	logger, _ := logger.GetLogger()
+
+	if err != nil {
+		logger.Sugar().Errorf("Can't close connection for network '%+v': %s", n, err.Error())
+	} else {
+		logger.Sugar().Infof("Connection closed for network '%+v'", n)
+	}
+
+	return err
+}