AriaLyy
8 years ago
parent
9bc87d75df
commit
9241c17513
@ -0,0 +1,175 @@ |
|||||||
|
/* |
||||||
|
* 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.inf; |
||||||
|
|
||||||
|
import android.support.annotation.NonNull; |
||||||
|
import com.arialyy.aria.core.AriaManager; |
||||||
|
import com.arialyy.aria.core.RequestEnum; |
||||||
|
import com.arialyy.aria.core.command.AbsCmd; |
||||||
|
import com.arialyy.aria.core.command.CmdFactory; |
||||||
|
import com.arialyy.aria.core.download.DownloadEntity; |
||||||
|
import com.arialyy.aria.core.upload.UploadEntity; |
||||||
|
import com.arialyy.aria.orm.DbEntity; |
||||||
|
import com.arialyy.aria.util.CommonUtil; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.List; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Set; |
||||||
|
|
||||||
|
/** |
||||||
|
* Created by Aria.Lao on 2017/2/28. |
||||||
|
*/ |
||||||
|
public class AbsTarget<ENTITY extends IEntity, TASK_ENTITY extends ITaskEntity> { |
||||||
|
protected ENTITY entity; |
||||||
|
protected TASK_ENTITY taskEntity; |
||||||
|
protected String targetName; |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取任务文件大小 |
||||||
|
* |
||||||
|
* @return -1，没有找到该任务 |
||||||
|
*/ |
||||||
|
public long getFileSize() { |
||||||
|
if (entity instanceof DownloadEntity) { |
||||||
|
DownloadEntity entity = DbEntity.findData(DownloadEntity.class, "downloadUrl=?", |
||||||
|
((DownloadEntity) this.entity).getDownloadUrl()); |
||||||
|
if (entity == null) { |
||||||
|
throw new NullPointerException("没有找到该任务"); |
||||||
|
} |
||||||
|
return entity.getFileSize(); |
||||||
|
} else if (entity instanceof UploadEntity) { |
||||||
|
UploadEntity entity = DbEntity.findData(UploadEntity.class, "filePath=?", |
||||||
|
((UploadEntity) this.entity).getFilePath()); |
||||||
|
if (entity == null) { |
||||||
|
throw new NullPointerException("没有找到该任务"); |
||||||
|
} |
||||||
|
return entity.getFileSize(); |
||||||
|
} |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取当前任务进度，如果任务存在，则返回当前进度 |
||||||
|
* |
||||||
|
* @return -1，没有找到该任务 |
||||||
|
*/ |
||||||
|
public long getCurrentProgress() { |
||||||
|
if (entity instanceof DownloadEntity) { |
||||||
|
DownloadEntity entity = DownloadEntity.findData(DownloadEntity.class, "downloadUrl=?", |
||||||
|
((DownloadEntity) this.entity).getDownloadUrl()); |
||||||
|
if (entity == null) { |
||||||
|
throw new NullPointerException("下载管理器中没有该任务"); |
||||||
|
} |
||||||
|
return entity.getCurrentProgress(); |
||||||
|
} else if (entity instanceof UploadEntity) { |
||||||
|
UploadEntity entity = DbEntity.findData(UploadEntity.class, "filePath=?", |
||||||
|
((UploadEntity) this.entity).getFilePath()); |
||||||
|
if (entity == null) { |
||||||
|
throw new NullPointerException("没有找到该任务"); |
||||||
|
} |
||||||
|
return entity.getCurrentProgress(); |
||||||
|
} |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 给url请求添加头部 |
||||||
|
* |
||||||
|
* @param key 头部key |
||||||
|
* @param header 头部value |
||||||
|
*/ |
||||||
|
protected void _addHeader(@NonNull String key, @NonNull String header) { |
||||||
|
taskEntity.headers.put(key, header); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 给url请求添加头部 |
||||||
|
* |
||||||
|
* @param headers Map<Key, Value> |
||||||
|
*/ |
||||||
|
protected void _addHeaders(Map<String, String> headers) { |
||||||
|
if (headers != null && headers.size() > 0) { |
||||||
|
Set<String> keys = headers.keySet(); |
||||||
|
for (String key : keys) { |
||||||
|
taskEntity.headers.put(key, headers.get(key)); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置请求类型 |
||||||
|
* |
||||||
|
* @param requestEnum {@link RequestEnum} |
||||||
|
*/ |
||||||
|
protected void _setRequestMode(RequestEnum requestEnum) { |
||||||
|
taskEntity.requestEnum = requestEnum; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 添加任务 |
||||||
|
*/ |
||||||
|
public void add() { |
||||||
|
AriaManager.getInstance(AriaManager.APP) |
||||||
|
.setCmd(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_CREATE)) |
||||||
|
.exe(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 开始下载 |
||||||
|
*/ |
||||||
|
public void start() { |
||||||
|
List<AbsCmd> cmds = new ArrayList<>(); |
||||||
|
cmds.add(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_CREATE)); |
||||||
|
cmds.add(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_START)); |
||||||
|
AriaManager.getInstance(AriaManager.APP).setCmds(cmds).exe(); |
||||||
|
cmds.clear(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 停止下载 |
||||||
|
*/ |
||||||
|
public void stop() { |
||||||
|
AriaManager.getInstance(AriaManager.APP) |
||||||
|
.setCmd(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_STOP)) |
||||||
|
.exe(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 恢复下载 |
||||||
|
*/ |
||||||
|
public void resume() { |
||||||
|
AriaManager.getInstance(AriaManager.APP) |
||||||
|
.setCmd(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_START)) |
||||||
|
.exe(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 取消下载 |
||||||
|
*/ |
||||||
|
public void cancel() { |
||||||
|
AriaManager.getInstance(AriaManager.APP) |
||||||
|
.setCmd(CommonUtil.createCmd(targetName, taskEntity, CmdFactory.TASK_CANCEL)) |
||||||
|
.exe(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 重新下载 |
||||||
|
*/ |
||||||
|
public void reStart() { |
||||||
|
cancel(); |
||||||
|
start(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,115 @@ |
|||||||
|
/* |
||||||
|
* 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.upload; |
||||||
|
|
||||||
|
import android.support.annotation.NonNull; |
||||||
|
import com.arialyy.aria.core.RequestEnum; |
||||||
|
import com.arialyy.aria.core.inf.AbsTarget; |
||||||
|
import com.arialyy.aria.core.queue.UploadTaskQueue; |
||||||
|
import com.arialyy.aria.orm.DbEntity; |
||||||
|
import java.util.Map; |
||||||
|
|
||||||
|
/** |
||||||
|
* Created by Aria.Lao on 2017/2/28. |
||||||
|
*/ |
||||||
|
public class UploadTarget extends AbsTarget<UploadEntity, UploadTaskEntity> { |
||||||
|
|
||||||
|
UploadTarget(UploadEntity entity, String targetName) { |
||||||
|
this.entity = entity; |
||||||
|
this.targetName = targetName; |
||||||
|
taskEntity = new UploadTaskEntity(entity); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置上传路径 |
||||||
|
* |
||||||
|
* @param uploadUrl 上传路径 |
||||||
|
*/ |
||||||
|
public UploadTarget setUploadUrl(@NonNull String uploadUrl) { |
||||||
|
taskEntity.uploadUrl = uploadUrl; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置服务器需要的附件key |
||||||
|
* |
||||||
|
* @param attachment 附件key |
||||||
|
*/ |
||||||
|
public UploadTarget setAttachment(@NonNull String attachment) { |
||||||
|
taskEntity.attachment = attachment; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置文件名 |
||||||
|
*/ |
||||||
|
public UploadTarget setFileName(String fileName) { |
||||||
|
entity.setFileName(fileName); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置上传文件类型 |
||||||
|
* |
||||||
|
* @param contentType <code>"multipart/form-data"<code/> |
||||||
|
*/ |
||||||
|
public UploadTarget setContentType(String contentType) { |
||||||
|
taskEntity.contentType = contentType; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 给url请求添加头部 |
||||||
|
* |
||||||
|
* @param key 头部key |
||||||
|
* @param header 头部value |
||||||
|
*/ |
||||||
|
public UploadTarget addHeader(@NonNull String key, @NonNull String header) { |
||||||
|
super._addHeader(key, header); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 给url请求添加头部 |
||||||
|
* |
||||||
|
* @param headers Map<Key, Value> |
||||||
|
*/ |
||||||
|
public UploadTarget addHeaders(Map<String, String> headers) { |
||||||
|
super._addHeaders(headers); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置请求类型 |
||||||
|
* |
||||||
|
* @param requestEnum {@link RequestEnum} |
||||||
|
*/ |
||||||
|
public UploadTarget setRequestMode(RequestEnum requestEnum) { |
||||||
|
super._setRequestMode(requestEnum); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
private UploadEntity getDownloadEntity(@NonNull String filePath) { |
||||||
|
return DbEntity.findData(UploadEntity.class, "filePath=?", filePath); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 是否在下载 |
||||||
|
*/ |
||||||
|
public boolean isUploading() { |
||||||
|
return UploadTaskQueue.getInstance().getTask(entity).isRunning(); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue