package com.wgcloud.controller;
|
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.github.pagehelper.PageInfo;
|
import com.wgcloud.config.CommonConfig;
|
import com.wgcloud.entity.CustomInfo;
|
import com.wgcloud.entity.CustomState;
|
import com.wgcloud.entity.HostGroup;
|
import com.wgcloud.entity.SystemInfo;
|
import com.wgcloud.service.*;
|
import com.wgcloud.util.*;
|
import com.wgcloud.util.staticvar.StaticKeys;
|
import org.apache.commons.lang3.StringUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.ui.Model;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @version v3.3
|
* @ClassName:CustomInfoController.java
|
* @author: http://www.wgstart.com
|
* @date: 2022年9月16日
|
* @Description: 自定义监控项管理
|
* @Copyright: 2019-2022 wgcloud. All rights reserved.
|
*/
|
@Controller
|
@RequestMapping("/customInfo")
|
public class CustomInfoController {
|
|
|
private static final Logger logger = LoggerFactory.getLogger(CustomInfoController.class);
|
|
@Resource
|
private CustomInfoService customInfoService;
|
|
@Resource
|
private CustomStateService customStateService;
|
|
@Resource
|
private LogInfoService logInfoService;
|
|
@Resource
|
private SystemInfoService systemInfoService;
|
|
@Resource
|
private DashboardService dashboardService;
|
|
@Resource
|
private ExcelExportService excelExportService;
|
@Resource
|
private HostGroupService hostGroupService;
|
|
@Autowired
|
private TokenUtils tokenUtils;
|
|
@Autowired
|
private CommonConfig commonConfig;
|
|
|
/**
|
* agent查询自定义监控项列表
|
*
|
* @return
|
*/
|
@ResponseBody
|
@RequestMapping(value = "agentList")
|
public String agentList(@RequestBody String paramBean) {
|
JSONObject agentJsonObject = (JSONObject) JSONUtil.parse(paramBean);
|
if (!tokenUtils.checkAgentToken(agentJsonObject)) {
|
logger.error(StaticKeys.TOKEN_ERROR);
|
return ResDataUtils.resetErrorJson(StaticKeys.TOKEN_ERROR);
|
}
|
Map<String, Object> params = new HashMap<String, Object>();
|
if (null == agentJsonObject.get("hostname") || StringUtils.isEmpty(agentJsonObject.get("hostname").toString())) {
|
return "";
|
}
|
params.put("hostname", agentJsonObject.get("hostname").toString());
|
try {
|
params.put("active", StaticKeys.ON_STATE);
|
List<CustomInfo> customInfoList = customInfoService.selectAllByParams(params);
|
//检测系统类型,检查是否含有敏感字符 begin
|
SystemInfo systemInfo = systemInfoService.selectByHostname(agentJsonObject.get("hostname").toString());
|
String shellToRunBlock = commonConfig.getShellToRunLinuxBlock();
|
if (systemInfo != null) {
|
String platForm = systemInfo.getPlatForm().toLowerCase();
|
if (platForm.contains("windows")) {
|
shellToRunBlock = commonConfig.getShellToRunWinBlock();
|
}
|
}
|
|
String blockKey = "";
|
List<CustomInfo> customInfoListResult = new ArrayList<>();
|
for (CustomInfo state : customInfoList) {
|
//检查是否含有敏感字符
|
blockKey = FormatUtil.haveBlockDanger(state.getCustomShell(), shellToRunBlock, "");
|
if (!StringUtils.isEmpty(blockKey)) {
|
logger.error(state.getCustomShell() + "自定义监控项指令含有敏感字符" + blockKey + "，不进行下发");
|
continue;
|
}
|
customInfoListResult.add(state);
|
}
|
//检测系统类型,检查是否含有敏感字符 end
|
return ResDataUtils.resetSuccessJson(customInfoListResult);
|
} catch (Exception e) {
|
logger.error("agent获取自定义监控项信息错误", e);
|
logInfoService.save("agent获取自定义监控项信息错误", e.toString(), StaticKeys.LOG_XTCZ);
|
return ResDataUtils.resetErrorJson(e.toString());
|
}
|
}
|
|
/**
|
* 根据条件查询自定义监控项列表
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "list")
|
public String customInfoList(CustomInfo customInfo, Model model, HttpServletRequest request) {
|
Map<String, Object> params = new HashMap<String, Object>();
|
try {
|
|
StringBuffer url = new StringBuffer();
|
String hostname = null;
|
if (!StringUtils.isEmpty(customInfo.getHostname())) {
|
hostname = customInfo.getHostname();
|
params.put("hostname", hostname.trim());
|
url.append("&hostname=").append(hostname);
|
}
|
if (!StringUtils.isEmpty(customInfo.getAccount())) {
|
params.put("account", customInfo.getAccount());
|
url.append("&account=").append(customInfo.getAccount());
|
}
|
if (!StringUtils.isEmpty(customInfo.getGroupId())) {
|
params.put("groupId", customInfo.getGroupId());
|
url.append("&groupId=").append(customInfo.getGroupId());
|
}
|
if (!StringUtils.isEmpty(customInfo.getState())) {
|
params.put("state", customInfo.getState());
|
url.append("&state=").append(customInfo.getState());
|
}
|
|
//校验是否需要添加过滤用户查询条件
|
HostUtil.addAccountquery(request, params);
|
|
PageInfo<CustomInfo> pageInfo = customInfoService.selectByParams(params, customInfo.getPage(), customInfo.getPageSize());
|
Map<String, Object> paramsLogInfo = new HashMap<String, Object>();
|
for (CustomInfo customInfo1 : pageInfo.getList()) {
|
//设置所属用户账号 begin
|
if (StaticKeys.TRUE_VAL.equals(commonConfig.getUserInfoManage())) {
|
customInfo1.setAccount(HostUtil.getAccount(customInfo1.getHostname()));
|
}
|
//设置所属用户账号 begin
|
|
customInfo1.setHostname(customInfo1.getHostname() + HostUtil.addRemark(customInfo1.getHostname()));
|
|
//设置累积告警次数 begin
|
if (StaticKeys.TRUE_VAL.equals(commonConfig.getShowWarnCount())) {
|
String warnQueryWd = customInfo1.getCustomName() + "，" + customInfo1.getHostname();
|
paramsLogInfo.clear();
|
paramsLogInfo.put("hostname", warnQueryWd);
|
paramsLogInfo.put("hostnameNe", "已恢复");
|
customInfo1.setWarnCount(logInfoService.countByParams(paramsLogInfo));
|
customInfo1.setWarnQueryWd(warnQueryWd);
|
}
|
//设置累积告警次数 end
|
|
}
|
//设置分组
|
List<HostGroup> hostGroupList = new ArrayList<>();
|
if (StaticKeys.TRUE_VAL.equals(commonConfig.getHostGroup())) {
|
hostGroupList = customInfoService.setGroupInList(pageInfo.getList());
|
}
|
|
//设置用户列表
|
HostUtil.addAccountListModel(model);
|
|
PageUtil.initPageNumber(pageInfo, model);
|
model.addAttribute("pageUrl", "/customInfo/list?1=1" + url.toString());
|
model.addAttribute("page", pageInfo);
|
model.addAttribute("customInfo", customInfo);
|
model.addAttribute("hostGroupList", hostGroupList);
|
} catch (Exception e) {
|
logger.error("查询自定义监控项信息错误", e);
|
logInfoService.save("查询自定义监控项信息错误", e.toString(), StaticKeys.LOG_XTCZ);
|
|
}
|
return "customInfo/list";
|
}
|
|
/**
|
* 保存主机分组信息
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@ResponseBody
|
@RequestMapping(value = "saveGroupId")
|
public String saveGroupId(Model model, HttpServletRequest request) {
|
try {
|
String ids = request.getParameter("ids");
|
String groupId = request.getParameter("groupId");
|
HostGroup hostGroup = hostGroupService.selectById(groupId);
|
if (!StringUtils.isEmpty(ids)) {
|
CustomInfo ho = null;
|
for (String id : ids.split(",")) {
|
ho = customInfoService.selectById(id);
|
ho.setGroupId(groupId);
|
customInfoService.updateById(ho);
|
logInfoService.save(HostUtil.getAccountByRequest(request).getAccount() + "设置自定义监控项分组：" + ho.getHostname() + "，" + ho.getCustomName(),
|
"分组：" + hostGroup.getGroupName(), StaticKeys.LOG_XTCZ);
|
}
|
}
|
} catch (Exception e) {
|
logger.error("保存自定义监控项分组信息错误", e);
|
logInfoService.save("保存自定义监控项分组信息错误", e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "redirect:/customInfo/list";
|
}
|
|
/**
|
* 保存应用监控信息
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "save")
|
public String saveCustomInfo(CustomInfo customInfo, Model model, HttpServletRequest request) {
|
String errorMsg = "保存自定义监控项错误";
|
try {
|
//检测指令有无敏感字符 begin
|
String blockKey = FormatUtil.haveBlockDanger(customInfo.getCustomShell(), commonConfig.getShellToRunLinuxBlock(), commonConfig.getShellToRunWinBlock());
|
if (!StringUtils.isEmpty(blockKey)) {
|
model.addAttribute("customInfo", customInfo);
|
model.addAttribute("msg", "下发指令含有敏感字符" + blockKey + "，请检查");
|
//敏感字符组装
|
getBlockStr(model);
|
//校验是否需要添加过滤用户查询条件
|
Map<String, Object> params = new HashMap<String, Object>();
|
HostUtil.addAccountquery(request, params);
|
List<SystemInfo> systemInfoList = systemInfoService.selectAllByParams(params);
|
model.addAttribute("systemInfoList", systemInfoList);
|
return "customInfo/add";
|
}
|
//检测指令有无敏感字符 end
|
|
if (StringUtils.isEmpty(customInfo.getId())) {
|
customInfoService.save(customInfo, request);
|
customInfoService.saveLog(request, StaticKeys.LOG_ADD, customInfo);
|
} else {
|
customInfoService.updateById(customInfo);
|
customInfoService.saveLog(request, StaticKeys.LOG_UPDATE, customInfo);
|
}
|
} catch (Exception e) {
|
logger.error(errorMsg, e);
|
logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "redirect:/customInfo/list";
|
}
|
|
/**
|
* 添加
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "edit")
|
public String edit(Model model, HttpServletRequest request) {
|
String errorMsg = "添加自定义监控项错误";
|
String id = request.getParameter("id");
|
CustomInfo customInfo = new CustomInfo();
|
try {
|
//校验是否需要添加过滤用户查询条件
|
Map<String, Object> params = new HashMap<String, Object>();
|
HostUtil.addAccountquery(request, params);
|
|
//敏感字符组装
|
getBlockStr(model);
|
|
List<SystemInfo> systemInfoList = systemInfoService.selectAllByParams(params);
|
model.addAttribute("systemInfoList", systemInfoList);
|
if (StringUtils.isEmpty(id)) {
|
model.addAttribute("customInfo", customInfo);
|
|
return "customInfo/add";
|
}
|
customInfo = customInfoService.selectById(id);
|
model.addAttribute("customInfo", customInfo);
|
} catch (Exception e) {
|
logger.error(errorMsg, e);
|
logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "customInfo/add";
|
}
|
|
|
/**
|
* 查看该监控项统计图
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "view")
|
public String viewChart(Model model, HttpServletRequest request) {
|
String errorMsg = "查看自定义监控项图表错误";
|
String id = request.getParameter("id");
|
String startTime = request.getParameter(StaticKeys.SEARCH_START_TIME);
|
String endTime = request.getParameter(StaticKeys.SEARCH_END_TIME);
|
String am = request.getParameter("am");
|
CustomInfo customInfo = new CustomInfo();
|
try {
|
customInfo = customInfoService.selectById(id);
|
//提取主机备注 begin
|
customInfo.setHostname(customInfo.getHostname() + HostUtil.addRemark(customInfo.getHostname()));
|
//提取主机备注 end
|
Map<String, Object> params = new HashMap<String, Object>();
|
dashboardService.setDateParam(am, startTime, endTime, params, model);
|
model.addAttribute("amList", dashboardService.getAmList());
|
params.put("customInfoId", customInfo.getId());
|
model.addAttribute("customInfo", customInfo);
|
List<CustomState> customStateList = customStateService.selectAllByParams(params);
|
// 设置图表的副标题，最高、平均、最低值
|
customStateService.setSubtitle(model, customStateList);
|
model.addAttribute("customStateList", JSONUtil.parseArray(customStateList));
|
} catch (Exception e) {
|
logger.error(errorMsg, e);
|
logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "customInfo/view";
|
}
|
|
|
/**
|
* 查看该应用统计图导出excel
|
*
|
* @param model
|
* @param request
|
* @return
|
*/
|
@RequestMapping(value = "chartExcel")
|
public void chartExcel(Model model, HttpServletRequest request, HttpServletResponse response) {
|
String errorMsg = "自定义监控项统计图导出excel错误";
|
String id = request.getParameter("id");
|
String startTime = request.getParameter(StaticKeys.SEARCH_START_TIME);
|
String endTime = request.getParameter(StaticKeys.SEARCH_END_TIME);
|
String am = request.getParameter("am");
|
try {
|
if (StringUtils.isEmpty(id)) {
|
response.setContentType("text/html;charset=UTF-8");
|
response.getOutputStream().write(StaticKeys.EXCEL_PARAM_ERROR.getBytes());
|
return;
|
}
|
Map<String, Object> params = new HashMap<String, Object>();
|
params.put("customInfoId", id);
|
dashboardService.setDateParam(am, startTime, endTime, params, model);
|
excelExportService.exportCustomExcel(params, response);
|
} catch (Exception e) {
|
logger.error(errorMsg, e);
|
logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
}
|
|
/**
|
* 删除自定义监控项
|
*
|
* @param model
|
* @param request
|
* @param redirectAttributes
|
* @return
|
*/
|
@RequestMapping(value = "del")
|
public String delete(Model model, HttpServletRequest request, RedirectAttributes redirectAttributes) {
|
String errorMsg = "删除自定义监控项信息错误";
|
CustomInfo customInfo = new CustomInfo();
|
try {
|
if (!StringUtils.isEmpty(request.getParameter("id"))) {
|
String[] ids = request.getParameter("id").split(",");
|
for (String id : ids) {
|
customInfo = customInfoService.selectById(id);
|
customInfoService.saveLog(request, StaticKeys.LOG_DEL, customInfo);
|
}
|
customInfoService.deleteById(request.getParameter("id").split(","));
|
}
|
} catch (Exception e) {
|
logger.error(errorMsg, e);
|
logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
|
}
|
return "redirect:/customInfo/list";
|
}
|
|
/**
|
* 提取已配置的敏感字符串
|
*
|
* @return
|
*/
|
private void getBlockStr(Model model) {
|
//敏感字符组装 begin
|
String blockStr = "";
|
if (!StringUtils.isEmpty(commonConfig.getShellToRunLinuxBlock())) {
|
blockStr = "[linux不能包含敏感字符:" + commonConfig.getShellToRunLinuxBlock() + "]";
|
}
|
if (!StringUtils.isEmpty(commonConfig.getShellToRunLinuxBlock())) {
|
blockStr += "[windows不能包含敏感字符:" + commonConfig.getShellToRunWinBlock() + "]";
|
}
|
model.addAttribute("blockStr", blockStr);
|
//敏感字符组装 end
|
}
|
|
|
}
|
