update from master

scaleoutsystems · Dec 7, 2023 · bf471e3 · bf471e3
1 parent 495699d
commit bf471e3
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/fedn/fedn/network/storage/statestore/mongostatestore.py b/fedn/fedn/network/storage/statestore/mongostatestore.py
@@ -808,17 +808,16 @@ def set_round_data(self, round_id, round_data):
         self.rounds.update_one({'round_id': round_id}, {
             '$set': {'round_data': round_data}}, True)
 
-    def update_client_status(self, client_name, status):
-        """Update client status in statestore.
-
+    def update_client_status(self, clients, status):
+        """ Update client status in statestore.
         :param client_name: The client name
         :type client_name: str
         :param status: The client status
         :type status: str
         :return: None
         """
         datetime_now = datetime.now()
-        filter_query = {"name": client_name}
+        filter_query = {"name": {"$in": clients}}
 
         update_query = {"$set": {"last_seen": datetime_now, "status": status}}
-        self.clients.update_one(filter_query, update_query)
+        self.clients.update_many(filter_query, update_query)