diff --git a/RepoDb.Core/RepoDb/Extensions/DbConnectionExtension.cs b/RepoDb.Core/RepoDb/Extensions/DbConnectionExtension.cs
index 4516d5028..711ba6b01 100644
--- a/RepoDb.Core/RepoDb/Extensions/DbConnectionExtension.cs
+++ b/RepoDb.Core/RepoDb/Extensions/DbConnectionExtension.cs
@@ -82,8 +82,6 @@ public static IDbConnection EnsureOpen(this IDbConnection connection)
return connection;
}
- // TODO: Ensure that the cancellation token is passed from the caller
-
///
/// Ensures the connection object is open in an asynchronous way.
///
@@ -319,6 +317,7 @@ internal static async Task> ExecuteQueryAsyncInternal(this
commandType: commandType,
commandTimeout: commandTimeout,
transaction: transaction,
+ cancellationToken: cancellationToken,
entityType: null,
dbFields: dbFields,
skipCommandArrayParametersCheck: skipCommandArrayParametersCheck))
@@ -836,6 +835,7 @@ private static async Task> ExecuteQueryAsyncInternalForType
commandType: commandType,
commandTimeout: commandTimeout,
transaction: transaction,
+ cancellationToken: cancellationToken,
entityType: typeof(TResult),
dbFields: dbFields,
skipCommandArrayParametersCheck: skipCommandArrayParametersCheck))
@@ -1096,6 +1096,7 @@ internal static async Task ExecuteReaderAsyncInternal(this IDbConne
commandType: commandType,
commandTimeout: commandTimeout,
transaction: transaction,
+ cancellationToken: cancellationToken,
entityType: entityType,
dbFields: dbFields,
skipCommandArrayParametersCheck: skipCommandArrayParametersCheck);
@@ -1263,6 +1264,7 @@ internal static async Task ExecuteNonQueryAsyncInternal(this IDbConnection
commandType: commandType,
commandTimeout: commandTimeout,
transaction: transaction,
+ cancellationToken: cancellationToken,
entityType: entityType,
dbFields: dbFields,
skipCommandArrayParametersCheck: skipCommandArrayParametersCheck))
@@ -1414,6 +1416,7 @@ internal static async Task