|
|
|
|
@ -39,8 +39,10 @@ import org.apache.http.util.EntityUtils;
|
|
|
|
|
import java.io.IOException; |
|
|
|
|
import java.net.URLEncoder; |
|
|
|
|
import java.nio.charset.StandardCharsets; |
|
|
|
|
import java.util.Arrays; |
|
|
|
|
import java.util.HashMap; |
|
|
|
|
import java.util.Map; |
|
|
|
|
import java.util.Objects; |
|
|
|
|
|
|
|
|
|
import javax.crypto.Mac; |
|
|
|
|
import javax.crypto.spec.SecretKeySpec; |
|
|
|
|
@ -61,6 +63,12 @@ public final class DingTalkSender {
|
|
|
|
|
private final String url; |
|
|
|
|
private final String keyword; |
|
|
|
|
private final String secret; |
|
|
|
|
private String msgType; |
|
|
|
|
|
|
|
|
|
private final String atMobiles; |
|
|
|
|
private final String atUserIds; |
|
|
|
|
private final Boolean atAll; |
|
|
|
|
|
|
|
|
|
private final Boolean enableProxy; |
|
|
|
|
|
|
|
|
|
private String proxy; |
|
|
|
|
@ -75,11 +83,17 @@ public final class DingTalkSender {
|
|
|
|
|
url = config.get(DingTalkParamsConstants.NAME_DING_TALK_WEB_HOOK); |
|
|
|
|
keyword = config.get(DingTalkParamsConstants.NAME_DING_TALK_KEYWORD); |
|
|
|
|
secret = config.get(DingTalkParamsConstants.NAME_DING_TALK_SECRET); |
|
|
|
|
msgType = config.get(DingTalkParamsConstants.NAME_DING_TALK_MSG_TYPE); |
|
|
|
|
|
|
|
|
|
atMobiles = config.get(DingTalkParamsConstants.NAME_DING_TALK_AT_MOBILES); |
|
|
|
|
atUserIds = config.get(DingTalkParamsConstants.NAME_DING_TALK_AT_USERIDS); |
|
|
|
|
atAll = Boolean.valueOf(config.get(DingTalkParamsConstants.NAME_DING_TALK_AT_ALL)); |
|
|
|
|
|
|
|
|
|
enableProxy = Boolean.valueOf(config.get(DingTalkParamsConstants.NAME_DING_TALK_PROXY_ENABLE)); |
|
|
|
|
if (Boolean.TRUE.equals(enableProxy)) { |
|
|
|
|
port = Integer.parseInt(config.get(DingTalkParamsConstants.NAME_DING_TALK_PORT)); |
|
|
|
|
proxy = config.get(DingTalkParamsConstants.NAME_DING_TALK_PROXY); |
|
|
|
|
user = config.get(DingTalkParamsConstants.DING_TALK_USER); |
|
|
|
|
user = config.get(DingTalkParamsConstants.NAME_DING_TALK_USER); |
|
|
|
|
password = config.get(DingTalkParamsConstants.NAME_DING_TALK_PASSWORD); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
@ -108,18 +122,7 @@ public final class DingTalkSender {
|
|
|
|
|
return RequestConfig.custom().setProxy(httpProxy).build(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private static String textToJsonString(String text) { |
|
|
|
|
Map<String, Object> items = new HashMap<>(); |
|
|
|
|
items.put("msgtype", "text"); |
|
|
|
|
Map<String, String> textContent = new HashMap<>(); |
|
|
|
|
byte[] byt = StringUtils.getBytesUtf8(text); |
|
|
|
|
String txt = StringUtils.newStringUtf8(byt); |
|
|
|
|
textContent.put("content", txt); |
|
|
|
|
items.put("text", textContent); |
|
|
|
|
return JSONUtils.toJsonString(items); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private static AlertResult checkSendDingTalkSendMsgResult(String result) { |
|
|
|
|
private AlertResult checkSendDingTalkSendMsgResult(String result) { |
|
|
|
|
AlertResult alertResult = new AlertResult(); |
|
|
|
|
alertResult.setStatus("false"); |
|
|
|
|
|
|
|
|
|
@ -144,6 +147,13 @@ public final class DingTalkSender {
|
|
|
|
|
return alertResult; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* send dingtalk msg handler |
|
|
|
|
* |
|
|
|
|
* @param title title |
|
|
|
|
* @param content content |
|
|
|
|
* @return |
|
|
|
|
*/ |
|
|
|
|
public AlertResult sendDingTalkMsg(String title, String content) { |
|
|
|
|
AlertResult alertResult; |
|
|
|
|
try { |
|
|
|
|
@ -160,18 +170,9 @@ public final class DingTalkSender {
|
|
|
|
|
|
|
|
|
|
private String sendMsg(String title, String content) throws IOException { |
|
|
|
|
|
|
|
|
|
StringBuilder text = new StringBuilder(); |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(keyword)) { |
|
|
|
|
text.append(keyword); |
|
|
|
|
text.append(":"); |
|
|
|
|
} |
|
|
|
|
text.append(title); |
|
|
|
|
text.append("\n"); |
|
|
|
|
text.append(content); |
|
|
|
|
|
|
|
|
|
String msgToJson = textToJsonString(text.toString()); |
|
|
|
|
String msg = generateMsgJson(title, content); |
|
|
|
|
|
|
|
|
|
HttpPost httpPost = constructHttpPost(org.apache.dolphinscheduler.spi.utils.StringUtils.isBlank(secret) ? url : generateSignedUrl(), msgToJson); |
|
|
|
|
HttpPost httpPost = constructHttpPost(org.apache.dolphinscheduler.spi.utils.StringUtils.isBlank(secret) ? url : generateSignedUrl(), msg); |
|
|
|
|
|
|
|
|
|
CloseableHttpClient httpClient; |
|
|
|
|
if (Boolean.TRUE.equals(enableProxy)) { |
|
|
|
|
@ -192,13 +193,119 @@ public final class DingTalkSender {
|
|
|
|
|
} finally { |
|
|
|
|
response.close(); |
|
|
|
|
} |
|
|
|
|
logger.info("Ding Talk send title :{},content : {}, resp: {}", title, content, resp); |
|
|
|
|
logger.info("Ding Talk send msg :{}, resp: {}", msg, resp); |
|
|
|
|
return resp; |
|
|
|
|
} finally { |
|
|
|
|
httpClient.close(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* generate msg json |
|
|
|
|
* |
|
|
|
|
* @param title title |
|
|
|
|
* @param content content |
|
|
|
|
* @return msg |
|
|
|
|
*/ |
|
|
|
|
private String generateMsgJson(String title, String content) { |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isBlank(msgType)) { |
|
|
|
|
msgType = DingTalkParamsConstants.DING_TALK_MSG_TYPE_TEXT; |
|
|
|
|
} |
|
|
|
|
Map<String, Object> items = new HashMap<>(); |
|
|
|
|
items.put("msgtype", msgType); |
|
|
|
|
Map<String, Object> text = new HashMap<>(); |
|
|
|
|
items.put(msgType, text); |
|
|
|
|
|
|
|
|
|
if (DingTalkParamsConstants.DING_TALK_MSG_TYPE_MARKDOWN.equals(msgType)) { |
|
|
|
|
generateMarkdownMsg(title, content, text); |
|
|
|
|
} else { |
|
|
|
|
generateTextMsg(title, content, text); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
setMsgAt(items); |
|
|
|
|
return JSONUtils.toJsonString(items); |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* generate text msg |
|
|
|
|
* |
|
|
|
|
* @param title title |
|
|
|
|
* @param content content |
|
|
|
|
* @param text text |
|
|
|
|
*/ |
|
|
|
|
private void generateTextMsg(String title, String content, Map<String, Object> text) { |
|
|
|
|
StringBuilder builder = new StringBuilder(title); |
|
|
|
|
builder.append("\n"); |
|
|
|
|
builder.append(content); |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(keyword)) { |
|
|
|
|
builder.append(" "); |
|
|
|
|
builder.append(keyword); |
|
|
|
|
} |
|
|
|
|
byte[] byt = StringUtils.getBytesUtf8(builder.toString()); |
|
|
|
|
String txt = StringUtils.newStringUtf8(byt); |
|
|
|
|
text.put("content", txt); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* generate markdown msg |
|
|
|
|
* |
|
|
|
|
* @param title title |
|
|
|
|
* @param content content |
|
|
|
|
* @param text text |
|
|
|
|
*/ |
|
|
|
|
private void generateMarkdownMsg(String title, String content, Map<String, Object> text) { |
|
|
|
|
StringBuilder builder = new StringBuilder(content); |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(keyword)) { |
|
|
|
|
builder.append(" "); |
|
|
|
|
builder.append(keyword); |
|
|
|
|
} |
|
|
|
|
builder.append("\n\n"); |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(atMobiles)) { |
|
|
|
|
Arrays.stream(atMobiles.split(",")).forEach(value -> { |
|
|
|
|
builder.append("@"); |
|
|
|
|
builder.append(value); |
|
|
|
|
builder.append(" "); |
|
|
|
|
}); |
|
|
|
|
} |
|
|
|
|
if (org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(atUserIds)) { |
|
|
|
|
Arrays.stream(atUserIds.split(",")).forEach(value -> { |
|
|
|
|
builder.append("@"); |
|
|
|
|
builder.append(value); |
|
|
|
|
builder.append(" "); |
|
|
|
|
}); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
byte[] byt = StringUtils.getBytesUtf8(builder.toString()); |
|
|
|
|
String txt = StringUtils.newStringUtf8(byt); |
|
|
|
|
text.put("title", title); |
|
|
|
|
text.put("text", txt); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* configure msg @person |
|
|
|
|
* |
|
|
|
|
* @param items items |
|
|
|
|
*/ |
|
|
|
|
private void setMsgAt(Map<String, Object> items) { |
|
|
|
|
Map<String, Object> at = new HashMap<>(); |
|
|
|
|
|
|
|
|
|
String[] atMobileArray = org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(atMobiles) ? atMobiles.split(",") : new String[0]; |
|
|
|
|
String[] atUserArray = org.apache.dolphinscheduler.spi.utils.StringUtils.isNotBlank(atUserIds) ? atUserIds.split(",") : new String[0]; |
|
|
|
|
boolean isAtAll = Objects.isNull(atAll) ? false : atAll; |
|
|
|
|
|
|
|
|
|
at.put("atMobiles", atMobileArray); |
|
|
|
|
at.put("atUserIds", atUserArray); |
|
|
|
|
at.put("isAtAll", isAtAll); |
|
|
|
|
|
|
|
|
|
items.put("at", at); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* generate sign url |
|
|
|
|
* |
|
|
|
|
* @return sign url |
|
|
|
|
*/ |
|
|
|
|
private String generateSignedUrl() { |
|
|
|
|
Long timestamp = System.currentTimeMillis(); |
|
|
|
|
String stringToSign = timestamp + "\n" + secret; |
|
|
|
|
|
JinYong Li
2 years ago
committed by
GitHub