Unverified Commit 46459544 authored by lgcareer's avatar lgcareer Committed by GitHub
Browse files

Merge pull request #288 from feloxx/feature/qiye_weixin 

Feature/qiye_weixin.
parents 9ddc9a04 1d624172

escheduler-alert/pom.xml

+11 −0
Original line number Diff line number Diff line
@@ -102,6 +102,17 @@ 
            <artifactId>escheduler-dao</artifactId>

        </dependency>



        <dependency>

            <groupId>org.apache.httpcomponents</groupId>

            <artifactId>httpcore</artifactId>

            <version>4.4.6</version>

        </dependency>

        <dependency>

            <groupId>org.apache.httpcomponents</groupId>

            <artifactId>httpclient</artifactId>

            <version>4.5.5</version>

        </dependency>



    </dependencies>



    <build>

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

+12 −0
Original line number Diff line number Diff line
@@ -128,4 +128,16 @@ public class Constants { 
    public static final String TH_END = "</th>";



    public static final int ALERT_SCAN_INTERVEL = 5000;



    public static final String ENTERPRISE_WECHAT_CORP_ID = "enterprise.wechat.corp.id";



    public static final String ENTERPRISE_WECHAT_SECRET = "enterprise.wechat.secret";



    public static final String ENTERPRISE_WECHAT_TOKEN_URL = "enterprise.wechat.token.url";



    public static final String ENTERPRISE_WECHAT_PUSH_URL = "enterprise.wechat.push.url";



    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";

}

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

0 → 100644
+167 −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.utils;



import com.alibaba.fastjson.JSON;



import com.google.common.reflect.TypeToken;

import org.apache.http.HttpEntity;

import org.apache.http.client.methods.CloseableHttpResponse;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.entity.StringEntity;

import org.apache.http.impl.client.CloseableHttpClient;

import org.apache.http.impl.client.HttpClients;

import org.apache.http.util.EntityUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



import java.io.IOException;

import java.util.Collection;

import java.util.Map;



import static cn.escheduler.alert.utils.PropertyUtils.getString;



/**

 * qiye weixin utils

 */

public class EnterpriseWeChatUtils {



    public static final Logger logger = LoggerFactory.getLogger(EnterpriseWeChatUtils.class);



    private static final String enterpriseWeChatCorpId = getString(Constants.ENTERPRISE_WECHAT_CORP_ID);



    private static final String enterpriseWeChatSecret = getString(Constants.ENTERPRISE_WECHAT_SECRET);



    private static final String enterpriseWeChatTokenUrl = getString(Constants.ENTERPRISE_WECHAT_TOKEN_URL);

    private String enterpriseWeChatTokenUrlReplace = enterpriseWeChatTokenUrl

            .replaceAll("\\$corpId", enterpriseWeChatCorpId)

            .replaceAll("\\$secret", enterpriseWeChatSecret);



    private static final String enterpriseWeChatPushUrl = getString(Constants.ENTERPRISE_WECHAT_PUSH_URL);



    private static final String enterpriseWeChatTeamSendMsg = getString(Constants.ENTERPRISE_WECHAT_TEAM_SEND_MSG);



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



    /**

     * get winxin token info

     * @return token string info

     * @throws IOException

     */

    public String getToken() throws IOException {

        String resp;



        CloseableHttpClient httpClient = HttpClients.createDefault();

        HttpGet httpGet = new HttpGet(enterpriseWeChatTokenUrlReplace);

        CloseableHttpResponse response = httpClient.execute(httpGet);

        try {

            HttpEntity entity = response.getEntity();

            resp = EntityUtils.toString(entity, "utf-8");

            EntityUtils.consume(entity);

        } finally {

            response.close();

        }



        Map<String, Object> map = JSON.parseObject(resp,

                new TypeToken<Map<String, Object>>() {

                }.getType());

        return map.get("access_token").toString();

    }



    /**

     * make team single weixin message

     * @param toParty

     * @param agentId

     * @param msg

     * @return weixin send message

     */

    public String makeTeamSendMsg(String toParty, String agentId, String msg) {

        return enterpriseWeChatTeamSendMsg.replaceAll("\\$toParty", toParty)

                .replaceAll("\\$agentId", agentId)

                .replaceAll("\\$msg", msg);

    }



    /**

     * make team multi weixin message

     * @param toParty

     * @param agentId

     * @param msg

     * @return weixin send message

     */

    public String makeTeamSendMsg(Collection<String> toParty, String agentId, String msg) {

        String listParty = FuncUtils.mkString(toParty, "|");

        return enterpriseWeChatTeamSendMsg.replaceAll("\\$toParty", listParty)

                .replaceAll("\\$agentId", agentId)

                .replaceAll("\\$msg", msg);

    }



    /**

     * make team single user message

     * @param toUser

     * @param agentId

     * @param msg

     * @return weixin send message

     */

    public String makeUserSendMsg(String toUser, String agentId, String msg) {

        return enterpriseWeChatUserSendMsg.replaceAll("\\$toUser", toUser)

                .replaceAll("\\$agentId", agentId)

                .replaceAll("\\$msg", msg);

    }



    /**

     * make team multi user message

     * @param toUser

     * @param agentId

     * @param msg

     * @return weixin send message

     */

    public String makeUserSendMsg(Collection<String> toUser, String agentId, String msg) {

        String listUser = FuncUtils.mkString(toUser, "|");

        return enterpriseWeChatUserSendMsg.replaceAll("\\$toUser", listUser)

                .replaceAll("\\$agentId", agentId)

                .replaceAll("\\$msg", msg);

    }



    /**

     * send weixin

     * @param charset

     * @param data

     * @param token

     * @return weixin resp, demo: {"errcode":0,"errmsg":"ok","invaliduser":""}

     * @throws IOException

     */

    public String sendQiyeWeixin(String charset, String data, String token) throws IOException {

        String enterpriseWeChatPushUrlReplace = enterpriseWeChatPushUrl.replaceAll("\\$token", token);



        CloseableHttpClient httpclient = HttpClients.createDefault();

        HttpPost httpPost = new HttpPost(enterpriseWeChatPushUrlReplace);

        httpPost.setEntity(new StringEntity(data, charset));

        CloseableHttpResponse response = httpclient.execute(httpPost);

        String resp;

        try {

            HttpEntity entity = response.getEntity();

            resp = EntityUtils.toString(entity, charset);

            EntityUtils.consume(entity);

        } finally {

            response.close();

        }

        logger.info("qiye weixin send [{}], param:{}, resp:{}", enterpriseWeChatPushUrl, data, resp);

        return resp;

    }



}

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

0 → 100644
+34 −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.utils;



public class FuncUtils {



    static public String mkString(Iterable<String> list, String split) {

        StringBuilder sb = new StringBuilder();

        boolean first = true;

        for (String item : list) {

            if (first)

                first = false;

            else

                sb.append(split);

            sb.append(item);

        }

        return sb.toString();

    }



}

escheduler-alert/src/main/resources/alert.properties

+7 −0
Original line number Diff line number Diff line
@@ -16,6 +16,13 @@ mail.smtp.ssl.enable=true 
#xls file path,need create if not exist

xls.file.path=/tmp/xls



# Enterprise WeChat configuration

enterprise.wechat.corp.id=xxxxxxx

enterprise.wechat.secret=xxxxxxx

enterprise.wechat.token.url=https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=$corpId&corpsecret=$secret

enterprise.wechat.push.url=https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=$token

enterprise.wechat.team.send.msg={\"toparty\":\"$toParty\",\"agentid\":\"$agentId\",\"msgtype\":\"text\",\"text\":{\"content\":\"$msg\"},\"safe\":\"0\"}

enterprise.wechat.user.send.msg={\"touser\":\"$toUser\",\"agentid\":\"$agentId\",\"msgtype\":\"text\",\"text\":{\"content\":\"$msg\"},\"safe\":\"0\"}