laoyuyu
6 years ago
parent
7f52271ee5
commit
f6792b225a
@ -1,44 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.download.wrapper; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.download.DGTaskWrapper; |
|
||||||
import com.arialyy.aria.core.download.DTaskWrapper; |
|
||||||
import com.arialyy.aria.orm.AbsWrapper; |
|
||||||
import com.arialyy.aria.orm.annotation.Many; |
|
||||||
import com.arialyy.aria.orm.annotation.One; |
|
||||||
import com.arialyy.aria.orm.annotation.Wrapper; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by laoyuyu on 2018/4/11. |
|
||||||
* 任务组任务实体和任务组任务实体的子任务实体对应关系 |
|
||||||
*/ |
|
||||||
@Wrapper |
|
||||||
public class DGSTEWrapper extends AbsWrapper { |
|
||||||
|
|
||||||
@One |
|
||||||
public DGTaskWrapper dgTaskEntity; |
|
||||||
|
|
||||||
@Many(parentColumn = "key", entityColumn = "groupName") |
|
||||||
public List<DTaskWrapper> subTaskEntity; |
|
||||||
|
|
||||||
@Override protected void handleConvert() { |
|
||||||
if (subTaskEntity != null && !subTaskEntity.isEmpty()) { |
|
||||||
dgTaskEntity.setSubTaskWrapper(subTaskEntity); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
@ -1,65 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.download.wrapper; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.download.DGTaskWrapper; |
|
||||||
import com.arialyy.aria.core.download.DownloadGroupEntity; |
|
||||||
import com.arialyy.aria.core.download.DTaskWrapper; |
|
||||||
import com.arialyy.aria.core.inf.AbsTaskWrapper; |
|
||||||
import com.arialyy.aria.orm.AbsWrapper; |
|
||||||
import com.arialyy.aria.orm.DbEntity; |
|
||||||
import com.arialyy.aria.orm.annotation.Many; |
|
||||||
import com.arialyy.aria.orm.annotation.One; |
|
||||||
import com.arialyy.aria.orm.annotation.Wrapper; |
|
||||||
import com.arialyy.aria.util.CommonUtil; |
|
||||||
import java.util.ArrayList; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by laoyuyu on 2018/3/30. |
|
||||||
* 任务组实体和任务组任务实体的关系 |
|
||||||
*/ |
|
||||||
@Wrapper |
|
||||||
public class DGTEWrapper extends AbsWrapper { |
|
||||||
|
|
||||||
@One |
|
||||||
public DownloadGroupEntity entity; |
|
||||||
|
|
||||||
@Many(parentColumn = "groupName", entityColumn = "key") |
|
||||||
private List<DGTaskWrapper> taskEntitys; |
|
||||||
|
|
||||||
public DGTaskWrapper taskEntity; |
|
||||||
|
|
||||||
@Override protected void handleConvert() { |
|
||||||
taskEntity = (taskEntitys == null || taskEntitys.isEmpty()) ? null : taskEntitys.get(0); |
|
||||||
if (taskEntity != null) { |
|
||||||
taskEntity.setEntity(entity); |
|
||||||
List<DTEWrapper> subWrappers = |
|
||||||
DbEntity.findRelationData(DTEWrapper.class, "DTaskWrapper.groupName=?", |
|
||||||
taskEntity.getKey()); |
|
||||||
if (subWrappers != null && !subWrappers.isEmpty()) { |
|
||||||
List<DTaskWrapper> temp = new ArrayList<>(); |
|
||||||
for (DTEWrapper dw : subWrappers) { |
|
||||||
if (dw.taskEntity.getRequestType() == AbsTaskWrapper.D_FTP) { |
|
||||||
dw.taskEntity.asFtp().setUrlEntity(CommonUtil.getFtpUrlInfo(dw.taskEntity.getUrl())); |
|
||||||
} |
|
||||||
temp.add(dw.taskEntity); |
|
||||||
} |
|
||||||
taskEntity.setSubTaskWrapper(temp); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
@ -1,46 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.download.wrapper; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.download.DownloadEntity; |
|
||||||
import com.arialyy.aria.core.download.DTaskWrapper; |
|
||||||
import com.arialyy.aria.orm.AbsWrapper; |
|
||||||
import com.arialyy.aria.orm.annotation.Many; |
|
||||||
import com.arialyy.aria.orm.annotation.One; |
|
||||||
import com.arialyy.aria.orm.annotation.Wrapper; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by laoyuyu on 2018/3/30. |
|
||||||
*/ |
|
||||||
@Wrapper |
|
||||||
public class DTEWrapper extends AbsWrapper { |
|
||||||
|
|
||||||
@One |
|
||||||
public DownloadEntity entity; |
|
||||||
|
|
||||||
@Many(parentColumn = "downloadPath", entityColumn = "key") |
|
||||||
private List<DTaskWrapper> taskEntitys = null; |
|
||||||
|
|
||||||
public DTaskWrapper taskEntity; |
|
||||||
|
|
||||||
@Override public void handleConvert() { |
|
||||||
taskEntity = (taskEntitys == null || taskEntitys.isEmpty()) ? null : taskEntitys.get(0); |
|
||||||
if (taskEntity != null) { |
|
||||||
taskEntity.setEntity(entity); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
@ -1,168 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.manager; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.download.DGTaskWrapper; |
|
||||||
import com.arialyy.aria.core.download.DTaskWrapper; |
|
||||||
import com.arialyy.aria.core.download.DownloadEntity; |
|
||||||
import com.arialyy.aria.core.download.DownloadGroupEntity; |
|
||||||
import com.arialyy.aria.core.download.wrapper.DGEWrapper; |
|
||||||
import com.arialyy.aria.core.download.wrapper.DGTEWrapper; |
|
||||||
import com.arialyy.aria.orm.DbEntity; |
|
||||||
import com.arialyy.aria.util.CommonUtil; |
|
||||||
import java.io.File; |
|
||||||
import java.util.ArrayList; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by Aria.Lao on 2017/11/1. |
|
||||||
* 任务实体工厂 |
|
||||||
*/ |
|
||||||
class DGTEFactory implements IGTEFactory<DownloadGroupEntity, DGTaskWrapper> { |
|
||||||
private static final String TAG = "DTEFactory"; |
|
||||||
private static volatile DGTEFactory INSTANCE = null; |
|
||||||
|
|
||||||
private DGTEFactory() { |
|
||||||
} |
|
||||||
|
|
||||||
public static DGTEFactory getInstance() { |
|
||||||
if (INSTANCE == null) { |
|
||||||
synchronized (DGTEFactory.class) { |
|
||||||
INSTANCE = new DGTEFactory(); |
|
||||||
} |
|
||||||
} |
|
||||||
return INSTANCE; |
|
||||||
} |
|
||||||
|
|
||||||
@Override public DGTaskWrapper getGTE(String groupName, List<String> urls) { |
|
||||||
DownloadGroupEntity entity = createDGroupEntity(groupName, urls); |
|
||||||
List<DGTEWrapper> wrapper = |
|
||||||
DbEntity.findRelationData(DGTEWrapper.class, "DGTaskWrapper.key=?", |
|
||||||
entity.getGroupName()); |
|
||||||
DGTaskWrapper gte; |
|
||||||
|
|
||||||
if (wrapper != null && !wrapper.isEmpty()) { |
|
||||||
gte = wrapper.get(0).taskEntity; |
|
||||||
if (gte == null) { |
|
||||||
// 创建新的任务组任务实体
|
|
||||||
gte = new DGTaskWrapper(); |
|
||||||
} |
|
||||||
} else { |
|
||||||
gte = new DGTaskWrapper(); |
|
||||||
} |
|
||||||
gte.setSubTaskWrapper(createDGSubTaskEntity(entity)); |
|
||||||
gte.setKey(entity.getGroupName()); |
|
||||||
gte.setEntity(entity); |
|
||||||
|
|
||||||
return gte; |
|
||||||
} |
|
||||||
|
|
||||||
@Override public DGTaskWrapper getFTE(String ftpUrl) { |
|
||||||
List<DGTEWrapper> wrapper = |
|
||||||
DbEntity.findRelationData(DGTEWrapper.class, "DGTaskWrapper.key=?", |
|
||||||
ftpUrl); |
|
||||||
DGTaskWrapper fte; |
|
||||||
|
|
||||||
if (wrapper != null && !wrapper.isEmpty()) { |
|
||||||
fte = wrapper.get(0).taskEntity; |
|
||||||
if (fte == null) { |
|
||||||
fte = new DGTaskWrapper(); |
|
||||||
DownloadGroupEntity dge = new DownloadGroupEntity(); |
|
||||||
dge.setGroupName(ftpUrl); |
|
||||||
fte.setEntity(dge); |
|
||||||
} else if (fte.getEntity() == null) { |
|
||||||
DownloadGroupEntity dge = new DownloadGroupEntity(); |
|
||||||
dge.setGroupName(ftpUrl); |
|
||||||
fte.setEntity(dge); |
|
||||||
} |
|
||||||
} else { |
|
||||||
fte = new DGTaskWrapper(); |
|
||||||
DownloadGroupEntity dge = new DownloadGroupEntity(); |
|
||||||
dge.setGroupName(ftpUrl); |
|
||||||
fte.setEntity(dge); |
|
||||||
} |
|
||||||
fte.setKey(ftpUrl); |
|
||||||
fte.asFtp().setUrlEntity(CommonUtil.getFtpUrlInfo(ftpUrl)); |
|
||||||
|
|
||||||
if (fte.getEntity().getSubEntities() == null) { |
|
||||||
fte.getEntity().setSubEntities(new ArrayList<DownloadEntity>()); |
|
||||||
} |
|
||||||
if (fte.getSubTaskWrapper() == null) { |
|
||||||
fte.setSubTaskWrapper(new ArrayList<DTaskWrapper>()); |
|
||||||
} |
|
||||||
return fte; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 创建任务组子任务的任务实体 |
|
||||||
*/ |
|
||||||
private List<DTaskWrapper> createDGSubTaskEntity(DownloadGroupEntity dge) { |
|
||||||
List<DTaskWrapper> list = new ArrayList<>(); |
|
||||||
for (DownloadEntity entity : dge.getSubEntities()) { |
|
||||||
DTaskWrapper taskEntity = new DTaskWrapper(); |
|
||||||
taskEntity.setEntity(entity); |
|
||||||
taskEntity.setKey(entity.getDownloadPath()); |
|
||||||
taskEntity.setGroupName(dge.getKey()); |
|
||||||
taskEntity.setGroupTask(true); |
|
||||||
taskEntity.setUrl(entity.getUrl()); |
|
||||||
list.add(taskEntity); |
|
||||||
} |
|
||||||
return list; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 查询任务组实体，如果数据库不存在该实体，则新创建一个新的任务组实体 |
|
||||||
*/ |
|
||||||
private DownloadGroupEntity createDGroupEntity(String groupName, List<String> urls) { |
|
||||||
List<DGEWrapper> wrapper = |
|
||||||
DbEntity.findRelationData(DGEWrapper.class, "DownloadGroupEntity.groupName=?", |
|
||||||
groupName); |
|
||||||
|
|
||||||
DownloadGroupEntity groupEntity; |
|
||||||
if (wrapper != null && !wrapper.isEmpty()) { |
|
||||||
groupEntity = wrapper.get(0).groupEntity; |
|
||||||
if (groupEntity == null) { |
|
||||||
groupEntity = new DownloadGroupEntity(); |
|
||||||
groupEntity.setSubEntities(createSubTask(groupName, urls)); |
|
||||||
} |
|
||||||
} else { |
|
||||||
groupEntity = new DownloadGroupEntity(); |
|
||||||
groupEntity.setSubEntities(createSubTask(groupName, urls)); |
|
||||||
} |
|
||||||
groupEntity.setGroupName(groupName); |
|
||||||
groupEntity.setUrls(urls); |
|
||||||
return groupEntity; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 创建子任务 |
|
||||||
*/ |
|
||||||
private List<DownloadEntity> createSubTask(String groupName, List<String> urls) { |
|
||||||
List<DownloadEntity> list = new ArrayList<>(); |
|
||||||
for (int i = 0, len = urls.size(); i < len; i++) { |
|
||||||
String url = urls.get(i); |
|
||||||
DownloadEntity entity = new DownloadEntity(); |
|
||||||
entity.setUrl(url); |
|
||||||
entity.setDownloadPath(groupName + "_" + i); |
|
||||||
int lastIndex = url.lastIndexOf(File.separator); |
|
||||||
entity.setFileName(url.substring(lastIndex + 1, url.length())); |
|
||||||
entity.setGroupName(groupName); |
|
||||||
entity.setGroupChild(true); |
|
||||||
list.add(entity); |
|
||||||
} |
|
||||||
return list; |
|
||||||
} |
|
||||||
} |
|
||||||
@ -0,0 +1,90 @@ |
|||||||
|
/* |
||||||
|
* 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.manager; |
||||||
|
|
||||||
|
import com.arialyy.aria.core.download.DGTaskWrapper; |
||||||
|
import com.arialyy.aria.core.download.DTaskWrapper; |
||||||
|
import com.arialyy.aria.core.download.DownloadEntity; |
||||||
|
import com.arialyy.aria.core.download.DownloadGroupEntity; |
||||||
|
import com.arialyy.aria.core.download.DGEntityWrapper; |
||||||
|
import com.arialyy.aria.orm.DbEntity; |
||||||
|
import com.arialyy.aria.util.CommonUtil; |
||||||
|
import com.arialyy.aria.util.DbDataHelper; |
||||||
|
import java.io.File; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* Created by Aria.Lao on 2017/11/1. 组合任务wrapper |
||||||
|
*/ |
||||||
|
class DGTaskWrapperFactory implements IGTEFactory<DownloadGroupEntity, DGTaskWrapper> { |
||||||
|
private static final String TAG = "DTaskWrapperFactory"; |
||||||
|
private static volatile DGTaskWrapperFactory INSTANCE = null; |
||||||
|
|
||||||
|
private DGTaskWrapperFactory() { |
||||||
|
} |
||||||
|
|
||||||
|
public static DGTaskWrapperFactory getInstance() { |
||||||
|
if (INSTANCE == null) { |
||||||
|
synchronized (DGTaskWrapperFactory.class) { |
||||||
|
INSTANCE = new DGTaskWrapperFactory(); |
||||||
|
} |
||||||
|
} |
||||||
|
return INSTANCE; |
||||||
|
} |
||||||
|
|
||||||
|
@Override public DGTaskWrapper getGTE(String groupName, List<String> urls) { |
||||||
|
DownloadGroupEntity entity = DbDataHelper.getHttpDGEntity(groupName, urls); |
||||||
|
DGTaskWrapper wrapper = new DGTaskWrapper(entity); |
||||||
|
wrapper.setSubTaskWrapper(createDGSubTaskWrapper(entity)); |
||||||
|
wrapper.setKey(entity.getGroupName()); |
||||||
|
return wrapper; |
||||||
|
} |
||||||
|
|
||||||
|
@Override public DGTaskWrapper getFTE(String ftpUrl) { |
||||||
|
DownloadGroupEntity entity = DbDataHelper.getFtpDGEntity(ftpUrl); |
||||||
|
DGTaskWrapper fte = new DGTaskWrapper(entity); |
||||||
|
|
||||||
|
fte.setKey(ftpUrl); |
||||||
|
fte.asFtp().setUrlEntity(CommonUtil.getFtpUrlInfo(ftpUrl)); |
||||||
|
|
||||||
|
if (fte.getEntity().getSubEntities() == null) { |
||||||
|
fte.getEntity().setSubEntities(new ArrayList<DownloadEntity>()); |
||||||
|
} |
||||||
|
if (fte.getSubTaskWrapper() == null) { |
||||||
|
fte.setSubTaskWrapper(new ArrayList<DTaskWrapper>()); |
||||||
|
} |
||||||
|
return fte; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建任务组子任务的任务实体 |
||||||
|
*/ |
||||||
|
private List<DTaskWrapper> createDGSubTaskWrapper(DownloadGroupEntity dge) { |
||||||
|
List<DTaskWrapper> list = new ArrayList<>(); |
||||||
|
for (DownloadEntity entity : dge.getSubEntities()) { |
||||||
|
DTaskWrapper taskEntity = new DTaskWrapper(entity); |
||||||
|
taskEntity.setKey(entity.getDownloadPath()); |
||||||
|
taskEntity.setGroupName(dge.getKey()); |
||||||
|
taskEntity.setGroupTask(true); |
||||||
|
taskEntity.setUrl(entity.getUrl()); |
||||||
|
list.add(taskEntity); |
||||||
|
} |
||||||
|
return list; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -1,47 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.wrapper; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.upload.UTaskWrapper; |
|
||||||
import com.arialyy.aria.core.upload.UploadEntity; |
|
||||||
import com.arialyy.aria.orm.AbsWrapper; |
|
||||||
import com.arialyy.aria.orm.annotation.Many; |
|
||||||
import com.arialyy.aria.orm.annotation.One; |
|
||||||
import com.arialyy.aria.orm.annotation.Wrapper; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by laoyuyu on 2018/3/30. |
|
||||||
*/ |
|
||||||
@Wrapper |
|
||||||
public class UTEWrapper extends AbsWrapper { |
|
||||||
|
|
||||||
@One |
|
||||||
public UploadEntity entity; |
|
||||||
|
|
||||||
@Many(parentColumn = "filePath", entityColumn = "key") |
|
||||||
private List<UTaskWrapper> taskEntitys = null; |
|
||||||
|
|
||||||
public UTaskWrapper taskEntity; |
|
||||||
|
|
||||||
@Override public void handleConvert() { |
|
||||||
//taskEntity.entity = (tEntity == null || tEntity.isEmpty()) ? null : tEntity.get(0);
|
|
||||||
taskEntity = (taskEntitys == null || taskEntitys.isEmpty()) ? null : taskEntitys.get(0); |
|
||||||
if (taskEntity != null) { |
|
||||||
taskEntity.setEntity(entity); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
@ -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.util; |
||||||
|
|
||||||
|
import com.arialyy.aria.core.common.RecordWrapper; |
||||||
|
import com.arialyy.aria.core.common.TaskRecord; |
||||||
|
import com.arialyy.aria.core.download.DGEntityWrapper; |
||||||
|
import com.arialyy.aria.core.download.DownloadEntity; |
||||||
|
import com.arialyy.aria.core.download.DownloadGroupEntity; |
||||||
|
import com.arialyy.aria.orm.DbEntity; |
||||||
|
import java.io.File; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* 数据库帮助类 |
||||||
|
*/ |
||||||
|
public class DbDataHelper { |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取任务记录 |
||||||
|
* |
||||||
|
* @param filePath 文件地址 |
||||||
|
* @return 没有记录返回null，有记录则返回任务记录 |
||||||
|
*/ |
||||||
|
public static TaskRecord getTaskRecord(String filePath) { |
||||||
|
List<RecordWrapper> records = |
||||||
|
DbEntity.findRelationData(RecordWrapper.class, "TaskRecord.filePath=?", filePath); |
||||||
|
if (records == null || records.size() == 0) { |
||||||
|
return null; |
||||||
|
} |
||||||
|
return records.get(0).taskRecord; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取组合任务实体 如果数据库不存在该实体，则新创建一个新的任务组实体 |
||||||
|
* |
||||||
|
* @param groupName 组合任务名称 |
||||||
|
* @param urls 子任务url列表 |
||||||
|
*/ |
||||||
|
public static DownloadGroupEntity getHttpDGEntity(String groupName, List<String> urls) { |
||||||
|
List<DGEntityWrapper> wrapper = |
||||||
|
DbEntity.findRelationData(DGEntityWrapper.class, "DownloadGroupEntity.groupName=?", |
||||||
|
groupName); |
||||||
|
|
||||||
|
DownloadGroupEntity groupEntity; |
||||||
|
if (wrapper != null && !wrapper.isEmpty()) { |
||||||
|
groupEntity = wrapper.get(0).groupEntity; |
||||||
|
if (groupEntity == null) { |
||||||
|
groupEntity = new DownloadGroupEntity(); |
||||||
|
groupEntity.setSubEntities(createHttpSubTask(groupName, urls)); |
||||||
|
} |
||||||
|
} else { |
||||||
|
groupEntity = new DownloadGroupEntity(); |
||||||
|
groupEntity.setSubEntities(createHttpSubTask(groupName, urls)); |
||||||
|
} |
||||||
|
groupEntity.setGroupName(groupName); |
||||||
|
groupEntity.setUrls(urls); |
||||||
|
return groupEntity; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建HTTP子任务实体 |
||||||
|
*/ |
||||||
|
private static List<DownloadEntity> createHttpSubTask(String groupName, List<String> urls) { |
||||||
|
List<DownloadEntity> list = new ArrayList<>(); |
||||||
|
for (int i = 0, len = urls.size(); i < len; i++) { |
||||||
|
String url = urls.get(i); |
||||||
|
DownloadEntity entity = new DownloadEntity(); |
||||||
|
entity.setUrl(url); |
||||||
|
entity.setDownloadPath(groupName + "_" + i); |
||||||
|
int lastIndex = url.lastIndexOf(File.separator); |
||||||
|
entity.setFileName(url.substring(lastIndex + 1)); |
||||||
|
entity.setGroupName(groupName); |
||||||
|
entity.setGroupChild(true); |
||||||
|
list.add(entity); |
||||||
|
} |
||||||
|
return list; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 通过Ftp下载地址获取组合任务实体 |
||||||
|
* |
||||||
|
* @param ftpUrl ftp下载地址 |
||||||
|
*/ |
||||||
|
public static DownloadGroupEntity getFtpDGEntity(String ftpUrl) { |
||||||
|
List<DGEntityWrapper> wrapper = |
||||||
|
DbEntity.findRelationData(DGEntityWrapper.class, "DownloadGroupEntity.groupName=?", |
||||||
|
ftpUrl); |
||||||
|
DownloadGroupEntity groupEntity; |
||||||
|
if (wrapper != null && !wrapper.isEmpty()) { |
||||||
|
groupEntity = wrapper.get(0).groupEntity; |
||||||
|
if (groupEntity == null) { |
||||||
|
groupEntity = new DownloadGroupEntity(); |
||||||
|
} |
||||||
|
} else { |
||||||
|
groupEntity = new DownloadGroupEntity(); |
||||||
|
} |
||||||
|
groupEntity.setGroupName(ftpUrl); |
||||||
|
return groupEntity; |
||||||
|
} |
||||||
|
} |
||||||
@ -1,42 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.util; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.common.RecordWrapper; |
|
||||||
import com.arialyy.aria.core.common.TaskRecord; |
|
||||||
import com.arialyy.aria.orm.DbEntity; |
|
||||||
import java.util.List; |
|
||||||
|
|
||||||
/** |
|
||||||
* 数据库帮助类 |
|
||||||
*/ |
|
||||||
public class DbHelper { |
|
||||||
|
|
||||||
/** |
|
||||||
* 获取任务记录 |
|
||||||
* |
|
||||||
* @param filePath 文件地址 |
|
||||||
* @return 没有记录返回null，有记录则返回任务记录 |
|
||||||
*/ |
|
||||||
public static TaskRecord getTaskRecord(String filePath) { |
|
||||||
List<RecordWrapper> records = |
|
||||||
DbEntity.findRelationData(RecordWrapper.class, "TaskRecord.filePath=?", filePath); |
|
||||||
if (records == null || records.size() == 0) { |
|
||||||
return null; |
|
||||||
} |
|
||||||
return records.get(0).taskRecord; |
|
||||||
} |
|
||||||
} |
|
||||||
Loading…
Reference in new issue