Apache Jmeter進階-RPC服務壓測

在上一節Apache Jmeter入門中，我們簡單介紹了Jmeter的使用方法。
本節中，我們將着重介紹Jmeter壓測，以Thrift RPC爲例子進行講解。

1、準備

我們依然採用Thrift RPC這節中的RPC服務作爲服務端程序並運行。

2、Jmeter測試類編寫

2.1 Jmeter測試類簡介

• 新建一個普通的Java工程
• 添加JMeter Maven依賴。

<dependency>
    <groupId>org.apache.jmeter</groupId>
    <artifactId>ApacheJMeter_core</artifactId>
    <version>2.13</version>
 </dependency>
 <dependency>
    <groupId>org.apache.jmeter</groupId>
    <artifactId>ApacheJMeter_java</artifactId>
    <version>2.13</version>
</dependency>

• 新建一個Java Class，如下例中的TestThriftByJmeter，並繼承AbstractJavaSamplerClient。
  AbstractJavaSamplerClient中默認實現了四個可以覆蓋的方法，分別是getDefaultParameters()，setupTest()，runTest()和teardownTest()方法。

  • getDefaultParameters 方法主要用於設置傳入界面的參數；
  • setupTest方法爲初始化方法，用於初始化性能測試時的每個線程；
  • runTest方法爲性能測試時的線程運行體；
  • teardownTest方法爲測試結束方法，用於結束性能測試中的每個線程。

2.2 RPC Thrift測試類編寫

ThriftClient

public class ThriftClient {
    private ComputeServer.Client client = null;
    private TTransport transport = null;

    protected ThriftClient(String ip, int port) {
        try {
            TTransport transport = new TFramedTransport(new TSocket(ip, port));
            transport.open();
            TProtocol protocol = new TBinaryProtocol(transport);
            client = new ComputeServer.Client(protocol);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    protected ComputeResponse getResponse(ComputeRequest request) {
        try {
            ComputeResponse response = client.getComputeResult(request);
            return response;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    protected void close() {
        if (transport != null && transport.isOpen()) {
            transport.close();
        }
    }
}

TestThriftByJmeter

public class TestThriftByJmeter extends AbstractJavaSamplerClient {
    private ThriftClient client;
    private ComputeRequest request;
    private ComputeResponse response;

    @Override
    public Arguments getDefaultParameters() {
        Arguments arguments = new Arguments();
        arguments.addArgument("ip", "127.0.0.1");
        arguments.addArgument("port", "9000");
        arguments.addArgument("X", "0");
        arguments.addArgument("Y", "0");
        arguments.addArgument("type", "0");
        return arguments;
    }

    @Override
    public void setupTest(JavaSamplerContext context) {
        try {
            // 獲取參數
            String ip = context.getParameter("ip");
            int port = context.getIntParameter("port");
            int x = context.getIntParameter("X");
            int y = context.getIntParameter("Y");
            ComputeType type = getComputeType(context.getIntParameter("type"));

            // 創建客戶端
            client = new ThriftClient(ip, port);
            // 設置request請求
            request = new ComputeRequest(x, y, type);
        } catch (Exception e) {
            e.printStackTrace();
        }
        super.setupTest(context);
    }

    public SampleResult runTest(JavaSamplerContext context) {
        SampleResult result = new SampleResult();
        result.sampleStart();
        try {
            long begin = System. currentTimeMillis();
            response = client.getResponse(request);
            long cost = (System.currentTimeMillis() - begin);
            System.out.println(response.toString() + " cost:[" + cost + "ms]");

            if (response == null) {
                result.setSuccessful(false);
                return result;
            }
            if (response.getErrorNo() == 0) {
                result.setSuccessful(true);
            } else {
                result.setSuccessful(false);
            }
        } catch (Exception e) {
            result.setSuccessful(false);
            e.printStackTrace();
        }  finally {
            result.sampleEnd();
        }
        return result;
    }

    public void tearDownTest(JavaSamplerContext context) {
        if (client != null) {
            client.close();
        }
        super.teardownTest(context);
    }

    ComputeType getComputeType(int type) {
        ComputeType computeType;
        switch (type) {
            case 0: computeType = ComputeType.ADD; break;
            case 1: computeType = ComputeType.SUB; break;
            case 2: computeType = ComputeType.MUL; break;
            case 3: computeType = ComputeType.DIV; break;
            default: computeType = ComputeType.ADD; break;
        }
        return computeType;
    }
}

代碼結構：

3、Jmeter測試

3.1 添加依賴

首先將2.2節中的代碼編譯打包，並使用命令mvn dependency:copy-dependencies -DoutputDirectory=lib將該代碼的Jar包和所依賴的包，添加到Jmeter的apache-jmeter-2.13\lib\ext目錄下。

3.2 Jmeter設置

打開Jmeter，設置如下信息。

在添加Java請求時，注意要選擇2.2節中的Jmeter測試類，下面的列表中可以看到參數和默認值。

3.3 測試

下面我們將進行持續一分鐘的的性能壓測，首先設置線程組，設置10個併發線程。

之後進行測試，測試結束後，可以在聚合報告中看到此次壓測的結果。

服務端日誌：