up

app/src/main/java/io/legado/app/ui/readbook/ReadBookViewModel.kt

@@ -136,48 +136,32 @@ class ReadBookViewModel(application: Application) : BaseViewModel(application) {
     }
 
     fun loadContent(book: Book, index: Int) {
-        synchronized(loadingLock) {
-            if (loadingChapters.contains(index)) return
-            loadingChapters.add(index)
-        }
+        if (addLoading(index)) {
             execute {
                 App.db.bookChapterDao().getChapter(book.bookUrl, index)?.let { chapter ->
                     BookHelp.getContent(book, chapter)?.let {
                         contentLoadFinish(chapter, it)
-                    synchronized(loadingLock) {
-                        loadingChapters.remove(index)
-                    }
+                        removeLoading(chapter.index)
                     } ?: download(book, chapter)
-            } ?: synchronized(loadingLock) {
-                loadingChapters.remove(index)
-            }
+                } ?: removeLoading(index)
             }.onError {
-            synchronized(loadingLock) {
-                loadingChapters.remove(index)
+                removeLoading(index)
             }
         }
     }
 
     private fun download(book: Book, index: Int) {
-        synchronized(loadingLock) {
-            if (loadingChapters.contains(index)) return
-            loadingChapters.add(index)
-        }
+        if (addLoading(index)) {
             execute {
                 App.db.bookChapterDao().getChapter(book.bookUrl, index)?.let { chapter ->
-                if (!BookHelp.hasContent(book, chapter)) {
-                    download(book, chapter)
+                    if (BookHelp.hasContent(book, chapter)) {
+                        removeLoading(chapter.index)
                     } else {
-                    synchronized(loadingLock) {
-                        loadingChapters.remove(index)
-                    }
-                }
-            } ?: synchronized(loadingLock) {
-                loadingChapters.remove(index)
+                        download(book, chapter)
                     }
+                } ?: removeLoading(index)
             }.onError {
-            synchronized(loadingLock) {
-                loadingChapters.remove(index)
+                removeLoading(index)
             }
         }
     }
@@ -187,22 +171,30 @@ class ReadBookViewModel(application: Application) : BaseViewModel(application) {
             ?.onSuccess(IO) { content ->
                 if (content.isNullOrEmpty()) {
                     contentLoadFinish(chapter, context.getString(R.string.content_empty))
-                    synchronized(loadingLock) {
-                        loadingChapters.remove(chapter.index)
-                    }
+                    removeLoading(chapter.index)
                 } else {
                     BookHelp.saveContent(book, chapter, content)
                     contentLoadFinish(chapter, content)
-                    synchronized(loadingLock) {
-                        loadingChapters.remove(chapter.index)
-                    }
+                    removeLoading(chapter.index)
                 }
             }?.onError {
                 contentLoadFinish(chapter, it.localizedMessage)
+                removeLoading(chapter.index)
+            }
+    }
+
+    private fun addLoading(index: Int): Boolean {
         synchronized(loadingLock) {
-                    loadingChapters.remove(chapter.index)
+            if (loadingChapters.contains(index)) return false
+            loadingChapters.add(index)
+            return true
         }
     }
+
+    private fun removeLoading(index: Int) {
+        synchronized(loadingLock) {
+            loadingChapters.remove(index)
+        }
     }
 
     private fun contentLoadFinish(chapter: BookChapter, content: String) {