JUC高併發編程(三)之模擬接口壓力測試

1.背景

接口壓力測試是產品上線前很重要的一項測試,我們可以使用很多開源工具測試,

當然我們也可以簡單的寫一個多線程併發測試案例

2.代碼

controller接口

 /**
     * 查詢訂單
     *
     * @return
     */
    @RequestMapping("/api/order")
    public Object product(Integer id) {
        // 爲了便於分析,設置一個線程號
        Thread.currentThread().setName("thread-" + id);
        log.info("查詢訂單-" + id);
        // 模擬隨機耗時
        ThreadUtil.sleepRandom();
        return "訂單編號-" + id;
    }

 

3.測試

測試代碼

package com.ldp.jucproject.controller;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpUtil;
import com.ldp.jucproject.utils.ThreadUtil;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;

import java.util.concurrent.CountDownLatch;

/**
 * @author 姿勢帝-博客園
 * @address https://www.cnblogs.com/newAndHui/
 * @WeChat 851298348
 * @create 11/06 10:18
 * @description
 */
@Slf4j
class OrderControllerTest {
    /**
     * 簡單測試
     */
    @Test
    void product01() {
        Integer id = 1;
        String url = "http://localhost:8001/api/order?id=" + id;
        HttpRequest request = HttpUtil.createGet(url);
        String response = request.execute().body();
        System.out.println("response=" + response);
    }

    /**
     * 模擬多個請求
     */
    @Test
    void product02() {
        for (int i = 1; i <= 100; i++) {
            Integer id = i;
            String url = "http://localhost:8001/api/order?id=" + id;
            HttpRequest request = HttpUtil.createGet(url);
            String response = request.execute().body();
            System.out.println("response=" + response);
        }
    }

    /**
     * 模擬多線程請求
     */
    @Test
    void product03() throws InterruptedException {
        for (int i = 1; i <= 100; i++) {
            ThreadUtil.sleepRandom();
            Integer id = i;
            new Thread(() -> {
                String url = "http://localhost:8001/api/order?id=" + id;
                System.out.println("待查詢訂單號=" + id);
                HttpRequest request = HttpUtil.createGet(url);
                String response = request.execute().body();
                System.out.println("response=" + response);
            }).start();
        }
        // 避免線程終止
        Thread.sleep(20 * 1000);
    }

    /**
     * 模擬多線程併發請求
     */
    @Test
    void product04() throws Exception {
        // 併發請求數
        int num = 100;
        CountDownLatch countDownLatch = new CountDownLatch(num);
        for (int i = 1; i <= num; i++) {
            ThreadUtil.sleepRandom();
            // 計數器減一
            countDownLatch.countDown();
            Integer id = i;
            new Thread(() -> {
                try {
                    String url = "http://localhost:8001/api/order?id=" + id;
                    // 等待計數器歸零,歸零前都是處於阻塞狀態
                    System.out.println("待查詢訂單號=" + id);
                    countDownLatch.await();
                    HttpRequest request = HttpUtil.createGet(url);
                    String response = request.execute().body();
                    System.out.println("response=" + response);
                } catch (Exception e) {
                    log.error("模擬併發出錯:{}", e);
                }
            }).start();
        }
        // 避免線程終止
        Thread.sleep(60 * 1000);
    }
}

 

4.完美!