Add timeout to Transport helper extension methods (#167)

Author: Mpdreamz

src/Elastic.Transport/ITransportHttpMethodExtensions.cs

@@ -2,6 +2,7 @@
 // Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
 // See the LICENSE file in the project root for more information
 
+using System;
 using System.Threading;
 using System.Threading.Tasks;
 using static Elastic.Transport.HttpMethod;
@@ -37,6 +38,8 @@ public static Task<TResponse> GetAsync<TResponse>(this ITransport transport, str
 		where TResponse : TransportResponse, new() =>
 		transport.RequestAsync<TResponse>(new EndpointPath(GET, pathAndQuery), postData: null, null, null, cancellationToken);
 
+
+
 	/// <summary>Perform a HEAD request</summary>
 	public static VoidResponse Head(this ITransport<ITransportConfiguration> transport, string path, RequestParameters parameters)
 		 => transport.Request<VoidResponse>(ToEndpointPath(HEAD, path, parameters, transport.Configuration), postData: null, null, null);
@@ -53,6 +56,16 @@ public static VoidResponse Head(this ITransport transport, string pathAndQuery)
 	public static Task<VoidResponse> HeadAsync(this ITransport transport, string pathAndQuery, CancellationToken cancellationToken = default)
 		=> transport.RequestAsync<VoidResponse>(new EndpointPath(HEAD, pathAndQuery), postData: null, null, null, cancellationToken);
 
+	/// <summary>Perform a HEAD request</summary>
+	public static VoidResponse Head(this ITransport transport, string pathAndQuery, TimeSpan timeout)
+		=> transport.Request<VoidResponse>(new EndpointPath(HEAD, pathAndQuery), postData: null, null, new RequestConfiguration { RequestTimeout = timeout });
+
+	/// <summary>Perform a HEAD request</summary>
+	public static Task<VoidResponse> HeadAsync(this ITransport transport, string pathAndQuery, TimeSpan timeout, CancellationToken cancellationToken = default)
+		=> transport.RequestAsync<VoidResponse>(new EndpointPath(HEAD, pathAndQuery), postData: null, null, new RequestConfiguration { RequestTimeout = timeout }, cancellationToken);
+
+
+
 	/// <summary>Perform a POST request</summary>
 	public static TResponse Post<TResponse>(this ITransport<ITransportConfiguration> transport, string path, PostData data, RequestParameters parameters)
 		where TResponse : TransportResponse, new() =>
@@ -74,6 +87,17 @@ public static Task<TResponse> PostAsync<TResponse>(this ITransport transport, st
 		where TResponse : TransportResponse, new() =>
 		transport.RequestAsync<TResponse>(new EndpointPath(POST, pathAndQuery), data, null, null, cancellationToken);
 
+	/// <summary>Perform a POST request</summary>
+	public static TResponse Post<TResponse>(this ITransport transport, string pathAndQuery, PostData data, TimeSpan timeout)
+		where TResponse : TransportResponse, new() =>
+		transport.Request<TResponse>(new EndpointPath(POST, pathAndQuery), data, null, new RequestConfiguration { RequestTimeout = timeout });
+
+	/// <summary>Perform a POST request</summary>
+	public static Task<TResponse> PostAsync<TResponse>(this ITransport transport, string pathAndQuery, PostData data, TimeSpan timeout, CancellationToken cancellationToken = default)
+		where TResponse : TransportResponse, new() =>
+		transport.RequestAsync<TResponse>(new EndpointPath(POST, pathAndQuery), data, null, new RequestConfiguration { RequestTimeout = timeout }, cancellationToken);
+
+
 	/// <summary>Perform a PUT request</summary>
 	public static TResponse Put<TResponse>(this ITransport<ITransportConfiguration> transport, string path, PostData data, RequestParameters parameters)
 		where TResponse : TransportResponse, new() =>
@@ -94,6 +118,17 @@ public static Task<TResponse> PutAsync<TResponse>(this ITransport transport, str
 		where TResponse : TransportResponse, new() =>
 		transport.RequestAsync<TResponse>(new EndpointPath(PUT, pathAndQuery), data, null, null, cancellationToken);
 
+	/// <summary>Perform a PUT request</summary>
+	public static TResponse Put<TResponse>(this ITransport transport, string pathAndQuery, PostData data, TimeSpan timeout)
+		where TResponse : TransportResponse, new() =>
+		transport.Request<TResponse>(new EndpointPath(PUT, pathAndQuery), data, null, new RequestConfiguration { RequestTimeout = timeout });
+
+	/// <summary>Perform a PUT request</summary>
+	public static Task<TResponse> PutAsync<TResponse>(this ITransport transport, string pathAndQuery, PostData data, TimeSpan timeout, CancellationToken cancellationToken = default)
+		where TResponse : TransportResponse, new() =>
+		transport.RequestAsync<TResponse>(new EndpointPath(PUT, pathAndQuery), data, null, new RequestConfiguration { RequestTimeout = timeout }, cancellationToken);
+
+
 	/// <summary>Perform a DELETE request</summary>
 	public static TResponse Delete<TResponse>(this ITransport<ITransportConfiguration> transport, string path, RequestParameters parameters, PostData? data = null)
 		where TResponse : TransportResponse, new() =>
@@ -105,12 +140,13 @@ public static Task<TResponse> DeleteAsync<TResponse>(this ITransport<ITransportC
 		transport.RequestAsync<TResponse>(ToEndpointPath(DELETE, path, parameters, transport.Configuration), data, null, null, cancellationToken);
 
 	/// <summary>Perform a DELETE request</summary>
-	public static TResponse Delete<TResponse>(this ITransport transport, string pathAndQuery, PostData? data = null)
+	public static TResponse Delete<TResponse>(this ITransport transport, string pathAndQuery, PostData? data = null, TimeSpan? timeout = null)
 		where TResponse : TransportResponse, new() =>
-		transport.Request<TResponse>(new EndpointPath(DELETE, pathAndQuery), data, null, null);
+		transport.Request<TResponse>(new EndpointPath(DELETE, pathAndQuery), data, null, timeout == null ? null : new RequestConfiguration { RequestTimeout = timeout });
 
 	/// <summary>Perform a DELETE request</summary>
-	public static Task<TResponse> DeleteAsync<TResponse>(this ITransport transport, string pathAndQuery, PostData? data = null, CancellationToken cancellationToken = default)
+	public static Task<TResponse> DeleteAsync<TResponse>(this ITransport transport, string pathAndQuery, PostData? data = null, TimeSpan? timeout = null, CancellationToken cancellationToken = default)
 		where TResponse : TransportResponse, new() =>
-		transport.RequestAsync<TResponse>(new EndpointPath(DELETE, pathAndQuery), data, null, null, cancellationToken);
+		transport.RequestAsync<TResponse>(new EndpointPath(DELETE, pathAndQuery), data, null, timeout == null ? null : new RequestConfiguration { RequestTimeout = timeout }, cancellationToken);
+
 }