Support RocketMQ5 benchmark using grpc sdk

README.md

@@ -23,6 +23,7 @@ This repository houses user-friendly, cloud-ready benchmarking suites for the fo
 * [Pravega](https://pravega.io/)
 * [RabbitMQ](https://www.rabbitmq.com/)
 * [Redis](https://redis.com/)
+* [MQTT](https://mqtt.org/)
 
 > More details could be found at the [official documentation](http://openmessaging.cloud/docs/benchmarks/).
 

benchmark-framework/pom.xml

@@ -106,6 +106,11 @@
             <artifactId>driver-rocketmq</artifactId>
             <version>${project.version}</version>
         </dependency>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>driver-rocketmq5</artifactId>
+            <version>${project.version}</version>
+        </dependency>
         <dependency>
             <groupId>com.beust</groupId>
             <artifactId>jcommander</artifactId>

driver-rocketmq5/pom.xml

@@ -0,0 +1,64 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+
+    Licensed under the Apache License, Version 2.0 (the "License");
+    you may not use this file except in compliance with the License.
+    You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+    Unless required by applicable law or agreed to in writing, software
+    distributed under the License is distributed on an "AS IS" BASIS,
+    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    See the License for the specific language governing permissions and
+    limitations under the License.
+
+-->
+<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>
+    <parent>
+        <groupId>io.openmessaging.benchmark</groupId>
+        <artifactId>messaging-benchmark</artifactId>
+        <version>0.0.1-SNAPSHOT</version>
+    </parent>
+
+    <artifactId>driver-rocketmq5</artifactId>
+
+    <properties>
+        <rocketmq-client-java-version>5.0.7</rocketmq-client-java-version>
+        <rocketmq.version>5.1.0</rocketmq.version>
+        <guava.version>29.0-jre</guava.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>driver-api</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>com.google.guava</groupId>
+            <artifactId>guava</artifactId>
+            <version>${guava.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.apache.rocketmq</groupId>
+            <artifactId>rocketmq-client-java</artifactId>
+            <version>${rocketmq-client-java-version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.rocketmq</groupId>
+            <artifactId>rocketmq-tools</artifactId>
+            <version>${rocketmq.version}</version>
+            <exclusions>
+                <exclusion>
+                    <groupId>ch.qos.logback</groupId>
+                    <artifactId>logback-classic</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+    </dependencies>
+
+</project>

driver-rocketmq5/rocketmq.yaml

@@ -0,0 +1,32 @@
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+name: RocketMQ5
+driverClass: io.openmessaging.benchmark.driver.rocketmq.RocketMQ5BenchmarkDriver
+
+# The RocketMQ nameserver address
+namesrvAddr: x.x.x.x:9876
+# The RocketMQ broker cluster name
+clusterName: rocketmq-broker-xxxx
+# The proxy address to connect for grpc client.
+grpcEndpoint: 127.0.0.1:8081
+# 是否生产定时消息，否：生产普通消息，是：生产定时/延时消息且通过delayTimeInSec设置延迟时间
+sendDelayMsg: false
+delayTimeInSec: 60
+# (Optional) The admin credential
+adminAccessKey: xxxx
+adminSecretKey: xxxx
+# (Optional) The credential that clients connect to proxy.
+accessKey: xxxxxx
+secretKey: xxxxx

driver-rocketmq5/src/main/java/io/openmessaging/benchmark/driver/rocketmq/RocketMQ5BenchmarkConsumer.java

@@ -0,0 +1,31 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.openmessaging.benchmark.driver.rocketmq;
+
+
+import io.openmessaging.benchmark.driver.BenchmarkConsumer;
+import org.apache.rocketmq.client.apis.consumer.PushConsumer;
+
+public class RocketMQ5BenchmarkConsumer implements BenchmarkConsumer {
+    private final PushConsumer rmqConsumer;
+
+    public RocketMQ5BenchmarkConsumer(final PushConsumer rmqConsumer) {
+        this.rmqConsumer = rmqConsumer;
+    }
+
+    @Override
+    public void close() throws Exception {
+        this.rmqConsumer.close();
+    }
+}