package com.wgcloud.controller;
|
|
import cn.hutool.json.JSONUtil;
|
import com.wgcloud.dto.TuopuNodeDto;
|
import com.wgcloud.entity.DceInfo;
|
import com.wgcloud.entity.SnmpInfo;
|
import com.wgcloud.entity.SystemInfo;
|
import com.wgcloud.service.DceInfoService;
|
import com.wgcloud.service.LogInfoService;
|
import com.wgcloud.service.SnmpInfoService;
|
import com.wgcloud.service.SystemInfoService;
|
import com.wgcloud.util.HostUtil;
|
import com.wgcloud.util.ThreadPoolUtil;
|
import com.wgcloud.util.TuopuUtil;
|
import com.wgcloud.util.staticvar.StaticKeys;
|
import org.apache.commons.lang3.StringUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.ui.Model;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @version v3.3
|
* @ClassName:TuopuController.java
|
* @author: http://www.wgstart.com
|
* @date: 2021年1月16日
|
* @Description: 网络拓扑图
|
* @Copyright: 2019-2021 wgcloud. All rights reserved.
|
*/
|
@Controller
|
@RequestMapping("/tuopu")
|
public class TuopuController {
|
|
|
private static final Logger logger = LoggerFactory.getLogger(TuopuController.class);
|
|
|
@Resource
|
private SystemInfoService systemInfoService;
|
@Resource
|
private DceInfoService dceInfoService;
|
@Resource
|
private SnmpInfoService snmpInfoService;
|
@Resource
|
private LogInfoService logInfoService;
|
|
|
private void testThread() {
|
Runnable runnable = () -> {
|
logger.info("TuopuController----------testThread");
|
};
|
ThreadPoolUtil.executor.execute(runnable);
|
}
|
|
/**
|
* 绘制主机拓扑图
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "tuopuListHost")
|
public String tuopuListHost(Model model, HttpServletRequest request) {
|
Map<String, Object> params = new HashMap<String, Object>();
|
try {
|
Map<String, String> systemInfoMap = new HashMap<>();
|
List<TuopuNodeDto> nodeList = new ArrayList<>();
|
|
//组装主机 begin
|
|
//校验是否添加过滤用户查询条件
|
HostUtil.addAccountquery(request, params);
|
|
List<SystemInfo> pageInfo = systemInfoService.selectAllByParams(params);
|
for (SystemInfo systemInfo : pageInfo) {
|
TuopuNodeDto dto = new TuopuNodeDto();
|
systemInfoMap.put(systemInfo.getHostname(), systemInfo.getHostname());
|
dto.setHostname(systemInfo.getHostname() + HostUtil.addRemark(systemInfo.getHostname()));
|
dto.setState(systemInfo.getState());
|
dto.setType("1");
|
nodeList.add(dto);
|
}
|
//组装主机 ends
|
|
TuopuUtil.initList(nodeList);
|
model.addAttribute("nodeList", JSONUtil.parseArray(nodeList));
|
|
} catch (Exception e) {
|
logger.error("主机拓扑图生成错误", e);
|
logInfoService.save("主机拓扑图生成错误", e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "tuopu";
|
}
|
|
|
/**
|
* 数通拓扑图绘制
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "tuopuListSt")
|
public String tuopuListSt(Model model, HttpServletRequest request) {
|
Map<String, Object> params = new HashMap<String, Object>();
|
try {
|
List<TuopuNodeDto> nodeList = new ArrayList<>();
|
|
//组装数通PING设备 begin
|
|
//校验是否添加过滤用户查询条件
|
HostUtil.addAccountquery(request, params);
|
|
List<DceInfo> pageInfoDce = dceInfoService.selectAllByParams(params);
|
for (DceInfo dceInfo : pageInfoDce) {
|
TuopuNodeDto dto = new TuopuNodeDto();
|
if (!StringUtils.isEmpty(dceInfo.getRemark())) {
|
dto.setHostname(dceInfo.getHostname() + "(" + dceInfo.getRemark() + ")");
|
} else {
|
dto.setHostname(dceInfo.getHostname());
|
}
|
if (dceInfo.getResTimes() < 0) {
|
dto.setState(StaticKeys.DOWN_STATE);
|
} else {
|
dto.setState(StaticKeys.ON_STATE);
|
}
|
dto.setType("2");
|
nodeList.add(dto);
|
}
|
//组装数通PING设备 end
|
|
|
//组装数通SNMP设备 begin
|
//暂存已添加过的snmp设备
|
List<SnmpInfo> snmpAddedList = new ArrayList<>();
|
|
//校验是否添加过滤用户查询条件
|
HostUtil.addAccountquery(request, params);
|
|
List<SnmpInfo> pageInfoSnmp = snmpInfoService.selectAllByParams(params);
|
for (SnmpInfo snmpInfo : pageInfoSnmp) {
|
for (SnmpInfo snmpInfoAdded : snmpAddedList) {
|
if (snmpInfo.getHostname().equals(snmpInfoAdded.getHostname())) {
|
continue;
|
}
|
}
|
TuopuNodeDto dto = new TuopuNodeDto();
|
if (!StringUtils.isEmpty(snmpInfo.getRemark())) {
|
dto.setHostname(snmpInfo.getHostname() + "(" + snmpInfo.getRemark() + ")");
|
} else {
|
dto.setHostname(snmpInfo.getHostname());
|
}
|
dto.setState(snmpInfo.getState());
|
dto.setType("2");
|
nodeList.add(dto);
|
}
|
//组装数通SNMP设备 end
|
|
//设置节点坐标
|
TuopuUtil.initList(nodeList);
|
model.addAttribute("nodeList", JSONUtil.parseArray(nodeList));
|
|
} catch (Exception e) {
|
logger.error("数通拓扑图生成错误", e);
|
logInfoService.save("数通拓扑图生成错误", e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "tuopu";
|
}
|
|
}
|
