IODEMO: non blocking connection establishment, CR(10)

openucx · Mar 22, 2021 · d2bbaf1 · d2bbaf1
1 parent 17f8007
commit d2bbaf1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/apps/iodemo/ucx_wrapper.cc b/test/apps/iodemo/ucx_wrapper.cc
@@ -831,6 +831,7 @@ void UcxConnection::connect_tag(UcxCallback *callback)
     } else {
         established(UCS_PTR_STATUS(rreq));
         if (rreq != NULL) {
+            // failed to receive
             return;
         }
     }
@@ -934,11 +935,11 @@ void UcxConnection::handle_connection_error(ucs_status_t status)
     _ucx_status = status;
 
     /* the upper layer should close the connection */
-    if (_establish_cb != NULL) {
+    if (is_established()) {
+        _context.handle_connection_error(this);
+    } else {
         (*_establish_cb)(status);
         _establish_cb = NULL;
-    } else {
-        _context.handle_connection_error(this);
     }
 }