About db-benchmarks open source test framework

Test framework

The test framework which is used on the backend of https://db-benchmarks.com is fully Open Source (AGPLv3 license) and can be found at https://github.com/db-benchmarks/db-benchmarks . Here’s what it does:

  • Automates data loading to the databases/search engines included in the repository.
  • Can run a database/search engine in Docker with a particular CPU/RAM constraint.
  • While testing:
    • Purges OS cache automatically
    • Automates purging database caches before each cold run
    • Restarts the database before each cold run
    • Looks after your CPU temperature to avoid throttling
    • Looks after the coefficient of variation while making queries and can stop as soon as:
      • The CV is low enough
      • And the number of queries made is sufficient
    • After starting a database/search engine, lets it do its warm-up stage (pre-read needed data from disk), stops waiting as soon as:
      • There’s no IO for a few seconds
      • And it can connect to the database/search engine
    • After stopping a database/search engine waits until it fully stops
    • Can accept different timeouts: start, warm-up, initial connection, getting info about the database/search engine, query
    • Can emulate one physical core which allows benchmarking algorithmic capabilities of databases more objectively (--limited)
    • Can accept all the values as command line arguments as well as environment variables for easier integration with CI systems
    • --test saves test results to file
    • --save saves test results from files to a remote database (neither of those that have been tested)
    • Tracks a lot of things while testing:
      • Server info: CPU, memory, running processes, filesystem, hostname
      • Current repository info to make sure there’s no local changes
      • Performance metrics: each query response time in microseconds, aggregated stats:
        • Coefficient of variation of all queries
        • Coefficient of variation of 80% fastest queries
        • Cold query’s response time
        • Avg(response times)
        • Avg(80% fastest queries’ response times)
        • Slowest query’s response time
      • Database/search engine info:
        • select count(*) and select * limit 1 to make sure the data collections are similar in different databases
        • internal database/search engine data structures status (chunks, shards, segments, partitions, parts, etc.)
  • Makes it easy to limit CPU/RAM consumption inside or outside the test (using environment variables cpuset and mem).
  • Allows to start each database/search engine easily the same way it’s started by the framework for manual testing and preparation of test queries.

Installation

Before you deploy the test framework, make sure you have the following:

  • Linux server fully dedicated to testing
  • Fresh CPU thermal paste to make sure your CPUs don’t throttle down
  • PHP 8 and:
    • curl module
    • mysqli module
  • docker
  • docker-compose
  • sensors to control CPU temperature to prevent throttling
  • dstat

To install:

  1. git clone from the repository:
    1
    2
    
    git clone git@github.com:db-benchmarks/db-benchmarks.git
    cd db-benchmarks
    
  2. Copy .env.example to .env
  3. Update mem and cpuset in .env with the default value of the memory (in megabytes) and CPUs the test framework can use for secondary tasks (data loading, getting info about databases)
  4. Tune JVM limits ES_JAVA_OPTS for your tests. Usually it’s size of allocated memory for Docker Machine

Get started

Prepare test

First you need to prepare a test:

Go to a particular test’s directory (all tests must be in directory ./tests), for example “hn_small”:

1
cd tests/hn_small

Run the init script:

1
./init

This will:

  • download the data collection from the Internet
  • build the tables/indices

Run test

Then run ../../test (it’s in the project root’s folder) to see the options:

 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
To run a particular test with specified engines, memory constraints and number of attempts and save the results locally:
	/perf/test_engines/test
	--test=test_name
	--engines={engine1:type,...,engineN}
	--memory=1024,2048,...,1048576 - memory constraints to test with, MB
	[--times=N] - max number of times to test each query, 100 by default
	[--dir=path] - if path is omitted - save to directory 'results' in the same dir where this file is located
	[--probe_timeout=N] - how long to wait for an initial connection, 30 seconds by default
	[--start_timeout=N] - how long to wait for a db/engine to start, 120 seconds by default
	[--warmup_timeout=N] - how long to wait for a db/engine to warmup after start, 300 seconds by default
	[--query_timeout=N] - max time a query can run, 900 seconds by default
	[--info_timeout=N] - how long to wait for getting info from a db/engine
	[--limited] - emulate one physical CPU core
	[--queries=/path/to/queries] - queries to test, ./tests/<test name>/test_queries by default
To save to db all results it finds by path
	/perf/test_engines/test
	--save=path/to/file/or/dir, all files in the dir recursively will be saved
	--host=HOSTNAME
	--port=PORT
	--username=USERNAME
	--password=PASSWORD
	--rm - remove after successful saving to database
	--skip_calm - avoid waiting until discs become calm
----------------------
Environment variables:
	All the options can be specified as environment variables, but you can't use the same option as an environment variables and as a command line argument at the same time.

And run the test:

1
../../test --test=hn_small --engines=elasticsearch,clickhouse --memory=16384

If you run your tests in local mode (development) and don’t care about tests inaccuracy you can avoid discs calm and cpu checks by setting parameter --skip_inaccuracy

1
../../test --test=hn_small --engines=elasticsearch,clickhouse --memory=16384 --skip_inaccuracy

Now you have test results in ./results/ (in the root of the repository), for example:

1
2
# ls results/
220401_054753

Save to db to visualize

You can now upload the results to the database for further visualization. The visualization tool, which is used on https://db-benchmarks.com/ , is also open source and can be found at https://github.com/db-benchmarks/ui.

Here’s how you can save the results:

1
username=login password=pass host=db.db-benchmarks.com port=443 save=./results ./test

or

1
./test --username=login --password=pass --host=db.db-benchmarks.com --port=443 --save=./results

Make pull request

We are eager to see your test results. If you believe they should be added to https://db-benchmarks.com, please make a pull request of your results to this repository.

Please keep the following in mind:

  • Your results should be located in the directory ./results.
  • If it’s a new test/engine, any other changes should be included in the same pull request.
  • It is important that we, and anyone else, should be able to reproduce your test and hopefully get similar results.

We will then:

  • Review your results to ensure they follow the testing principles.
  • Reproduce your test on our hardware to ensure they are comparable with other tests.
  • Discuss any arising questions with you.
  • And, if everything checks out, we will merge your pull request.

Directory structure

 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
39
40
41
42
43
44
45
46
47
48
49
.
  |-.env                                    <- you need to update "mem" here
  |-test                                    <- the executable file which you need to run to test or save test results
  |-plugins                                 <- plugins directory: if you decide to extend the framework by adding one more database / search engine to test you need to put it into this directory
  |  |-elasticsearch.php                    <- Elasticsearch plugin
  |  |-manticoresearch.php                  <- Manticore Search plugin
  |  |-clickhouse.php                       <- ClickHouse plugin
  |  |-mysql.php                            <- MySQL plugin
  |-README.md                               <- you are reading this file
  |-tests                                   <- tests directory
  |  |-hn                                   <- Hackernews test
  |  |  |-prepare_csv                       <- Here we prepare the data collection, it's done in ./tests/hn/init
  |  |  |-description                       <- Test description which is included into test results and then is to be used when the results are visualized
  |  |  |-manticore                         <- In this dir happens everything related to testing Manticore Search WRT the current test (Hackernews)
  |  |  |  |-init                           <- This is a common script which should be in every <test>/<database> directory which is responsible for generating all for the <database>
  |  |  |-ch                                <- "Hackernews test -> ClickHouse" directory
  |  |  |  |-data_limited                   <- This will be mounted to ClickHouse Docker if the docker-compose is run with env. var. suffix=_limited 
  |  |  |  |-post_load_limited.sh           <- This is a hook which is triggered after the data load, called by ./init in the same directory
                                               Note, there's no post_load.sh (with no suffix), which means that no hook will be called in this case.
  |  |  |  |-data                           <- This is another ClickHouse directory, no suffix means the docker-compose should be run with suffix= (empty value)
  |  |  |  |-init                           <- ClickHouse's init script
  |  |  |-es                                <- "Hackernews test -> Elasticsearch" directory
  |  |  |  |-logstash_limited               <- Logstash config dir for type "limited", hence suffix "_limited"
  |  |  |  |  |-post_load.sh                
  |  |  |  |  |-logstash.conf               <- Logstash config
  |  |  |  |  |-template.json               <- Logstash template
  |  |  |  |  |-jvm.options                 <- Logstash jvm options
  |  |  |  |-elasticsearch_limited.yml      <- Elasticsearch config for type "limited"
  |  |  |  |-logstash                       <- Logstash config dir for the default type
  |  |  |  |  |-logstash.conf               
  |  |  |  |  |-template.json
  |  |  |  |  |-jvm.options
  |  |  |  |-logstash_tuned                 <- Logstash config dir for type "tuned"
  |  |  |  |  |-post_load.sh
  |  |  |  |  |-logstash.conf
  |  |  |  |  |-template.json
  |  |  |  |  |-jvm.options
  |  |  |  |-elasticsearch.yml
  |  |  |  |-elasticsearch_tuned.yml
  |  |  |  |-init
  |  |  |-test_queries                      <- All test queries for the current test are here
  |  |  |-test_info_queries                 <- And here should be those queries that are called to get info about the data collection
  |  |  |-data                              <- Prepared data collection here
  |  |  |-init                              <- Main initialization script for the test
  |  |-taxi                                 <- Another test: Taxi rides, similar structure
  |  |-hn_small                             <- Another test: non-multiplied Hackernews dataset, similar structure
  |  |-logs10m                              <- Another test: Nginx logs, similar structure
  |-docker-compose.yml                      <- docker-compose config: responsible for starting / stopping the databases / search engines
  |-results                                 <- test results, the results you see on https://db-benchmarks.com/ can be found here and you can use ./test --save to visualize them yourself

How to start a particular database / search engine with a particular dataset

1
test=logs10m cpuset="0,1" mem=32768 suffix=_tuned docker-compose up elasticsearch

will:

  • start Elasticsearch to test “logs10m” with the following settings:
  • suffix=_tuned: maps ./tests/logs10m/es/data/idx_tuned as a data directory
  • mem=32768 limits RAM to 32GB, if not specified the default will be used from file .env
  • cpuset="0,1": Elasticsearch’s container will be running only on CPU cores 0 and 1 (which may be the first whole physical CPU)

To stop - just CTRL-C.

Notes