Add function to get number of manga chapters

Suwayomi · Jul 16, 2023 · 80f431b · 80f431b
1 parent 7447d53
commit 80f431b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/server/src/main/kotlin/suwayomi/tachidesk/graphql/mutations/ChapterMutation.kt b/server/src/main/kotlin/suwayomi/tachidesk/graphql/mutations/ChapterMutation.kt
@@ -122,7 +122,7 @@ class ChapterMutation {
         val (clientMutationId, mangaId) = input
 
         return future {
-            val numberOfCurrentChapters = transaction { ChapterTable.select { ChapterTable.manga eq mangaId }.count().toInt() }
+            val numberOfCurrentChapters = Chapter.getCountOfMangaChapters(mangaId)
             Chapter.fetchChapterList(mangaId)
             numberOfCurrentChapters
         }.thenApply { numberOfCurrentChapters ->

diff --git a/server/src/main/kotlin/suwayomi/tachidesk/manga/impl/Chapter.kt b/server/src/main/kotlin/suwayomi/tachidesk/manga/impl/Chapter.kt
@@ -62,6 +62,10 @@ object Chapter {
         }
     }
 
+    fun getCountOfMangaChapters(mangaId: Int): Int {
+        return transaction { ChapterTable.select { ChapterTable.manga eq mangaId }.count().toInt() }
+    }
+
     private suspend fun getSourceChapters(mangaId: Int): List<ChapterDataClass> {
         val chapterList = fetchChapterList(mangaId)
 
@@ -113,8 +117,7 @@ object Chapter {
             url = manga.url
         }
 
-        val numberOfCurrentChapters =
-            transaction { ChapterTable.select { ChapterTable.manga eq mangaId }.count().toInt() }
+        val numberOfCurrentChapters = getCountOfMangaChapters(mangaId)
         val chapterList = source.fetchChapterList(sManga).awaitSingle()
 
         // Recognize number for new chapters.