Commit 94a8ade8 authored by cdp's avatar cdp
Browse files

Add qiye_weixin send feature.

parent a362954a

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
@@ -126,4 +126,16 @@ public class Constants { 
    public static final String TH_END = "</th>";



    public static final int ALERT_SCAN_INTERVEL = 5000;



    public static final String QIYE_WEIXIN_CORP_ID = "qiye.weixin.corp.id";



    public static final String QIYE_WEIXIN_SECRET = "qiye.weixin.secret";



    public static final String QIYE_WEIXIN_TOKEN_URL = "qiye.weixin.token.url";



    public static final String QIYE_WEIXIN_PUSH_URL = "qiye.weixin.push.url";



    public static final String QIYE_WEIXIN_TEAM_SEND_MSG = "qiye.weixin.team.send.msg";



    public static final String QIYE_WEIXIN_USER_SEND_MSG = "qiye.weixin.user.send.msg";

}

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

0 → 100644
+18 −0
Original line number Diff line number Diff line 
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/java/cn/escheduler/alert/utils/QiyeWeixinUtils.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 QiyeWeixinUtils {



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



    private static final String qiyeWeixinCorpId = getString(Constants.QIYE_WEIXIN_CORP_ID);



    private static final String qiyeWeixinSecret = getString(Constants.QIYE_WEIXIN_SECRET);



    private static final String qiyeWeixinTokenUrl = getString(Constants.QIYE_WEIXIN_TOKEN_URL);

    private String qiyeWeixinTokenUrlReplace = qiyeWeixinTokenUrl

            .replaceAll("\\$weixinCorpId", qiyeWeixinCorpId)

            .replaceAll("\\$weixinSecret", qiyeWeixinSecret);



    private static final String qiyeWeixinPushUrl = getString(Constants.QIYE_WEIXIN_PUSH_URL);



    private static final String qiyeWeixinTeamSendMsg = getString(Constants.QIYE_WEIXIN_TEAM_SEND_MSG);



    private static final String qiyeWeixinUserSendMsg = getString(Constants.QIYE_WEIXIN_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(qiyeWeixinTokenUrlReplace);

        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 qiyeWeixinTeamSendMsg.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 qiyeWeixinTeamSendMsg.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 qiyeWeixinUserSendMsg.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 qiyeWeixinUserSendMsg.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 qiyeWeixinPushUrlReplace = qiyeWeixinPushUrl.replaceAll("\\$weixinToken", token);



        CloseableHttpClient httpclient = HttpClients.createDefault();

        HttpPost httpPost = new HttpPost(qiyeWeixinPushUrlReplace);

        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:{}", qiyeWeixinPushUrl, data, resp);

        return resp;

    }



}

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

+7 −0
Original line number Diff line number Diff line
@@ -11,6 +11,13 @@ mail.passwd=xxxxxxx 
#xls file path,need create if not exist

xls.file.path=/opt/xls



# qiye weixin configuration

qiye.weixin.corp.id=xxxxxxx

qiye.weixin.secret=xxxxxxx

qiye.weixin.token.url=https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=$weixinCorpId&corpsecret=$weixinSecret

qiye.weixin.push.url=https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=$weixinToken

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

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