Merge pull request #963 from kebby/master
[mono.git] / mcs / class / System.Net.Http / System.Net.Http / HttpClient.cs
index 839f073cc2b6f5ef6a57d9be0a2158b5edd6c49e..41ece441a4053fb43e565d57f3d174d30ab92fa4 100644 (file)
 
 using System.Threading;
 using System.Net.Http.Headers;
+using System.Threading.Tasks;
+using System.IO;
 
 namespace System.Net.Http
 {
-       public class HttpClient : IDisposable
+       public class HttpClient : HttpMessageInvoker
        {
                static readonly TimeSpan TimeoutDefault = TimeSpan.FromSeconds (100);
 
                Uri base_address;
-               CancellationTokenSource cancellation_token;
+               CancellationTokenSource cts;
                bool disposed;
-               readonly HttpMessageHandler handler;
                HttpRequestHeaders headers;
-               int buffer_size;
+               long buffer_size;
                TimeSpan timeout;
 
                public HttpClient ()
-                       : this (null)
+                       : this (new HttpClientHandler (), true)
                {
                }
-
+               
                public HttpClient (HttpMessageHandler handler)
+                       : this (handler, true)
+               {
+               }
+
+               public HttpClient (HttpMessageHandler handler, bool disposeHandler)
+                       : base (handler, disposeHandler)
                {
-                       this.handler = handler ?? new HttpClientHandler ();
-                       buffer_size = 0x10000;
+                       buffer_size = int.MaxValue;
                        timeout = TimeoutDefault;
+                       cts = new CancellationTokenSource ();
                }
 
                public Uri BaseAddress {
@@ -70,7 +77,7 @@ namespace System.Net.Http
                        }
                }
 
-               public int MaxResponseContentBufferSize {
+               public long MaxResponseContentBufferSize {
                        get {
                                return buffer_size;
                        }
@@ -96,83 +103,138 @@ namespace System.Net.Http
 
                public void CancelPendingRequests ()
                {
-                       if (cancellation_token != null)
-                               cancellation_token.Cancel ();
-
-                       cancellation_token = new CancellationTokenSource ();
-               }
-               public void Dispose ()
-               {
-                       Dispose (true);
+                       // Cancel only any already running requests not any new request after this cancellation
+                       using (var c = Interlocked.Exchange (ref cts, new CancellationTokenSource ()))
+                               c.Cancel ();
                }
-               
-               protected virtual void Dispose (bool disposing)
+
+               protected override void Dispose (bool disposing)
                {
                        if (disposing && !disposed) {
                                disposed = true;
 
-                               if (cancellation_token != null)
-                                       cancellation_token.Dispose ();
+                               cts.Dispose ();
                        }
+                       
+                       base.Dispose (disposing);
+               }
+
+               public Task<HttpResponseMessage> DeleteAsync (string requestUri)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Delete, requestUri));
+               }
+
+               public Task<HttpResponseMessage> DeleteAsync (string requestUri, CancellationToken cancellationToken)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Delete, requestUri), cancellationToken);
+               }
+
+               public Task<HttpResponseMessage> DeleteAsync (Uri requestUri)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Delete, requestUri));
                }
 
-               public HttpResponseMessage Delete (string requestUri)
+               public Task<HttpResponseMessage> DeleteAsync (Uri requestUri, CancellationToken cancellationToken)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Delete, requestUri));
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Delete, requestUri), cancellationToken);
                }
 
-               public HttpResponseMessage Delete (Uri requestUri)
+               public Task<HttpResponseMessage> GetAsync (string requestUri)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Delete, requestUri));
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri));
                }
 
-               public HttpResponseMessage Get (string requestUri)
+               public Task<HttpResponseMessage> GetAsync (string requestUri, CancellationToken cancellationToken)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Get, requestUri));
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), cancellationToken);
                }
 
-               public HttpResponseMessage Get (Uri requestUri)
+               public Task<HttpResponseMessage> GetAsync (string requestUri, HttpCompletionOption completionOption)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Get, requestUri));
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), completionOption);
                }
 
-               public HttpResponseMessage Post (string requestUri, HttpContent content)
+               public Task<HttpResponseMessage> GetAsync (string requestUri, HttpCompletionOption completionOption, CancellationToken cancellationToken)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content });
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), completionOption, cancellationToken);
                }
 
-               public HttpResponseMessage Put (Uri requestUri, HttpContent content)
+               public Task<HttpResponseMessage> GetAsync (Uri requestUri)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content });
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri));
                }
 
-               public HttpResponseMessage Put (string requestUri, HttpContent content)
+               public Task<HttpResponseMessage> GetAsync (Uri requestUri, CancellationToken cancellationToken)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content });
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), cancellationToken);
                }
 
-               public HttpResponseMessage Post (Uri requestUri, HttpContent content)
+               public Task<HttpResponseMessage> GetAsync (Uri requestUri, HttpCompletionOption completionOption)
                {
-                       return Send (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content });
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), completionOption);
                }
 
-               public HttpResponseMessage Send (HttpRequestMessage request)
+               public Task<HttpResponseMessage> GetAsync (Uri requestUri, HttpCompletionOption completionOption, CancellationToken cancellationToken)
                {
-                       return Send (request, HttpCompletionOption.ResponseContentRead, CancellationToken.None);
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Get, requestUri), completionOption, cancellationToken);
                }
 
-               public HttpResponseMessage Send (HttpRequestMessage request, HttpCompletionOption completionOption)
+               public Task<HttpResponseMessage> PostAsync (string requestUri, HttpContent content)
                {
-                       return Send (request, completionOption, CancellationToken.None);
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content });
                }
 
-               public HttpResponseMessage Send (HttpRequestMessage request, CancellationToken cancellationToken)
+               public Task<HttpResponseMessage> PostAsync (string requestUri, HttpContent content, CancellationToken cancellationToken)
                {
-                       return Send (request, HttpCompletionOption.ResponseContentRead, cancellationToken);
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content }, cancellationToken);
                }
 
-               public HttpResponseMessage Send (HttpRequestMessage request, HttpCompletionOption completionOption, CancellationToken cancellationToken)
+               public Task<HttpResponseMessage> PostAsync (Uri requestUri, HttpContent content)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content });
+               }
+
+               public Task<HttpResponseMessage> PostAsync (Uri requestUri, HttpContent content, CancellationToken cancellationToken)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Post, requestUri) { Content = content }, cancellationToken);
+               }
+
+               public Task<HttpResponseMessage> PutAsync (Uri requestUri, HttpContent content)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content });
+               }
+
+               public Task<HttpResponseMessage> PutAsync (Uri requestUri, HttpContent content, CancellationToken cancellationToken)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content }, cancellationToken);
+               }
+
+               public Task<HttpResponseMessage> PutAsync (string requestUri, HttpContent content)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content });
+               }
+
+               public Task<HttpResponseMessage> PutAsync (string requestUri, HttpContent content, CancellationToken cancellationToken)
+               {
+                       return SendAsync (new HttpRequestMessage (HttpMethod.Put, requestUri) { Content = content }, cancellationToken);
+               }
+
+               public Task<HttpResponseMessage> SendAsync (HttpRequestMessage request)
+               {
+                       return SendAsync (request, HttpCompletionOption.ResponseContentRead, CancellationToken.None);
+               }
+
+               public Task<HttpResponseMessage> SendAsync (HttpRequestMessage request, HttpCompletionOption completionOption)
+               {
+                       return SendAsync (request, completionOption, CancellationToken.None);
+               }
+
+               public override Task<HttpResponseMessage> SendAsync (HttpRequestMessage request, CancellationToken cancellationToken)
+               {
+                       return SendAsync (request, HttpCompletionOption.ResponseContentRead, cancellationToken);
+               }
+
+               public Task<HttpResponseMessage> SendAsync (HttpRequestMessage request, HttpCompletionOption completionOption, CancellationToken cancellationToken)
                {
                        if (request == null)
                                throw new ArgumentNullException ("request");
@@ -185,24 +247,87 @@ namespace System.Net.Http
                                        throw new InvalidOperationException ("The request URI must either be an absolute URI or BaseAddress must be set");
 
                                request.RequestUri = base_address;
-                       }
+                       } else if (!request.RequestUri.IsAbsoluteUri) {
+                               if (base_address == null)
+                                       throw new InvalidOperationException ("The request URI must either be an absolute URI or BaseAddress must be set");
 
-                       try {
-                               if (cancellation_token == null)
-                                       cancellation_token = new CancellationTokenSource ();
+                               request.RequestUri = new Uri (base_address, request.RequestUri);
+                       }
 
-                               using (var cts = CancellationTokenSource.CreateLinkedTokenSource (cancellation_token.Token, cancellationToken)) {
-                                       cts.CancelAfter (timeout);
+                       if (headers != null) {
+                               request.Headers.AddHeaders (headers);
+                       }
 
-                                       var response = handler.Send (request, cts.Token);
-                                       if (response == null)
-                                               throw new InvalidOperationException ("Handler failed to return a response");
+                       return SendAsyncWorker (request, completionOption, cancellationToken);
+               }
 
-                                       return response;
+               async Task<HttpResponseMessage> SendAsyncWorker (HttpRequestMessage request, HttpCompletionOption completionOption, CancellationToken cancellationToken)
+               {
+                       using (var lcts = CancellationTokenSource.CreateLinkedTokenSource (cts.Token, cancellationToken)) {
+                               lcts.CancelAfter (timeout);
+
+                               var task = base.SendAsync (request, lcts.Token);
+                               if (task == null)
+                                       throw new InvalidOperationException ("Handler failed to return a value");
+                                       
+                               var response = await task.ConfigureAwait (false);
+                               if (response == null)
+                                       throw new InvalidOperationException ("Handler failed to return a response");
+
+                               //
+                               // Read the content when default HttpCompletionOption.ResponseContentRead is set
+                               //
+                               if (response.Content != null && (completionOption & HttpCompletionOption.ResponseHeadersRead) == 0) {
+                                       await response.Content.LoadIntoBufferAsync (MaxResponseContentBufferSize).ConfigureAwait (false);
                                }
-                       } finally {
-                               cancellation_token.Dispose ();
-                               cancellation_token = null;
+                                       
+                               return response;
+                       }
+               }
+
+               public async Task<byte[]> GetByteArrayAsync (string requestUri)
+               {
+                       using (var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false)) {
+                               resp.EnsureSuccessStatusCode ();
+                               return await resp.Content.ReadAsByteArrayAsync ().ConfigureAwait (false);
+                       }
+               }
+
+               public async Task<byte[]> GetByteArrayAsync (Uri requestUri)
+               {
+                       using (var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false)) {
+                               resp.EnsureSuccessStatusCode ();
+                               return await resp.Content.ReadAsByteArrayAsync ().ConfigureAwait (false);
+                       }
+               }
+
+               public async Task<Stream> GetStreamAsync (string requestUri)
+               {
+                       var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false);
+                       resp.EnsureSuccessStatusCode ();
+                       return await resp.Content.ReadAsStreamAsync ().ConfigureAwait (false);
+               }
+
+               public async Task<Stream> GetStreamAsync (Uri requestUri)
+               {
+                       var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false);
+                       resp.EnsureSuccessStatusCode ();
+                       return await resp.Content.ReadAsStreamAsync ().ConfigureAwait (false);
+               }
+
+               public async Task<string> GetStringAsync (string requestUri)
+               {
+                       using (var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false)) {
+                               resp.EnsureSuccessStatusCode ();
+                               return await resp.Content.ReadAsStringAsync ().ConfigureAwait (false);
+                       }
+               }
+
+               public async Task<string> GetStringAsync (Uri requestUri)
+               {
+                       using (var resp = await GetAsync (requestUri, HttpCompletionOption.ResponseContentRead).ConfigureAwait (false)) {
+                               resp.EnsureSuccessStatusCode ();
+                               return await resp.Content.ReadAsStringAsync ().ConfigureAwait (false);
                        }
                }
        }