Linux流量复制工具

  对于一些有并发要求的业务,特别是对接外部流量时,产品上线前一定要做的就是压力测试,但是常规的压力测试并不能覆盖所有情况。以gemeter、ab,、webbench、http_load为例,这些通过模拟请求的压测工具,只能发送特定的参数,对于一些参数异常导致的程序处理异常是无法考虑到的,所以就需要一款能复制真实流量,并且不影响线上业务的工具。
流量复制工具有很多,例如Gor、tcpreplay、tcpcopy等,这些工具贴合真实场景,能模拟真实流量,并支持流量的放大或缩小,更容易测试出程序的瓶颈和潜在问题。

几款流量复制工具:

  1. gor: https://github.com/buger/goreplay
  2. tcpreplay: https://github.com/appneta/tcpreplay
  3. tcpcopy: https://github.com/session-replay-tools/tcpcopy
  4. Nginx模块ngx_http_mirror_module,在Nginx 1.13.4中开始引入,使用前请检查nginx版本

Nginx模块ngx_http_mirror_module

配置如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
server {
listen 8080;
access_log /home/work/log/nginx/org.log;
root html/org;
}

server {
listen 8081;
access_log /home/work/log/nginx/mir.log ;
root html/mir;
}

upstream backend {
server 127.0.0.1:8080;
}

upstream test_backend {
server 127.0.0.1:8081;
}

server {
listen 80;
server_name localhost;

#charset koi8-r;
#access_log logs/host.access.log main;

location / {
mirror /mirror;
proxy_pass http://backend;
}

location /mirror {
internal;
proxy_pass http://test_backend$request_uri;
}

}

流量放大,配置两个mirror即可

1
2
3
4
5
location / {
mirror /mirror;
mirror /mirror;
proxy_pass http://backend;
}

使用是很方便,但是线上nginx一般都承载了不止一个业务,修改nginx配置后需要nginx -s reload来使之生效,这种操作在线上还是尽量需要避免的。

gor https://github.com/buger/goreplay

Gor概述

  Gor 是用 Golang 写的一个 HTTP 实时流量复制工具。功能更强大,支持流量的放大、缩小,频率限制,还支持把请求记录到文件,方便回放和分析,也支持和 ElasticSearch 集成,将流量存入 ES 进行实时分析。

下载安装,可以下载编译好的二进制文件直接使用

1
2
> wget https://github.com/buger/goreplay/releases/download/v0.16.1/gor_0.16.1_x64.tar.gz
> tar xzvf gor_0.16.1_x64.tar.gz

流量复制到文件

可以将流量复制到文件,然后再对他们进行回放。回放的时候,流量会维持原始的时间间隔。如果你使用了百分比来进行速率限制,那么回放的速率会相应的增加或减少。有了这种速率限制,gor就可以用来进行压力测试。

1
2
3
4
5
#write to file
> gor --input-raw :80 --output-file requests_origin.gor

#read from file
> gor --input-file requests_origin.gor --output-http "http://localhost:8081"

可以使用时间戳命名录制文件,默认情况下,文件是按“块”存储的,即文件大小到达上限后,添加后缀,并新建另一个文件,如下

1
2
3
4
5
6
7
> gor --input-raw :80 --output-file %Y%m%d.gor
#append false

20140608_0.gor
20140608_1.gor
20140609_0.gor
20140609_1.gor

默认是按“块”存储文件的方式,但是可以参数配置,–output-file-append,使用之后如下

1
2
3
4
5
> gor --input-raw :80 --output-file %Y%m%d.gor --output-file-append
#append true

20140608.gor
20140609.gor

时间格式化文件名的配置说明:

1
2
3
4
5
6
7
%Y: year including the century (at least 4 digits)
%m: month of the year (01..12)
%d: Day of the month (01..31)
%H: Hour of the day, 24-hour clock (00..23)
%M: Minute of the hour (00..59)
%S: Second of the minute (00..60)
默认格式是%Y%m%d%H

流量回放

目前,这种方式只支持”input-file”,而且只能用百分比去控制回放速率。请注意,这个回放的速率比例是相对于input的。即按照录下来的流量的时间戳去进行回放。

以2倍速率回放
gor --input-file "requests_origin.gor|200%" --output-http "http://localhost:8081"

如果“input-flie”是多个文件,可以用正则去匹配,
gor --input-file "requests_origin*.gor|200%" --output-http "http://localhost:8081"

配合如下配置参数,可以更好进行压力测试
--input-file-loop 重复循环执行input-file
--exit-after 30s 在30s后停止,可以控制压力测试的时间。分钟的单位是m

Gor常用命令

简单的HTTP流量复制
> gor --input-raw :80 --output-http "http://localhost:8081"

HTTP流量复制频率控制(获取每秒超过10个请求)
> gor --input-tcp :28020 --output-http "http://localhost:8081|10"

HTTP流量复制缩小
> gor --input-raw :80 --output-tcp "http://localhost:8081|10%"

HTTP流量记录到本地文件
> gor --input-raw :80 --output-file requests_origin.gor

HTTP流量回放和压测
> gor --input-file "requests_origin.gor|200%" --output-http "http://localhost:8081"

HTTP流量过滤复制

1
> gor --input-raw :8080 --output-http http://localhost:8081 --output-http-url-regexp ^www.

自定义一些流量复制的参数

1
> gor --input-raw :80 --output-http http://localhost:8081 --http-allow-method POST --http-set-header 'User-Agent: Gor' -output-http-workers=1 -http-allow-url test.php

将流量复制两份到不同的测试服务

1
> gor --input-tcp :8080 --output-http "http://localhost:8081" --output-http "http://localhost:8082"

将流量像负载均衡一样分配到不同的服务器

1
> gor --input-tcp :8080 --output-http "http://localhost:8081" --output-http "http://localhost:8082" --split-output true

Gor配置参数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
> gor --help

-http-allow-header value
gor --input-raw :8080 --output-http localhost:8081 --http-allow-header api-version:v1.1
用一个正则表达式来匹配http头部,如果请求的头部没有匹配上,则被拒绝

-http-allow-method value
gor --input-raw :8080 --output-http localhost:8081 --http-allow-method GET
类似于一个白名单机制来允许通过的http请求方法,除此之外的方法都被拒绝.

-http-allow-url value
gor --input-raw :8080 --output-http localhost:8081 --http-allow-url ^www
一个正则表达式用来匹配url, 用来过滤完全匹配的的url,在此之外的都被过滤掉

-http-disallow-header value
gor --input-raw :8080 --output-http localhost:8081 --http-disallow-header "User-Agent: Replayed by Gor"
用一个正则表达式来匹配http头部,匹配到的请求会被拒绝掉

-http-disallow-url value
gor --input-raw :8080 --output-http localhost:8081 --http-disallow-url ^www
用一个正则表达式来匹配url,如果请求匹配上了,则会被拒绝

-http-set-header value
gor --input-raw :8080 --output-http localhost:8081 --http-set-header 'User-Agent: Gor'
设置头信息,如果已经存在会覆盖

-http-set-param value
gor --input-raw :8080 --output-http localhost:8081 --http-set-param api_key=v1.1
设置请求参数,如果已经存在会覆盖

更多参数请查阅官方文档 https://github.com/buger/goreplay/wiki
JouyPub wechat
欢迎订阅「K叔区块链」 - 专注于区块链技术学习