// 当引擎被批准时，其路径也应该随着用户设定的新路径发生了变化，修改之
// Topo节点也会自动随之而修改
void autoUpdateNodesForEngine(String legacyPath, String path) {
if (StringUtils.isNotBlank(path) && !StringUtils.equals(legacyPath, path)) {
try {
nodeService.updateNodesPath(legacyPath, path);
} catch (NodeException e) {
logger.warn(e.getMessage(), e);
}
}
}
@Override
public MonitorEngine register(MonitorEngine engine) throws ResourceException {
// 先执行自动批准动作
performApproval(engine);
// 如果是从监控服务器下载下来的引擎，关联了相应的engine id，则直接更新
if (StringUtils.isNotBlank(engine.getEngineId())) {
MonitorEngine legacy = findByEngineId(engine.getEngineId());
// id/name/label 保留数据库中的值
engine.setId(legacy.getId());
engine.setName(legacy.getName());
engine.setLabel(legacy.getLabel());
// properties 合并
if (legacy.getProperties() != null) {
if (engine.getProperties() == null) {
engine.setProperties(legacy.getProperties());
} else {
engine.getProperties().putAll(legacy.getProperties());
}
}
return update(legacy, engine);
} else if (isDefault(engine)) { // 还有一种常见情况，就是缺省引擎（与监控服务器安装在同一个地址)，而后没有配置engineId
MonitorEngine defaultEngine = getRepository().findDefaultEngine();
engine.setEngineId(defaultEngine.getEngineId());
return update(defaultEngine, engine);
} else {
// 否则创建一个新的实例
return create(engine);
}
}
/**
*
*
* <h2>实际执行添加引擎的动作</h2>
*
* <p>有如下添加方式（对应两种情况）：
*
* <ol>
* <li>管理员通过界面主动新建一个监控引擎（而后由脚本安装） 用户输入了如下信息
* <ul>
* <li>引擎路径（安装目标路径）
* <li>引擎地址（也是主机地址）
* <li>主机类型
* <li>主机登录信息
* </ul>
* 程序在本步骤保持新的引擎对象以及其所在主机信息之后，后继应该使用该主机信息，将引擎安装到相应目录去
* <li>用户在特定机器上安装引擎（而后引擎启动时，通过监控服务器SPI自动注册）
* <ul>
* <li>引擎路径（安装目标路径）
* <li>引擎地址（也是主机地址）
* <li>进程pids
* <li>主机类型
* </ul>
* </ol>
*
* 开发者可以通过监控引擎是否有pids值判断是否是界面输入的监控引擎
*
* @param engine 新注册的引擎对象
* @throws ResourceException
*/
@Override
protected void performCreate(MonitorEngine engine) throws ResourceException {
boolean manual;
manual = engine.getPids() == null || engine.getPids().isEmpty();
// 新建的引擎，没有真的引擎实例与之关联，所以，必定 不可用, 待批准
engine.setEngineId(UUID.randomUUID().toString());
engine.setAvailability(Availability.Unavailable);
engine.setConfigStatus(ConfigStatus.Unknown);
engine.setPerformance(Performance.Unknown);
engine.setApproveStatus(ApproveStatus.Requested);
// 没有预先创建，自动注册的监控引擎没有name，需要为其生成
// @see dnt.monitor.engine.support.IdentityManager#createLocalEngine(String)
if (StringUtils.isBlank(engine.getName())) {
int maxPending = getRepository().countMaxPending();
maxPending++;
engine.setLabel("新监控引擎#" + maxPending);
String name = "pending_" + maxPending;
engine.setName(name);
engine.setApiToken(null); // 防止提交进来
}
// 本函数中没有包括创建该引擎所在主机，这个动作是在引擎创建后，
// 由 SetupEngineAfterCreation这个handler完成
logger.debug(
"An engine from {} created in {} mode", engine.getAddress(), (manual ? "manual" : "auto"));
super.performCreate(engine);
}
protected boolean isDefault(MonitorEngine engine) {
// 判断引擎的名称，这个是由管理员自行设置的
if (engine.isDefault()) return true;
// 在某些本机场景下, server address = localhost, engine address = 192.168.12.63
SocketAddress serverAddr = IpUtils.parseSocketAddress(serverService.getServer().getAddress());
SocketAddress engineAddr = IpUtils.parseSocketAddress(engine.getAddress());
return StringUtils.equals(serverAddr.getHost(), engineAddr.getHost());
}
@Override
protected void validateOnUpdate(MonitorEngine exist, MonitorEngine resource)
throws ValidationException {
super.validateOnUpdate(exist, resource);
// IP地址变了
if (!StringUtils.equals(exist.getAddress(), resource.getAddress())) {
MonitorEngine another = findByAddress(resource.getAddress());
if (another != null) {
throw new ValidationException(
"There is another engine with address " + resource.getAddress());
}
}
}
@Override
public BoundaryCronTrigger create(TimeInterval frequency, Schedule schedule, TimeSpan window) {
if (frequency.getMilliseconds() % 15 * DateUtils.MILLIS_PER_SECOND != 0) {
throw new IllegalArgumentException(
"The frequency " + frequency + " can't be divided by system minimal interval(15s)");
}
int pos, val;
String interval = frequency.getInterval();
if (interval.contains("h")) {
pos = 2;
val = Integer.valueOf(interval.substring(0, interval.indexOf('h')));
} else if (interval.contains("m")) {
pos = 1;
val = Integer.valueOf(interval.substring(0, interval.indexOf('m')));
} else if (interval.contains("s")) {
pos = 0;
val = Integer.valueOf(interval.substring(0, interval.indexOf('s')));
} else {
throw new IllegalArgumentException(
"The interval " + interval + " is invalid, we only support h/m/s");
}
// 如果连系统都没有设置默认的监控的频度，那么我们默认设置为所有时间
String[] cron =
schedule == null
? new String[] {"*", "*", "*", "*", "*", "*"}
: schedule.getCron().split("\\s");
Granularity granularity = getGranularity(frequency);
// 不同的频度，还需要削峰填谷
// 在频度之前的每个字段，先归置为0
for (int i = 0; i < pos; i++) {
cron[i] = granularity.getSegment(i);
}
// 在小时这里，schedule和frequency的偏移量可能出现冲突
// 原则是，保留 schedule的限制
cron[pos] = granularity.getSegment(pos) + "/" + val;
// 向上计数，下个该间隔的计划自然向上偏移
granularity.stepUp();
String expression = StringUtils.join(cron, " ");
BoundaryCronTrigger trigger = new BoundaryCronTrigger(expression);
trigger.include(schedule);
trigger.exclude(window);
return trigger;
}
