fix(rust): fixed app reporting inlet as disconnected when it was ac…

…tually connected
build-trust · Nov 17, 2023 · 1821aad · 1821aad
1 parent 0b82836
commit 1821aad
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/implementations/rust/ockam/ockam_app_lib/src/incoming_services/commands.rs b/implementations/rust/ockam/ockam_app_lib/src/incoming_services/commands.rs
@@ -41,13 +41,15 @@ impl AppState {
                 match result {
                     Ok(port) => {
                         if let Some(service) = guard.find_mut_by_id(service.id()) {
-                            service.set_port(port)
+                            if let Some(port) = port {
+                                service.set_port(port)
+                            }
                         }
                     }
                     Err(err) => {
                         warn!(%err, "Failed to refresh TCP inlet for accepted invitation");
                         if let Some(service) = guard.find_mut_by_id(service.id()) {
-                            service.set_port(None)
+                            service.remove_port()
                         }
                     }
                 }

diff --git a/implementations/rust/ockam/ockam_app_lib/src/incoming_services/state.rs b/implementations/rust/ockam/ockam_app_lib/src/incoming_services/state.rs
@@ -217,8 +217,12 @@ impl IncomingService {
         self.enabled
     }
 
-    pub fn set_port(&mut self, port: Option<Port>) {
-        self.port = port;
+    pub fn set_port(&mut self, port: Port) {
+        self.port = Some(port);
+    }
+
+    pub fn remove_port(&mut self) {
+        self.port = None;
     }
 
     pub fn enable(&mut self) {