fix: make get hardware info function thread-safe (#1956)

Co-authored-by: vansangpfiev <[email protected]>
janhq · Feb 11, 2025 · 4eba4ee · 4eba4ee
1 parent 36222f1
commit 4eba4ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/engine/services/hardware_service.cc b/engine/services/hardware_service.cc
@@ -35,6 +35,7 @@ bool TryConnectToServer(const std::string& host, int port) {
 
 HardwareInfo HardwareService::GetHardwareInfo() {
   // append active state
+  std::lock_guard<std::mutex> l(mtx_);
   auto gpus = cortex::hw::GetGPUInfo();
   auto res = db_service_->LoadHardwareList();
   if (res.has_value()) {

diff --git a/engine/services/hardware_service.h b/engine/services/hardware_service.h
@@ -2,6 +2,7 @@
 #include <stdint.h>
 #include <string>
 #include <vector>
+#include <mutex>
 
 #include "common/hardware_config.h"
 #include "database_service.h"
@@ -39,4 +40,5 @@ class HardwareService {
  private:
   std::shared_ptr<DatabaseService> db_service_ = nullptr;
   std::optional<cortex::hw::ActivateHardwareConfig> ahc_;
+  std::mutex mtx_;
 };