軟體效能測試分析與調優實踐之路-JMeter對RPC服務的效能壓測分析與調優-手稿節選

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
   <groupId>jmeter.tools</groupId>
    <artifactId>jmeter-grpc</artifactId>
    <packaging>jar</packaging>
    <version>1.0-SNAPSHOT</version>
    <properties>
        <grpc.version>1.27.0</grpc.version>
    </properties>
    <dependencies>
        <dependency>
            <groupId>io.grpc</groupId>
            <artifactId>grpc-netty</artifactId>
            <version>${grpc.version}</version>
        </dependency>
        <dependency>
            <groupId>io.grpc</groupId>
            <artifactId>grpc-protobuf</artifactId>
            <version>${grpc.version}</version>
        </dependency>
        <dependency>
            <groupId>io.grpc</groupId>
            <artifactId>grpc-stub</artifactId>
            <version>${grpc.version}</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.jmeter/ApacheJMeter_java -->
        <dependency>
            <groupId>org.apache.jmeter</groupId>
            <artifactId>ApacheJMeter_java</artifactId>
            <version>3.0</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.jmeter/ApacheJMeter_core -->
        <dependency>
            <groupId>org.apache.jmeter</groupId>
            <artifactId>ApacheJMeter_core</artifactId>
            <version>3.0</version>
        </dependency>
    </dependencies>
    <build>
        <plugins>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <configuration>
                    <source>${java.version}</source>
                    <target>${java.version}</target>
                    <skip>true</skip>
                    <encoding>${project.build.sourceEncoding}</encoding>
                </configuration>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-dependency-plugin</artifactId>
                <version>2.8</version>
                <executions>
                    <execution>
                        <id>copy-dependencies</id>
                        <phase>package</phase>
                        <goals>
                            <goal>copy-dependencies</goal>
                        </goals>
                        <configuration>
                           <outputDirectory>${project.build.directory}</outputDirectory>
                            <overWriteReleases>true</overWriteReleases>
                            <overWriteSnapshots>true</overWriteSnapshots>
                            <overWriteIfNewer>true</overWriteIfNewer>
                            <useSubDirectoryPerType>true</useSubDirectoryPerType>
                            <includeArtifactIds>
                                guava
                            </includeArtifactIds>
                            <silent>true</silent>
                        </configuration>
                    </execution>
                </executions>
            </plugin>
            <plugin>
                <artifactId>maven-assembly-plugin</artifactId>
                <configuration>
                    <appendAssemblyId>false</appendAssemblyId>
                    <descriptorRefs>
                        <descriptorRef>jar-with-dependencies</descriptorRef>
                    </descriptorRefs>
                </configuration>
            </plugin>
        </plugins>
        <defaultGoal>compile</defaultGoal>
    </build>
</project>

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;

public class ExampleSample implements JavaSamplerClient {
    @Override
    public void setupTest(JavaSamplerContext javaSamplerContext) {
        //初始化方法，對資料進行初始化，該方法只會執行一次
    }

    @Override
    public SampleResult runTest(JavaSamplerContext javaSamplerContext) {
        //Sample的請求的具體實現
        return null;
    }

    @Override
    public void teardownTest(JavaSamplerContext javaSamplerContext) {
        //資料或者資源銷燬介面，一般用於壓測停止時，需要做的動作。
    }

    @Override
    public Arguments getDefaultParameters() {
        //引數設定方法，一般用於設定傳遞引數
        return null;
    }
}

syntax = "proto3";
package com.zyq.example.cas.management.grpc;
message RequestData {
  string text = 1;
}
message ResponseData {
  string text = 1;
}
service StreamService {
  //rpc服務的方法
  rpc SimpleFun(RequestData) returns (ResponseData){}
}

import com.cf.cas.management.grpc.Example;
import com.cf.cas.management.grpc.StreamServiceGrpc;
import com.google.gson.Gson;
import io.grpc.ManagedChannel;
import io.grpc.ManagedChannelBuilder;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;

import java.util.HashMap;
import java.util.Map;

/**
 * Created by zyq on 2020/3/4.
 */
public class GrpcJmeter implements JavaSamplerClient {
    private String userAccount;
    private String password;
    private String address;
    private Integer port;
    @Override
    public void setupTest(JavaSamplerContext javaSamplerContext) {

    }
    @Override
    public SampleResult runTest(JavaSamplerContext javaSamplerContext) {
        SampleResult results = new SampleResult();
        userAccount = javaSamplerContext.getParameter("userAccount"); // 獲取在JMeter中設定的引數值
        password = javaSamplerContext.getParameter("password"); // 獲取在JMeter中設定的引數值
        address = javaSamplerContext.getParameter("address"); // 獲取在JMeter中設定的引數值
        port =Integer.valueOf(javaSamplerContext.getParameter("port")) ; // 獲取在JMeter中設定的引數值
        results.sampleStart();// JMeter 開始統計響應時間標記
        ManagedChannel channel=null;
        try {
            //grpc呼叫的具體實現
             channel = ManagedChannelBuilder.forAddress(address, port).usePlaintext().build();
            StreamServiceGrpc.StreamServiceBlockingStub stub = StreamServiceGrpc.newBlockingStub(channel);
            Map<String,Object> map = new HashMap<>();
            map.put("userAccount",userAccount);
            map.put("password",password);
            Gson gson = new Gson();
            Example.RequestData requestData = Example.RequestData.newBuilder().setText(gson.toJson(map)).build();
            Example.ResponseData responseData = stub.simpleFun(requestData);
            //設定請求的資料，這裡設定後，在JMeter的察看結果樹中才可顯示
            results.setRequestHeaders(gson.toJson(map));
            if(null!=responseData && null!=responseData.getText() && responseData.getText().contains("success")){
                results.setSuccessful(true);
            }
            else {
                results.setSuccessful(false);
            }
            //設定響應的資料，這裡設定後，在JMeter的察看結果樹中才可顯示
            results.setResponseMessage(responseData.getText());
            results.setResponseData(responseData.getText(),"UTF-8");
        } catch (Exception e) {
            results.setSuccessful(false);
            e.printStackTrace();
        }
        finally {
            if(null!=channel){
                channel.shutdown();
            }
            results.sampleEnd();// JMeter 結束統計響應時間標記
        }
        return results;
    }
    @Override
    public void teardownTest(JavaSamplerContext javaSamplerContext) {
    }
    @Override
    public Arguments getDefaultParameters() {
        Arguments params = new Arguments();
        params.addArgument("userAccount", "zyq");//設定引數，並賦予預設值
        params.addArgument("password", "111");//設定引數，並賦予預設值
        params.addArgument("address", "127.0.0.1");//設定引數，並賦予預設值
        params.addArgument("port", "8883");//設定引數，並賦予預設值
        return params;
    }
}