|
|
|
@ -29,8 +29,10 @@ class DownloadService : BaseService() { |
|
|
|
|
private val handler = Handler() |
|
|
|
|
private var runnable: Runnable = Runnable { upDownload() } |
|
|
|
|
private val downloadMap = hashMapOf<String, LinkedHashSet<BookChapter>>() |
|
|
|
|
private val downloadCount = hashMapOf<String, DownloadCount>(); |
|
|
|
|
private val finalMap = hashMapOf<String, LinkedHashSet<BookChapter>>() |
|
|
|
|
private var notificationContent = "正在启动下载" |
|
|
|
|
|
|
|
|
|
private val notificationBuilder by lazy { |
|
|
|
|
val builder = NotificationCompat.Builder(this, AppConst.channelIdDownload) |
|
|
|
|
.setSmallIcon(R.drawable.ic_download) |
|
|
|
@ -97,6 +99,11 @@ class DownloadService : BaseService() { |
|
|
|
|
finalMap.remove(bookUrl) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private fun updateNotification(downloadCount:DownloadCount, totalCount: Int, content: String){ |
|
|
|
|
notificationContent = |
|
|
|
|
"进度:${downloadCount.downloadFinishedCount}/$totalCount,成功:${downloadCount.successCount},$content" |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private fun download() { |
|
|
|
|
val task = Coroutine.async(this, context = searchPool) { |
|
|
|
|
downloadMap.forEach { entry -> |
|
|
|
@ -106,6 +113,9 @@ class DownloadService : BaseService() { |
|
|
|
|
val bookSource = |
|
|
|
|
App.db.bookSourceDao().getBookSource(book.origin) ?: return@async |
|
|
|
|
val webBook = WebBook(bookSource) |
|
|
|
|
|
|
|
|
|
downloadCount[entry.key] = DownloadCount() |
|
|
|
|
|
|
|
|
|
entry.value.forEach { chapter -> |
|
|
|
|
if (!isActive) return@async |
|
|
|
|
if (downloadMap.containsKey(book.bookUrl)) { |
|
|
|
@ -116,16 +126,19 @@ class DownloadService : BaseService() { |
|
|
|
|
scope = this, |
|
|
|
|
context = searchPool |
|
|
|
|
) |
|
|
|
|
.onStart { |
|
|
|
|
notificationContent = chapter.title |
|
|
|
|
} |
|
|
|
|
//.onStart { |
|
|
|
|
// notificationContent = "启动:" + chapter.title |
|
|
|
|
//} |
|
|
|
|
.onSuccess(IO) { content -> |
|
|
|
|
content?.let { |
|
|
|
|
downloadCount[entry.key]?.increaseSuccess() |
|
|
|
|
BookHelp.saveContent(book, chapter, content) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
.onFinally(IO) { |
|
|
|
|
synchronized(this@DownloadService) { |
|
|
|
|
downloadCount[entry.key]?.increaseFinished() |
|
|
|
|
downloadCount[entry.key]?.let { updateNotification(it, entry.value.size, chapter.title) } |
|
|
|
|
val chapterMap = |
|
|
|
|
finalMap[book.bookUrl] |
|
|
|
|
?: linkedSetOf<BookChapter>().apply { |
|
|
|
@ -135,9 +148,14 @@ class DownloadService : BaseService() { |
|
|
|
|
if (chapterMap.size == entry.value.size) { |
|
|
|
|
downloadMap.remove(book.bookUrl) |
|
|
|
|
finalMap.remove(book.bookUrl) |
|
|
|
|
downloadCount.remove(entry.key) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} else{ |
|
|
|
|
//无需下载的,设置为增加成功 |
|
|
|
|
downloadCount[entry.key]?.increaseSuccess() |
|
|
|
|
downloadCount[entry.key]?.increaseFinished() |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
@ -176,4 +194,17 @@ class DownloadService : BaseService() { |
|
|
|
|
val notification = builder.build() |
|
|
|
|
startForeground(AppConst.notificationIdDownload, notification) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
class DownloadCount{ |
|
|
|
|
@Volatile public var downloadFinishedCount = 0 // 下载完成的条目数量 |
|
|
|
|
@Volatile public var successCount = 0 //下载成功的条目数量 |
|
|
|
|
|
|
|
|
|
fun increaseSuccess(){ |
|
|
|
|
++successCount; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
fun increaseFinished(){ |
|
|
|
|
++downloadFinishedCount; |
|
|
|
|
} |
|
|
|
|
} |