From d7093a57a7b9c9772b3e4e858d46350bebc351e2 Mon Sep 17 00:00:00 2001
From: Ian Kent <email@iankent.co.uk>
Date: Sun, 21 Dec 2014 09:18:19 +0000
Subject: [PATCH] Fix #18

---
 MailHog-Server/api/v1.go | 28 ++++++++++------------------
 Makefile                 |  1 +
 2 files changed, 11 insertions(+), 18 deletions(-)

diff --git a/MailHog-Server/api/v1.go b/MailHog-Server/api/v1.go
index 41ccabc..f3683e2 100644
--- a/MailHog-Server/api/v1.go
+++ b/MailHog-Server/api/v1.go
@@ -11,18 +11,16 @@ import (
 	"github.com/ian-kent/go-log/log"
 	gotcha "github.com/ian-kent/gotcha/app"
 	"github.com/ian-kent/gotcha/http"
+
+	"github.com/ian-kent/goose"
 )
 
 type APIv1 struct {
-	config         *config.Config
-	eventlisteners []*EventListener
-	app            *gotcha.App
+	config *config.Config
+	app    *gotcha.App
 }
 
-type EventListener struct {
-	session *http.Session
-	ch      chan []byte
-}
+var stream *goose.EventStream
 
 type ReleaseConfig struct {
 	Email string
@@ -33,11 +31,11 @@ type ReleaseConfig struct {
 func CreateAPIv1(conf *config.Config, app *gotcha.App) *APIv1 {
 	log.Println("Creating API v1")
 	apiv1 := &APIv1{
-		config:         conf,
-		eventlisteners: make([]*EventListener, 0),
-		app:            app,
+		config: conf,
+		app:    app,
 	}
 
+	stream = goose.NewEventStream()
 	r := app.Router
 
 	r.Get("/api/v1/messages/?", apiv1.messages)
@@ -68,19 +66,13 @@ func CreateAPIv1(conf *config.Config, app *gotcha.App) *APIv1 {
 func (apiv1 *APIv1) broadcast(json string) {
 	log.Println("[APIv1] BROADCAST /api/v1/events")
 	b := []byte(json)
-	for _, l := range apiv1.eventlisteners {
-		log.Printf("Sending to connection: %s\n", l.session.Request.RemoteAddr)
-		l.ch <- b
-	}
+	stream.Notify("data", b)
 }
 
 func (apiv1 *APIv1) eventstream(session *http.Session) {
 	log.Println("[APIv1] GET /api/v1/events")
 
-	apiv1.eventlisteners = append(apiv1.eventlisteners, &EventListener{
-		session,
-		session.Response.EventStream(),
-	})
+	stream.AddReceiver(session.Response.GetWriter())
 }
 
 func (apiv1 *APIv1) messages(session *http.Session) {
diff --git a/Makefile b/Makefile
index 0800f16..db09b26 100644
--- a/Makefile
+++ b/Makefile
@@ -24,6 +24,7 @@ deps:
 	go get github.com/ian-kent/gotcha/gotcha
 	go get github.com/ian-kent/go-log/log
 	go get github.com/ian-kent/envconf
+	go get github.com/ian-kent/goose
 	go get github.com/jteeuwen/go-bindata/...
 	go get labix.org/v2/mgo
 	# added to fix travis issues