package com.wgcloud.common;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import cn.hutool.json.JSONArray;
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.wgcloud.config.CommonConfig;
|
import com.wgcloud.dto.HostWarnDiyDto;
|
import com.wgcloud.util.UUIDUtil;
|
import com.wgcloud.util.msg.WarnPools;
|
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.boot.ApplicationArguments;
|
import org.springframework.boot.ApplicationRunner;
|
import org.springframework.boot.system.ApplicationHome;
|
import org.springframework.stereotype.Component;
|
|
import javax.servlet.ServletContext;
|
import java.io.File;
|
import java.nio.charset.Charset;
|
|
/**
|
* @version v3.3
|
* @ClassName:ApplicationContextHelper.java
|
* @author: http://www.wgstart.com
|
* @date: 2021年4月22日
|
* @Description: springboot 容器加载完成后执行一些初始化工作
|
* @Copyright: 2019-2021 wgcloud. All rights reserved.
|
*/
|
@Component
|
public class ApplicationStartListener implements ApplicationRunner {
|
private Logger logger = LoggerFactory.getLogger(ApplicationStartListener.class);
|
@Autowired
|
CommonConfig commonConfig;
|
@Autowired
|
private ServletContext servletContext;
|
|
@Override
|
public void run(ApplicationArguments args) throws Exception {
|
//自动闭合左侧菜单
|
servletContext.setAttribute("sidebarCollapse", commonConfig.getSidebarCollapse());
|
//是否开启使用主机机组
|
servletContext.setAttribute("hostGroup", commonConfig.getHostGroup());
|
//是否开启使用用户管理
|
servletContext.setAttribute("userInfoManage", commonConfig.getUserInfoManage());
|
//是否在列表页面显示告警次数
|
servletContext.setAttribute("showWarnCount", commonConfig.getShowWarnCount());
|
//每页显示多少条数据，读取配置文件中的pageSize
|
StaticKeys.PAGE_SIZE = commonConfig.getPageSize();
|
//初始化程序运行所在目录
|
ApplicationHome h = new ApplicationHome(getClass());
|
String jarPath = h.getSource().getParentFile().toString();
|
logger.debug("jar包路径-----------------" + jarPath);
|
StaticKeys.JAR_PATH = jarPath;
|
//初始化下告警缓存map，这样MEM_WARN_MAP就不为null
|
WarnPools.isExpireWarnTime("TEST", commonConfig.getWarnCacheTimes());
|
//初始化服务接口告警map，ping设备告警map，连续失败多少次再发送告警
|
WarnPools.initWarnCountMap();
|
//serverInfoId，只有master节点才给服务标识一个serverInfoId，只是标识，无其他作用
|
if (StaticKeys.NODE_MASTER.equals(commonConfig.getNodeType())) {
|
servletContext.setAttribute("serverInfoId", UUIDUtil.getUUID());
|
}
|
try {
|
//解析主机告警个性化配置json begin
|
String path = System.getProperty("user.dir");
|
File file = new File(path + "/config/hostWarnDiy.json");
|
if (file.exists()) {
|
logger.info("发现并解析hostWarnDiy.json");
|
JSONArray hostJsonArray = JSONUtil.readJSONArray(file, Charset.forName("utf-8"));
|
if (!CollectionUtil.isEmpty(hostJsonArray)) {
|
for (int i = 0; i < hostJsonArray.size(); i++) {
|
JSONObject jsonObject = JSONUtil.parseObj(hostJsonArray.get(i));
|
HostWarnDiyDto hostWarnDto = JSONUtil.toBean(jsonObject, HostWarnDiyDto.class);
|
if (StringUtils.isEmpty(hostWarnDto.getHostname())) {
|
continue;
|
}
|
StaticKeys.HOST_WARN_MAP.put(hostWarnDto.getHostname(), hostWarnDto);
|
}
|
}
|
}
|
//解析主机告警个性化配置json end
|
} catch (Exception e) {
|
logger.error("hostWarnDiy.json解析错误：", e);
|
}
|
|
try {
|
servletContext.setAttribute("webSsh", commonConfig.getWebSsh());
|
if (StaticKeys.TRUE_VAL.equals(commonConfig.getWebSsh())) {
|
logger.info("NettyServer服务启动，端口：" + commonConfig.getWebSshPort());
|
NettyServer.start(commonConfig.getWebSshPort());
|
}
|
} catch (Exception e) {
|
logger.error("NettyServer服务启动错误：", e);
|
}
|
|
|
}
|
|
|
}
|
