为HttpClient模块提供POST方法的支持

LanbingIce · Sep 28, 2023 · 5ecdd51 · 5ecdd51
1 parent 12da5d9
commit 5ecdd51
Showing 1 changed file with 23 additions and 20 deletions.
diff --git a/IsaacSocket/Modules/HttpClientModule.cs b/IsaacSocket/Modules/HttpClientModule.cs
@@ -201,36 +201,39 @@ internal override void ReceiveMemoryMessage(byte[] message)
         {
             request = new()
             {
+                Content = new ByteArrayContent(body),
+                Method = action == ActionType.GET_REQUEST ? HttpMethod.Get : HttpMethod.Post,
                 RequestUri = new(url)
             };
             foreach (var kvp in headers)
             {
-                request.Headers.Add(kvp.Key, kvp.Value);
+                try
+                {
+                    request.Headers.Add(kvp.Key, kvp.Value);
+                }
+                catch (InvalidOperationException)
+                {
+                    request.Content.Headers.Add(kvp.Key, kvp.Value);
+                }
             }
         }
         catch (Exception e)
         {
             Callback(CallbackType.MEMORY_MESSAGE_GENERATED, SerializeFaulted(id, e.Message));
             return;
         }
-        switch (action)
-        {
-            case ActionType.GET_REQUEST:
-                request.Method = HttpMethod.Get;
-                Interlocked.Increment(ref taskCounter);
-                client.SendAsync(request, cancellationTokenSource.Token).ContinueWith(task =>
-                 {
-                     if (task.IsCompletedSuccessfully)
-                     {
-                         Callback(CallbackType.MEMORY_MESSAGE_GENERATED, SerializeResponse(id, task.Result));
-                     }
-                     else
-                     {
-                         Callback(CallbackType.MEMORY_MESSAGE_GENERATED, SerializeFaulted(id, task.Exception?.Message ?? ""));
-                     }
-                     Interlocked.Decrement(ref taskCounter);
-                 });
-                break;
-        }
+        Interlocked.Increment(ref taskCounter);
+        client.SendAsync(request, cancellationTokenSource.Token).ContinueWith(task =>
+         {
+             if (task.IsCompletedSuccessfully)
+             {
+                 Callback(CallbackType.MEMORY_MESSAGE_GENERATED, SerializeResponse(id, task.Result));
+             }
+             else
+             {
+                 Callback(CallbackType.MEMORY_MESSAGE_GENERATED, SerializeFaulted(id, task.Exception?.Message ?? ""));
+             }
+             Interlocked.Decrement(ref taskCounter);
+         });
     }
 }