Allow for different message types

api.go

@@ -13,17 +13,24 @@ import (
 	"github.com/gorilla/websocket"
 )
 
+const (
+	msgTypeStore string = "store"
+)
+
 var (
 	socketSubscriptions     = map[string]func(msg interface{}) error{}
 	socketSubscriptionsLock = new(sync.RWMutex)
 )
 
-func sendAllSockets(msg interface{}) error {
+func sendAllSockets(msgType string, msg interface{}) error {
 	socketSubscriptionsLock.RLock()
 	defer socketSubscriptionsLock.RUnlock()
 
 	for _, hdl := range socketSubscriptions {
-		if err := hdl(msg); err != nil {
+		if err := hdl(map[string]interface{}{
+			"payload": msg,
+			"type":    msgType,
+		}); err != nil {
 			return errors.Wrap(err, "submit message")
 		}
 	}
@@ -80,7 +87,10 @@ func handleUpdateSocket(w http.ResponseWriter, r *http.Request) {
 		}
 	}()
 
-	if err := conn.WriteJSON(store); err != nil {
+	if err := conn.WriteJSON(map[string]interface{}{
+		"payload": store,
+		"type":    msgTypeStore,
+	}); err != nil {
 		log.WithError(err).Error("Unable to send initial state")
 		return
 	}

app.js

@@ -67,7 +67,15 @@ const app = new Vue({
       }
       this.socket.onmessage = evt => {
         const data = JSON.parse(evt.data)
+
+        switch (data.type) {
+          case 'store':
             this.store = data
+            break
+
+          default:
+            console.log(`Unhandled message type ${data.type}`, data)
+        }
       }
       this.socket.onopen = evt => {
         this.conn.avail = true

main.go

@@ -63,6 +63,7 @@ func main() {
 	registerAPI(router)
 
 	router.HandleFunc("/{file:(?:app.js|overlay.html)}", func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Cache-Control", "no-cache")
 		http.ServeFile(w, r, path.Join(cfg.AssetDir, mux.Vars(r)["file"]))
 	})
 
@@ -85,7 +86,7 @@ func main() {
 	for {
 		select {
 		case <-timerForceSync.C:
-			if err := sendAllSockets(store); err != nil {
+			if err := sendAllSockets(msgTypeStore, store); err != nil {
 				log.WithError(err).Error("Unable to send store to all sockets")
 			}
 

stats.go

@@ -77,7 +77,7 @@ func updateFollowers() error {
 	}
 
 	return errors.Wrap(
-		sendAllSockets(store),
+		sendAllSockets(msgTypeStore, store),
 		"update all sockets",
 	)
 }

webhook.go

@@ -98,7 +98,7 @@ func handleWebHookPush(w http.ResponseWriter, r *http.Request) {
 		logger.WithError(err).Error("Unable to update persistent store")
 	}
 
-	if err := sendAllSockets(store); err != nil {
+	if err := sendAllSockets(msgTypeStore, store); err != nil {
 		logger.WithError(err).Error("Unable to send update to all sockets")
 	}
 }