botchain/control
1
0
Fork 0
mirror of https://github.com/BotChain-Robots/control.git synced 2026-03-09 16:22:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix some concurrency issues causing unreliability

Browse Source
This commit is contained in:
Johnathon Slightham
2026-02-28 22:36:54 -05:00
parent 9066ccd721
commit 657dc42833
2 changed files with 37 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib_c_control.cpp
View File

@@ -30,7 +30,11 @@ LIB_API void cleanup() {
LIB_API int send_angle_control(int module_id, int angle) {
if (const auto maybe_module = robot_controller->getModule(module_id)) {
const auto module = (*maybe_module).lock();
if (module) {
module->actuate(angle);
} else {
spdlog::warn("[c_control] send_angle_control: module {} has expired", module_id);
}
}
return 0;
}

51
src/libcontrol.cpp
View File

@@ -35,10 +35,15 @@ using namespace std::chrono_literals;
RobotController::~RobotController() {
m_stop_thread = true;
if (m_metadata_loop.joinable())
m_metadata_loop.join();
if (m_transmit_loop.joinable())
m_transmit_loop.join();
if (m_configuration_loop.joinable())
m_configuration_loop.join();
if (m_sensor_loop.joinable())
m_sensor_loop.join();
if (m_expiry_looop.joinable())
m_expiry_looop.join();
}
@@ -85,15 +90,17 @@ std::optional<std::weak_ptr<Module>> RobotController::getModule(uint8_t device_i
void RobotController::resetModules() {
std::unique_lock module_lock(m_module_lock);
std::unique_lock conn_lock(m_connection_lock);
m_id_to_module.erase(m_id_to_module.begin(), m_id_to_module.end());
m_connection_map.erase(m_connection_map.begin(), m_connection_map.end());
m_id_to_module.clear();
m_connection_map.clear();
}
void RobotController::fetchDirectlyConnectedModules(bool block) {
spdlog::info("[Control] Fetching modules from network");
auto t = std::thread([&] {
auto out = m_messaging_interface->find_connected_modules(
std::chrono::milliseconds(SCAN_DURATION_MS));
// Capture m_messaging_interface by value (shared_ptr) so the detached
// thread does not hold a dangling reference to 'this'.
auto messaging = m_messaging_interface;
auto t = std::thread([messaging] {
auto out = messaging->find_connected_modules(std::chrono::milliseconds(SCAN_DURATION_MS));
spdlog::info("[Control] Found {} modules on the network", out.size());
});
@@ -189,25 +196,21 @@ void RobotController::expiry_loop() {
}
}
// todo
// Remove connections
// for (auto it = m_connection_map.begin(); it != m_connection_map.end();) {
// // Remove it->x connections
// if (delete_modules.contains(it->first)) {
// it = m_connection_map.erase(it);
// } else {
// ++it;
// }
// // Remove x->it connections
// for (auto it2 = it->second.begin(); it2 != it->second.end();) {
// if (delete_modules.contains(it2->to_module_id)) {
// it2 = it->second.erase(it2);
// } else {
// ++it2;
// }
// }
// }
// Remove connections involving expired modules.
for (auto it = m_connection_map.begin(); it != m_connection_map.end();) {
if (delete_modules.contains(it->first)) {
it = m_connection_map.erase(it);
} else {
// Remove individual connections pointing to an expired module.
auto &conns = it->second;
conns.erase(std::remove_if(conns.begin(), conns.end(),
[&](const Flatbuffers::ModuleConnectionInstance &c) {
return delete_modules.contains(c.to_module_id);
}),
conns.end());
++it;
}
}
}
}