update

Adjusting the job handle name. Migrating the m_FlushSendJobHandle.Complete(); outside of the driver IsCreated check to avoid any potential job safety issues.
Unity-Technologies · Oct 31, 2024 · 85e7b2b · 85e7b2b
1 parent c0d2cbf
commit 85e7b2b
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/com.unity.netcode.gameobjects/Runtime/Transports/UTP/UnityTransport.cs b/com.unity.netcode.gameobjects/Runtime/Transports/UTP/UnityTransport.cs
@@ -944,8 +944,11 @@ private bool ProcessEvent()
             return false;
         }
 
+        private JobHandle m_FlushSendJobHandle;
+
         internal override void EarlyUpdate()
         {
+            m_FlushSendJobHandle.Complete();
             if (m_Driver.IsCreated)
             {
                 if (m_ProtocolType == ProtocolType.RelayUnityTransport && m_Driver.GetRelayConnectionStatus() == RelayConnectionStatus.AllocationInvalid)
@@ -959,10 +962,6 @@ internal override void EarlyUpdate()
 
                 m_Driver.ScheduleUpdate().Complete();
 
-                // Schedule a flush send as the last transport action for the
-                // current frame.
-                m_Driver.ScheduleFlushSend(default).Complete();
-
                 while (AcceptConnection() && m_Driver.IsCreated)
                 {
                     ;
@@ -984,11 +983,9 @@ internal override void PostLateUpdate()
                     SendBatchedMessages(kvp.Key, kvp.Value);
                 }
 
-                m_Driver.ScheduleUpdate();
-
                 // Schedule a flush send as the last transport action for the
                 // current frame.
-                m_Driver.ScheduleFlushSend(default);
+                m_FlushSendJobHandle = m_Driver.ScheduleFlushSend(default);
             }
 
 #if MULTIPLAYER_TOOLS_1_0_0_PRE_7