mirror of https://github.com/longtai-cn/hippo4j
imyzt
3 years ago
parent
5494c035c6
commit
7915df10a4
@ -0,0 +1,30 @@
|
||||
package cn.hippo4j.starter.alarm.lark;
|
||||
|
||||
/**
|
||||
* lark alarm constants.
|
||||
*
|
||||
* @author imyzt
|
||||
* @date 2021-11-23 19:29
|
||||
*/
|
||||
public class LarkAlarmConstants {
|
||||
|
||||
/**
|
||||
* lark bot url
|
||||
*/
|
||||
public static final String LARK_BOT_URL = "https://open.feishu.cn/open-apis/bot/v2/hook/";
|
||||
|
||||
/**
|
||||
* lark 报警 json文件路径
|
||||
*/
|
||||
public static final String ALARM_JSON_PATH = "classpath:properties/lark/alarm.json";
|
||||
|
||||
/**
|
||||
* lark 配置变更通知 json文件路径
|
||||
*/
|
||||
public static final String NOTICE_JSON_PATH = "classpath:properties/lark/notice.json";
|
||||
|
||||
/**
|
||||
* lark at format
|
||||
*/
|
||||
public static final String LARK_AT_FORMAT = "<at email=''>%s</at>";
|
||||
}
|
||||
@ -0,0 +1,187 @@
|
||||
package cn.hippo4j.starter.alarm.lark;
|
||||
|
||||
import cn.hippo4j.common.model.InstanceInfo;
|
||||
import cn.hippo4j.common.model.PoolParameterInfo;
|
||||
import cn.hippo4j.starter.alarm.NotifyDTO;
|
||||
import cn.hippo4j.starter.alarm.NotifyPlatformEnum;
|
||||
import cn.hippo4j.starter.alarm.SendMessageHandler;
|
||||
import cn.hippo4j.starter.core.DynamicThreadPoolExecutor;
|
||||
import cn.hippo4j.starter.core.GlobalThreadPoolManage;
|
||||
import cn.hippo4j.starter.toolkit.thread.QueueTypeEnum;
|
||||
import cn.hippo4j.starter.toolkit.thread.RejectedTypeEnum;
|
||||
import cn.hippo4j.starter.wrapper.DynamicThreadPoolWrapper;
|
||||
import cn.hutool.core.date.DateUtil;
|
||||
import cn.hutool.core.io.FileUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.http.HttpRequest;
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.SneakyThrows;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.util.ResourceUtils;
|
||||
|
||||
import java.io.FileNotFoundException;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.Arrays;
|
||||
import java.util.concurrent.BlockingQueue;
|
||||
import java.util.concurrent.ThreadPoolExecutor;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import static cn.hippo4j.starter.alarm.lark.LarkAlarmConstants.*;
|
||||
|
||||
/**
|
||||
* Send lark notification message.
|
||||
*
|
||||
* @author imyzt
|
||||
* @date 2021/11/22 21:12
|
||||
*/
|
||||
@Slf4j
|
||||
@AllArgsConstructor
|
||||
public class LarkSendMessageHandler implements SendMessageHandler {
|
||||
|
||||
private final String active;
|
||||
|
||||
private final InstanceInfo instanceInfo;
|
||||
|
||||
@Override
|
||||
public String getType() {
|
||||
return NotifyPlatformEnum.LARK.name();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendAlarmMessage(NotifyDTO notifyConfig, DynamicThreadPoolExecutor pool) {
|
||||
larkAlarmSendMessage(notifyConfig, pool);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendChangeMessage(NotifyDTO notifyConfig, PoolParameterInfo parameter) {
|
||||
larkChangeSendMessage(notifyConfig, parameter);
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
private void larkAlarmSendMessage(NotifyDTO notifyConfig, DynamicThreadPoolExecutor pool) {
|
||||
String afterReceives = getReceives(notifyConfig);
|
||||
|
||||
BlockingQueue<Runnable> queue = pool.getQueue();
|
||||
|
||||
String larkAlarmJson = getLarkJson(ALARM_JSON_PATH);
|
||||
|
||||
String text = String.format(larkAlarmJson,
|
||||
// 环境
|
||||
active.toUpperCase(),
|
||||
// 线程池ID
|
||||
pool.getThreadPoolId(),
|
||||
// 应用名称
|
||||
instanceInfo.getAppName(),
|
||||
// 实例信息
|
||||
instanceInfo.getIdentify(),
|
||||
// 报警类型
|
||||
notifyConfig.getTypeEnum(),
|
||||
// 核心线程数
|
||||
pool.getCorePoolSize(),
|
||||
// 最大线程数
|
||||
pool.getMaximumPoolSize(),
|
||||
// 当前线程数
|
||||
pool.getPoolSize(),
|
||||
// 活跃线程数
|
||||
pool.getActiveCount(),
|
||||
// 最大任务数
|
||||
pool.getLargestPoolSize(),
|
||||
// 线程池任务总量
|
||||
pool.getCompletedTaskCount(),
|
||||
// 队列类型名称
|
||||
queue.getClass().getSimpleName(),
|
||||
// 队列容量
|
||||
queue.size() + queue.remainingCapacity(),
|
||||
// 队列元素个数
|
||||
queue.size(),
|
||||
// 队列剩余个数
|
||||
queue.remainingCapacity(),
|
||||
// 拒绝策略名称
|
||||
pool.getRejectedExecutionHandler().getClass().getSimpleName(),
|
||||
// 拒绝策略次数
|
||||
pool.getRejectCount(),
|
||||
// 告警姓名
|
||||
afterReceives,
|
||||
// 当前时间
|
||||
DateUtil.now(),
|
||||
// 报警频率
|
||||
notifyConfig.getInterval()
|
||||
);
|
||||
|
||||
execute(notifyConfig, text);
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
private void larkChangeSendMessage(NotifyDTO notifyConfig, PoolParameterInfo parameter) {
|
||||
String threadPoolId = parameter.getTpId();
|
||||
DynamicThreadPoolWrapper poolWrap = GlobalThreadPoolManage.getExecutorService(threadPoolId);
|
||||
if (poolWrap == null) {
|
||||
log.warn("Thread pool is empty when sending change notification, threadPoolId :: {}", threadPoolId);
|
||||
return;
|
||||
}
|
||||
|
||||
String afterReceives = getReceives(notifyConfig);
|
||||
|
||||
ThreadPoolExecutor customPool = poolWrap.getExecutor();
|
||||
|
||||
String larkNoticeJson = getLarkJson(NOTICE_JSON_PATH);
|
||||
|
||||
/**
|
||||
* hesitant e.g. ➲ ➜ ⇨ ➪
|
||||
*/
|
||||
String text = String.format(larkNoticeJson,
|
||||
// 环境
|
||||
active.toUpperCase(),
|
||||
// 线程池名称
|
||||
threadPoolId,
|
||||
// 应用名称
|
||||
instanceInfo.getAppName(),
|
||||
// 实例信息
|
||||
instanceInfo.getIdentify(),
|
||||
// 核心线程数
|
||||
customPool.getCorePoolSize() + " ➲ " + parameter.getCoreSize(),
|
||||
// 最大线程数
|
||||
customPool.getMaximumPoolSize() + " ➲ " + parameter.getMaxSize(),
|
||||
// 线程存活时间
|
||||
customPool.getKeepAliveTime(TimeUnit.SECONDS) + " ➲ " + parameter.getKeepAliveTime(),
|
||||
// 阻塞队列
|
||||
QueueTypeEnum.getBlockingQueueNameByType(parameter.getQueueType()),
|
||||
// 阻塞队列容量
|
||||
(customPool.getQueue().size() + customPool.getQueue().remainingCapacity()) + " ➲ " + parameter.getCapacity(),
|
||||
// 拒绝策略
|
||||
customPool.getRejectedExecutionHandler().getClass().getSimpleName(),
|
||||
RejectedTypeEnum.getRejectedNameByType(parameter.getRejectedType()),
|
||||
// 告警姓名
|
||||
afterReceives,
|
||||
// 当前时间
|
||||
DateUtil.now()
|
||||
);
|
||||
|
||||
execute(notifyConfig, text);
|
||||
}
|
||||
|
||||
private String getLarkJson(String filePath) throws FileNotFoundException {
|
||||
return FileUtil.readString(ResourceUtils.getFile(filePath), StandardCharsets.UTF_8);
|
||||
}
|
||||
|
||||
private String getReceives(NotifyDTO notifyConfig) {
|
||||
if (StrUtil.isBlank(notifyConfig.getReceives())) {
|
||||
return "";
|
||||
}
|
||||
return Arrays.stream(notifyConfig.getReceives().split(","))
|
||||
.map(receive -> String.format(LARK_AT_FORMAT, receive))
|
||||
.collect(Collectors.joining(" "));
|
||||
}
|
||||
|
||||
private void execute(NotifyDTO notifyConfig, String text) {
|
||||
String serverUrl = LARK_BOT_URL + notifyConfig.getSecretKey();
|
||||
|
||||
try {
|
||||
HttpRequest.post(serverUrl).body(text).execute();
|
||||
} catch (Exception ex) {
|
||||
log.error("Lark failed to send message", ex);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue