update random_device and find

OpenAtomFoundation · Nov 8, 2023 · 7aafd54 · 7aafd54
1 parent 20d29c4
commit 7aafd54
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/helper.h b/src/helper.h
@@ -7,7 +7,7 @@
 
 #pragma once
 
-#include <cstdlib>
+#include <random>
 #include <vector>
 #include "pstring.h"
 
@@ -56,7 +56,11 @@ inline typename HASH::const_iterator FollyRandomHashMember(const HASH& container
   }
 
   it = container.cbegin();
-  size_t randomIdx = random() % container.size();
+  std::random_device rd;
+  std::mt19937 gen(rd());
+  std::uniform_int_distribution<> dis(0, container.size() - 1);
+
+  size_t randomIdx = dis(gen);
   while (randomIdx > 0) {
     randomIdx--;
     ++it;

diff --git a/src/store.cc b/src/store.cc
@@ -210,7 +210,7 @@ size_t PStore::BlockedClients::UnblockClient(PClient* client) {
   const auto& keys = client->WaitingKeys();
 
   for (const auto& key : keys) {
-    Clients clients = blockedClients_[key];
+    Clients clients = blockedClients_.find(key)->second;
     assert(!clients.empty());
 
     for (auto it(clients.begin()); it != clients.end(); ++it) {