feat: 优化代码

5 years ago · 8fce75a788
parent 6b31896cf5
commit 8fce75a788
1 changed files with 5 additions and 4 deletions
--- a/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookActivity.kt
@ -95,10 +95,11 @@ class ArrangeBookActivity : VMBaseActivity<ArrangeBookViewModel>(R.layout.activi
    private fun initBookData() {
        booksLiveData?.removeObservers(this)
        booksLiveData =
-            if (groupId == -1) {
-                App.db.bookDao().observeAll()
-            } else {
-                App.db.bookDao().observeByGroup(groupId)
+            when (groupId) {
+                -1 -> App.db.bookDao().observeAll()
+                -2 -> App.db.bookDao().observeLocal()
+                -3 -> App.db.bookDao().observeAudio()
+                else -> App.db.bookDao().observeByGroup(groupId)
            }
        booksLiveData?.observe(this, Observer {
            adapter.selectedBooks.clear()