org.knowm.xchange.coinsuper.utils.RestSignUtil Maven / Gradle / Ivy

Go to download
package org.knowm.xchange.coinsuper.utils;

import java.security.MessageDigest;
import java.util.Iterator;
import java.util.Map;
import java.util.TreeMap;

/** */
public class RestSignUtil {

  /**
   * 生成API传参签名
   *
   * @param param
   * @param secretkey
   * @return
   */
  public static  String generateSign(RestRequestParam param, String secretkey) {
    TreeMap paramsMap = new TreeMap();
    paramsMap.put("accesskey", param.getCommon().getAccesskey());
    paramsMap.put("timestamp", Long.toString(param.getCommon().getTimestamp()));

    //    for (Map.Entry entry : param.getData())
    //    {
    //        paramsMap.put(entry.getKey(), entry.getValue().toString());
    //    }
    paramsMap.put("secretkey", secretkey);
    return generateSignByRule(paramsMap);
  }

  /**
   * @param md5Str
   * @return
   */
  private static String getMD5(String md5Str) {
    try {
      MessageDigest md5 = MessageDigest.getInstance("MD5");
      byte[] bytes = md5.digest(md5Str.getBytes("utf-8"));
      StringBuffer sb = new StringBuffer();
      String temp = "";
      for (byte b : bytes) {
        temp = Integer.toHexString(b & 0XFF);
        sb.append(temp.length() == 1 ? "0" + temp : temp);
      }
      return sb.toString();
    } catch (Exception e) {
      return null;
    }
  }

  /**
   * 把需要验签的参数封装到TreeMap中,生成加密sign串
   *
   * @param map
   * @return
   */
  private static String generateSignByRule(TreeMap map) {
    String sign = "";
    Iterator> it = map.entrySet().iterator();
    while (it.hasNext()) {
      Map.Entry entry = it.next();
      if (it.hasNext()) {
        if (entry.getValue() != null && !"".equals(entry.getValue().toString().trim())) {
          sign += entry.getKey() + "=" + entry.getValue() + "&";
        }
      } else {
        if (entry.getValue() != null && !"".equals(entry.getValue().toString().trim())) {
          sign += entry.getKey() + "=" + entry.getValue();
        } else {
          sign = sign.substring(0, sign.length() - 1);
        }
      }
    }
    return getMD5(sign);
  }
}