add Enterprise WeChat alert to alert sender (ae03a375) · Commits · SUMMER2020 / students / proj-2002251

escheduler-alert/src/main/java/cn/escheduler/alert/manager/EnterpriseWeChatManager.java

0 → 100644

+57 −0

Original line number	Diff line number	Diff line
		/*
		* Licensed to the Apache Software Foundation (ASF) under one or more
		* contributor license agreements. See the NOTICE file distributed with
		* this work for additional information regarding copyright ownership.
		* The ASF licenses this file to You under the Apache License, Version 2.0
		* (the "License"); you may not use this file except in compliance with
		* the License. You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*/
		package cn.escheduler.alert.manager;

		import cn.escheduler.alert.utils.Constants;
		import cn.escheduler.alert.utils.EnterpriseWeChatUtils;
		import cn.escheduler.dao.model.Alert;
		import org.slf4j.Logger;
		import org.slf4j.LoggerFactory;

		import java.io.IOException;
		import java.util.Arrays;
		import java.util.HashMap;
		import java.util.List;
		import java.util.Map;

		/**
		* Enterprise WeChat Manager
		*/
		public class EnterpriseWeChatManager {
		private static final Logger logger = LoggerFactory.getLogger(MsgManager.class);
		/**
		* Enterprise We Chat send
		* @param alert
		*/
		public Map<String,Object> send(Alert alert, String token){
		Map<String,Object> retMap = new HashMap<>();
		retMap.put(Constants.STATUS, false);
		String agentId = EnterpriseWeChatUtils.enterpriseWeChatAgentId;
		String users = EnterpriseWeChatUtils.enterpriseWeChatUsers;
		List<String> userList = Arrays.asList(users.split(","));
		logger.info("send message {}",alert);
		String msg = EnterpriseWeChatUtils.makeUserSendMsg(userList, agentId,EnterpriseWeChatUtils.markdownByAlert(alert));
		try {
		EnterpriseWeChatUtils.sendEnterpriseWeChat(Constants.UTF_8, msg, token);
		} catch (IOException e) {
		logger.error(e.getMessage(),e);
		}
		retMap.put(Constants.STATUS, true);
		return retMap;
		}

		}

escheduler-alert/src/main/java/cn/escheduler/alert/runner/AlertSender.java

+9 −0

Original line number	Diff line number	Diff line
		@@ -17,7 +17,9 @@
		package cn.escheduler.alert.runner;

		import cn.escheduler.alert.manager.EmailManager;
		import cn.escheduler.alert.manager.EnterpriseWeChatManager;
		import cn.escheduler.alert.utils.Constants;
		import cn.escheduler.alert.utils.EnterpriseWeChatUtils;
		import cn.escheduler.common.enums.AlertStatus;
		import cn.escheduler.common.enums.AlertType;
		import cn.escheduler.dao.AlertDao;
		@@ -40,6 +42,7 @@ public class AlertSender{
		private static final Logger logger = LoggerFactory.getLogger(AlertSender.class);

		private static final EmailManager emailManager= new EmailManager();
		private static final EnterpriseWeChatManager weChatManager= new EnterpriseWeChatManager();


		private List<Alert> alertList;
		@@ -109,6 +112,12 @@ public class AlertSender{
		if (flag){
		alertDao.updateAlert(AlertStatus.EXECUTION_SUCCESS, "execution success", alert.getId());
		logger.info("alert send success");
		try {
		String token = EnterpriseWeChatUtils.getToken();
		weChatManager.send(alert,token);
		} catch (Exception e) {
		logger.error(e.getMessage(),e);
		}
		}else {
		alertDao.updateAlert(AlertStatus.EXECUTION_FAILURE,String.valueOf(retMaps.get(Constants.MESSAGE)),alert.getId());
		logger.info("alert send error : {}" , String.valueOf(retMaps.get(Constants.MESSAGE)));

escheduler-alert/src/main/java/cn/escheduler/alert/utils/Constants.java

+4 −0

Original line number	Diff line number	Diff line
		@@ -144,4 +144,8 @@ public class Constants {
		public static final String ENTERPRISE_WECHAT_TEAM_SEND_MSG = "enterprise.wechat.team.send.msg";

		public static final String ENTERPRISE_WECHAT_USER_SEND_MSG = "enterprise.wechat.user.send.msg";

		public static final String ENTERPRISE_WECHAT_AGENT_ID = "enterprise.wechat.agent.id";

		public static final String ENTERPRISE_WECHAT_USERS = "enterprise.wechat.users";
		}

escheduler-alert/src/main/java/cn/escheduler/alert/utils/EnterpriseWeChatUtils.java

+6 −2

Original line number	Diff line number	Diff line
		@@ -60,6 +60,10 @@ public class EnterpriseWeChatUtils {

		private static final String enterpriseWeChatUserSendMsg = getString(Constants.ENTERPRISE_WECHAT_USER_SEND_MSG);

		public static final String enterpriseWeChatAgentId = getString(Constants.ENTERPRISE_WECHAT_AGENT_ID);

		public static final String enterpriseWeChatUsers = getString(Constants.ENTERPRISE_WECHAT_USERS);

		/**
		* get Enterprise WeChat token info
		* @return token string info
		@@ -140,14 +144,14 @@ public class EnterpriseWeChatUtils {
		}

		/**
		* send weixin
		* send Enterprise WeChat
		* @param charset
		* @param data
		* @param token
		* @return Enterprise WeChat resp, demo: {"errcode":0,"errmsg":"ok","invaliduser":""}
		* @throws IOException
		*/
		public static String sendQiyeWeixin(String charset, String data, String token) throws IOException {
		public static String sendEnterpriseWeChat(String charset, String data, String token) throws IOException {
		String enterpriseWeChatPushUrlReplace = enterpriseWeChatPushUrl.replaceAll("\\$token", token);

		CloseableHttpClient httpclient = HttpClients.createDefault();