微信小程序客服消息token验证

最近公司项目要求, 搞个微信小程序的客服功能!!! 因为这个要牵扯到token的验证, 当时我就极力反对做这个功能(因为之前自己玩的时候设置token一直没搞好,心里有些阴影_,), 不过既然是公司的业务需要, 自己又是负责这方面的, 也只能顶着头皮上咯; 在网上也找了好多, 不知道为什么, 每次在网上找的, 都不能直接用, 还得自己各种调试, 看文档; 都上网拷别人代码了, 肯定不想自己再看文档了>, <.... 下面不废话了, 直接上代码吧, 我这个人就是喜欢简单粗暴直接的,~ 哈哈哈哈!

1534820253(1).jpg

package controller;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import util.SignUtil;

/**
 * 小程序消息模板 token校验
 * @author MaoLG
 *
 * @2018-8-17下午4:48:33
 */
@RestController
@RequestMapping("")//这里填自己的访问路径, 要和url(访问地址一致)
public class PushMessage {

    
    @RequestMapping("")//上面是前半段, 这是后半段, 不多做解释了
    public void session(HttpServletRequest request, HttpServletResponse response){
        boolean isGet = request.getMethod().toLowerCase().equals("get");
        PrintWriter print;
        if (isGet) {//下面从请求中获取校验需要的参数
            // 微信加密签名
            String signature = request.getParameter("signature");
            // 时间戳
            String timestamp = request.getParameter("timestamp");
            // 随机数
            String nonce = request.getParameter("nonce");
            // 随机字符串
            String echostr = request.getParameter("echostr");
            // 通过检验signature对请求进行校验，若校验成功则原样返回echostr，表示接入成功，否则接入失败
            if (signature != null && SignUtil.checkSignature(signature, timestamp, nonce)) {
                try {
                    print = response.getWriter();
                    print.write(echostr);
                    print.flush();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
    
}
//上面代码是controller层
-----------------------------------------------------------------------------------------------------
//我是分割线
-----------------------------------------------------------------------------------------------------
//下面为工具类
package util;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;

/**
 * 验证签名使用的工具
 * @author MaoLG
 *
 * @2018-8-17下午4:54:34
 */
public class SignUtil {
    
    private  String token = "token";//token(令牌)和网页上配置的一致
    
    public static boolean checkSignature(String signature, String timestamp, String nonce) {  
        
        //与token 比较
        String[] arr = new String[] { token, timestamp, nonce };  
 
        // 将token、timestamp、nonce三个参数进行字典排序  
        Arrays.sort(arr); 
        
        StringBuilder content = new StringBuilder();  
        
        for (int i = 0; i < arr.length; i++) {  
            content.append(arr[i]);  
        }
        
        MessageDigest md = null;  
        String tmpStr = null;  
  
        try {  
            md = MessageDigest.getInstance("SHA-1");  
            
            // 将三个参数字符串拼接成一个字符串进行sha1加密
            byte[] digest = md.digest(content.toString().getBytes());
            tmpStr = byteToStr(digest); 
        } catch (NoSuchAlgorithmException e) {  
            ErrorLogUtil.log(e);
            e.printStackTrace();  
        }  
 
        content = null;  
        // 将sha1加密后的字符串可与signature对比  
        return tmpStr != null ? tmpStr.equals(signature.toUpperCase()) : false;  
    }  
    
    /**
     * 将字节数组转换为十六进制字符串
     * 
     * @param byteArray
     * @return
     */
    private static String byteToStr(byte[] byteArray) {
        String strDigest = "";
        for (int i = 0; i < byteArray.length; i++) {
            strDigest += byteToHexStr(byteArray[i]);
        }
        return strDigest;
    }
    
    /**
     * 将字节转换为十六进制字符串
     * 
     * @param mByte
     * @return
     */
    private static String byteToHexStr(byte mByte) {
        char[] Digit = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };
        char[] tempArr = new char[2];
        tempArr[0] = Digit[(mByte >>> 4) & 0X0F];
        tempArr[1] = Digit[mByte & 0X0F];
        String s = new String(tempArr);
        return s;
    }
    public static void sort(String a[]) {
        for (int i = 0; i < a.length - 1; i++) {
            for (int j = i + 1; j < a.length; j++) {
                if (a[j].compareTo(a[i]) < 0) {
                    String temp = a[i];
                    a[i] = a[j];
                    a[j] = temp;
                }
            }
        }
    }
}

好啦, 就这两块代码, 拷过去就直接用了!