X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/2a276c273c21a1752d541bf87e074c5d1c27e30f..2124c559da8f164a43a3da987b399b6ac6125035:/daemon.go

diff --git a/daemon.go b/daemon.go
index 3a54b1a..958d6f7 100644
--- a/daemon.go
+++ b/daemon.go
@@ -364,8 +364,15 @@ func daemonServe(ctx context.Context, listener net.Listener, bot *TelegramBot, w
 		loops = append(loops, loop)
 	}
 
+	messages, err := openDaemonMessages(loops)
+	if err != nil {
+		return errors.Join(fmt.Errorf("open agent messages: %w", err), listener.Close())
+	}
+
 	hub.snapshot = func(ctx context.Context, snapshot func() error) error {
-		return daemonSnapshot(ctx, loops, snapshot)
+		return messages.Snapshot(func() error {
+			return daemonSnapshot(ctx, loops, snapshot)
+		})
 	}
 
 	var workers sync.WaitGroup