Merge remote-tracking branch 'origin/v2.0.3_dev' into v2.0.3_dev

# Conflicts:
#	src/main/java/com/dispose/mapper/DeviceTaskMapper.java
#	src/main/java/com/dispose/service/impl/DeviceTaskManagerServiceImpl.java
This commit is contained in:
HuangXin 2020-11-13 17:10:13 +08:00
commit c250fd01b2
2 changed files with 54 additions and 26 deletions

View File

@ -7,7 +7,12 @@ import com.dispose.common.ErrorCode;
import com.dispose.common.Helper;
import com.dispose.common.IpAddrType;
import com.dispose.common.NetflowDirection;
import com.dispose.common.DisposeDeviceType;
import com.dispose.manager.DisposeTaskManager;
import com.dispose.mapper.DisposeDeviceMapper;
import com.dispose.pojo.dto.protocol.base.ProtocolRespDTO;
import com.dispose.pojo.entity.DeviceTask;
import com.dispose.pojo.entity.DisposeTask;
import com.dispose.pojo.po.MulReturnType;
import com.dispose.pojo.vo.DeviceFirewareInfo;
import com.pengxin.dispose.common.PengXinDisposeAbilityRsp;
@ -24,9 +29,11 @@ import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import javax.annotation.Nullable;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
/**
* The type Peng xin ability.
@ -35,6 +42,17 @@ import java.util.List;
*/
@Slf4j
public class PengXinAbilityImpl implements DisposeAbility {
/**
* The Dispose task manager.
*/
@Resource
private DisposeTaskManager disposeTaskManager;
/**
* The Dispose device mapper.
*/
@Resource
private DisposeDeviceMapper disposeDeviceMapper;
/**
* The Protect ip v 4.
@ -423,15 +441,25 @@ public class PengXinAbilityImpl implements DisposeAbility {
try {
String url = urlRootPath + "dispose_device/task/get";
DisposeTask task = disposeTaskManager.getDisposeTaskById(taskId);
if (task == null) {
return new MulReturnType<>(ErrorCode.ERR_NOSUCHTASK, -1L);
}
//根据设备ID获取设备信息判断任务列表中的设备为鹏信获取externId
List<DeviceTask> dev = task.getDeviceTask().stream().
filter(v -> disposeDeviceMapper.getDeviceById(v.getDeviceId()).getDeviceType() == DisposeDeviceType.PENGXIN_PLATFORM).collect(Collectors.toList());
ProtocolRespDTO<PengXinTaskStatusRsp> rspInfo = restfulInterface.getDeviceTaskStatus(url, token,
new String[] {String.valueOf(taskId)});
new String[]{String.valueOf(dev.get(0).getExternId())});
// 判断是否token过期
if (rspInfo != null && rspInfo.getCode() == HttpServletResponse.SC_UNAUTHORIZED) {
// 重新登录获取 token
upgradeToken();
rspInfo = restfulInterface.getDeviceTaskStatus(urlRootPath, token,
new String[] {String.valueOf(taskId)});
rspInfo = restfulInterface.getDeviceTaskStatus(url, token,
new String[]{String.valueOf(dev.get(0).getExternId())});
}
if (rspInfo != null && rspInfo.getCode() == ErrorCode.ERR_OK.getHttpCode()) {

View File

@ -90,7 +90,7 @@
<update id="changeTaskDevStatus">
UPDATE device_task
SET status = #{status}
SET devStatus = #{devStatus}
WHERE id = #{id}
</update>