SpringBoot 钉钉机器人消息推送 SpringBoot实现钉钉机器人消息推送的代码实例

时间:2021-04-21 LYX6666 人气:0

零、前言

上一次做消息推送，是微信公众号的定时消息通知。
由于自己当时的水平不够，加上企鹅家的开发文档普遍不太友好，导致根本看不懂文档在写什么，不得不去看第三方博客来学习公众号的开发。
这次就不一样了，昨天刚看了一下，阿里的开发文档比鹅厂要清晰的多，而且在同一功能上，使用了多种语言作为示例代码，可以说很友好了。可能这就是阿里和鹅厂的区别吧...辣鸡文档和好文档的区别...
本着“授之以渔”的态度，写了这篇文章，作为官方文档的补充。

一、在群里添加机器人

在群设置的智能群助手中添加自定义机器人，它长这个样子：

比较关键的一步，是进行安全设置。
加密方式一共有三种，既可以选择一种也可以使用多种方式组合：

自定义关键词
加签
IP地址

各种加密方式的介绍，详见官网：https://ding-doc.dingtalk.com/doc

为了让博客起到效果，我们选择相对安全、也比较难的加签方式。
选择加签之后，把密钥复制出来，然后就可以点确定了。

二、构建请求地址和内容

先看看官方文档怎么描述加签的：

第一步，把timestamp+"\n"+密钥当做签名字符串，使用HmacSHA256算法计算签名，然后进行Base64 encode，最后再把签名参数再进行urlEncode，得到最终的签名（需要使用UTF-8字符集）。
第二步，把 timestamp和第一步得到的签名值拼接到URL中。

官方的解释很高大上，其实原理很简单，就是把机器人密钥加密后，放在URL的参数中，所以我们需要分别获取时间戳和密钥，组合一下，加密一下，再拼接一下就好了，如图：

~~I have a Pen,~~
~~I have an Apple,~~
~~Oh~ Applepen~~~

官方给出了这样的示例代码：

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import org.apache.commons.codec.binary.Base64;
import java.net.URLEncoder;

public class Test {
 public static void main(String[] args) throws Exception {
  Long timestamp = System.currentTimeMillis();
  String secret = "this is secret";

  String stringToSign = timestamp + "\n" + secret;
  Mac mac = Mac.getInstance("HmacSHA256");
  mac.init(new SecretKeySpec(secret.getBytes("UTF-8"), "HmacSHA256"));
  byte[] signData = mac.doFinal(stringToSign.getBytes("UTF-8"));
  String sign = URLEncoder.encode(new String(Base64.encodeBase64(signData)),"UTF-8");
  System.out.println(sign);
 }
}

然而，org.apache.commons.codec.binary.Base64不是Java的内置类，也就是说，示例代码并不能直接拿过来用：

查了一下，发现Java8中内置的java.util已经包含了Base64，因此用它替换掉原来的codec，无需再引入第三方包：

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.util.Base64;
import java.net.URLEncoder;

public class ding {
 public static void main(String[] args) throws Exception {
  //获取时间戳
  Long timestamp = System.currentTimeMillis();
  //定义密钥
  String secret = "this is secret";
  //把时间戳和密钥拼接成字符串，中间加入一个换行符
  String stringToSign = timestamp + "\n" + secret;
  //声明一个Mac对象，用来操作字符串
  Mac mac = Mac.getInstance("HmacSHA256");
  //初始化Mac对象，设置Mac对象操作的字符串是UTF-8类型，加密方式是SHA256
  mac.init(new SecretKeySpec(secret.getBytes("UTF-8"), "HmacSHA256"));
  //把字符串转化成字节形式
  byte[] signData = mac.doFinal(stringToSign.getBytes("UTF-8"));
  //新建一个Base64编码对象
  Base64.Encoder encoder = Base64.getEncoder();
  //把上面的字符串进行Base64加密后再进行URL编码
  String sign = URLEncoder.encode(new String(encoder.encodeToString(signData)),"UTF-8");
  //分别输出时间戳和加密信息
  System.out.println(timestamp);
  System.out.println(sign);
 }
}

用最笨的方法，在终端执行一下看看：

成功输出了时间戳和验证信息。

我们测试上述代码的时候，可以手动拼接URL，直接发起请求：
（URL一共有三个参数：access_token、timestamp、sign,需要换成自己的，也就是上面终端输出的结果）

//替换参数后，在终端执行
curl 'https://oapi.dingtalk.com/robot/send?access_token=70c168d03e73728ef36abea63c3c10048cbd054913cfeb×tamp=1584607421017&sign=gJ3l4mhnlMuHxK1qFUx1kKUSdjuCNntsdG%2Bv%2BTCrLQM%3D' \
 -H 'Content-Type: application/json' \
 -d '{"msgtype": "text", 
  "text": {
    "content": "我就是我, 是不一样的烟火"
  },
  "sign": "gJ3l4mhnlMuHxK1qFUx1kKUSdjuCNntsdG%2Bv%2BTCrLQM%3D"
  }'

然后就出现了：

经过测试，代码正常运行，接下来就是部署到生产环境了。

三、部署代码

我们需要先找一下Spring如何发起HTTP请求。

以前，笔者只用过前台的HttpClient，对于后台的HTTP工具并不了解。

一开始尝试用Spring内置的RestTemplate，去网上查了它的用法，写了一堆代码，但怎么也不成功。由于从来没用过RestTemplate，也没耐心去看它的源码，于是放弃。

后来，只能老老实实的用apache的httpClient，查了一下用法，虽然有点麻烦，很多操作没法自动完成，但还算通俗易懂，而且它的包托管在Maven上，导入很方便。

<dependency>
 <groupId>org.apache.httpcomponents</groupId>
 <artifactId>httpclient</artifactId>
 <version>4.5.9</version>
</dependency>

httpClient的使用很灵活，这里使用的是POST方式，有一个参数，发起POST请求时，必须将字符集编码设置成UTF-8。

粗略步骤如图：

直接来一段稍微改一下就能用的代码：

public class DingService {
 //请求地址以及access_token
 String Webhook = "https://oapi.dingtalk.com/robot/send?access_token=YOUR TOKEN";
 //密钥
 String secret = "YOUR SECRET";

 /*
 ** 生成时间戳和验证信息
 */
 
 public String encode() throws Exception {
  //获取时间戳
  Long timestamp = System.currentTimeMillis();
  //把时间戳和密钥拼接成字符串，中间加入一个换行符
  String stringToSign = timestamp + "\n" + this.secret;
  //声明一个Mac对象，用来操作字符串
  Mac mac = Mac.getInstance("HmacSHA256");
  //初始化，设置Mac对象操作的字符串是UTF-8类型，加密方式是SHA256
  mac.init(new SecretKeySpec(this.secret.getBytes("UTF-8"), "HmacSHA256"));
  //把字符串转化成字节形式
  byte[] signData = mac.doFinal(stringToSign.getBytes("UTF-8"));
  //新建一个Base64编码对象
  Base64.Encoder encoder = Base64.getEncoder();
  //把上面的字符串进行Base64加密后再进行URL编码
  String sign = URLEncoder.encode(new String(encoder.encodeToString(signData)),"UTF-8");
  System.out.println(timestamp);
  System.out.println(sign);
  String result = "×tamp=" + timestamp + "&sign=" + sign;
  return result;
 };

 /* param: message 要发送的信息
 ** return: void 无返回值
 ** 作用：把传入的message发送给钉钉机器人*/

 public void dingRequest(String message){
  CloseableHttpClient httpClient = HttpClientBuilder.create().build();
  String url = null;
  try {
   url = this.Webhook + this.encode();
  } catch (Exception e) {
   e.printStackTrace();
  }
  HttpPost httpPost = new HttpPost(url);
  //设置http的请求头，发送json字符串，编码UTF-8
  httpPost.setHeader("Content-Type", "application/json;charset=utf8");
  //生成json对象传入字符
  JSONObject result = new JSONObject();
  JSONObject text = new JSONObject();
  text.put("content", message);
  result.put("text", text);
  result.put("msgtype", "text");
  String jsonString = JSON.toJSONString(result);
  StringEntity entity = new StringEntity(jsonString, "UTF-8");
  //设置http请求的内容
  httpPost.setEntity(entity);
  // 响应模型
  CloseableHttpResponse response = null;
  try {
   // 由客户端执行(发送)Post请求
   response = httpClient.execute(httpPost);
   // 从响应模型中获取响应实体
   HttpEntity responseEntity = response.getEntity();
   System.out.println("响应状态为:" + response.getStatusLine());
   if (responseEntity != null) {
    System.out.println("响应内容长度为:" + responseEntity.getContentLength());
    System.out.println("响应内容为:" + EntityUtils.toString(responseEntity));
   }
  } catch (Exception e) {
   e.printStackTrace();
  } finally {
   try {
    // 释放资源
    if (httpClient != null) {
     httpClient.close();
    }
    if (response != null) {
     response.close();
    }
   } catch (Exception e) {
    e.printStackTrace();
   }
  }
 }
}

总结

其实消息推送的功能并不难，只是由于初次接触，需要查很多的文档，在这个过程中，锻炼了文本阅读能力和独立解决问题的能力。

参考资料

Java如何进行Base64的编码(Encode)与解码(Decode)

Spring RestTemplate介绍

Spring--Http请求--HttpClient

加载全部内容