jmeter使用beanshell完成签名计算，附与python代码对比
签名计算过程：
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+"×tamp="+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("数据处理失败，原因为:\n%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得方法，希望各位大佬支个招，能用更好得方式实现
