一、 .proto文件編譯成.jar文件

相關依賴下載詳見：將 message.proto 編譯成 .jar文件

1.依賴于java編譯環境
2.依賴protoc編譯jar包

編譯目錄

1.創建一個根目錄：protobuf
2.在protobuf下創建build、output、lib、src目錄

lib：放 protobuf-java-4.31.1.jar
src：放 xxx.proto

準備.proto 文件

將準備好的.protp 文件放入src目錄下

syntax = "proto3"; option java_package = "com.message.proto";
option java_outer_classname = "MessageProBuf";package Message;//消息的頭信息
message MessageHead
{string from = 1;string to = 2;string messageId = 3;int32 chatType = 4;}//消息的 body 
message ChatMessage
{MessageHead messageHead=1;string fromUserId = 2;string toUserId= 3;int64 timeSend= 4;int32 type= 5;string content=6;
}

將.proto文件編譯.java 文件

執行命令：

cd .\src\
protoc --java_out=../output message.proto

命令執行完成后，根據.proto 文件中的 java_package、java_outer_classname 生成對應的包目錄和文件名，示例如下：
在這里插入圖片描述

使用javac命令將MessageProBuf.java編譯成.class文件

這里依賴protobuf-java-4.31.1.jar 庫，所以需要根目錄下創建一個lib目錄，并將protobuf-java-4.31.1.jar文件放到lib目錄下

cd ..\lib\
javac -encoding UTF-8 -cp ".;protobuf-java-4.31.1.jar" -d ..\build ..\output\com\message\proto\MessageProBuf.java 

使用 jar 命令將所有 .class 文件打包成 .jar 文件

cd ..\build\
jar cf message-protobuf.jar com/

示例完整命令匯總

:: .proto文件編譯為.java文件
cd .\src\
protoc --java_out=../output message.proto:: .java文件編譯為.class文件
cd ..\lib\
javac -encoding UTF-8 -cp ".;protobuf-java-4.31.1.jar" -d ..\build ..\output\com\message\proto\MessageProBuf.java :: .class文件編譯為.jar包
cd ..\build\
jar cf message-protobuf.jar com/

二、編寫 .groovy文件編譯成.jar文件

相關依賴下載詳見：JMeter groovy 編譯成.jar 文件

編譯目錄

1.創建一個根目錄：groovy
2.在protobuf下創建build、lib、src目錄

lib：放 message-protobuf.jar、protobuf-java-4.31.1.jar 等jar包
src：放 .groovy 文件

準備 MessageBuilder.groovy （消息構建文件）

// 導入 .proto 文件中的包名和類名
import com.message.proto.MessageProBuf
java_package ：com.message.proto
java_outer_classname：MessageProBuf

import com.message.proto.MessageProBuf  // .proto 中的java_package、java_outer_classname
import com.google.protobuf.MessageLite/*** 消息構建工具類*/
class MessageBuilder {/*** 構建單聊消息* @param fromUserId 發送者ID* @param fromUserName 發送者名稱* @param toUserId 接收者ID* @param toUserName 接收者名稱* @param content 消息內容* @param messageId 消息ID（可選）* @return 構建完成的 MessageLite 對象*/static MessageLite buildChatMessage(String fromUserId, String fromUserName,String toUserId, String toUserName,String content, String messageId = null) {if (messageId == null) {messageId = UUID.randomUUID().toString()}return MessageProBuf.ChatMessage.newBuilder().setMessageHead(MessageProBuf.MessageHead.newBuilder().setFrom("${fromUserId}/pc").setTo(toUserId).setMessageId(messageId).setChatType(1)) // 單聊.setFromUserId(fromUserId).setFromUserName(fromUserName).setToUserId(toUserId).setToUserName(toUserName).setContent(content).setType(1) // 文本消息.setTimeSend(System.currentTimeMillis()).setIsReadDel(false).setIsEncrypt(false).setSeqNo(-1).build()}// ...  構建其他消息類型省略
}

準備 ProtobufParser.groovy （加密解密文件）

import com.message.proto.MessageProBuf
import com.google.protobuf.InvalidProtocolBufferException
import com.google.protobuf.MessageLite
import java.util.function.Function
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Loggerclass ProtobufParser {private static final Logger log = LogManager.getLogger(ProtobufParser::class)// 支持擴展的命令解析映射private static final Map<Integer, Function<byte[], MessageLite>> MESSAGE_PARSERS = new HashMap<>()static {MESSAGE_PARSERS.put(10, { data -> MessageProBuf.ChatMessage.parseFrom(data) });// 更多指令碼和解析函數的映射可以添加到這里}/*** 構建加密后的消息字節數組* @param command 指令碼* @param message Protobuf 消息體* @return 拼接后的字節數組*/static byte[] encryptMessage(int command, MessageLite message) {if (message == null) {throw new IllegalArgumentException("Message cannot be null")}byte[] buffer = message.toByteArray()byte[] bytes = new byte[buffer.length + 1]bytes[0] = (byte) commandSystem.arraycopy(buffer, 0, bytes, 1, buffer.length)return bytes}/*** 解析響應數據中的 Protobuf 消息* @param responseData 響應字節數組數據* @return 解析后的 MessageLite 對象（可擴展）*/static Optional<MessageLite> parseProtobufMessage(byte[] responseData) {if (responseData == null || responseData.length < 1) {throw new IllegalArgumentException("Response data is null or empty")}int command = responseData[0] & 0xFFbyte[] messageData = Arrays.copyOfRange(responseData, 1, responseData.length)try {Function<byte[], MessageLite> parser = MESSAGE_PARSERS.get(command)if (parser != null) {MessageLite message = parser.apply(messageData)log.info("Decoded message: ${message.toString()}")return Optional.of(message)} else {log.warn("Unknown command: $command")return Optional.empty()}} catch (InvalidProtocolBufferException e) {log.error("Failed to parse Protobuf message", e)throw new RuntimeException("Failed to parse Protobuf message with command: $command", e)}}
}

編譯 Groovy 文件為 .class

1、將MessageBuilder.groovy 構件成 MessageBuilder.class 文件到指定的build/message 目錄下
2、將ProtobufParser.groovy 構件成 ProtobufParser.class 文件到指定的build/protobuf 目錄下

:: 切換到目錄
cd groovygroovyc -cp "lib/*" src/MessageBuilder.groovy -d build/messagegroovyc -cp "lib/*" src/ProtobufParser.groovy -d build/protobuf

將目錄build/message 、build/protobuf 目錄下所有的.class 文件分別構建成 .jar文件

:: 切換到目錄
cd groovyjar cf MessageBuilder.jar -C build/message .jar cf ProtobufParser.jar -C build/protobuf .

示例完整命令匯總

PS D:\System\Desktop\csdn> cd groovy
PS D:\System\Desktop\csdn\groovy> groovyc -cp "lib/*" src/MessageBuilder.groovy -d build/message
PS D:\System\Desktop\csdn\groovy> groovyc -cp "lib/*" src/ProtobufParser.groovy -d build/protobuf
PS D:\System\Desktop\csdn\groovy> jar cf MessageBuilder.jar -C build/message .
PS D:\System\Desktop\csdn\groovy> jar cf ProtobufParser.jar -C build/protobuf .
PS D:\System\Desktop\csdn\groovy>

三、JMeter 使用加解密

將 protobuf-java-4.31.1.jar 文件復制到 jmeter lib 目錄下

將message-protobuf.jar、MessageBuilder.jar、ProtobufParser.jar 三個文件復制到lib/ext 目錄下

四、重啟JMeter

五、在JMeter 中使用

在JMeter 中添加線程組

在線程組上添加取樣器/JSR223 Sampler

在JSR223 Sampler 中的script 腳本添加如下代碼

import com.message.proto.MessageProBuf;
import com.google.protobuf.MessageLite;
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Logger// 創建日志記錄器
Logger logger = LogManager.getLogger("test-protpbuf-jsr223-sampler")// 單聊消息
MessageLite chatMessage = MessageBuilder.buildChatMessage("${fromUserId}","fromUserName","${userId}","toUserName","我是xxx"); // 構建消息體
byte[] chatMessageEncryptedBytes = ProtobufParser.encryptMessage(10,chatMessage); // 加密
logger.info("單聊消息byte: {}", chatMessageEncryptedBytes)
def chatMessageDecrypt = ProtobufParser.parseProtobufMessage(chatMessageEncryptedBytes); // 消息解密

設置顯示日志視圖

運行測試

運行結果