自动化完成1000个用户的登录并获取token并生成tokens.txt文件

自动化完成1000个用户的登录并获取token并生成tokens.txt文件

• 写作背景

  在我学习使用redis实现秒杀功能的过程中，在编写完秒杀代码后，需要使用Jmeter实际测试1000个用户进行秒杀，由于秒杀功能需要在用户登录完成后才能实现，用户是否登录是由登录拦截器实现的，用户在登录成功后，后台会生成一个token然后发送给浏览器，每次发送秒杀请求时，登录拦截器都会判断浏览器中是否有token，没有token就说明用户未登录，不能进行秒杀，只有含有token的秒杀请求才会被执行，具体流程图如下：
  现在存在的问题是：我们想要发送/voucher-order/seckill/id请求，就先需要实现/user/code和/user/login请求，而这两个请求实现很简单，但是需要实现1000次，同时获取这1000给token用于Jmeter压力测试。这总不可能手动测试把，所以就需要通过代码实现了（详情见P69）

• 代码实现

  核心实现思路：通过编写一个测试类，然后使用MockMvc 发送请求，最终将获取到的token写入tokens.txt文件中（如果您有更好的方法，请留言告知在下，在下不胜感激）

  package com.hmdp.shop;import cn.hutool.core.lang.Assert;
  import cn.hutool.core.thread.ThreadUtil;
  import com.fasterxml.jackson.databind.ObjectMapper;
  import com.hmdp.dto.LoginFormDTO;
  import com.hmdp.dto.Result;
  import com.hmdp.entity.User;
  import com.hmdp.service.IUserService;
  import lombok.SneakyThrows;
  import lombok.extern.slf4j.Slf4j;
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
  import org.springframework.boot.test.context.SpringBootTest;
  import org.springframework.http.MediaType;
  import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  import org.springframework.test.web.servlet.MockMvc;
  import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
  import org.springframework.test.web.servlet.result.MockMvcResultMatchers;import javax.annotation.Resource;
  import java.io.BufferedWriter;
  import java.io.File;
  import java.io.FileOutputStream;
  import java.io.OutputStreamWriter;
  import java.nio.charset.StandardCharsets;
  import java.util.List;
  import java.util.concurrent.CopyOnWriteArrayList;
  import java.util.concurrent.CountDownLatch;
  import java.util.concurrent.ExecutorService;
  import java.util.stream.Collectors;/*** @author ghp* @date 2023/2/7* @title* @description*/
  @SpringBootTest
  @RunWith(SpringJUnit4ClassRunner.class)
  @AutoConfigureMockMvc
  @Slf4j
  public class GenerateToken {@Resourceprivate MockMvc mockMvc;@Resourceprivate IUserService userService;@Resourceprivate ObjectMapper mapper;@Test// 忽视异常@SneakyThrowspublic void login() {// 查询数据库得到1000个号码List<String> phoneList = userService.lambdaQuery().select(User::getPhone).last("limit 1000").list().stream().map(User::getPhone).collect(Collectors.toList());// 使用线程池，线程池总放入1000个号码，提高效率ExecutorService executorService = ThreadUtil.newExecutor(phoneList.size());// 创建List集合，存储生成的token。多线程下使用CopyOnWriteArrayList，实现读写分离，保障线程安全（ArrayList不能保障线程安全）List<String> tokenList = new CopyOnWriteArrayList<>();// 创建CountDownLatch（线程计数器）对象，用于协调线程间的同步CountDownLatch countDownLatch = new CountDownLatch(phoneList.size());// 遍历phoneList，发送请求，然后将获取的token写入tokenList中phoneList.forEach(phone -> {executorService.execute(() -> {try {// 发送获取验证码的请求，获取验证码String codeJson = mockMvc.perform(MockMvcRequestBuilders.post("/user/code").queryParam("phone", phone)).andExpect(MockMvcResultMatchers.status().isOk()).andReturn().getResponse().getContentAsString();// 将返回的JSON字符串反序列化为Result对象Result result = mapper.readerFor(Result.class).readValue(codeJson);Assert.isTrue(result.getSuccess(), String.format("获取“%s”手机号的验证码失败", phone));String code = result.getData().toString();// 创建一个登录表单// 使用建造者模式构建 登录信息对象，我这里就没有使用了，我是直接使用new（效率较低不推荐使用）
  //                    LoginFormDTO formDTO = LoginFormDTO.builder().code(code).phone(phone).build();LoginFormDTO formDTO = new LoginFormDTO();formDTO.setCode(code);formDTO.setPhone(phone);// 将LoginFormDTO对象序列化为JSONString json = mapper.writeValueAsString(formDTO);// 发送登录请求，获取token// 发送登录请求，获取返回信息（JSON字符串，其中包含token）String tokenJson = mockMvc.perform(MockMvcRequestBuilders.post("/user/login").content(json).contentType(MediaType.APPLICATION_JSON)).andExpect(MockMvcResultMatchers.status().isOk()).andReturn().getResponse().getContentAsString();// 将JSON字符串反序列化为Result对象result = mapper.readerFor(Result.class).readValue(tokenJson);Assert.isTrue(result.getSuccess(), String.format("获取“%s”手机号的token失败,json为“%s”", phone, json));String token = result.getData().toString();tokenList.add(token);// 线程计数器减一countDownLatch.countDown();} catch (Exception e) {e.printStackTrace();}});});// 线程计数器为0时，表示所有线程执行完毕，此时唤醒主线程countDownLatch.await();// 关闭线程池executorService.shutdown();Assert.isTrue(tokenList.size() == phoneList.size());// 所有线程都获取了token，此时将所有的token写入tokens.txt文件中writeToTxt(tokenList, "\\tokens.txt");log.info("程序执行完毕！");}/*** 生成tokens.txt文件* @param list* @param suffixPath* @throws Exception*/private static void writeToTxt(List<String> list, String suffixPath) throws Exception {// 1. 创建文件File file = new File(System.getProperty("user.dir") + "\\src\\main\\resources" + suffixPath);if (!file.exists()) {file.createNewFile();}// 2. 输出BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), StandardCharsets.UTF_8));for (String content : list) {bw.write(content);bw.newLine();}bw.close();log.info("tokens.txt文件生成完毕！");}
  }
  

  测试结果：
  
  
  注意点：

  1. 需要使用前后端协议，将后端返回数据统一封装到Result类中
  2. 在获取验证码时，一定要记得将验证码封装到Result中，即：Result.ok(code)，但在实际业务中是Result.ok()，因为生成的验证码不可能直接返回给浏览器吧，所以在测试完成后，一定要记得改回来