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

原创 已于 2024-06-15 03:26:17 修改 · 置顶 · 1k 阅读 · 1 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#jmeter #压力测试 #java #开源软件 #测试工具 #源代码管理 #交互

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

前言

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("port")) ; // 获取在JMeter中设置的参数值
        routingKey = javaSamplerContext.getParameter("routingKey");
        routingSecret = javaSamplerContext.getParameter("routingSecret");

        String userAgent = USER_AGENT_PREFIX;
        try {
            userAgent += "_" + InetAddress.getLocalHost().getHostName();
        } catch (final UnknownHostException e) {
            //System.out.println("Unable to determine local hostname. Defaulting gRPC user agent to {}.", new Object[]{USER_AGENT_PREFIX}, e);
        }

        final NettyChannelBuilder nettyChannelBuilder = NettyChannelBuilder.forAddress(address, port)
                // supports both gzip and plaintext, but will compress by default.
                .compressorRegistry(CompressorRegistry.getDefaultInstance())
                .decompressorRegistry(DecompressorRegistry.getDefaultInstance())
                .maxInboundMessageSize(50000)
                .userAgent(userAgent)
                .usePlaintext();

        channel = nettyChannelBuilder.build();

        //sendCountDownLatch = new CountDownLatch(1);
        //channel = ManagedChannelBuilder.forAddress(address, port).usePlaintext().build();
        //EdgeRpcServiceGrpc.EdgeRpcServiceBlockingStub stub = EdgeRpcServiceGrpc.newBlockingStub(channel);
        org.apache.nifi.processors.grpc.gen.edge.v1.EdgeRpcServiceGrpc.EdgeRpcServiceStub stub = org.apache.nifi.processors.grpc.gen.edge.v1.EdgeRpcServiceGrpc.newStub(channel);
        inputStream = stub.withCompression("gzip").handleMsgs(initOutputStream(results,countDownLatch,routingKey, this::onUplinkResponse,this::onError));
        inputStream.onNext(org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsg.newBuilder()
                .setMsgType(org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsgType.CONNECT_RPC_MESSAGE)
                .setConnectRequestMsg(org.apache.nifi.processors.grpc.gen.edge.v1.ConnectRequestMsg.newBuilder()
                        .setEdgeRoutingKey(routingKey)
                        .setEdgeSecret(routingSecret)
                        .setEdgeVersion(org.apache.nifi.processors.grpc.gen.edge.v1.EdgeVersion.V_3_6_0)
                        .setMaxInboundMessageSize(419820)
                        .build())
                .build());
    }
    @Override
    public SampleResult runTest(JavaSamplerContext javaSamplerContext) {

        results.sampleStart();// JMeter 开始统计响应时间标记

        try {
            String sourceMsg = "{'key1':'value1','key2':'value1','key3':'value1','key4':'value1','key5':'value1','key6':'value1','key7':'value1','key8':'value1','key9':'value1','key11':'value1','key12':'value1','key13':'value1','key14':'value1','key15':'value1','key16':'value1','key17':'value1','key18':'value1','key19':'value1','key21':'valuie2"+ Math.random()+ "'}";

            results.setRequestHeaders(sourceMsg);
            results.setSamplerData(sourceMsg);
            results.setDataType(SampleResult.TEXT);
            org.apache.nifi.processors.grpc.gen.edge.v1.UplinkMsg msg = buildUpLinkMsg(sourceMsg);

            org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsg requestMsg = org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsg.newBuilder()
                    .setMsgType(org.apache.nifi.processors.grpc.gen.edge.v1.RequestMsgType.UPLINK_RPC_MESSAGE)
                    .setUplinkMsg(msg)
                    .build();
            inputStream.onNext(requestMsg);

            results.setSentBytes(JSON.toJSONString(requestMsg).getBytes().length);

            countDownLatch.await(10, TimeUnit.MILLISECONDS);
            //results.setSuccessful(true);
        } catch (Exception e){
            e.printStackTrace();
            results.setSuccessful(false);
            results.setResponseCode("500");
            results.setResponseMessage(e.getMessage());
            //results.setResponseData(("exception," + e.getMessage()).getBytes());
            //results.setResponseData(e.getMessage(),"UTF-8");
        } finally {
            /*if(Objects.nonNull(inputStream)){
                inputStream.onCompleted();
                inputStream = null;
            }
            if (channel!=null){
                channel.shutdown();
            }*/
            //results.sampleEnd();
        }
        results.sampleEnd();

        return results;
    }
    @Override
    public void teardownTest(JavaSamplerContext javaSamplerContext) {
        if(Objects.nonNull(inputStream)){
            inputStream.onCompleted();
            inputStream = null;
        }
        if (channel!=null){
            channel.shutdown();
        }
    }
    @Override
    public Arguments getDefaultParameters() {
        Arguments params = new Arguments();
        params.addArgument("userAccount", "key55555");//设置参数，并赋予默认值
        params.addArgument("password", "s55555");//设置参数，并赋予默认值
        params.addArgument("routingKey", "key55555");//设置参数，并赋予默认值
        params.addArgument("routingSecret", "s55555");//设置参数，并赋予默认值
        params.addArgument("address", "127.0.0.1");//设置参数，并赋予默认值
        params.addArgument("port", "7070");//设置参数，并赋予默认值
        return params;
    }



    private static StreamObserver<org.apache.nifi.processors.grpc.gen.edge.v1.ResponseMsg> initOutputStream(SampleResult results,CountDownLatch countDownLatch,String edgeKey,
                                                                                                            Consumer<org.apache.nifi.processors.grpc.gen.edge.v1.UplinkResponseMsg> onUplinkResponse, Consumer<Exception> onError) {
        return new StreamObserver<org.apache.nifi.processors.grpc.gen.edge.v1.ResponseMsg>() {
            @Override
            public void onNext(org.apache.nifi.processors.grpc.gen.edge.v1.ResponseMsg responseMsg) {
                if (responseMsg.hasConnectResponseMsg()) {
                    org.apache.nifi.processors.grpc.gen.edge.v1.ConnectResponseMsg connectResponseMsg = responseMsg.getConnectResponseMsg();

                    /*if(Objects.nonNull(results)){
                        results.setSuccessful(true);
                    }*/
                    System.out.println("收到服务端连接回复消息" + connectResponseMsg);
                } else if (responseMsg.hasUplinkResponseMsg()) {
                    countDownLatch.countDown();
                    System.out.println("收到服务端响应消息" + responseMsg.getUplinkResponseMsg());
                    if(responseMsg.getUplinkResponseMsg().getSuccess()){
                        results.setSuccessful(true);
                        results.setResponseCodeOK();
                        results.setResponseMessage(String.valueOf(responseMsg.getUplinkResponseMsg().getSuccess()));
                        results.setResponseData(JSON.toJSONString(responseMsg.getUplinkResponseMsg()),"UTF-8");
                    } else {
                        results.setSuccessful(false);
                        results.setResponseCode("500");
                        results.setResponseMessage(responseMsg.getUplinkResponseMsg().getErrorMsg());
                        results.setResponseData(JSON.toJSONString(responseMsg.getUplinkResponseMsg()),"UTF-8");
                    }
                    //onUplinkResponse.accept(responseMsg.getUplinkResponseMsg());
                }
            }

            @Override
            public void onError(Throwable t) {
                t.printStackTrace();
                countDownLatch.countDown();
                //onError.accept(new RuntimeException(t));
                System.out.println("断开了" + edgeKey);
                results.setSuccessful(false);
                results.setResponseMessage(t.getMessage());
                results.setResponseData("exception," + t.getMessage(), "UTF-8");
                //results.setResponseData(t.getMessage(),"UTF-8");

            }

            @Override
            public void onCompleted() {
                countDownLatch.countDown();
                System.out.println("[{}] Stream was closed and completed successfully!" + edgeKey);
            }
        };
    }

    private void onError(final Exception e){
        results.setSuccessful(false);
        results.setResponseMessage(e.getMessage());
        results.setResponseData(e.getMessage(),"UTF-8");
    }

    private void onUplinkResponse(org.apache.nifi.processors.grpc.gen.edge.v1.UplinkResponseMsg msg) {
        //设置响应的数据，这里设置后，在JMeter的察看结果树中才可显示
        results.setSuccessful(true);
        results.setResponseMessage(JSON.toJSONString(msg));
        results.setResponseData(JSON.toJSONString(msg),"UTF-8");
    }

    private org.apache.nifi.processors.grpc.gen.edge.v1.UplinkMsg buildUpLinkMsg(String jsonMeg) throws JsonProcessingException {
        UUID entityId = UUID.randomUUID();
        org.apache.nifi.processors.grpc.gen.edge.v1.EntityDataProto.Builder builder = org.apache.nifi.processors.grpc.gen.edge.v1.EntityDataProto.newBuilder()
                .setEntityIdMSB(entityId.getMostSignificantBits())
                .setEntityIdLSB(entityId.getLeastSignificantBits())
                .setEntityType("device");
        long ts = System.currentTimeMillis();
        builder.setJsonData(jsonMeg);
        org.apache.nifi.processors.grpc.gen.edge.v1.EntityDataProto entityDataProto = builder.build();
        return  org.apache.nifi.processors.grpc.gen.edge.v1.UplinkMsg.newBuilder()
                .setUplinkMsgId(EdgeUtils.nextPositiveInt())
                .addEntityData(entityDataProto)
                .build();
    }

}

源码下载

[https://download.youkuaiyun.com/download/gzcsschen/89427899]

源码如何使用

编译源码

在这里插入图片描述

拷贝编译好的jar 到 jmeter 目录下的 /lib/ext/目录下

在这里插入图片描述

重新启动jmeter，在add java request 里面会多一个 grpc的选项，并且会自动识别proto文件

在这里插入图片描述

选择grpc 选项

在这里插入图片描述

用jmeter-grpc-request性能测试的严重问题
主要分享测试的学习资源，帮助快速了解测试行业，帮助想转行、进阶、小白成长为高级测试工程师。
03-07 1万+
起因 今日收到一个同事的求救信息，说正在做gRPC接口测试，用的是jmeter的一个第三方插件，叫jmeter-grpc-request，平日用着挺好用的，今天设置了100个线程，持续跑，结果才跑了5000来个请求，就卡住了。 卡住了？什么是卡住了呢？ 我仔细问了，才知道是jmeter整个没有响应了，只能强行杀进程才能停止。这是怎么回事呢？ 场景重现 我问同事要了jmeter的脚本文件，并且下载了这个gRPC取样器的插件，在我本机试了一下，果然，线程数量很少时候，运行正常，但是数量多了一些（仅仅到了50.
jmeter 压测数据库
m0_61066945的博客
02-26 5090
基础压测指导，业务并发分配方法。支持mysql、oracl、pg等压测
Apache JMeter 5.6.3 性能测试工具全解析与实战应用
最新发布
weixin_34064233的博客
10-16 676
Apache JMeter 是一款开源的性能测试工具，最初设计用于Web应用测试，现已扩展支持HTTP、HTTPS、FTP、JDBC、SOAP、REST等多种协议。其核心基于Java开发，具备跨平台能力，支持多线程并发模拟，能够对服务器、网络或对象施加高负载以评估系统性能。以开发一个简单的为例：@Overridetry {
apache-jmeter-5.6.2.tgz
09-18
apache-jmeter-5.6.2.tgz
apache-jmeter-5.6.3
01-08
apache-jmeter-5.6.3.zip apache-jmeter-5.6.3.tgz apache-jmeter-5.6.3_src.zip apache-jmeter-5.6.3_src.tgz
用Jmeter对GRPC接口进行测试
m0_65120252的博客
04-12 2535
jmeter：可以进行接口测试也可以进行性能测试Jmeter 环境安装 看之前的博文他是一个功能强大的JMeter Grpc插件，可用于测试任何gRPC服务器，它不需要生成gRPC类或编译服务的protos二进制文件，相较于其他RPC框架，gRPC是一个高性能、开源和通用的RPC框架，基于HTTP/2标准，并使用Protocol Buffers（protobuf）作为接口定义语言（IDL）。简单配置。
Jmeter 实现 grpc服务 压测
小小晓晓阳的博客
06-09 4628
使用jmeter实现 grpc服务压力测试
jmeter5.6.3压测工具 jmeter压测工具 jmeter压测工具
06-21
JMeter 5.6.3是其一个版本，它提供了丰富的功能，可以帮助开发者、测试人员评估系统的稳定性和性能。以下是对JMeter 5.6.3版本的一些详细知识点介绍： 1. **组件体系**： JMeter由多种组件构成，包括采样器...
jmeter 自定义grpc 压测扩展实现源码
06-13
jmeter 自定义grpc 压测扩展实现源码，适用对grpc双向通信进行压力测试，其他grpc模式实现都非常一致。调整下代码即可实现。 使用方法：编译工程打包成jar文件，然后放到jmeter的jar目录下，以我本地为例： D:\soft...
JMeter自定义gRPC压测插件开发与实践
3. 启动JMeter应用程序，就可以在JMeter界面中使用这个自定义的gRPC压测插件了。 开发此类自定义扩展，需要具备以下知识点： 1. **JMeter插件开发**: JMeter插件开发是扩展JMeter功能的一种方式，它允许开发者在...
jmeter插件grpc插件
07-02
jmeter插件grpc插件
Apacha Jmeter 5.6.3
01-08
Apacha Jmeter 5.6.3
jmeter进行GRPC性能测试教程
qq_41958041的博客
12-20 1409
将下载文件解压后放到目录下 /lib/ext下即可。
【Jmeter】Jmeter的gRPC调用
Joe Throne
08-09 2103
Jmeter测试gRPC
史上最细gRPC(Go)入门教程(十)------gRPC压测工具ghz
探索云原生
05-15 3319
本文主要介绍了 gRPC 压测工具 ghz ，包括 ghz 的安装、使用及压测计划制定等。 gRPC 系列相关代码见 Github 1. 安装 可以直接在Release页面下载二进制文件，也可以 clone 仓库手动编译。 下载解压后即可使用 # 下载 $ wget https://github.91chifun.workers.dev/https:
GRPC 压测探索
weixin_45675776的博客
05-08 1046
基于应用认证中心对HTTP与GRPC两种调用方式性能的对比，探索一个既支持HTTP，又支持GRPC的压测工具。
JMeter gRPC 请求插件使用教程
gitblog_00410的博客
08-08 1006
JMeter gRPC 请求插件使用教程 jmeter-grpc-requestJMeter gRPC Request load test plugin for gRPC项目地址:https://gitcode.com/gh_mirrors/jm/jmeter-grpc-request 项目介绍 JMeter gRPC 请求插件是一个用于 gRPC 负载测试的插件。它允许用户通过 JMeter ...
GRPC接口压测 --- ghz工具
谬也的博客
07-29 2732
GRPC接口的压测工具也很少，找来找去好像也就ghz比较能用。 该工具的官方说明文档：https://ghz.sh/docs/usage 举个例子： ghz -n 100 -c 10 --insecure --proto ./api.proto --call api.HelloWorld.Say -d '[{"Message":"hello"},{"Message":"hi"}]' 127.0.0.1:9092 结果 参数说明： -n: 请求数 -z: 请求持续时间 -t: 超时时间 -x:
jmeter-grpc-plugin
arjick的专栏
05-28 6769
###背景与目标 通过该插件实现针对grpc接口的测试，并且可以利用jmeter的所有测试特性。 在 https://github.com/yangliCypressTest/jmeter-grpc-plugin的基础上做了一部分的修改。 ###工程优势 无需编写独立的测试工程 有大量的报表支持实现多维度的指标监测 可以编写测试用例，测试用例便于传播 基本实现编码 ###入门教程 ####一 ...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

gzcsschen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值