Add CancellationToken to Repository methods

Author: A-Programmer

src/KSFramework/GenericRepository/GenericRepository.cs

@@ -23,20 +23,24 @@ public GenericRepository(DbContext context) : base(context)
     /// Gets an entity by its primary key.
     /// </summary>
     /// <param name="id">The entity ID.</param>
+    /// <param name="cancellationToken"></param>
     /// <returns>The entity if found; otherwise, null.</returns>
-    public async ValueTask<TEntity?> GetByIdAsync(object id)
+    public async ValueTask<TEntity?> GetByIdAsync(object id,
+        CancellationToken cancellationToken = default)
     {
-        return await DbSet.FindAsync(id);
+        return await DbSet.FindAsync(id, cancellationToken);
     }
 
     /// <summary>
     /// Gets all entities.
     /// </summary>
     /// <param name="asNoTracking">Whether to disable tracking for better performance.</param>
+    /// <param name="cancellationToken"></param>
     /// <returns>A list of all entities.</returns>
-    public async Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true)
+    public async Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true,
+        CancellationToken cancellationToken = default)
     {
-        return await AsQueryable(asNoTracking).ToListAsync();
+        return await AsQueryable(asNoTracking).ToListAsync(cancellationToken);
     }
 
     /// <summary>
@@ -48,7 +52,11 @@ public async Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true)
     /// <param name="orderBy">Property name to order by.</param>
     /// <param name="desc">Order descending if true.</param>
     /// <returns>A paginated list of entities.</returns>
-    public async Task<PaginatedList<TEntity>> GetPagedAsync(int pageIndex, int pageSize, Expression<Func<TEntity, bool>>? where = null, string? orderBy = "", bool desc = false)
+    public async Task<PaginatedList<TEntity>> GetPagedAsync(int pageIndex,
+        int pageSize,
+        Expression<Func<TEntity, bool>>? where = null,
+        string? orderBy = "",
+        bool desc = false)
     {
         var query = ApplyWhere(AsQueryable(), where);
         return await PaginatedList<TEntity>.CreateAsync(query, pageIndex, pageSize, where, orderBy, desc);
@@ -84,28 +92,34 @@ public IEnumerable<TEntity> Find(Expression<Func<TEntity, bool>> predicate, bool
     /// Gets a single entity that matches the given predicate or null.
     /// </summary>
     /// <param name="predicate">The filter expression.</param>
+    /// <param name="cancellationToken"></param>
     /// <returns>The matching entity or null.</returns>
-    public async Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate)
+    public async Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate,
+        CancellationToken cancellationToken = default)
     {
-        return await DbSet.SingleOrDefaultAsync(predicate);
+        return await DbSet.SingleOrDefaultAsync(predicate, cancellationToken);
     }
 
     /// <summary>
     /// Adds a new entity asynchronously.
     /// </summary>
     /// <param name="entity">The entity to add.</param>
-    public async Task AddAsync(TEntity entity)
+    /// <param name="cancellationToken"></param>
+    public async Task AddAsync(TEntity entity,
+        CancellationToken cancellationToken = default)
     {
-        await DbSet.AddAsync(entity);
+        await DbSet.AddAsync(entity, cancellationToken);
     }
 
     /// <summary>
     /// Adds a range of entities asynchronously.
     /// </summary>
     /// <param name="entities">The entities to add.</param>
-    public async Task AddRangeAsync(IEnumerable<TEntity> entities)
+    /// <param name="cancellationToken"></param>
+    public async Task AddRangeAsync(IEnumerable<TEntity> entities,
+        CancellationToken cancellationToken = default)
     {
-        await DbSet.AddRangeAsync(entities);
+        await DbSet.AddRangeAsync(entities, cancellationToken);
     }
 
     /// <summary>
@@ -139,9 +153,11 @@ public void RemoveRange(IEnumerable<TEntity> entities)
     /// Determines whether any entity exists that matches the specified predicate.
     /// </summary>
     /// <param name="predicate">The condition to check.</param>
+    /// <param name="cancellationToken"></param>
     /// <returns>True if at least one entity exists; otherwise, false.</returns>
-    public async Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate)
+    public async Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate,
+        CancellationToken cancellationToken = default)
     {
-        return await DbSet.AnyAsync(predicate);
+        return await DbSet.AnyAsync(predicate, cancellationToken);
     }
 }

src/KSFramework/GenericRepository/IGenericRepository.cs

@@ -14,14 +14,14 @@ public interface IGenericRepository<TEntity> where TEntity : class
     /// </summary>
     /// <param name="id">The unique identifier of the entity.</param>
     /// <returns>A task representing the asynchronous operation, containing the entity if found; otherwise, null.</returns>
-    ValueTask<TEntity?> GetByIdAsync(object id);
+    ValueTask<TEntity?> GetByIdAsync(object id, CancellationToken cancellationToken = default);
 
     /// <summary>
     /// Asynchronously retrieves all entities.
     /// </summary>
     /// <param name="asNoTracking">Whether to track entities in change tracker.</param>
     /// <returns>A task containing all entities.</returns>
-    Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true);
+    Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true, CancellationToken cancellationToken = default);
 
     /// <summary>
     /// Asynchronously retrieves a paginated list of entities with optional filtering and ordering.
@@ -68,19 +68,21 @@ PaginatedList<TEntity> GetPaged(
     /// </summary>
     /// <param name="predicate">The condition to match.</param>
     /// <returns>A task containing a single entity or null.</returns>
-    Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate);
+    Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate, CancellationToken cancellationToken = default);
 
     /// <summary>
     /// Asynchronously adds an entity to the repository.
     /// </summary>
     /// <param name="entity">The entity to add.</param>
-    Task AddAsync(TEntity entity);
+    /// <param name="cancellationToken"></param>
+    Task AddAsync(TEntity entity, CancellationToken cancellationToken = default);
 
     /// <summary>
     /// Asynchronously adds multiple entities to the repository.
     /// </summary>
     /// <param name="entities">The entities to add.</param>
-    Task AddRangeAsync(IEnumerable<TEntity> entities);
+    /// <param name="cancellationToken"></param>
+    Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default);
 
     /// <summary>
     /// Updates an existing entity in the repository.
@@ -104,6 +106,8 @@ PaginatedList<TEntity> GetPaged(
     /// Asynchronously checks whether any entity matches the given predicate.
     /// </summary>
     /// <param name="predicate">The condition to match.</param>
+    /// <param name="cancellationToken"></param>
     /// <returns>True if any entity matches; otherwise, false.</returns>
-    Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate);
+    Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate,
+        CancellationToken cancellationToken = default);
 }

src/KSFramework/GenericRepository/IRepository.cs

@@ -9,10 +9,13 @@ namespace KSFramework.GenericRepository;
 /// <typeparam name="TEntity">The entity type.</typeparam>
 public interface IRepository<TEntity> where TEntity : class
 {
-    Task AddAsync(TEntity entity);
-    Task AddRangeAsync(IEnumerable<TEntity> entities);
+    Task AddAsync(TEntity entity,
+        CancellationToken cancellationToken = default);
+    Task AddRangeAsync(IEnumerable<TEntity> entities,
+        CancellationToken cancellationToken = default);
     IEnumerable<TEntity> Find(Expression<Func<TEntity, bool>> predicate, bool asNoTracking = true);
-    Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true);
+    Task<IEnumerable<TEntity>> GetAllAsync(bool asNoTracking = true,
+        CancellationToken cancellationToken = default);
     Task<PaginatedList<TEntity>> GetPagedAsync(
         int pageIndex,
         int pageSize,
@@ -25,10 +28,13 @@ PaginatedList<TEntity> GetPaged(
         Expression<Func<TEntity, bool>>? where = null,
         string? orderBy = "",
         bool desc = false);
-    ValueTask<TEntity?> GetByIdAsync(object id);
+    ValueTask<TEntity?> GetByIdAsync(object id,
+        CancellationToken cancellationToken = default);
     void Remove(TEntity entity);
     void RemoveRange(IEnumerable<TEntity> entities);
-    Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate);
-    Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate);
+    Task<TEntity?> SingleOrDefaultAsync(Expression<Func<TEntity, bool>> predicate,
+        CancellationToken cancellationToken = default);
+    Task<bool> IsExistValueForPropertyAsync(Expression<Func<TEntity, bool>> predicate,
+        CancellationToken cancellationToken = default);
     void Update(TEntity entity);
 }