laoyuyu
2 years ago
parent
1415de11c6
commit
e0c9b0c373
@ -0,0 +1,51 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.aria.http.download |
||||
|
||||
import com.arialyy.aria.core.inf.IBlockManager |
||||
|
||||
/** |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 11:37 2023/3/12 |
||||
**/ |
||||
internal interface ITaskAdapterDelegate { |
||||
/** |
||||
* 任务是否正在执行 |
||||
* |
||||
* @return `true` 任务正在执行 |
||||
*/ |
||||
fun isRunning(): Boolean |
||||
|
||||
/** |
||||
* 取消 |
||||
*/ |
||||
fun cancel() |
||||
|
||||
/** |
||||
* 停止 |
||||
*/ |
||||
fun stop() |
||||
|
||||
/** |
||||
* 开始 |
||||
*/ |
||||
fun start() |
||||
|
||||
fun setBlockManager(blockManager: IBlockManager) |
||||
|
||||
fun getBlockManager(): IBlockManager |
||||
} |
||||
@ -0,0 +1,100 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.aria.http.download |
||||
|
||||
import android.os.Looper |
||||
import com.arialyy.aria.core.DuaContext |
||||
import com.arialyy.aria.core.inf.IBlockManager |
||||
import com.arialyy.aria.core.task.DBlockManager |
||||
import com.arialyy.aria.core.task.TaskResp |
||||
import com.arialyy.aria.core.task.ThreadTaskManager2 |
||||
import com.arialyy.aria.exception.AriaException |
||||
import com.arialyy.aria.http.HttpTaskOption |
||||
import kotlinx.coroutines.Dispatchers |
||||
import kotlinx.coroutines.launch |
||||
import timber.log.Timber |
||||
|
||||
/** |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 11:32 2023/3/12 |
||||
**/ |
||||
internal class SingleTaskDelegate(val adapter: HttpDTaskAdapter) : ITaskAdapterDelegate { |
||||
private val blockManager = DBlockManager(adapter.getTask()) |
||||
|
||||
init { |
||||
ThreadTaskManager2.putThreadManager(adapter.getTask().taskId, blockManager) |
||||
} |
||||
|
||||
override fun isRunning(): Boolean { |
||||
return blockManager.isRunning() |
||||
} |
||||
|
||||
override fun cancel() { |
||||
if (blockManager.isCanceled()) { |
||||
Timber.w("task already canceled, taskId: ${adapter.getTask().taskId}") |
||||
return |
||||
} |
||||
DuaContext.duaScope.launch(Dispatchers.IO) { |
||||
ThreadTaskManager2.stopThreadTask(adapter.getTask().taskId, true) |
||||
} |
||||
} |
||||
|
||||
override fun stop() { |
||||
if (blockManager.isStopped()) { |
||||
Timber.w("task already stopped, taskId: ${adapter.getTask().taskId}") |
||||
return |
||||
} |
||||
DuaContext.duaScope.launch(Dispatchers.IO) { |
||||
ThreadTaskManager2.stopThreadTask(adapter.getTask().taskId) |
||||
} |
||||
} |
||||
|
||||
override fun start() { |
||||
adapter.getTask().getTaskOption(HttpTaskOption::class.java).taskInterceptor.let { |
||||
if (it.isNotEmpty()) { |
||||
adapter.addInterceptors(it) |
||||
} |
||||
} |
||||
DuaContext.duaScope.launch(Dispatchers.IO) { |
||||
Looper.prepare() |
||||
blockManager.setLooper() |
||||
adapter.addCoreInterceptor(HttpDCheckInterceptor()) |
||||
adapter.addCoreInterceptor(TimerInterceptor()) |
||||
adapter.addCoreInterceptor(HttpDHeaderInterceptor()) |
||||
adapter.addCoreInterceptor(HttpDBlockInterceptor()) |
||||
adapter.addCoreInterceptor(HttpBlockThreadInterceptor()) |
||||
val resp = adapter.interceptor() |
||||
if (resp == null || resp.code != TaskResp.CODE_SUCCESS) { |
||||
adapter.getTask().getTaskOption(HttpTaskOption::class.java).eventListener.onFail( |
||||
false, |
||||
AriaException("start task fail, task interrupt, code: ${resp?.code ?: TaskResp.CODE_INTERRUPT}") |
||||
) |
||||
blockManager.stop() |
||||
return@launch |
||||
} |
||||
Looper.loop() |
||||
} |
||||
} |
||||
|
||||
override fun setBlockManager(blockManager: IBlockManager) { |
||||
throw UnsupportedOperationException("Single task does not support setting up a block manager") |
||||
} |
||||
|
||||
override fun getBlockManager(): IBlockManager { |
||||
return blockManager |
||||
} |
||||
} |
||||
@ -0,0 +1,51 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.aria.http.download |
||||
|
||||
import com.arialyy.aria.core.inf.IBlockManager |
||||
|
||||
/** |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 11:35 2023/3/12 |
||||
**/ |
||||
internal class SubTaskDelegate(val adapter: HttpDTaskAdapter) : ITaskAdapterDelegate { |
||||
private lateinit var blockManager: IBlockManager |
||||
|
||||
override fun isRunning(): Boolean { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun cancel() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun stop() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun start() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun setBlockManager(blockManager: IBlockManager) { |
||||
this.blockManager = blockManager |
||||
} |
||||
|
||||
override fun getBlockManager(): IBlockManager { |
||||
return blockManager |
||||
} |
||||
} |
||||
@ -0,0 +1,73 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.dua.group |
||||
|
||||
import com.arialyy.aria.core.common.TaskOption |
||||
import com.arialyy.aria.core.task.ITask |
||||
import com.arialyy.aria.core.task.ITaskInterceptor |
||||
import com.arialyy.aria.core.task.SingleDownloadTask |
||||
import com.arialyy.aria.core.task.TaskChain |
||||
import com.arialyy.aria.core.task.TaskResp |
||||
import com.arialyy.aria.http.HttpTaskOption |
||||
import com.arialyy.aria.http.download.HttpDTaskAdapter |
||||
import com.arialyy.aria.orm.entity.DGEntity |
||||
|
||||
/** |
||||
* Subtasks do not support chunking |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 08:51 2023/3/12 |
||||
**/ |
||||
internal class HttpDGSubTaskInterceptor : ITaskInterceptor { |
||||
private lateinit var task: ITask |
||||
private lateinit var optionAdapter: HttpDGOptionAdapter |
||||
private lateinit var taskOption: HttpTaskOption |
||||
|
||||
override suspend fun interceptor(chain: TaskChain): TaskResp { |
||||
task = chain.getTask() |
||||
taskOption = task.getTaskOption(HttpTaskOption::class.java) |
||||
optionAdapter = taskOption.getOptionAdapter(HttpDGOptionAdapter::class.java) |
||||
val subList = createSubTask(chain) |
||||
|
||||
|
||||
return TaskResp(TaskResp.CODE_SUCCESS) |
||||
} |
||||
|
||||
private fun startSubTask(subList: List<SingleDownloadTask>) { |
||||
subList.forEach { |
||||
|
||||
} |
||||
} |
||||
|
||||
private fun createSubTask(chain: TaskChain): List<SingleDownloadTask> { |
||||
val subTaskList = mutableListOf<SingleDownloadTask>() |
||||
task.taskState.getEntity(DGEntity::class.java).subList.forEach { |
||||
val tp = TaskOption() |
||||
tp.sourUrl = it.sourceUrl |
||||
tp.savePathDir = taskOption.savePathDir |
||||
tp.threadNum = 1 |
||||
tp.eventListener = HttpSubListener() |
||||
val subTask = SingleDownloadTask(tp) |
||||
val subAdapter = HttpDTaskAdapter(true) |
||||
subAdapter.setBlockManager(HttpSubBlockManager(chain.blockManager.handler)) |
||||
|
||||
subTask.adapter = subAdapter |
||||
subTaskList.add(subTask) |
||||
} |
||||
return subTaskList |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,52 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.dua.group |
||||
|
||||
import android.os.Handler |
||||
import com.arialyy.aria.core.inf.IBlockManager |
||||
import com.arialyy.aria.core.task.IThreadTask |
||||
import com.arialyy.aria.orm.entity.BlockRecord |
||||
|
||||
/** |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 10:04 2023/3/12 |
||||
**/ |
||||
class HttpSubBlockManager(val handler: Handler) : IBlockManager { |
||||
private val unfinishedBlock = mutableListOf<BlockRecord>() |
||||
private var blockNum: Int = 1 |
||||
|
||||
override fun putUnfinishedBlock(record: BlockRecord) { |
||||
unfinishedBlock.add(record) |
||||
} |
||||
|
||||
override fun getUnfinishedBlockList(): List<BlockRecord> { |
||||
return unfinishedBlock |
||||
} |
||||
|
||||
override fun getHandler() = handler |
||||
|
||||
override fun start(threadTaskList: List<IThreadTask>) { |
||||
threadTaskList.forEach { tt -> |
||||
tt.run() |
||||
} |
||||
} |
||||
|
||||
override fun setBlockNum(blockNum: Int) { |
||||
this.blockNum = blockNum |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,58 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package com.arialyy.dua.group |
||||
|
||||
import com.arialyy.aria.core.listener.IEventListener |
||||
import com.arialyy.aria.exception.AriaException |
||||
|
||||
/** |
||||
* @Author laoyuyu |
||||
* @Description |
||||
* @Date 10:13 2023/3/12 |
||||
**/ |
||||
internal class HttpSubListener : IEventListener { |
||||
override fun onPre() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onStart(startLocation: Long) { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onResume(resumeLocation: Long) { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onProgress(currentLocation: Long) { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onStop(stopLocation: Long) { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onComplete() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onCancel() { |
||||
TODO("Not yet implemented") |
||||
} |
||||
|
||||
override fun onFail(needRetry: Boolean, e: AriaException?) { |
||||
TODO("Not yet implemented") |
||||
} |
||||
} |
||||
Loading…
Reference in new issue