在网上查In-Memory NoSQL性能测试的资料时,偶然间发现了这个性能测试工具YCSB,全称为“Yahoo! Cloud Serving Benchmark”。它内置了对常见NoSQL数据库和数据网格产品的支持,如Cassandra、MongoDB、HBase、Redis、Infinispan等等很多的主流产品,而且不仅安装使用简单,还能自由扩展测试数据类型和支持的数据库产品。
1 编译安装
首先说一下YCSB的安装。直接下载tar.gz分发包的话其实也不需要安装什么,解压后通过bin目录下的脚本执行即可(依赖python和JRE)。但由于tar.gz分发包比较大,一百多M,而且我这里总是连接超时,于是直接下载master库的源代码,自己编译。编译方法也很简单,因为工程是Maven管理的,所以只要有Maven环境就可以直接在YCSB根目录下运行package任务,之后就能在distribution工程的target中看到ycsb-0.1.4.tar.gz。最后将编译出的包拷贝到自己的环境如虚拟机里,就能使用了。本机调试也是可以的,只需在IDE中配置好Client.main()方法的启动参数就行了。
2 使用
现在来说一下如何使用YCSB,以Redis的性能测试为例,分为1.初始化测试数据 (loading阶段)和2.执行测试操作(transaction阶段)。首先进行初始化,先保证Redis服务已经启动,然后在YCSB根目录下执行:
> bin/ycsb load redis -s -P workloads/workloada -p "redis.host=127.0.0.1" -p "redis.port=6379"
之后我们在Redis中就能看到1000条数据,因为workloada中定义了recordcount=1000。每一个key都是hash类型,我们可以重执行上面的命令,将redis改为basic,basic是一种dummy客户端,实际并不操作任何数据库,而是将YCSB对数据库的各种操作的key-value输出到控制台,这样我们就能看到这些key-value是怎么插入到Redis的,value都是什么样子的了。
继续开始真正的测试:
> bin/ycsb run redis -s -P workloads/workloada -p "redis.host=127.0.0.1" -p "redis.port=6379" \
-threads 128 -p "operationcount=10000" -p "measurementtype=timeseries" \
-p "timeseries.granularity=5000"
简单解释一下这些参数:-threads是模拟的线程数,-p是传入的参数,可以是客户端的参数,可以是YCSB内部组件的参数,例如measurementtype会配置Measurements输出时间序列而不是直方图,也可以覆盖workload文件中预定义的参数,例如上面的operationcount。还有一个常用的参数是-target,指的是每秒最大操作数,当一秒内到达这个阈值,线程就会休眠1毫秒。
输出的内容大概是这个样子,主要关注总的执行时间、吞吐量,以及读和写的平均延迟:
[OVERALL], RunTime(ms), 60058.0
[OVERALL], Throughput(ops/sec), 27049.768557061507
[UPDATE], Operations, 812803
[UPDATE], AverageLatency(us), 2275.102034564341
[UPDATE], MinLatency(us), 616
[UPDATE], MaxLatency(us), 64614
[UPDATE], Return=0, 812800
[UPDATE], 0, 4529.805933250927
[UPDATE], 5000, 2208.521014693651
[UPDATE], 10000, 2148.7565276549426
[UPDATE], 15000, 2139.8536001184116
[UPDATE], 20000, 2205.0305368941204
[UPDATE], 25000, 2165.1238092653402
[UPDATE], 30000, 2180.7067298448014
[UPDATE], 35000, 2188.5334523973074
[UPDATE], 40000, 2145.7207987586858
[UPDATE], 45000, 2199.350677041509
[UPDATE], 50000, 2175.385860213585
[UPDATE], 55000, 2156.2515744608663
[READ], Operations, 811752
[READ], AverageLatency(us), 2304.3717539346994
[READ], MinLatency(us), 629
[READ], MaxLatency(us), 75380
[READ], Return=0, 811751
[READ], 0, 4634.609504308506
[READ], 5000, 2233.603889043769
[READ], 10000, 2179.2526166711627
[READ], 15000, 2165.0853859227886
[READ], 20000, 2234.4603871649133
[READ], 25000, 2178.2390324907783
[READ], 30000, 2208.70719241602
[READ], 35000, 2207.444112422369
[READ], 40000, 2173.767667054904
[READ], 45000, 2223.0628439478996
[READ], 50000, 2200.857439151695
[READ], 55000, 2178.257027425664
3 源码分析
YCSB的核心类都在Maven子模块core中,主要有Client、ClientThread、CoreWorkload、DB实现类等。
其主流程是:python脚本bin/ycsb会调用Client类的main()方法,初始化Workload和DB的实现类,并根据-threads参数创建多个Client线程,-s参数创建一个Status线程。每个线程会循环recordcount或operationcount(在workloada中定义)除以threads次,并根据参数bin/ycsb load/run决定是初始化数据doInsert()还是执行读写操作doTransaction()。
4 添加新数据库
1.新建module:在YCSB根目录pom.xml下添加一个Maven子module,artifactId为xxx-binding(ycsb脚本会根据这个名字后缀扫描jar包,加入到classpath中),而目录名则为前缀xxx。
2.配置pom.xml:打开新建子module的pom.xml,加入
3.新建DB子类:在此子module中新建package和类,继承DB类。
4.打包module:执行package任务,在target下会生成一个xxx-binding.jar。拷贝到YCSB包下xxx-binding/lib目录下。
5.执行测试:可以通过-db指定我们的DB类,或在ycsb Python脚本中给我们的类加个别名。