fix: Prevent close on a nil channel (#49)

Author: keelerm84

server.go

@@ -351,6 +351,10 @@ func (s *subscription) send(e eventOrComment) bool {
 //
 // This should be called only from the Server.run() goroutine.
 func (s *subscription) close() {
+	if s.out == nil {
+		return
+	}
+
 	close(s.out)
 	s.out = nil
 }

server_test.go

@@ -5,6 +5,7 @@ import (
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
+	"sync"
 	"testing"
 	"time"
 
@@ -53,6 +54,27 @@ func TestNewServerHandlerRespondsAfterClose(t *testing.T) {
 	}
 }
 
+func TestServerHandlesLoadsOfPendingTasks(t *testing.T) {
+	channel := "test"
+	server := NewServer()
+	httpServer := httptest.NewServer(server.Handler(channel))
+	defer httpServer.Close()
+
+	var wg sync.WaitGroup
+	wg.Add(1)
+
+	go func() {
+		_, _ = http.Get(httpServer.URL)
+		wg.Wait()
+	}()
+
+	server.Register(channel, &testServerRepository{})
+	for i := 0; i < 1000; i++ {
+		server.PublishComment([]string{channel}, "my comment")
+	}
+	wg.Done()
+}
+
 func TestServerHandlerReceivesPublishedEvents(t *testing.T) {
 	channel := "test"
 	server := NewServer()