程序员的简单接口压测

原创

liangtu33 2019-03-27 16:13:54 ©著作权

文章标签 restful mysql 压测 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者liangtu33的原创作品，请联系作者获取转载授权，否则将追究法律责任

接口必然需要压测，但有时候往往没有专门压测人员，程序员自己动手。以实名接口为例，传入姓名、×××号上有返回一致、不一致等信息，一致信息则插入用户信息表，下次调用时先查询该表，如果有直接返回结果。

1.restful接口使用springboot很容易搭建接口，过程略

2.造数据 2.1java批量插入数据，略 2.2存储过程造数据 2.3sql造数据

-- 建表
drop table if exists `t_identity_info`;
create table `t_identity_info` (
  `id` int(11) not null auto_increment comment '主键id',
  `real_name` varchar(20) default null comment '姓名',
  `id_card` varchar(18) not null comment '×××号',
  `create_time` varchar(19) not null comment '创建时间',
  primary key (`id`),
  key `index_id_card` (`id_card`)
) engine=innodb auto_increment=1 default charset=utf8 comment='用户信息表';


-- 存储过程
drop procedure if exists insert_t_identity_info_data;

delimiter ;;
create procedure insert_t_identity_info_data()
begin 
declare i int;
declare tempName varchar(4);
declare tempAreaCode varchar(6);
declare tempDate varchar(8);

set i = 0;
while i < 10000 do
set tempAreaCode = concat((CEILING(RAND()*1000 mod 50) + 10),(CEILING(RAND()*1000 mod 50) + 10),(CEILING(RAND()*1000 mod 50) + 10)) ; 
set tempDate = DATE_FORMAT(date_add(now(), interval - CEILING(RAND()*1000000 mod 10000) day),'%Y%m%d');
set tempName = concat(substring('赵钱孙李周吴郑王冯陈褚卫蒋沈韩',CEILING(RAND()*1000 mod 15),1),
substring('嘉懿煜城懿轩烨伟苑博伟泽熠彤鸿',CEILING(RAND()*1000 mod 15),1));

insert into t_identity_info (real_name,id_card,create_time) values 
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now()),
(tempName,concat(tempDate,CEILING(RAND()*100000) mod 9999+1000),now());
select count(1) from t_identity_info;
set i = i +1; 

end while; 
end 
;;
DELIMITER ;

-- 调用存储过程
call insert_t_identity_info_data();

-- 循环造数据，执行一次数据量*2
insert into t_identity_info(real_name,id_card,create_time) select real_name,id_card,now() from t_identity_info;

完成接口的httpclient

3.压测运行spring boot 运行n个接口客户端

注意：不要在本机压测，数据库、服务、客户端都应在服务器上，压测至少半个小时，根据tps启动客户端个数、每个客户端的数量；检测服务器的内存、cpu；考虑网络瓶颈。

package com.ld.api;

import java.util.Date;
import java.util.Map;

import org.apache.commons.lang3.RandomStringUtils;
import org.apache.commons.lang3.time.DateFormatUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.alibaba.fastjson.JSONObject;
import com.ld.util.LoggerUtil;

@RestController
@RequestMapping("/identity")
public class IdentityApi {

	private static final Logger logger = LoggerFactory.getLogger(LoggerUtil.class);

	@PostMapping("/check")
	public String check(@RequestBody Map<String, String> map) {
		String transNo = getTransNoByType("identityCheck");
		long beginTime = System.currentTimeMillis();
		logger.info("transNo = {} , map = {}", transNo, map);

		// 查询db
		sleepSomeTime();

		// 返回结果
		String result = null;

		JSONObject jsonObject = new JSONObject();
		try {
			jsonObject.put("transNo", transNo);
			jsonObject.put("state", System.currentTimeMillis() % 2 + 1);
			jsonObject.put("costTime", System.currentTimeMillis() - beginTime);// 接口耗费时间
		} catch (Exception e) {
			e.printStackTrace();
		}

		result = jsonObject.toString();
		logger.info("transNo = {} , result = {}", transNo, result);
		return result;
	}

	/**
	 * 通过流水号串联日志，方便定位问题
	 * 
	 * @param transNoType
	 * @return 2019年3月27日
	 */
	private String getTransNoByType(String transNoType) {
		return transNoType + "_" + DateFormatUtils.format(new Date(), "yyMMddHHmm") + "_"
				+ RandomStringUtils.randomAlphabetic(4);
	}

	/**
	 * 随机休眠
	 * 
	 * 2019年3月27日
	 */
	private void sleepSomeTime() {
		try {
			Thread.sleep(System.currentTimeMillis() % 300);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
	}

}

package com.ld;

import org.apache.http.HttpEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;
import org.json.JSONObject;

public class IdentityCheck {
	public static void main(String[] args) {
		try {

			for (int i = 0; i < 100; i++) {
				check();
			}

		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static void check() throws Exception {
		CloseableHttpClient httpclient = HttpClientBuilder.create().build();
		String loginUrl = "http://localhost:8080/identity/check";

		HttpPost httppost = new HttpPost(loginUrl);
		httppost.addHeader("Content-Type", "application/json;charset=UTF-8");

		JSONObject jsonObject = new JSONObject();
		jsonObject.put("idCard", "131024199901010011");
		jsonObject.put("realName", "李四");

		StringEntity stringEntity = new StringEntity(jsonObject.toString(), "utf-8");
		httppost.setEntity(stringEntity);

		CloseableHttpResponse response = httpclient.execute(httppost);
		System.out.println("response = " + response.toString());

		HttpEntity entity = response.getEntity();
		String jsonStr = EntityUtils.toString(entity, "utf-8");
		System.out.println("jsonStr = " + jsonStr);

		httppost.releaseConnection();
	}

}