From 05ecd0105d4e723517bc269c9b42bc650743ee89 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Wed, 20 Jan 2021 20:51:10 +0100
Subject: Changing local state via service context

---
 src/network.cpp | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

(limited to 'src/network.cpp')

diff --git a/src/network.cpp b/src/network.cpp
index eab0f67..c0bcc2a 100644
--- a/src/network.cpp
+++ b/src/network.cpp
@@ -550,16 +550,22 @@ void Server::Peer::updateService(ReplyBuilder & reply)
 	for (auto & x : serviceQueue) {
 		if (auto ref = std::get<1>(x)->check(reply)) {
 			if (lpeer) {
-				Service::Context ctx(new Service::Context::Priv {
-					.ref = *ref,
-					.peer = erebos::Peer(lpeer),
-				});
-
-				for (auto & svc : server.services)
-					if (svc->uuid() == std::get<UUID>(x)) {
-						svc->handle(ctx);
-						break;
+				server.localHead.update([&] (const Stored<LocalState> & local) {
+					Service::Context ctx(new Service::Context::Priv {
+						.ref = *ref,
+						.peer = erebos::Peer(lpeer),
+						.local = local,
+					});
+
+					for (auto & svc : server.services) {
+						if (svc->uuid() == std::get<UUID>(x)) {
+							svc->handle(ctx);
+							break;
+						}
 					}
+
+					return ctx.local();
+				});
 			}
 		} else {
 			next.push_back(std::move(x));
-- 
cgit v1.2.3