From bcaba7612398a31c8f9ca7d40bcfe66b0c25a62d Mon Sep 17 00:00:00 2001
From: Henauxg <19689618+Henauxg@users.noreply.github.com>
Date: Wed, 18 Jan 2023 15:54:37 +0100
Subject: [client] Use shared reception tasks

---
 src/client.rs | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

(limited to 'src/client.rs')

diff --git a/src/client.rs b/src/client.rs
index 51cc0ea..0efe051 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -148,9 +148,17 @@ impl Client {
         connection.open_channel(ChannelType::UnorderedReliable)?;
         connection.open_channel(ChannelType::Unreliable)?;
 
+        self.last_gen_id += 1;
+        let connection_id = self.last_gen_id;
+        self.connections.insert(connection_id, connection);
+        if self.default_connection_id.is_none() {
+            self.default_connection_id = Some(connection_id);
+        }
+
         // Async connection
         self.runtime.spawn(async move {
             connection_task(
+                connection_id,
                 config,
                 cert_mode,
                 to_sync_client_send,
@@ -162,13 +170,6 @@ impl Client {
             .await
         });
 
-        self.last_gen_id += 1;
-        let connection_id = self.last_gen_id;
-        self.connections.insert(connection_id, connection);
-        if self.default_connection_id.is_none() {
-            self.default_connection_id = Some(connection_id);
-        }
-
         Ok((connection_id, ordered_reliable_id))
     }
 
-- 
cgit v1.2.3