Fix NativeArrayContent

kochounoyume · Jan 1, 2025 · 062d6e3 · 062d6e3
1 parent 221ca9b
commit 062d6e3
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/Packages/MinimalUtility/Runtime/WebRequest/NativeArrayContent.cs b/Packages/MinimalUtility/Runtime/WebRequest/NativeArrayContent.cs
@@ -1,8 +1,12 @@
-using System;
+#if ENABLE_WEBREQUEST && ENABLE_UNITASK
+#nullable enable
+
+using System;
 using System.IO;
 using System.Net;
 using System.Net.Http;
 using System.Threading.Tasks;
+using Cysharp.Threading.Tasks;
 using Unity.Collections;
 
 namespace MinimalUtility.WebRequest
@@ -21,10 +25,12 @@ public NativeArrayContent(NativeArray<byte>.ReadOnly data, IDisposable handler)
             _handler = handler;
         }
 
-        protected override Task SerializeToStreamAsync(Stream stream, TransportContext _)
+        protected override async Task SerializeToStreamAsync(Stream stream, TransportContext _)
         {
-            stream.Write(_data.AsReadOnlySpan());
-            return Task.CompletedTask;
+            await using (UniTask.ReturnToMainThread())
+            {
+                stream.Write(_data.AsReadOnlySpan());
+            }
         }
 
         protected override bool TryComputeLength(out long length)
@@ -42,4 +48,5 @@ protected override void Dispose(bool disposing)
             base.Dispose(disposing);
         }
     }
-}
+}
+#endif