增加HeadersAttribute特性

Author: xljiulang

WebApiClient/Attributes/ParameterAttributes/HeadersAttribute.cs

@@ -0,0 +1,36 @@
+using System;
+using System.Threading.Tasks;
+using WebApiClient.Contexts;
+using WebApiClient.Interfaces;
+
+namespace WebApiClient.Attributes
+{
+    /// <summary>
+    /// 表示参数值作为请求头   
+    /// 对于复杂类型的参数值，将拆解作为多个Header
+    /// </summary>
+    [AttributeUsage(AttributeTargets.Parameter, AllowMultiple = false, Inherited = true)]
+    public class HeadersAttribute : Attribute, IApiParameterAttribute
+    {
+        /// <summary>
+        /// http请求之前
+        /// </summary>
+        /// <param name="context">上下文</param>
+        /// <param name="parameter">特性关联的参数</param>
+        /// <exception cref="HttpApiConfigException"></exception>
+        /// <returns></returns>
+        public async Task BeforeRequestAsync(ApiActionContext context, ApiParameterDescriptor parameter)
+        {
+            var options = context.HttpApiConfig.FormatOptions;
+            var formatter = context.HttpApiConfig.KeyValueFormatter;
+            var keyValues = formatter.Serialize(parameter, options);
+
+            foreach (var kv in keyValues)
+            {
+                var name = kv.Key.Replace("_", "-");
+                var header = new HeaderAttribute(name, kv.Value);
+                await header.BeforeRequestAsync(context);
+            }
+        }
+    }
+}

WebApiClient/HttpApiRequestMessage.cs

@@ -158,6 +158,7 @@ public async Task AddFormFieldAsync(IEnumerable<KeyValuePair<string, string>> ke
             {
                 this.EnsureMediaTypeEqual(mediaType);
                 formBody = await this.Content.ReadAsByteArrayAsync();
+                this.Content.Dispose();
             }
 
             var bytesContent = MergeFields(formBody, keyValues);

WebApiClientTest/Attributes/ParameterAttributes/HeadersAttributeTest.cs

@@ -0,0 +1,48 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Net.Http;
+using System.Threading.Tasks;
+using WebApiClient;
+using WebApiClient.Attributes;
+using WebApiClient.Contexts;
+using Xunit;
+
+
+namespace WebApiClientTest.Attributes.ParameterAttributes
+{
+    public class HeadersAttributeTest
+    {
+        public interface IMyApi : IDisposable
+        {
+            ITask<HttpResponseMessage> PostAsync(object headers);
+        }
+
+        [Fact]
+        public async Task IApiParameterAttributeTest()
+        {
+            var context = new ApiActionContext
+            {
+                HttpApiConfig = new HttpApiConfig(),
+                RequestMessage = new HttpApiRequestMessage
+                {
+                    RequestUri = new Uri("http://www.mywebapi.com"),
+                    Method = HttpMethod.Post
+                },
+                ApiActionDescriptor = ApiDescriptorCache.GetApiActionDescriptor(typeof(IMyApi).GetMethod("PostAsync"))
+            };
+
+            var parameter = context.ApiActionDescriptor.Parameters[0];
+            parameter.Value = new { @class = 123, User_Agent = "WebApiClient" };
+
+            var attr = new HeadersAttribute();
+            await attr.BeforeRequestAsync(context, parameter);
+
+            context.RequestMessage.Headers.TryGetValues("User-Agent", out IEnumerable<string> values);
+            Assert.Equal("WebApiClient", values.FirstOrDefault());
+
+            context.RequestMessage.Headers.TryGetValues("class", out IEnumerable<string> cValues);
+            Assert.Equal("123", cValues.FirstOrDefault());
+        }
+    }
+}