fix: adds cancellation token parameter

Signed-off-by: Vincent Biret <[email protected]>
microsoft · Aug 27, 2024 · 275120e · 275120e
1 parent 1b2094d
commit 275120e
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/src/abstractions/NativeResponseWrapper.cs b/src/abstractions/NativeResponseWrapper.cs
@@ -5,6 +5,7 @@
 using System;
 using System.Collections.Generic;
 using System.ComponentModel;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace Microsoft.Kiota.Abstractions
@@ -58,15 +59,17 @@ public class NativeResponseWrapper
         /// <param name="q">The query parameters of the request</param>
         /// <param name="h">The request headers of the request</param>
         /// <param name="o">Request options</param>
+        /// <param name="cancellationToken">The cancellation token</param>
         /// <returns></returns>
         public static async Task<NativeResponseType?> CallAndGetNativeTypeAsync<ModelType, NativeResponseType, QueryParametersType>(
-                Func<Action<QueryParametersType>?, Action<IDictionary<string, string>>?, IEnumerable<IRequestOption>?, IResponseHandler, Task<ModelType>> originalCall,
+                Func<Action<QueryParametersType>?, Action<IDictionary<string, string>>?, IEnumerable<IRequestOption>?, IResponseHandler, CancellationToken, Task<ModelType>> originalCall,
                 Action<QueryParametersType>? q = default,
                 Action<IDictionary<string, string>>? h = default,
-                IEnumerable<IRequestOption>? o = default) where NativeResponseType : class
+                IEnumerable<IRequestOption>? o = default,
+                CancellationToken cancellationToken = default) where NativeResponseType : class
         {
             var responseHandler = new NativeResponseHandler();
-            await originalCall.Invoke(q, h, o, responseHandler);
+            await originalCall.Invoke(q, h, o, responseHandler, cancellationToken);
             return responseHandler.Value as NativeResponseType;
         }
 
@@ -79,15 +82,17 @@ public class NativeResponseWrapper
         /// <param name="q">The query parameters of the request</param>
         /// <param name="h">The request headers of the request</param>
         /// <param name="o">Request options</param>
+        /// <param name="cancellationToken">The cancellation token</param>
         public static async Task<NativeResponseType?> CallAndGetNativeTypeAsync<ModelType, NativeResponseType, QueryParametersType, RequestBodyType>(
-                Func<RequestBodyType, Action<QueryParametersType>?, Action<IDictionary<string, string>>?, IEnumerable<IRequestOption>?, IResponseHandler, Task<ModelType>> originalCall,
+                Func<RequestBodyType, Action<QueryParametersType>?, Action<IDictionary<string, string>>?, IEnumerable<IRequestOption>?, IResponseHandler, CancellationToken, Task<ModelType>> originalCall,
                 RequestBodyType requestBody,
                 Action<QueryParametersType>? q = default,
                 Action<IDictionary<string, string>>? h = default,
-                IEnumerable<IRequestOption>? o = default) where NativeResponseType : class
+                IEnumerable<IRequestOption>? o = default,
+                CancellationToken cancellationToken = default) where NativeResponseType : class
         {
             var responseHandler = new NativeResponseHandler();
-            await originalCall.Invoke(requestBody, q, h, o, responseHandler);
+            await originalCall.Invoke(requestBody, q, h, o, responseHandler, cancellationToken);
             return responseHandler.Value as NativeResponseType;
         }
     }