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