org.dromara.jpom.func.system.service.ClusterInfoService Maven / Gradle / Ivy
/*
* Copyright (c) 2019 Of Him Code Technology Studio
* Jpom is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/
package org.dromara.jpom.func.system.service;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.SystemClock;
import cn.hutool.core.net.NetUtil;
import cn.hutool.core.net.url.UrlBuilder;
import cn.hutool.core.util.StrUtil;
import cn.hutool.db.Entity;
import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.Method;
import cn.keepbx.jpom.event.IAsyncLoad;
import cn.keepbx.jpom.model.JsonMessage;
import com.alibaba.fastjson2.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.dromara.jpom.JpomApplication;
import org.dromara.jpom.common.JpomManifest;
import org.dromara.jpom.common.ServerConst;
import org.dromara.jpom.common.i18n.I18nMessageUtil;
import org.dromara.jpom.configuration.ClusterConfig;
import org.dromara.jpom.cron.CronUtils;
import org.dromara.jpom.func.assets.server.MachineDockerServer;
import org.dromara.jpom.func.assets.server.MachineNodeServer;
import org.dromara.jpom.func.assets.server.MachineSshServer;
import org.dromara.jpom.func.system.model.ClusterInfoModel;
import org.dromara.jpom.model.user.UserModel;
import org.dromara.jpom.service.h2db.BaseDbService;
import org.dromara.jpom.service.system.WorkspaceService;
import org.dromara.jpom.system.ServerConfig;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
/**
* @author bwcx_jzy
* @since 2023/8/19
*/
@Service
@Slf4j
public class ClusterInfoService extends BaseDbService implements IAsyncLoad, Runnable {
private final ClusterConfig clusterConfig;
private static final String TASK_ID = "system_monitor_cluster";
private final WorkspaceService workspaceService;
/**
* 是否为多集群
*/
private boolean multiServer = false;
public ClusterInfoService(ServerConfig serverConfig,
WorkspaceService workspaceService) {
this.clusterConfig = serverConfig.getCluster();
this.workspaceService = workspaceService;
}
/**
* 获取当前集群
*
* @return 集群信息
*/
public ClusterInfoModel getCurrent() {
ClusterInfoModel clusterInfoModel = this.getByKey(JpomManifest.getInstance().getInstallId());
Assert.notNull(clusterInfoModel, I18nMessageUtil.get("i18n.cluster_does_not_exist.97a4"));
return clusterInfoModel;
}
/**
* 是否为多服务，集群模式
*
* @return true
*/
public boolean isMultiServer() {
return multiServer;
}
@Override
public void startLoad() {
// 启动心跳检测
int heartSecond = clusterConfig.getHeartSecond();
ScheduledExecutorService scheduler = JpomApplication.getScheduledExecutorService();
scheduler.scheduleWithFixedDelay(this, 0, heartSecond, TimeUnit.SECONDS);
// 判断是否为多集群模式
this.multiServer = this.count() > 1;
}
@Override
public void run() {
int heartSecond = clusterConfig.getHeartSecond();
try {
CronUtils.TaskStat taskStat = CronUtils.getTaskStat(TASK_ID, StrUtil.format(I18nMessageUtil.get("i18n.execution_frequency.d014"), heartSecond));
taskStat.onStart();
// 判断是否为多集群模式
this.multiServer = this.count() > 1;
//
JpomManifest jpomManifest = JpomManifest.getInstance();
String installId = jpomManifest.getInstallId();
ClusterInfoModel byKey = this.getByKey(installId);
if (byKey == null) {
// 初始安装
this.insert(this.createDefault(installId));
// 自动绑定默认数据
this.bindDefault(installId);
return;
}
// 更新数据
ClusterInfoModel clusterInfoModel = new ClusterInfoModel();
clusterInfoModel.setId(byKey.getId());
if (!StrUtil.equals(byKey.getClusterId(), clusterConfig.getId())) {
log.warn(I18nMessageUtil.get("i18n.cluster_id_changed.6e49"), byKey.getClusterId(), clusterConfig.getId());
clusterInfoModel.setClusterId(clusterConfig.getId());
}
clusterInfoModel.setLocalHostName(NetUtil.getLocalHostName());
clusterInfoModel.setLastHeartbeat(SystemClock.now());
clusterInfoModel.setJpomVersion(jpomManifest.getVersion());
// 测试
try {
String url = byKey.getUrl();
if (StrUtil.isEmpty(url)) {
clusterInfoModel.setStatusMsg(I18nMessageUtil.get("i18n.address_not_configured.f2eb"));
} else {
this.testUrl(url);
clusterInfoModel.setStatusMsg("OK");
}
} catch (Exception e) {
clusterInfoModel.setStatusMsg(e.getMessage());
}
this.updateById(clusterInfoModel);
// 检查是否重复
Entity entity = Entity.create();
entity.set("clusterId", clusterConfig.getId());
entity.set("id", StrUtil.format(" <> {}", installId));
List clusterInfoModels = this.listByEntity(entity);
if (CollUtil.isNotEmpty(clusterInfoModels)) {
for (ClusterInfoModel infoModel : clusterInfoModels) {
log.error(I18nMessageUtil.get("i18n.cluster_id_conflict.45b7"), clusterConfig.getId(), infoModel.getId(), infoModel.getName());
}
}
// 通知任务结束
taskStat.onSucceeded();
} catch (Throwable throwable) {
CronUtils.TaskStat taskStat = CronUtils.getTaskStat(TASK_ID, StrUtil.format(I18nMessageUtil.get("i18n.execution_frequency.d014"), heartSecond));
taskStat.onFailed(TASK_ID, throwable);
}
}
private void testUrl(String url) {
//
UrlBuilder urlBuilder = UrlBuilder.ofHttp(url);
urlBuilder.addPath(ServerConst.CHECK_SYSTEM);
HttpRequest httpRequest = HttpRequest.of(urlBuilder).timeout(30 * 1000).method(Method.GET);
try {
JSONObject jsonObject = httpRequest.thenFunction(httpResponse -> {
String body = httpResponse.body();
return JSONObject.parseObject(body);
});
int code = jsonObject.getIntValue(JsonMessage.CODE);
Assert.state(code == JsonMessage.DEFAULT_SUCCESS_CODE, () -> {
String msg = jsonObject.getString(JsonMessage.MSG);
msg = StrUtil.emptyToDefault(msg, jsonObject.toString());
return StrUtil.format(I18nMessageUtil.get("i18n.cluster_status_code_exception.9d89"), code, msg);
});
//
JSONObject data = jsonObject.getJSONObject("data");
Assert.notNull(data, I18nMessageUtil.get("i18n.cluster_response_incorrect.c08a"));
boolean expression = data.containsKey("routerBase") && data.containsKey("extendPlugins");
Assert.state(expression, I18nMessageUtil.get("i18n.incorrect_cluster_address.893f"));
} catch (Exception e) {
log.error(I18nMessageUtil.get("i18n.check_cluster_info_exception.7b0c"), e);
throw new IllegalArgumentException(I18nMessageUtil.get("i18n.cluster_address_check_exception.cd92") + e.getMessage());
}
}
/**
* 自动帮忙集群相关的默认数据
*
* @param installId 安装Id
*/
private void bindDefault(String installId) {
long count = this.count();
if (count != 1) {
log.debug(I18nMessageUtil.get("i18n.multiple_clusters_exist.196b"));
return;
}
// 所以工作空间自动绑定集群Id
String sql = "update " + workspaceService.getTableName() + " set clusterInfoId=?";
workspaceService.execute(sql, installId);
// 获取所有的资产分组
List list = this.listLinkGroups();
String join = CollUtil.join(list, StrUtil.COMMA);
//
ClusterInfoModel clusterInfoModel = new ClusterInfoModel();
clusterInfoModel.setId(installId);
clusterInfoModel.setLinkGroup(join);
this.updateById(clusterInfoModel);
}
/**
* 查询集群可以管理的分组名
*
* @return list
*/
public List listLinkGroups() {
MachineDockerServer machineDockerServer = SpringUtil.getBean(MachineDockerServer.class);
MachineNodeServer machineNodeServer = SpringUtil.getBean(MachineNodeServer.class);
MachineSshServer machineSshServer = SpringUtil.getBean(MachineSshServer.class);
List nodeGroup = machineNodeServer.listGroupName();
List sshGroup = machineSshServer.listGroupName();
List dockerGroup = machineDockerServer.listGroupName();
//
List all = new ArrayList<>();
CollUtil.addAll(all, nodeGroup);
CollUtil.addAll(all, sshGroup);
CollUtil.addAll(all, dockerGroup);
//
all = all.stream()
.distinct()
.collect(Collectors.toList());
return all;
}
/**
* 创建默认的集群数据
*
* @param installId 系统安装 id
* @return 默认数据
*/
private ClusterInfoModel createDefault(String installId) {
ClusterInfoModel clusterInfoModel = new ClusterInfoModel();
clusterInfoModel.setId(installId);
clusterInfoModel.setName(I18nMessageUtil.get("i18n.default_cluster.38cf"));
clusterInfoModel.setCreateUser(UserModel.SYSTEM_ADMIN);
clusterInfoModel.setClusterId(clusterConfig.getId());
clusterInfoModel.setLastHeartbeat(SystemClock.now());
return clusterInfoModel;
}
/**
* 判断集群是否在线
*
* @param clusterInfoModel 集群信息
* @return true 在线
*/
public boolean online(ClusterInfoModel clusterInfoModel) {
if (clusterInfoModel == null) {
return false;
}
Long lastHeartbeat = clusterInfoModel.getLastHeartbeat();
if (lastHeartbeat == null) {
return false;
}
long millis = TimeUnit.SECONDS.toMillis(clusterConfig.getHeartSecond());
return lastHeartbeat > SystemClock.now() - millis;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy