基于Groovy编写Ngrinder脚本常用方法
1、生成随机字符串（import org.apache.commons.lang.RandomStringUtils） 数字：RandomStringUtils.randomNumeric(length); 字母：RandomStringUtils.randomAlphabetic(length); 字母加数字：RandomStringUtils.randomAlphanumeric(length); 所有ASCCII字符：RandomStringUtils.randomAscii(length); 自定义混合字符：RandomStringUtils.randomAscii(length, string); 2、生成随机数字：（import java.util.concurrent.ThreadLocalRandom;） 数字：int random_number = ThreadLocalRandom.current().nextInt(min_num, max_num); 3、获取项目数据文件路径 common项目："/resources/account.txt" maven项目：Thread.currentThread().getContextClassLoader().getResource("account.txt").getPath(); maven项目获取文件内容：ReflectionUtils.getCallingClass(0).getResourceAsStream("/account.txt").getText("UTF-8") 4、读取文件： txt每行单数据： String[] file_arrary = new File("/resources/account.txt") as String[]; String file_data = file_arrary[arrary_index]; txt每行双数据： String[] file_arrary = new File("/resources/account.txt") as String[]; String data_one = file_arrary[arrary_index].split(",")[0]; String data_two = file_arrary[arrary_index].split(",")[1]; 另一种方法： List<String> reqDataArrList = new File(dataFilePath).readLines() String data_one = reqDataArrList.get(arrary_index).split(",")[0]; String data_two = reqDataArrList.get(arrary_index).split(",")[1]; txt每行多数据可参考双数据方法。也可以参考json方式存储： BufferedReader txt_content=new BufferedReader(new FileReader(new File("/resources/account.txt"))) data_json = new JSONObject() String text_line = "" while(( text_line=txt_content.readLine())!=null){ data_json.put(text_line.split(",")[0],text_line.split(",")[1]) } String data_one = data_json.keys[0] String data_two = data_json.getString(data_one) 5、写入文件： 覆盖写入： def write = new File(file_path, file_name).newPrintWriter(); write.write(write_text); write.flush(); write.close() 追加写入： def write = new File(file_path, file_name).newPrintWriter(); write.append(write_text); write.flush(); write.close() 6、json文件的数据处理(import org.ngrinder.recorder.RecorderUtils) json文件读取： String json_str = new File(file_path).getText("UTF-8") def json_object = RecorderUtils.parseRequestToJson(json_str) 长度：json_object.length() 关键字：json_object.keys() 添加元素：json_object.put(name, value) 修改元素：json_object.put(name, value) 删除元素：json_object.remove(name, value) 获取对应value：json_object.getString(name) 7、字符串的处理 字符串截取：String new_str = old_str[0..3] 字符串替换：String string = str.replace("old","new") 字符串统计：int count = string.count("char") 字符串转化：int int_num = Integer.parseInt(string) 1、设置多个请求事务(即多个test方法) 1)设置多个静态Gtest对象： public static GTest test1 public static GTest test2 2)实例化多个Gtest对象： test1 = new GTest(1, "test1"); test2 = new GTest(2, "test2"); 3)监听多个test请求： test1.record(this, "test1") test2.record(this, "test2") 4)定义多个test方法： public void test1(){ grinder.logger.info("---ones: {}---", grinder.threadNumber+1) } public void test2(){ grinder.logger.info("---twos: {}---", grinder.threadNumber+1) } 2、Ngrinder定义请求参数集： add方法： List<NVPair> paramList = new ArrayList<NVPair>(); paramList.add(new NVPair("name", "value")); paramList.add(new NVPair("name", "value")); params = paramList.toArray(); new方法： params = [new NVPair("name", "value"), new NVPair("name", "value")]; 3、Ngrinder处理日志： 日志级别(三种常见)： grinder.logger.info("----before process.----"); grinder.logger.warn("----before process.----"); grinder.logger.error("----before process.----"); 日志限定(仅打印error级别) ： 1)导入依赖包 import ch.qos.logback.classic.Level; import org.slf4j.LoggerFactory; 2)设定级别 @BeforeThread LoggerFactory.getLogger("worker").setLevel(Level.ERROR); 3)设置打印语句 @test grinder.logger.error("----error.----"); 日志输出(输出所有进程日志)：将每个agent的.ngrinder_agent/agent.conf中一项修改为agent.all_logs=true 日志打印：打印变量：grinder.logger.error("{},{}",variable1,variable2); // 换行或缩进可在""中加\n或\t 4、Ngrinder的cookie处理 1) 登录产生cookie @BeforeThread login_get_cookie(); // 调用登录方法 cookies = CookieModule.listAllCookies(HTTPPluginControl.getThreadHTTPClientContext()); // 配置cookie管理器 2) 读取控制器中cookie @Before cookies.each { CookieModule.addCookie(it, HTTPPluginControl.getThreadHTTPClientContext()) } 5、Ngrinder请求方式： 1)通过url加参数直接访问： post方法： HTTPResponse result = request.POST("http://192.168.2.135:8080/blogs", params, headers) get方法： HTTPResponse result = request.GET("http://192.168.2.135:8080/blogs", params, headers) 参数是json：设置请求头参数{"Content-Type": "application/json"} 2)通过参数化所有请求数据为json对象(导入import org.ngrinder.recorder.RecorderUtils) HTTPResponse result = RecorderUtils.sendBy(request, req_data_json) HTTPResponse result = RecorderUtils.sendBy(request, req_data_json) 6、Ngringer的test运行次数设定(将总运行测试次数按百分比例分配到相应test)： 1)引用依赖包： import net.grinder.scriptengine.groovy.junit.annotation.RunRate 2)设置运行次数百分比(所有test设定的比例值不够100，那不满的部分不运行，比如设定总比80，只运行这80部分)： @RunRate(50) // 数字代表百分比 @Test public void test1(){} @RunRate(50) // 数字代表百分比 @Test public void test2(){} 7、Ngringer获取设置的加压机总数、进程总数、线程总数等信息： int tota_agents = Integer.parseInt(grinder.getProperties().get("grinder.agents").toString()) // 设置的总加压机数 int total_processes = Integer.parseInt(grinder.properties().get("grinder.processes").toString()) // 设置的总进程数 int total_threads = Integer.parseInt(grinder.properties().get("grinder.threads").toString()) // 设置的总线程数 int total_runs = Integer.parseInt(grinder.properties().get("grinder.runs").toString()) // 设置的总运行次数(若设置的是运行时长，则得到0) 8、Ngringer获取当前运行的加压机编号、进程编号、线程编号等信息(都从0递增)： int agent_number = grinder.agentNumber // 当前运行的加压机编号 int process_number = grinder.processNumber // 当前运行的进程编号 int thread_number = grinder.threadNumber // 当前运行的线程编号 int run_number = grinder.runNumber // 当前运行的运行次数编号 9、Ngringer获取唯一递增值方法(从1递增，不重复)： // 传递接口参数runNumber(即def runNumber = grinder.runNumber) private int getIncrementId(int runNumber){ // 获取压力机总数、进程总数、线程总数 int totalAgents = Integer.parseInt(grinder.getProperties().get("grinder.agents").toString()) int totalProcess = Integer.parseInt(grinder.getProperties().get("grinder.processes").toString()) int totalThreads = Integer.parseInt(grinder.getProperties().get("grinder.threads").toString()) // 获取当前压力机数、进程数、线程数 int agentNum = grinder.agentNumber int processNum = grinder.processNumber int threadNum = grinder.threadNumber // 获取唯一递增数id int incrementId = agentNum * totalProcess * totalThreads + processNum * totalThreads + threadNum + totalAgents * totalProcess * totalThreads * runNumber return incrementId } 10、Ngringer根据唯一递增值获取参数化文件中的唯一行号： 1)需要设置静态变量：private enum WhenOutOfValues { AbortVuser, ContinueInCycleManner, ContinueWithLastValue } 2)传递接口参数fileDataList(即def fileDataList = new File(dataFilePath).readLines()) private int getLineNum(def fileDataList) { // 获取当前运行数、数据读取行数、数据最大行数 int counter = getIncrementId(grinder.runNumber) int lineNum = counter + 1 int maxLineNum = fileDataList.size() - 1 // 读取最大值的判断处理 WhenOutOfValues outHandler = WhenOutOfValues.AbortVuser if (lineNum > maxLineNum) { if(outHandler.equals(WhenOutOfValues.AbortVuser)) { lineNum = maxLineNum //grinder.stopThisWorkerThread() } else if (outHandler.equals(WhenOutOfValues.ContinueInCycleManner)) { lineNum = (lineNum - 1) % maxLineNum + 1 } else if (outHandler.equals(WhenOutOfValues.ContinueWithLastValue)) { lineNum = maxLineNum } } return lineNum } 11、Ngrinder日志输出配置的测试信息：(import java.text.SimpleDateFormat) public static String getTestInfo(){ String time_string = "" // 获取压测时设置的进程总数、线程总数、运行次数并在log中打印 int all_process = grinder.getProperties().getInt("grinder.processes", 1) // 设置的总进程数 int all_threads = grinder.getProperties().getInt("grinder.threads", 1) // 设置的总线程数 int all_runs = grinder.getProperties().getInt("grinder.runs", 1) // 设置的总运行次数(若设置的是运行时长，则得到0) int all_duration = grinder.getProperties().getLong("grinder.duration", 1) // 设置的总运行时长(若设置的是运行次数，则得到0) // 格式化时间毫秒输出(输出格式00:00:00) SimpleDateFormat formatter = new SimpleDateFormat("HH:mm:ss") formatter.setTimeZone(TimeZone.getTimeZone("GMT+00:00")) String all_duration_str = formatter.format(all_duration) if (all_duration_str.equals("00:00:00")) time_string = "Test information: the processes is "+all_process+", the threads is "+all_threads+", the run count is "+all_runs+"." else time_string = "Test information: the processes is "+all_process+", the threads is "+all_threads+", the run time is "+all_duration_str+"." return time_string } 12、Ngrinder打印所有的配置信息 String property = grinder.getProperties(); grinder.logger.info("------- {}", property) ; 13、Ngrinder获取请求返回值： HTTPResponse result = request.POST("http://192.168.2.135:8080/blogs", params, headers) 返回的文本：grinder.logger.info("----{}----", result.getText()) // 或者result.text 返回的状态码：grinder.logger.info("----{}----", result.getStatusCode()) // 或者result.statusCode 返回的url：grinder.logger.info("----{}----", result.getEffectiveURI()) 返回的请求头所有参数：grinder.logger.info("---\n{}---", result) 返回的请求头某参数：grinder.logger.info("----{}---- ", result.getHeader("Content-type")) 14、Ngrinder返回值的匹配： 匹配状态码：assertThat(result.getStatusCode(), is(200)) 匹配包含文本：assertThat(result.getText(), containsString("success")) 15、Ngrinder获取所有虚拟用户数： public int getVusers() { int totalAgents = Integer.parseInt(grinder.getProperties().get("grinder.agents").toString()); int totalProcesses = Integer.parseInt(grinder.getProperties().get("grinder.processes").toString()); int totalThreads = Integer.parseInt(grinder.getProperties().get("grinder.threads").toString()); int vusers = totalAgents * totalProcesses * totalThreads; return vusers; } 16、Ngrinder的断言和error日志输出 if (result.statusCode == 301 || result.statusCode == 302) { grinder.logger.error("Possible error: {} expected: <200> but was: <{}>.",result.getEffectiveURI(),result.statusCode); } else { assertEquals((String)result.getEffectiveURI(), result.statusCode, 200) assertThat((String)result.getEffectiveURI(), result.statusCode, is(200)) }
参考博文：https://www.cnblogs.com/fxcity/p/11251671.html
