lyy
2 years ago
parent
b378d9e3f2
commit
3cc9b09055
@ -0,0 +1,157 @@ |
|||||||
|
/* |
||||||
|
* 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.core.task |
||||||
|
|
||||||
|
import android.os.Handler.Callback |
||||||
|
import android.os.Looper |
||||||
|
import com.arialyy.aria.core.inf.IThreadStateManager |
||||||
|
import com.arialyy.aria.core.listener.IEventListener |
||||||
|
import com.arialyy.aria.exception.AriaException |
||||||
|
import timber.log.Timber |
||||||
|
import java.util.concurrent.atomic.AtomicInteger |
||||||
|
|
||||||
|
class BlockManager( |
||||||
|
val mListener: IEventListener, |
||||||
|
val looper: Looper, |
||||||
|
private val blockNum: Int, |
||||||
|
val blockParentDir: String |
||||||
|
) : IThreadStateManager { |
||||||
|
private val blockList = mutableListOf<BlockState>() |
||||||
|
private val canceledNum = AtomicInteger(0) // 已经取消的线程的数 |
||||||
|
private val stoppedNum = AtomicInteger(0) // 已经停止的线程数 |
||||||
|
|
||||||
|
private val failedNum = AtomicInteger(0) // 失败的线程数 |
||||||
|
|
||||||
|
private val completedNum = AtomicInteger(0) // 完成的线程数 |
||||||
|
|
||||||
|
private var progress: Long = 0 //当前总进度 |
||||||
|
|
||||||
|
private val callback = Callback { msg -> |
||||||
|
checkLooper() |
||||||
|
when (msg.what) { |
||||||
|
IThreadStateManager.STATE_STOP -> { |
||||||
|
stoppedNum.getAndIncrement() |
||||||
|
if (isStopped) { |
||||||
|
quitLooper() |
||||||
|
} |
||||||
|
} |
||||||
|
IThreadStateManager.STATE_CANCEL -> { |
||||||
|
canceledNum.getAndIncrement() |
||||||
|
if (isCanceled) { |
||||||
|
quitLooper() |
||||||
|
} |
||||||
|
} |
||||||
|
IThreadStateManager.STATE_FAIL -> { |
||||||
|
failedNum.getAndIncrement() |
||||||
|
if (hasFailedBlock()) { |
||||||
|
val b = msg.data |
||||||
|
mListener.onFail( |
||||||
|
b.getBoolean(IThreadStateManager.DATA_RETRY, false), |
||||||
|
b.getSerializable(IThreadStateManager.DATA_ERROR_INFO) as AriaException? |
||||||
|
) |
||||||
|
quitLooper() |
||||||
|
} |
||||||
|
} |
||||||
|
IThreadStateManager.STATE_COMPLETE -> { |
||||||
|
completedNum.getAndIncrement() |
||||||
|
if (isCompleted) { |
||||||
|
Timber.d("isComplete, completeNum = %s", completedNum) |
||||||
|
if (mTaskRecord.isBlock || mTaskRecord.threadNum === 1) { |
||||||
|
if (mergeFile()) { |
||||||
|
mListener.onComplete() |
||||||
|
} else { |
||||||
|
mListener.onFail(false, null) |
||||||
|
} |
||||||
|
quitLooper() |
||||||
|
break |
||||||
|
} |
||||||
|
mListener.onComplete() |
||||||
|
quitLooper() |
||||||
|
} |
||||||
|
} |
||||||
|
IThreadStateManager.STATE_RUNNING -> { |
||||||
|
val b = msg.data |
||||||
|
if (b != null) { |
||||||
|
val len = b.getLong(IThreadStateManager.DATA_ADD_LEN, 0) |
||||||
|
progress += len |
||||||
|
} |
||||||
|
} |
||||||
|
IThreadStateManager.STATE_UPDATE_PROGRESS -> if (msg.obj == null) { |
||||||
|
progress = updateBlockProgress() |
||||||
|
} else if (msg.obj is Long) { |
||||||
|
progress = msg.obj as Long |
||||||
|
} |
||||||
|
} |
||||||
|
false |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 退出looper循环 |
||||||
|
*/ |
||||||
|
private fun quitLooper() { |
||||||
|
looper.quit() |
||||||
|
} |
||||||
|
|
||||||
|
fun addBlockState(state: BlockState) { |
||||||
|
blockList.add(state) |
||||||
|
} |
||||||
|
|
||||||
|
fun getBlockState(): BlockState { |
||||||
|
return blockList.removeFirst() |
||||||
|
} |
||||||
|
|
||||||
|
override fun hasFailedBlock(): Boolean { |
||||||
|
Timber.d("isFailed, blockBum = ${blockNum}, completedNum = ${completedNum.get()}, ") |
||||||
|
return failedNum.get() != 0 |
||||||
|
} |
||||||
|
|
||||||
|
override fun isCompleted(): Boolean { |
||||||
|
// Timber.d("isCompleted, blockBum = ${blockNum}, completedNum = ${completedNum.get()}, ") |
||||||
|
return completedNum.get() == blockNum |
||||||
|
} |
||||||
|
|
||||||
|
override fun getCurrentProgress(): Long { |
||||||
|
return progress |
||||||
|
} |
||||||
|
|
||||||
|
override fun updateCurrentProgress(currentProgress: Long) { |
||||||
|
progress = currentProgress |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 所有子线程是否都已经停止 |
||||||
|
*/ |
||||||
|
override fun isStopped(): Boolean { |
||||||
|
// Timber.d("isStopped, blockBum = ${blockNum}, stoppedNum = ${stoppedNum.get()}, completedNum = ${completedNum.get()}, failedNum = ${failedNum.get()}") |
||||||
|
return stoppedNum.get() + completedNum.get() + failedNum.get() == blockNum |
||||||
|
} |
||||||
|
|
||||||
|
override fun isCanceled(): Boolean { |
||||||
|
// Timber.d("isStopped, blockBum = ${blockNum}, canceledNum = $canceledNum") |
||||||
|
return canceledNum.get() == blockNum |
||||||
|
} |
||||||
|
|
||||||
|
override fun getHandlerCallback(): Callback { |
||||||
|
return callback |
||||||
|
} |
||||||
|
|
||||||
|
private fun mergeFile(): Boolean { |
||||||
|
if (blockNum == 1){ |
||||||
|
|
||||||
|
return true |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,30 @@ |
|||||||
|
/* |
||||||
|
* 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.core.task |
||||||
|
|
||||||
|
class BlockState(val blockId: Int, val blockPath: String) { |
||||||
|
|
||||||
|
companion object { |
||||||
|
/** |
||||||
|
* dir/.fileName.blockId |
||||||
|
*/ |
||||||
|
const val BLOCK_PATH = "%s/.%s.%d" |
||||||
|
|
||||||
|
const val BLOCK_SIZE = 1024 * 1024 * 5 |
||||||
|
} |
||||||
|
|
||||||
|
var curProgress: Long = 0 |
||||||
|
} |
||||||
Loading…
Reference in new issue