流量回放平台(支持http+grpc,支持pb格式)

本文介绍了自研的流量回放平台,该平台弥补了开源工具对HTTP协议和protobuf支持的不足,支持http+grpc协议,并能处理pb格式数据。文章详细讲解了平台的背景、架构、重点实现,包括流量录制、解析、请求引擎等关键环节,并提及了多线程控制和QPS管理策略。
摘要由CSDN通过智能技术生成

流量回放平台(支持http+grpc,支持pb格式)

背景

目前开源的工具例如tcpcopy、goreplay大多只支持http协议的流量回放,而且不支持pb,所以采用自研的方式。

架构图

一、整体架构:
在这里插入图片描述
二、压测引擎
在这里插入图片描述
三、流量录制推送
在这里插入图片描述

重点实现

1、流量录制:
流量录制使用tcpdump把流量保存到本地/mq中
2、流量解析:
因为有pb格式的数据。所以借助了python的 dpkt包对流量进行了解析组装,根据请求的url不同进行区分,然后放到消息队列中
在这里插入图片描述

3、请求引擎:
http:正常使用各种httpclient
grpc:1、调用服务自身grpc请求client 2、使用通用client
4、多线程处理,qps控制
开启多线程,消费消息队列数据,然后进行发压,qps通过每秒控制消息队列数据量进行控制。
5、流量回放时间时间控制:
根据设定时间,计时控制压测引擎使用时间

代码:

比较多,下次再写

Java + gRPC + grpc-gateway 的实践主要分为以下几个步骤: 1. 定义 proto 文件 在 proto 文件中定义需要调用的服务以及方法,同时指定请求和响应的数据类型。例如: ``` syntax = "proto3"; package example; service ExampleService { rpc ExampleMethod (ExampleRequest) returns (ExampleResponse) {} } message ExampleRequest { string example_field = 1; } message ExampleResponse { string example_field = 1; } ``` 2. 使用 protoc 编译 proto 文件 使用 protoc 编译 proto 文件,生成 Java 代码。例如: ``` protoc --java_out=./src/main/java ./example.proto ``` 3. 实现 gRPC 服务 在 Java 代码中实现定义的 gRPC 服务,例如: ``` public class ExampleServiceImpl extends ExampleServiceGrpc.ExampleServiceImplBase { @Override public void exampleMethod(ExampleRequest request, StreamObserver<ExampleResponse> responseObserver) { // 实现具体逻辑 ExampleResponse response = ExampleResponse.newBuilder().setExampleField("example").build(); responseObserver.onNext(response); responseObserver.onCompleted(); } } ``` 4. 启动 gRPC 服务器 使用 gRPC 提供的 ServerBuilder 构建 gRPC 服务器,并启动服务器。例如: ``` Server server = ServerBuilder.forPort(8080).addService(new ExampleServiceImpl()).build(); server.start(); ``` 5. 集成 grpc-gateway 使用 grpc-gateway 可以将 gRPC 服务转换为 HTTP/JSON API。在 proto 文件中添加以下内容: ``` import "google/api/annotations.proto"; service ExampleService { rpc ExampleMethod (ExampleRequest) returns (ExampleResponse) { option (google.api.http) = { post: "/example" body: "*" }; } } ``` 在 Java 代码中添加以下内容: ``` Server httpServer = ServerBuilder.forPort(8081).addService(new ExampleServiceImpl()).build(); httpServer.start(); String grpcServerUrl = "localhost:8080"; String httpServerUrl = "localhost:8081"; ProxyServerConfig proxyConfig = new ProxyServerConfig(grpcServerUrl, httpServerUrl, "/example"); HttpProxyServer httpProxyServer = new HttpProxyServer(proxyConfig); httpProxyServer.start(); ``` 6. 测试 使用 HTTP/JSON API 调用 gRPC 服务,例如: ``` POST http://localhost:8081/example Content-Type: application/json { "example_field": "example" } ``` 以上就是 Java + gRPC + grpc-gateway 的实践步骤。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值