package com.wgcloud.controller;
|
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.wgcloud.entity.CustomInfo;
|
import com.wgcloud.entity.CustomState;
|
import com.wgcloud.service.CustomInfoService;
|
import com.wgcloud.service.DockerInfoService;
|
import com.wgcloud.service.LogInfoService;
|
import com.wgcloud.service.SystemInfoService;
|
import com.wgcloud.util.FormatUtil;
|
import com.wgcloud.util.ThreadPoolUtil;
|
import com.wgcloud.util.TokenUtils;
|
import com.wgcloud.util.msg.WarnMailUtil;
|
import com.wgcloud.util.staticvar.BatchData;
|
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.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
import javax.annotation.Resource;
|
import java.util.Date;
|
import java.util.Iterator;
|
|
/**
|
* @version v3.3
|
* @ClassName:AgentCustomGoController.java
|
* @author: http://www.wgstart.com
|
* @date: 2022年9月16日
|
* @Description: 自定义监控项处理，默认每10分钟提交一次
|
* @Copyright: 2019-2022 wgcloud. All rights reserved.
|
*/
|
@Controller
|
@RequestMapping("/agentCustomGo")
|
public class AgentCustomGoController {
|
|
|
private static final Logger logger = LoggerFactory.getLogger(AgentCustomGoController.class);
|
|
|
@Resource
|
private LogInfoService logInfoService;
|
@Resource
|
private SystemInfoService systemInfoService;
|
@Autowired
|
private DockerInfoService dockerInfoService;
|
@Autowired
|
private CustomInfoService customInfoService;
|
@Autowired
|
private TokenUtils tokenUtils;
|
|
@ResponseBody
|
@RequestMapping("/minTask")
|
public JSONObject minTask(@RequestBody String paramBean) {
|
JSONObject agentJsonObject = (JSONObject) JSONUtil.parse(paramBean);
|
logger.debug("agent上报自定义监控项数据-------------" + agentJsonObject.toString());
|
JSONObject resultJson = new JSONObject();
|
if (!tokenUtils.checkAgentToken(agentJsonObject)) {
|
logger.error(StaticKeys.TOKEN_ERROR);
|
resultJson.set("result", StaticKeys.TOKEN_ERROR);
|
return resultJson;
|
}
|
Date nowtime = new Date();
|
|
try {
|
JSONObject customInfosJson = agentJsonObject.getJSONObject("customInfos");
|
if (customInfosJson == null) {
|
logger.error("customInfoList is null");
|
resultJson.set("result", "error：customInfoList is null");
|
return resultJson;
|
}
|
Iterator iter = customInfosJson.keySet().iterator();
|
while (iter.hasNext()) {
|
String id = (String) iter.next();
|
JSONObject jsonObject = customInfosJson.getJSONObject(id);
|
//监控项执行结果值
|
String customValue = jsonObject.getStr("customValue");
|
|
CustomState state = new CustomState();
|
state.setCreateTime(nowtime);
|
state.setCustomInfoId(id);
|
state.setCustomValue(customValue);
|
if (!StringUtils.isEmpty(customValue)) {
|
BatchData.CUSTOM_STATE_LIST.add(state);
|
}
|
|
CustomInfo info = new CustomInfo();
|
info.setId(id);
|
info.setCreateTime(nowtime);
|
info.setCustomValue(customValue);
|
info.setState(StaticKeys.ON_STATE);
|
BatchData.CUSTOM_INFO_LIST.add(info);
|
|
//判断告警表达式是否成立，发邮件 begin
|
if (!StringUtils.isEmpty(customValue)) {
|
Runnable runnable = () -> {
|
try {
|
CustomInfo customInfo = customInfoService.selectById(info.getId());
|
if (customInfo != null) {
|
Double customValueDouble = Double.valueOf(customValue);
|
Boolean result = FormatUtil.validateExpression(customInfo.getResultExp(), customValueDouble);
|
if (result) {
|
info.setState(StaticKeys.DOWN_STATE);
|
customInfo.setCustomValue(customValue);
|
WarnMailUtil.sendCustomInfoDown(customInfo, customInfo.getAccount());
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
|
};
|
ThreadPoolUtil.executor.execute(runnable);
|
}
|
//判断告警表达式是否成立，发邮件 end
|
}
|
|
resultJson.set("result", "success");
|
} catch (Exception e) {
|
logger.error("解析自定义监控项上报数据错误", e);
|
resultJson.set("result", "error：" + e.toString());
|
}
|
return resultJson;
|
}
|
|
|
}
|
