package com.wgcloud.controller;
|
|
import cn.hutool.json.JSONArray;
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.wgcloud.config.CommonConfig;
|
import com.wgcloud.entity.DbInfo;
|
import com.wgcloud.entity.DbTable;
|
import com.wgcloud.entity.DbTableCount;
|
import com.wgcloud.service.DbInfoService;
|
import com.wgcloud.service.DbTableService;
|
import com.wgcloud.service.LogInfoService;
|
import com.wgcloud.util.ResDataUtils;
|
import com.wgcloud.util.ThreadPoolUtil;
|
import com.wgcloud.util.TokenUtils;
|
import com.wgcloud.util.msg.WarnMailUtil;
|
import com.wgcloud.util.msg.WarnPools;
|
import com.wgcloud.util.staticvar.BatchData;
|
import com.wgcloud.util.staticvar.StaticKeys;
|
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.*;
|
|
/**
|
* @version v3.3
|
* @ClassName:AgentDbTableGoController.java
|
* @author: http://www.wgstart.com
|
* @date: 2022年9月16日
|
* @Description: server-backup监控数据表提交处理，默认每60分钟提交一次
|
* @Copyright: 2019-2022 wgcloud. All rights reserved.
|
*/
|
@Controller
|
@RequestMapping("/agentDbTableGo")
|
public class AgentDbTableGoController {
|
|
|
private static final Logger logger = LoggerFactory.getLogger(AgentDbTableGoController.class);
|
|
|
@Resource
|
private LogInfoService logInfoService;
|
@Autowired
|
private DbInfoService dbInfoService;
|
@Autowired
|
private DbTableService dbTableService;
|
@Autowired
|
private TokenUtils tokenUtils;
|
@Autowired
|
private CommonConfig commonConfig;
|
|
@ResponseBody
|
@RequestMapping("/minTask")
|
public String minTask(@RequestBody String paramBean) {
|
JSONObject agentJsonObject = (JSONObject) JSONUtil.parse(paramBean);
|
logger.debug("server-backup监控数据表上报数据-------------" + agentJsonObject.toString());
|
if (!tokenUtils.checkAgentToken(agentJsonObject)) {
|
logger.error(StaticKeys.TOKEN_ERROR);
|
return ResDataUtils.resetErrorJson(StaticKeys.TOKEN_ERROR);
|
}
|
Date nowtime = new Date();
|
|
try {
|
//数据源状态数据
|
JSONArray dbInfosJsonArr = agentJsonObject.getJSONArray("dbInfosUpdate");
|
//数据表状态数据
|
JSONArray dbTablesJsonArr = agentJsonObject.getJSONArray("dbTablesUpdate");
|
if (dbInfosJsonArr == null) {
|
logger.error("dbInfosUpdate is null");
|
return ResDataUtils.resetErrorJson("dbInfosUpdate is null");
|
}
|
//处理数据源状态数据
|
List<DbInfo> dbInfoList = JSONUtil.toList(dbInfosJsonArr, DbInfo.class);
|
for (DbInfo dbInfo : dbInfoList) {
|
//之前存贮的数据源旧数据
|
DbInfo dbInfoSaved = dbInfoService.selectById(dbInfo.getId());
|
//只有数据源在线时候，才更新时间，连接失败就不更新时间
|
if (StaticKeys.ON_STATE.equals(dbInfo.getDbState())) {
|
dbInfo.setCreateTime(nowtime);
|
if (null != WarnPools.MEM_WARN_MAP && null != WarnPools.MEM_WARN_MAP.get(dbInfo.getId())) {
|
Runnable runnable = () -> {
|
WarnMailUtil.sendDbDown(dbInfoSaved, false);
|
};
|
ThreadPoolUtil.executor.execute(runnable);
|
}
|
} else {
|
Runnable runnable = () -> {
|
WarnMailUtil.sendDbDown(dbInfoSaved, true);
|
};
|
ThreadPoolUtil.executor.execute(runnable);
|
}
|
dbInfoService.updateById(dbInfo);
|
}
|
|
//处理数据表上报数据
|
if (null != dbTablesJsonArr) {
|
List<DbTable> dbTableList = JSONUtil.toList(dbTablesJsonArr, DbTable.class);
|
for (DbTable dbTable : dbTableList) {
|
dbTable.setCreateTime(nowtime);
|
|
DbTableCount dbTableCount = new DbTableCount();
|
dbTableCount.setCreateTime(nowtime);
|
dbTableCount.setDbTableId(dbTable.getId());
|
dbTableCount.setTableCount(dbTable.getTableCount());
|
BatchData.DBTABLE_COUNT_LIST.add(dbTableCount);
|
}
|
Map<String, Object> params = new HashMap<>();
|
List<DbInfo> dbInfos = new ArrayList<>();
|
if (StaticKeys.TRUE_VAL.equals(commonConfig.getUserInfoManage())) {
|
dbInfos = dbInfoService.selectAllByParams(params);
|
}
|
dbTableService.warnCheckExp(dbTableList, dbInfos);
|
dbTableService.updateRecord(dbTableList);
|
}
|
|
} catch (Exception e) {
|
logger.error("解析server-backup监控数据表上报数据错误", e);
|
return ResDataUtils.resetErrorJson(e.toString());
|
}
|
return ResDataUtils.resetSuccessJson(null);
|
}
|
|
|
}
|
