Jmeter 5.6.3 自定义GRPC 压测插件，配置，使用，

置顶

gzcsschen

已于 2024-06-15 03:26:17 修改

阅读量858

点赞数 1

文章标签： jmeter 压力测试 java 开源软件测试工具源代码管理交互

于 2024-06-13 11:47:30 首次发布

本文链接：https://blog.csdn.net/gzcsschen/article/details/139532087

版权

前言

jmeter 版本 apache-jmeter-5.6.3

说明

jmeter 自定义grpc 压测扩展实现源码，适用对grpc双向通信进行压力测试，其他grpc模式实现都非常一致。调整下代码即可实现。

使用方法：编译工程打包成jar文件，然后放到jmeter的jar目录下，以我本地为例： D:\soft\apache-jmeter-5.6.3\lib\ext

启动jmeter 进行测试

三、配置与使用
1）添加线程组：右键单击测试计划 → 添加 → 线程(用户) → 线程组

在这里插入图片描述

2）添加GRPC Request：右键单击新建的线程组 → 添加 → 取样器 → GRPC Request

在这里插入图片描述

3）填写请求信息：主机、端口、proto文件夹、rpc方法、请求数据

在这里插入图片描述

必填项：
在这里插入图片描述

java插件核心代码

package org.apache.nifi.processors.grpc;

import com.alibaba.fastjson.JSON;
import com.fasterxml.jackson.core.JsonProcessingException;
import io.grpc.CompressorRegistry;
import io.grpc.DecompressorRegistry;
import io.grpc.ManagedChannel;
import io.grpc.netty.NettyChannelBuilder;
import io.grpc.stub.StreamObserver;
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.io.UnsupportedEncodingException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;

/**
 * Created by zyq on 2020/3/4.
 */
public class GrpcJmeter implements JavaSamplerClient {
   
    private String userAccount;
    private String password;
    private String address;
    private Integer port;

    private String routingKey;
    private String routingSecret;

    private SampleResult results;

    private StreamObserver<org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsg> inputStream;

    private ManagedChannel channel;

    private CountDownLatch countDownLatch;


    private static final String USER_AGENT_PREFIX = "NiFi_invokeGRPC";



    @Override
    public void setupTest(JavaSamplerContext javaSamplerContext) {
   

        countDownLatch = new CountDownLatch(1);
        results = new SampleResult();

        userAccount = javaSamplerContext.getParameter("userAccount"); // 获取在JMeter中设置的参数值
        password = javaSamplerContext.getParameter("password"); // 获取在JMeter中设置的参数值
        address = javaSamplerContext.getParameter("address"); // 获取在JMeter中设置的参数值
        port =Integer.valueOf(javaSamplerContext.getParameter(