fix(rust): fix BleTransport initialization race condition

build-trust · Feb 18, 2022 · 6d753cb · 6d753cb
1 parent b2db7b4
commit 6d753cb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/implementations/rust/ockam/ockam_transport_ble/src/router/mod.rs b/implementations/rust/ockam/ockam_transport_ble/src/router/mod.rs
@@ -88,8 +88,6 @@ impl Worker for BleRouter {
     type Message = RouterMessage;
 
     async fn initialize(&mut self, ctx: &mut Context) -> Result<()> {
-        debug!("Registering Ble router for type = {}", crate::BLE);
-        ctx.register(crate::BLE, ctx.address()).await?;
         ctx.set_cluster(crate::CLUSTER_NAME).await?;
         Ok(())
     }
@@ -134,9 +132,12 @@ impl BleRouter {
 
         let handle = router.create_self_handle(ctx).await?;
 
-        trace!("BleRouter start_worker({:?})", addr.clone());
+        trace!("Start Ble router for address = {:?}", addr.clone());
         ctx.start_worker(addr.clone(), router).await?;
 
+        trace!("Registering Ble router for type = {}", crate::BLE);
+        ctx.register(crate::BLE, ctx.address()).await?;
+
         Ok(handle)
     }
 }