Ensure watching of db document changes

lib/wire.go

@@ -6,6 +6,10 @@ package lib
 import (
 	"context"
 	"log"
+	"os"
+	"os/signal"
+	"sync"
+	"syscall"
 
 	"ttt/internal/base"
 	"ttt/internal/game"
@@ -24,6 +28,8 @@ func Configure(appID string, db *db.Context, jwt *jwt.Context, mux *http.Mux) {
 	user.Configure(baseWire)
 
 	ensureIndexes(baseWire)
+
+	go ensureWatchers(baseWire)
 }
 
 func ensureIndexes(base base.Wire) {
@@ -41,3 +47,32 @@ func ensureIndexes(base base.Wire) {
 		}
 	}
 }
+
+func ensureWatchers(base base.Wire) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	var watchers = []db.Watcher{
+		base.AppStore(),
+	}
+
+	var waitGroup sync.WaitGroup
+
+	done := make(chan os.Signal, 1)
+	signal.Notify(done, syscall.SIGINT, syscall.SIGTERM)
+
+	for _, watcher := range watchers {
+		waitGroup.Add(1)
+
+		go func(w db.Watcher) {
+			defer waitGroup.Done()
+
+			err := w.Watch(ctx)
+			log.Println(err)
+		}(watcher)
+	}
+
+	<-done
+	cancel()
+	waitGroup.Wait()
+}