From a5b25878c78d9791aca373b65c8943ced0256cc0 Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Fri, 18 Dec 2020 13:26:49 +0000
Subject: [PATCH] Make sure we wake up for our own key changes

---
 syncapi/consumers/keychange.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/syncapi/consumers/keychange.go b/syncapi/consumers/keychange.go
index 128f6011e..59cd583d1 100644
--- a/syncapi/consumers/keychange.go
+++ b/syncapi/consumers/keychange.go
@@ -112,7 +112,8 @@ func (s *OutputKeyChangeEventConsumer) onMessage(msg *sarama.ConsumerMessage) er
 		log.WithError(err).Error("syncapi: failed to QuerySharedUsers for key change event from key server")
 		return err
 	}
-	// TODO: f.e queryRes.UserIDsToCount : notify users by waking up streams
+	// make sure we get our own key updates too!
+	queryRes.UserIDsToCount[output.UserID] = 1
 	posUpdate := types.StreamingToken{
 		DeviceListPosition: types.LogPosition{
 			Offset:    msg.Offset,