Pre Merge pull request !36 from venus-suite/master
venus-suite
4 years ago
committed by
Gitee
commit
8ab81b91e3
@ -0,0 +1,47 @@
|
||||
package com.xxl.job.admin.core.alarm;
|
||||
|
||||
import com.xxl.job.admin.core.util.I18nUtil;
|
||||
|
||||
/**
|
||||
* @author tomj2ee 2022-02-20 11:11:23
|
||||
*/
|
||||
public enum AlarmTypeEnum {
|
||||
/**
|
||||
* 邮件
|
||||
*/
|
||||
EMAIL(1,I18nUtil.getString("jobinfo_field_email")),
|
||||
|
||||
/**
|
||||
* webhook
|
||||
*/
|
||||
WEBHOOK(2,I18nUtil.getString("jobinfo_field_webhook"));
|
||||
|
||||
|
||||
|
||||
private String title;
|
||||
private int alarmType;
|
||||
|
||||
AlarmTypeEnum(int alarmType,String title) {
|
||||
this.alarmType=alarmType;
|
||||
this.title = title;
|
||||
}
|
||||
|
||||
public String getTitle() {
|
||||
return title;
|
||||
}
|
||||
|
||||
public int getAlarmType() {
|
||||
return alarmType;
|
||||
}
|
||||
|
||||
|
||||
public static AlarmTypeEnum match(String name, AlarmTypeEnum defaultItem){
|
||||
for (AlarmTypeEnum item: AlarmTypeEnum.values()) {
|
||||
if (item.name().equals(name)) {
|
||||
return item;
|
||||
}
|
||||
}
|
||||
return defaultItem;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,144 @@
|
||||
package com.xxl.job.admin.core.alarm.impl;
|
||||
|
||||
import com.xxl.job.admin.core.alarm.AlarmTypeEnum;
|
||||
import com.xxl.job.admin.core.alarm.JobAlarm;
|
||||
import com.xxl.job.admin.core.alarm.msg.FeiShuTextMsgReq;
|
||||
import com.xxl.job.admin.core.alarm.msg.TextMsgReq;
|
||||
import com.xxl.job.admin.core.conf.XxlJobAdminConfig;
|
||||
import com.xxl.job.admin.core.model.XxlJobGroup;
|
||||
import com.xxl.job.admin.core.model.XxlJobInfo;
|
||||
import com.xxl.job.admin.core.model.XxlJobLog;
|
||||
import com.xxl.job.admin.core.util.I18nUtil;
|
||||
import com.xxl.job.core.biz.model.ReturnT;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.http.*;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.util.StringUtils;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
import java.text.MessageFormat;
|
||||
import java.util.Arrays;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Set;
|
||||
|
||||
/**
|
||||
* job alarm by webhook
|
||||
*
|
||||
* @author hcmyfs@163.com 2022-02-20
|
||||
*/
|
||||
@Component
|
||||
public class WebHookJobAlarm implements JobAlarm {
|
||||
|
||||
private RestTemplate restTemplate;
|
||||
private static Logger log = LoggerFactory.getLogger(WebHookJobAlarm.class);
|
||||
|
||||
/**
|
||||
* fail alarm
|
||||
*
|
||||
* @param jobLog
|
||||
*/
|
||||
@Override
|
||||
public boolean doAlarm(XxlJobInfo info, XxlJobLog jobLog) {
|
||||
boolean alarmResult = true;
|
||||
String br = "\r\n";
|
||||
restTemplate = new RestTemplate();
|
||||
// send monitor webhook
|
||||
if (info != null && info.getAlarmType() == AlarmTypeEnum.WEBHOOK.getAlarmType() &&
|
||||
info.getAlarmUrl() != null && info.getAlarmUrl().trim().length() > 0) {
|
||||
|
||||
// alarmContent
|
||||
String alarmContent = "Alarm Job LogId=" + jobLog.getId();
|
||||
if (jobLog.getTriggerCode() != ReturnT.SUCCESS_CODE) {
|
||||
alarmContent += br + "TriggerMsg=" + br + jobLog.getTriggerMsg();
|
||||
}
|
||||
if (jobLog.getHandleCode() > 0 && jobLog.getHandleCode() != ReturnT.SUCCESS_CODE) {
|
||||
alarmContent += br + "HandleCode=" + jobLog.getHandleMsg();
|
||||
}
|
||||
|
||||
// email info
|
||||
XxlJobGroup group = XxlJobAdminConfig.getAdminConfig().getXxlJobGroupDao().load(Integer.valueOf(info.getJobGroup()));
|
||||
//String personal = I18nUtil.getString("admin_name_full");
|
||||
String title = I18nUtil.getString("jobconf_monitor");
|
||||
String content = MessageFormat.format(loadEmailJobAlarmTemplate(),
|
||||
group != null ? group.getTitle() : "null",
|
||||
info.getId(),
|
||||
info.getJobDesc(),
|
||||
alarmContent);
|
||||
content=content.replace("<br>","\n");
|
||||
content=content.replace("<span style=\"color:#00c0ef;\" >","");
|
||||
content=content.replace("</span>","");
|
||||
|
||||
Set<String> urlList = new HashSet<String>(Arrays.asList(info.getAlarmUrl().split(",")));
|
||||
for (String webHookUrl : urlList) {
|
||||
// make mail
|
||||
try {
|
||||
sendRobot(webHookUrl, content, null);
|
||||
} catch (Exception e) {
|
||||
log.error(">>>>>>>>>>> xxl-job, job fail alarm webhook send error, JobLogId:{}", jobLog.getId(), e);
|
||||
alarmResult = false;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
return alarmResult;
|
||||
}
|
||||
|
||||
/**
|
||||
* load webhook job alarm template
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
private static final String loadEmailJobAlarmTemplate() {
|
||||
String mailBodyTemplate = I18nUtil.getString("jobconf_monitor_detail") + "\n" +
|
||||
"" + I18nUtil.getString("jobinfo_field_jobgroup") + ":{0}\n" +
|
||||
"" + I18nUtil.getString("jobinfo_field_id") + ":{1}\n" +
|
||||
"" + I18nUtil.getString("jobinfo_field_jobdesc") + ":{2}\n" +
|
||||
"" + I18nUtil.getString("jobconf_monitor_alarm_title") + ":" + I18nUtil.getString("jobconf_monitor_alarm_type") + "\n" +
|
||||
"" + I18nUtil.getString("jobconf_monitor_alarm_content") + ":{3}\n";
|
||||
|
||||
return mailBodyTemplate;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 发送消息
|
||||
*
|
||||
* @param msg
|
||||
* @return
|
||||
*/
|
||||
public boolean sendRobot(String url, String msg, List<String> usersList) {
|
||||
if (!StringUtils.hasText(url)) {
|
||||
log.error("sendRobot url 格式错误！");
|
||||
throw new RuntimeException("sendRobot url 格式错误！");
|
||||
}
|
||||
if (!url.startsWith("http")) {
|
||||
throw new RuntimeException("sendRobot url 格式错误！");
|
||||
}
|
||||
//创建请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
ResponseEntity<String> responseEntity = null;
|
||||
if (url.contains("open.feishu.cn")) {
|
||||
FeiShuTextMsgReq textMsgReq = new FeiShuTextMsgReq();
|
||||
textMsgReq.withContent(msg);
|
||||
HttpEntity<FeiShuTextMsgReq> entity = new HttpEntity<>(textMsgReq, headers);
|
||||
responseEntity = restTemplate.postForEntity(url, entity, String.class);
|
||||
} else {
|
||||
TextMsgReq textMsgReq = new TextMsgReq();
|
||||
textMsgReq.withContent(msg);
|
||||
if (!CollectionUtils.isEmpty(usersList)) {
|
||||
textMsgReq.withUserList(usersList);
|
||||
}
|
||||
HttpEntity<TextMsgReq> entity = new HttpEntity<>(textMsgReq, headers);
|
||||
responseEntity = restTemplate.postForEntity(url, entity, String.class);
|
||||
}
|
||||
|
||||
String body = responseEntity.getBody();
|
||||
log.info("sendRobot={}", body);
|
||||
return responseEntity.getStatusCode() == HttpStatus.OK;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,62 @@
|
||||
package com.xxl.job.admin.core.alarm.msg;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonIgnore;
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
|
||||
/**
|
||||
* 飞书消息
|
||||
*/
|
||||
public class FeiShuTextMsgReq {
|
||||
|
||||
public FeiShuTextMsgReq() {
|
||||
List<Map<String, Object>> elements = new ArrayList<>();
|
||||
Map<String, Boolean> config = new HashMap<>();
|
||||
config.put("wide_screen_mode", true);
|
||||
card.put("config", config);
|
||||
card.put("elements",elements);
|
||||
text.put("tag", "markdown");
|
||||
elements.add(text);
|
||||
}
|
||||
|
||||
|
||||
private Map<String,Object> card=new HashMap<>();
|
||||
|
||||
@JsonIgnore
|
||||
private Map<String,Object> text=new HashMap<>();
|
||||
|
||||
@JsonProperty(value = "msg_type")
|
||||
private String msgType="interactive";
|
||||
|
||||
/**
|
||||
* 设置用户
|
||||
*
|
||||
* @param content
|
||||
*/
|
||||
@JsonIgnore
|
||||
public FeiShuTextMsgReq withContent(String content) {
|
||||
text.put("content", content);
|
||||
return this;
|
||||
}
|
||||
|
||||
public String getMsgType() {
|
||||
return msgType;
|
||||
}
|
||||
|
||||
public void setMsgType(String msgType) {
|
||||
this.msgType = msgType;
|
||||
}
|
||||
|
||||
public Map<String, Object> getCard() {
|
||||
return card;
|
||||
}
|
||||
|
||||
public void setCard(Map<String, Object> card) {
|
||||
this.card = card;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,63 @@
|
||||
package com.xxl.job.admin.core.alarm.msg;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonIgnore;
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
|
||||
/**
|
||||
* 企业微信消息
|
||||
*/
|
||||
public class TextMsgReq {
|
||||
|
||||
// {
|
||||
// "msgtype": "text",
|
||||
// "text": {
|
||||
// "content": ""
|
||||
// "mentioned_mobile_list":[]
|
||||
// }
|
||||
// }
|
||||
|
||||
@JsonProperty(value = "msgtype")
|
||||
private String msgType = "text";
|
||||
|
||||
@JsonProperty(value = "text")
|
||||
private Map<String, Object> text = new HashMap<>();
|
||||
|
||||
@JsonIgnore
|
||||
public List<String> userList;
|
||||
|
||||
/**
|
||||
* 设置用户
|
||||
*
|
||||
* @param content
|
||||
*/
|
||||
@JsonIgnore
|
||||
public TextMsgReq withContent(String content) {
|
||||
text.put("content", content);
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置@人
|
||||
*
|
||||
* @param userList
|
||||
*/
|
||||
@JsonIgnore
|
||||
public TextMsgReq withUserList(List<String> userList) {
|
||||
text.put("mentioned_mobile_list", userList);
|
||||
return this;
|
||||
}
|
||||
|
||||
public String getMsgType() {
|
||||
return msgType;
|
||||
}
|
||||
|
||||
public void setMsgType(String msgType) {
|
||||
this.msgType = msgType;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue