From 5988d10a0b467e2b8171c4cd86a151947d85a291 Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Thu, 6 Feb 2020 21:30:44 +0100
Subject: [PATCH] Remove downstream conn from list on disconnect

---
 server.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/server.go b/server.go
index 92c7820..f231908 100644
--- a/server.go
+++ b/server.go
@@ -3,6 +3,7 @@ package jounce
 import (
 	"fmt"
 	"net"
+	"sync"
 
 	"gopkg.in/irc.v3"
 )
@@ -42,6 +43,7 @@ type Server struct {
 	Logger    Logger
 	Upstreams []Upstream // TODO: per-user
 
+	lock            sync.Mutex
 	downstreamConns []*downstreamConn
 }
 
@@ -74,11 +76,21 @@ func (s *Server) Serve(ln net.Listener) error {
 		}
 
 		conn := newDownstreamConn(s, netConn)
-		s.downstreamConns = append(s.downstreamConns, conn)
 		go func() {
+			s.lock.Lock()
+			s.downstreamConns = append(s.downstreamConns, conn)
+			s.lock.Unlock()
 			if err := conn.readMessages(); err != nil {
 				conn.logger.Printf("failed to handle messages: %v", err)
 			}
+			s.lock.Lock()
+			for i, c := range s.downstreamConns {
+				if c == conn {
+					s.downstreamConns = append(s.downstreamConns[:i], s.downstreamConns[i+1:]...)
+					break
+				}
+			}
+			s.lock.Unlock()
 		}()
 	}
 }