lywsvip/MixFile
1
0
Fork 0
mirror of https://github.com/InvertGeek/MixFile.git synced 2026-05-22 08:37:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update DownloadRoute.kt

Browse Source
This commit is contained in:
Evan You
2025-04-27 14:06:09 +08:00
parent 6878ba0201
commit db1c6415e6
1 changed files with 45 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
app/src/main/java/com/donut/mixfile/server/core/routes/api/DownloadRoute.kt
View File

@@ -17,6 +17,7 @@ import io.ktor.server.response.header
import io.ktor.server.response.respondBytesWriter
import io.ktor.server.response.respondText
import io.ktor.server.routing.RoutingHandler
import io.ktor.utils.io.ByteWriteChannel
import io.ktor.utils.io.close
import io.ktor.utils.io.writeFully
import kotlinx.coroutines.Deferred
@@ -79,67 +80,64 @@ suspend fun MixFileServer.respondMixFile(call: ApplicationCall, shareInfo: MixSh
}
contentLength = mixFile.fileSize - range.first
}
responseDownloadFileStream(
call = call,
fileList = fileList,
contentLength = contentLength,
shareInfo = shareInfo,
mixFile = mixFile,
referer = referer,
name = name
)
call.respondBytesWriter(
contentType = name.parseFileMimeType(),
contentLength = contentLength
) {
writeMixFileToByteChannel(
shareInfo = shareInfo,
mixFile = mixFile,
fileList = fileList,
referer = referer,
channel = this
)
}
}
private suspend fun MixFileServer.responseDownloadFileStream(
call: ApplicationCall,
fileList: List<Pair<String, Int>>,
contentLength: Long,
suspend fun MixFileServer.writeMixFileToByteChannel(
shareInfo: MixShareInfo,
mixFile: MixFile,
fileList: List<Pair<String, Int>> = mixFile.fileList.map { it to 0 },
referer: String = shareInfo.referer,
name: String = shareInfo.fileName
channel: ByteWriteChannel,
) {
coroutineScope {
val chunkSize = mixFile.chunkSize
val chunkSizeMB = chunkSize / 1.mb
val taskCount = downloadTaskCount / chunkSizeMB.coerceAtLeast(1)
val fileListToWrite = fileList.toMutableList()
call.respondBytesWriter(
contentType = name.parseFileMimeType(),
contentLength = contentLength
) {
val sortedTask = SortedTask(taskCount.coerceAtLeast(1))
val tasks = mutableListOf<Deferred<Unit>>()
while (!isClosedForWrite && fileListToWrite.isNotEmpty()) {
val currentFile = fileListToWrite.removeAt(0)
val taskOrder = -fileListToWrite.size
sortedTask.prepareTask(taskOrder)
tasks.add(async {
val (url, range) = currentFile
val dataBytes = try {
shareInfo.fetchFile(url, httpClient, referer)
val sortedTask = SortedTask(taskCount.coerceAtLeast(1))
val tasks = mutableListOf<Deferred<Unit>>()
while (!channel.isClosedForWrite && fileListToWrite.isNotEmpty()) {
val currentFile = fileListToWrite.removeAt(0)
val taskOrder = -fileListToWrite.size
sortedTask.prepareTask(taskOrder)
tasks.add(async {
val (url, range) = currentFile
val dataBytes = try {
shareInfo.fetchFile(url, httpClient, referer)
} catch (e: Exception) {
channel.close(e)
throw e
}
sortedTask.addTask(taskOrder) {
val dataToWrite = when {
range == 0 -> dataBytes
range < 0 -> dataBytes.copyOfRange(0, -range) //一般无 < 0 的情况
else -> dataBytes.copyOfRange(range, dataBytes.size)
}
try {
channel.writeFully(dataToWrite)
onDownloadData(dataToWrite)
} catch (e: Exception) {
close(e)
channel.close(e)
throw e
}
sortedTask.addTask(taskOrder) {
val dataToWrite = when {
range == 0 -> dataBytes
range < 0 -> dataBytes.copyOfRange(0, -range) //一般无 < 0 的情况
else -> dataBytes.copyOfRange(range, dataBytes.size)
}
try {
writeFully(dataToWrite)
onDownloadData(dataToWrite)
} catch (e: Exception) {
close(e)
throw e
}
}
sortedTask.execute()
})
}
tasks.awaitAll()
}
sortedTask.execute()
})
}
tasks.awaitAll()
}
}