From ee1a0264eb78ed4b3c1043c2b018201df580f7ca Mon Sep 17 00:00:00 2001
From: Ahmet <ahmet.cotur@hotmail.com>
Date: Wed, 13 Jan 2021 15:22:31 +0300
Subject: [PATCH] Update MongoPageRepository.cs

---
 .../Volo/CmsKit/MongoDB/Pages/MongoPageRepository.cs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/modules/cms-kit/src/Volo.CmsKit.MongoDB/Volo/CmsKit/MongoDB/Pages/MongoPageRepository.cs b/modules/cms-kit/src/Volo.CmsKit.MongoDB/Volo/CmsKit/MongoDB/Pages/MongoPageRepository.cs
index 23cc68709b..c1ed599705 100644
--- a/modules/cms-kit/src/Volo.CmsKit.MongoDB/Volo/CmsKit/MongoDB/Pages/MongoPageRepository.cs
+++ b/modules/cms-kit/src/Volo.CmsKit.MongoDB/Volo/CmsKit/MongoDB/Pages/MongoPageRepository.cs
@@ -20,12 +20,14 @@ namespace Volo.CmsKit.MongoDB.Pages
 
         public virtual async Task<int> GetCountAsync(string filter = null, CancellationToken cancellationToken = default)
         {
-            return await (await GetMongoQueryableAsync())
+            cancellationToken = GetCancellationToken(cancellationToken);
+            
+            return await (await GetMongoQueryableAsync(cancellationToken))
                 .WhereIf<Page, IMongoQueryable<Page>>(
                     !filter.IsNullOrWhiteSpace(),
                     u =>
                         u.Title.Contains(filter)
-                ).CountAsync(GetCancellationToken(cancellationToken));
+                ).CountAsync(cancellationToken);
         }
 
         public virtual async Task<List<Page>> GetListAsync(
@@ -35,7 +37,9 @@ namespace Volo.CmsKit.MongoDB.Pages
             string sorting = null,
             CancellationToken cancellationToken = default)
         {
-            return await (await GetMongoQueryableAsync())
+            cancellationToken = GetCancellationToken(cancellationToken);
+            
+            return await (await GetMongoQueryableAsync(cancellationToken))
                 .WhereIf<Page, IMongoQueryable<Page>>(
                     !filter.IsNullOrWhiteSpace(),
                     u =>
@@ -44,7 +48,7 @@ namespace Volo.CmsKit.MongoDB.Pages
                 .OrderBy(sorting ?? nameof(Page.Title))
                 .As<IMongoQueryable<Page>>()
                 .PageBy<Page, IMongoQueryable<Page>>(skipCount, maxResultCount)
-                .ToListAsync(GetCancellationToken(cancellationToken));
+                .ToListAsync(cancellationToken);
         }
 
         public virtual Task<Page> GetByUrlAsync(string url, CancellationToken cancellationToken = default)