jmeter使用beanshell完成签名计算，附与python代码对比

Posted 2020-11-27 wan了个蛋

签名计算过程：

1.ticket计算：时间戳加+随机数字拼接后md5加密

2.组装公共参数+ticket+时间戳+业务参数

beanshell代码实现：

import java.util.*;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Random;
import java.util.Date;
import java.text.SimpleDateFormat;
import net.sf.json.JSONObject;
//net.sf.json.JSONObject 将Map转换为JSON方法

//获取shop_id
String shop_id=vars.get("ebai_shopId");
//log.info("shop_id1233:"+shop_id+"");
//获取时间戳
long timestamp=new Date().getTime();
vars.put("timestamp",timestamp+"");
//log.info("timestamp:"+timestamp+"");

//中文转unicode编码
public static String gbEncoding( String gbString) {
char[] utfBytes = gbString.toCharArray();
String unicodeBytes = "";
for (int i = 0; i < utfBytes.length; i++) {
String hexB = Integer.toHexString(utfBytes[i]);
if (hexB.length() <= 2) {
hexB = "00" + hexB;
}
unicodeBytes = unicodeBytes + "\u" + hexB;
}
return unicodeBytes;
}

//md5加密方法
private static String getMD5(String input) {
try {
MessageDigest md = MessageDigest.getInstance("MD5");
// log.info("input:"+input+"");
byte[] messageDigest = md.digest(input.getBytes());
BigInteger number = new BigInteger(1, messageDigest);
String hashtext = number.toString(16);
while (hashtext.length() < 32) {
hashtext = "0" + hashtext;
}
return hashtext.toUpperCase();
} catch (NoSuchAlgorithmException e) {
throw null;
}
}
//获取ticket方法
public static String getTicket() {
final int random = new Random().nextInt(Integer.MAX_VALUE);
final String md5 = getMD5(String.valueOf(timestamp + random));
StringBuilder sb = new StringBuilder(md5);
// 在左起第8、12、16、20位字符后面添加减号，注意是针对md5
final StringBuilder result = sb.insert(8, "-").insert(13, "-").insert(18, "-").insert(23, "-");
return result.toString().toUpperCase();
}
//将时间戳转换为时间字符串年月日，时分秒
public static String stampToDate(String s){
String res;
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
long lt = new Long(s);
Date date = new Date(lt);
res = simpleDateFormat.format(date);
return res;
}
//获取时间：年月日时间秒
String timestamp_str=timestamp+"";
String time_str=stampToDate(timestamp_str);
vars.put("time_str",time_str);
////获取ticket
String ticket=getTicket();
//公共请求参数
String data_str="data=";
String pub_request_data="&source="+shop_id+"&ticket="+ticket+"&timestamp="+timestamp_str+"";
String secret_key="111";

//获取商家信息
//处理请求参数
Map commercialInfo_map=new HashMap();
commercialInfo_map.put("shop_id", shop_id) ;
JSONObject commercialInfo_str = JSONObject.fromObject(commercialInfo_map);
//log.info("commercialInfo_str:"+commercialInfo_str+"");
//1.组装请求参数
String request_data_commercialInfo=""+data_str+commercialInfo_str+pub_request_data+"";
vars.put("request_data_commercialInfo",request_data_commercialInfo);
//log.info("request_data_commercialInfo:"+request_data_commercialInfo+"");
//2.拼接加密字符串
String commercialInfo_Authorization=""+request_data_commercialInfo+secret_key+"";
//log.info("Authorization_commercialInfo:"+Authorization_commercialInfo+"");
//3.获取加密后的签名
String Authorization_commercialInfo=getMD5(commercialInfo_Authorization);
vars.put("Authorization_commercialInfo",Authorization_commercialInfo);
//log.info("Authorization_commercialInfo:"+Authorization_commercialInfo+"");

 

python代码实现：

# ‘‘‘生成md5加密字符串‘‘‘
def md5_Encry(self, str_in,upper=None):
    str_out = hashlib.md5()  # 采用md5加密
    str_out.update(str(str_in).encode(encoding=‘utf-8‘))
    md5=str_out.hexdigest()
    if upper==True:
        return md5.upper()
    return  md5

def str_to_time(self, time_to_str=‘1970-01-01 00:00:00‘):  # 字符串转换为时间戳
    try:
        if time_to_str==‘1970-01-01 00:00:00‘:
            time_to_str=self.time_to_str()
        d = datetime.datetime.strptime(time_to_str, "%Y-%m-%d %H:%M:%S")
        t = d.timetuple()
        timeStamp = int(time.mktime(t))
        # print(timeStamp)
        return timeStamp
    except Exception as error:
        print("数据处理失败，原因为:
%s" % (error))

def get_timestamp(self):  # 输出当前时间的13位时间戳
    current_milli_time = lambda: int(round(time.time() * 1000))  # 输出13位时间戳,round:对浮点数进行四舍五入
    return str(current_milli_time())

# ticket计算
    def kuaishou_ticket_calc(self):
        random_int=random.randint(1,100)
        md5_string=self.requestData.md5_Encry(‘{}{}‘.format(self.timestamp,random_int),upper=True)
        list_md5=list(md5_string)
        list_md5.insert(8,‘-‘),list_md5.insert(13,‘-‘),list_md5.insert(18,‘-‘),list_md5.insert(23,‘-‘)
        ticket_str=‘‘.join(list_md5)
        # print(ticket_str)
        return ticket_str

    # # 授权参数计算
    def kuaishou_sign_calc(self,sign_data=None):
        if sign_data==None:
            # 将参数转换为dict并排序
            str_dict = self.requestData.strToDict(self.public_data, space_one=‘=‘, space_two=‘&‘)
            str_dict = self.requestData.sortedDict(str_dict)
            # 将dict转换为str
            sort_str = self.requestData.dictToStr(str_dict)
            self.data=‘data={}&{}‘.format(self.data,sort_str)
            input_auth_str = ‘{}{}‘.format(self.data,self.secret_key)
        else:
            input_auth_str=sign_data
        print(input_auth_str)
        auth_str=self.requestData.md5_Encry(input_auth_str,upper=True)
        print(auth_str)
        return auth_str

    # 获取商户信息与资质信息
    def kuaishou_openshop_merchandiseInfo(self,data):
        url=self.yaml[‘gld_api_host‘]+‘/api/snack/open/kuaishou/commercial/info‘
        self.data=json.dumps(data)
        self.headers[‘Authorization‘]=self.kuaishou_sign_calc()
        requets=self.request.run_main(‘post‘,url=url,data=self.data,headers=self.headers)
        print(json.dumps(requets,indent=2,ensure_ascii=False))

　　遇到得坑：中文需要进行unicode编码，不然一直提示签名错误，坑了我好久

　　计算签名时，python中的json.dumps方法会自动对json中得中文进行unicode编码，而beanshell中的hashmap转换为json对象时则不会，需要需要手动转换编码

　　不知道java中有没有类似json.dumps得方法，希望各位大佬支个招，能用更好得方式实现