如何使用JMeter 中beanshell sample实现 POST 请求并处理响应

在 JMeter 的性能测试中，可以通过 Java 代码来实现复杂的 POST 请求并处理响应。
以下是一个详细的示例代码解析：

import java.io.OutputStreamWriter;
import java.io.InputStreamReader;
import java.io.BufferedReader;
import java.net.HttpURLConnection;
import java.net.URL;

public class HttpPostExample {

    public static void main(String[] args) {
        try {
            // 设置请求的 URL
            String urlStr = "http://127.0.0.1:1234/login";
            URL url = new URL(urlStr);

            // 打开连接
            HttpURLConnection connection = (HttpURLConnection) url.openConnection();

            // 设置请求方法为 POST
            connection.setRequestMethod("POST");

            // 设置请求头
            connection.setRequestProperty("Content-Type", "application/json");
            connection.setRequestProperty("Accept", "application/json");

            // 允许写入请求体
            connection.setDoOutput(true);

            // 请求体内容
            String jsonInputString = "{\"username\": \"admin\", \"password\": \"123456\"}";

            // 写入请求体
            OutputStreamWriter writer = null;
            try {
                writer = new OutputStreamWriter(connection.getOutputStream(), "UTF-8");
                writer.write(jsonInputString);
                writer.flush();
            } catch (Exception e) {
                System.err.println("Error writing request body: " + e.getMessage());
            } finally {
                if (writer != null) {
                    try {
                        writer.close();
                    } catch (Exception e) {
                        System.err.println("Error closing writer: " + e.getMessage());
                    }
                }
            }

            // 获取响应代码
            int responseCode = 0;
            try {
                responseCode = connection.getResponseCode();
                System.out.println("Response Code: " + responseCode);
            } catch (Exception e) {
                System.err.println("Error getting response code: " + e.getMessage());
            }

            // 读取响应内容
            BufferedReader in = null;
            StringBuilder response = new StringBuilder();
            try {
                in = new BufferedReader(new InputStreamReader(connection.getInputStream(), "UTF-8"));
                String inputLine;
                while ((inputLine = in.readLine()) != null) {
                    response.append(inputLine);
                }
            } catch (Exception e) {
                System.err.println("Error reading response: " + e.getMessage());
            } finally {
                if (in != null) {
                    try {
                        in.close();
                    } catch (Exception e) {
                        System.err.println("Error closing reader: " + e.getMessage());
                    }
                }
            }

            // 将响应内容打印到 JMeter 日志
            System.out.println("Response: " + response.toString());

            // 将响应数据存储到 JMeter 变量中，以便在 View Results Tree 中显示
            // SampleResult.setResponseData(response.toString(), "UTF-8");

        } catch (Exception e) {
            System.err.println("Error: " + e.getMessage());
        }
    }
}

解析示例代码

引言

在 JMeter 的性能测试中，通常需要模拟各种类型的 HTTP 请求，其中包括 POST 请求。假如你想研究下beanshell sampler，那本文将演示如何使用 Java 编写的代码，在 JMeter 中实现一个简单的 POST 请求，并处理服务器的响应。

设置请求的 URL

String urlStr = "http://127.0.0.1:1234/login";
URL url = new URL(urlStr);

首先，定义要发送 POST 请求的目标 URL 。

打开连接并设置请求方法为 POST

HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setRequestMethod("POST");

通过 openConnection 方法获取到与目标 URL 的连接，并设置请求方法为 POST 。

设置请求头和允许写入请求体

connection.setRequestProperty("Content-Type", "application/json");
connection.setRequestProperty("Accept", "application/json");
connection.setDoOutput(true);

设置请求头信息，包括 Content-Type 和 Accept ，指定服务器要求的数据格式为 JSON，并允许向请求体写入数据。

写入请求体内容

String jsonInputString = "{\"username\": \"admin\", \"password\": \"123456\"}";

OutputStreamWriter writer = null;
try {
    writer = new OutputStreamWriter(connection.getOutputStream(), "UTF-8");
    writer.write(jsonInputString);
    writer.flush();
} catch (Exception e) {
    System.err.println("Error writing request body: " + e.getMessage());
} finally {
    if (writer != null) {
        try {
            writer.close();
        } catch (Exception e) {
            System.err.println("Error closing writer: " + e.getMessage());
        }
    }
}

在这里，定义了请求体的 JSON 数据，并将其写入到连接的输出流中。同时，对可能出现的异常进行了处理，并在结束时关闭了输出流。

获取响应代码和读取响应内容

int responseCode = 0;
try {
    responseCode = connection.getResponseCode();
    System.out.println("Response Code: " + responseCode);
} catch (Exception e) {
    System.err.println("Error getting response code: " + e.getMessage());
}

BufferedReader in = null;
StringBuilder response = new StringBuilder();
try {
    in = new BufferedReader(new InputStreamReader(connection.getInputStream(), "UTF-8"));
    String inputLine;
    while ((inputLine = in.readLine()) != null) {
        response.append(inputLine);
    }
} catch (Exception e) {
    System.err.println("Error reading response: " + e.getMessage());
} finally {
    if (in != null) {
        try {
            in.close();
        } catch (Exception e) {
            System.err.println("Error closing reader: " + e.getMessage());
        }
    }
}

获取服务器返回的响应代码，并记录在日志中。然后，使用 BufferedReader 逐行读取服务器返回的响应内容，并将其添加到 StringBuilder 中。同样处理了可能出现的异常，并在结束时关闭了输入流。

打印响应内容到日志和存储响应数据

System.out.println("Response: " + response.toString());
// SampleResult.setResponseData(response.toString(), "UTF-8");

最后，将响应内容打印到控制台日志中。如果需要在 JMeter 中查看响应内容，可以使用 SampleResult.setResponseData 方法将响应数据存储到 JMeter 变量中，以便在 View Results Tree 中显示。