https://docs.pingcap.com/zh/tidb/stable/tiup-bench

在测试数据库性能时，经常需要对数据库进行压测，为了满足这一需求，TiUP 集成了 bench 组件。TiUP bench 组件提供多种压测的 workloads，命令分别如下：

其中 tpcc, tpch, ch, rawsql 支持如下命令行参数。ycsb 使用方法较为不同，它主要通过 properties 文件进行配置，详见 go-ycsb 使用说明。

• --host 和 --port 支持以逗号分隔传入多个值，以启用客户端负载均衡。例如，当指定 --host 172.16.4.1,172.16.4.2 --port 4000,4001 时，负载程序将以轮询调度的方式连接到 172.16.4.1:4000, 172.16.4.1:4001, 172.16.4.2:4000, 172.16.4.2:4001 这 4 个实例上。
• --conn-params 需要符合 query string 格式，不同数据库支持不同参数，如：
  • --conn-params tidb_isolation_read_engines='tiflash' 设置 TiDB 通过 TiFlash 进行查询。
  • --conn-params sslmode=disable 设置连接 PostgreSQL 不启用加密。
• 当运行 CH-benCHmark 时，可以通过 --ap-host, --ap-port, --ap-conn-params 来指定独立的 TiDB 实例用于 OLAP 查询。

下文分别介绍如何使用 TiUP 运行 TPC-C, TPC-H 以及 YCSB 测试。

使用 TiUP 运行 TPC-C 测试

TiUP bench 组件支持如下运行 TPC-C 测试的命令和参数：

TPC-C 测试步骤

以下为简化后的关键步骤。完整的测试流程可以参考如何对 TiDB 进行 TPC-C 测试

1. 通过 HASH 使用 4 个分区创建 4 个仓库：

   tiup bench tpcc --warehouses 4 --parts 4 prepare

2. 运行 TPC-C 测试：

   tiup bench tpcc --warehouses 4 --time 10m run

3. 检查一致性：

   tiup bench tpcc --warehouses 4 check

4. 清理数据：

   tiup bench tpcc --warehouses 4 cleanup

当需要测试大数据集时，直接写入数据通常较慢，此时可以使用如下命令生成 CSV 数据集，然后通过 TiDB Lightning 导入数据。

• 生成 CSV 文件：

  tiup bench tpcc --warehouses 4 prepare --output-dir data --output-type=csv

• 为指定的表生成 CSV 文件：

  tiup bench tpcc --warehouses 4 prepare --output-dir data --output-type=csv --tables history,orders

使用 TiUP 运行 TPC-H 测试

TiUP bench 组件支持如下运行 TPC-H 测试的命令和参数：

TPC-H 测试步骤

1. 准备数据：

   tiup bench tpch --sf=1 prepare

2. 运行 TPC-H 测试，根据是否检查结果执行相应命令：

   • 检查结果：

     tiup bench tpch --count=22 --sf=1 --check=true run

   • 不检查结果：

     tiup bench tpch --count=22 --sf=1 run

3. 清理数据：

   tiup bench tpch cleanup

使用 TiUP 运行 YCSB 测试

你可以使用 TiUP 对 TiDB 和 TiKV 节点分别进行 YCSB 测试。

测试 TiDB

1. 准备数据：

   tiup bench ycsb load tidb -p tidb.instances="127.0.0.1:4000" -p recordcount=10000

2. 运行 YCSB 测试：

   # 默认读写比例为 95:5 tiup bench ycsb run tidb -p tidb.instances="127.0.0.1:4000" -p operationcount=10000

测试 TiKV

1. 准备数据：

   tiup bench ycsb load tikv -p tikv.pd="127.0.0.1:2379" -p recordcount=10000

2. 运行 YCSB 测试：

   # 默认读写比例为 95:5 tiup bench ycsb run tikv -p tikv.pd="127.0.0.1:2379" -p operationcount=10000

使用 TiUP 运行 RawSQL 测试

你可以将 OLAP 查询写到 SQL 文件中，通过 tiup bench rawsql 执行测试，步骤如下：

1. 准备数据和需要执行的查询：

   -- 准备数据 CREATE TABLE t (a int); INSERT INTO t VALUES (1), (2), (3); -- 构造查询，保存为 demo.sql SELECT a, sleep(rand()) FROM t WHERE a < 4*rand();

2. 运行 RawSQL 测试：

   tiup bench rawsql run --count 60 --query-files demo.sql