laoyuyu
5 years ago
parent
8158c58d6e
commit
b1a2f232fa
@ -0,0 +1,112 @@ |
||||
/* |
||||
* 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.ftp.upload; |
||||
|
||||
import aria.apache.commons.net.ftp.FTPFile; |
||||
import com.arialyy.aria.core.TaskRecord; |
||||
import com.arialyy.aria.core.ThreadRecord; |
||||
import com.arialyy.aria.core.common.RecordHandler; |
||||
import com.arialyy.aria.core.upload.UTaskWrapper; |
||||
import com.arialyy.aria.util.ALog; |
||||
import com.arialyy.aria.util.RecordUtil; |
||||
import java.util.ArrayList; |
||||
|
||||
/** |
||||
* 上传任务记录处理器 |
||||
*/ |
||||
final class FtpURecordHandler extends RecordHandler { |
||||
private FTPFile ftpFile; |
||||
|
||||
FtpURecordHandler(UTaskWrapper wrapper) { |
||||
super(wrapper); |
||||
} |
||||
|
||||
void setFtpFile(FTPFile ftpFile) { |
||||
this.ftpFile = ftpFile; |
||||
} |
||||
|
||||
@Override public void handlerTaskRecord(TaskRecord record) { |
||||
if (record.threadRecords == null || record.threadRecords.isEmpty()) { |
||||
record.threadRecords = new ArrayList<>(); |
||||
record.threadRecords.add( |
||||
createThreadRecord(record, 0, ftpFile == null ? 0 : ftpFile.getSize(), getFileSize())); |
||||
} |
||||
|
||||
if (ftpFile != null) { |
||||
//远程文件已完成
|
||||
if (ftpFile.getSize() == getFileSize()) { |
||||
record.threadRecords.get(0).isComplete = true; |
||||
ALog.d(TAG, "FTP服务器上已存在该文件【" + ftpFile.getName() + "】"); |
||||
} else if (ftpFile.getSize() == 0) { |
||||
getWrapper().setNewTask(true); |
||||
ALog.d(TAG, "FTP服务器上已存在该文件【" + ftpFile.getName() + "】，但文件长度为0，重新上传该文件"); |
||||
} else { |
||||
ALog.w(TAG, "FTP服务器已存在未完成的文件【" |
||||
+ ftpFile.getName() |
||||
+ "，size: " |
||||
+ ftpFile.getSize() |
||||
+ "】" |
||||
+ "尝试从位置：" |
||||
+ (ftpFile.getSize() - 1) |
||||
+ "开始上传"); |
||||
getWrapper().setNewTask(false); |
||||
|
||||
// 修改记录
|
||||
ThreadRecord threadRecord = record.threadRecords.get(0); |
||||
//修改本地保存的停止地址为服务器上对应文件的大小
|
||||
threadRecord.startLocation = ftpFile.getSize() - 1; |
||||
} |
||||
} else { |
||||
ALog.d(TAG, "FTP服务器上不存在该文件"); |
||||
getWrapper().setNewTask(true); |
||||
ThreadRecord tr = record.threadRecords.get(0); |
||||
tr.startLocation = 0; |
||||
tr.endLocation = getFileSize(); |
||||
tr.isComplete = false; |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public ThreadRecord createThreadRecord(TaskRecord record, int threadId, long startL, long endL) { |
||||
ThreadRecord tr; |
||||
tr = new ThreadRecord(); |
||||
tr.taskKey = record.filePath; |
||||
tr.threadId = threadId; |
||||
tr.startLocation = startL; |
||||
tr.isComplete = false; |
||||
tr.threadType = getWrapper().getEntity().getTaskType(); |
||||
tr.endLocation = getFileSize(); |
||||
tr.blockLen = RecordUtil.getBlockLen(getFileSize(), threadId, record.threadNum); |
||||
return tr; |
||||
} |
||||
|
||||
@Override public TaskRecord createTaskRecord(int threadNum) { |
||||
TaskRecord record = new TaskRecord(); |
||||
record.fileName = getEntity().getFileName(); |
||||
record.filePath = getEntity().getFilePath(); |
||||
record.threadRecords = new ArrayList<>(); |
||||
record.threadNum = threadNum; |
||||
record.isBlock = false; |
||||
record.taskType = getWrapper().getEntity().getTaskType(); |
||||
record.isGroupRecord = getEntity().isGroupChild(); |
||||
|
||||
return record; |
||||
} |
||||
|
||||
@Override public int initTaskThreadNum() { |
||||
return 1; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue