stress test for the hasura graphql engine

this script allows you to configure and run workloads against a running graphql-engine instance, and produces an animated visualisation of memory usage. both bursty and sustained workloads are possible, and workloads can be configured to get heavier or lighter over time. in addition, it supports running workloads that do (for example) the same amount of "work" but with different amounts of concurrency.

workload structure

test workloads are of the following form, parameterised by the given options, which are all specified in the config file:

• we run loop_count loops
  • each loop consists of bursts_per_loop bursts
    • each burst consists of requests_per_burst requests
      • the request is a graphql file located at the path given by payload
      • after each request, we wait for request_delay seconds
    • once all requests are sent, the burst is complete
    • if wait_for_bursts_to_complete, we wait for graphql-engine to respond to all requests made as part of the burst here
    • then we wait for burst_delay seconds
• then we wait for all bursts to complete, so all requests made as part of the loop have been responded to by graphql-engine
• then we wait for loop_delay before starting a new loop

requests_per_burst and bursts_per_loop are configured in a way that lets them linearly ramp up (or down) over time to simulate increasingly heavy workloads, with _min and _incr options in the config file. requests_per_burst is changed after each burst, and reset at the start of a new loop. bursts_per_loop is changed after each loop.

setting the _incr option to zero allows for workloads of constant intensity.

visualisation format

• the marked regions on the graph correspond to periods in which a burst is being sent
• each short line marker is a response received from graphql-engine
• each green line corresponds to a burst which has been fully serviced by graphql-engine
• the heading is a concise description of the workload parameters:

  rpb_min(+rpb_incr) reqs + req_delay
> bpl_min(+bpl_incr) bursts + burst_delay
> loop_count loops + loop_delay

workload quality

bursty workloads can be simulated with fewer, larger bursts and a large burst delay. for sustained load tests, burst delay can be set to zero.

to implement concurrency scaling, one can do, e.g.

bursts_per_loop_min = [2, 4, 8, 16]
requests_per_burst_min = [16, 8, 4, 2]

this will run the same number of requests in each loop, but with decreasing concurrency.

to avoid overloading the server with too many pending requests, set wait_for_bursts_to_complete to true.

setup instructions

stress.py expects a graphql-engine pid to monitor for memory usage, and a configuration file that defines the parameters for the test. by default, it expects a graphql-engine instance on port 9080.

# install python deps
$ pip install -r requirements.txt

# set up a postgres db on port 7432
$ docker run --rm --name hge2 -e "POSTGRES_PASSWORD=password" -p 7432:5432 -d postgres -c "log_statement=all"

# import the data dump
$ PGPASSWORD=password psql -h 172.17.0.1 -U postgres -d postgres -p 7432 --single-transaction -f dump.sql

# start hge, e.g. for 1.3.2 (client repro version):
$ docker run --rm -p 9080:8080 hasura/graphql-engine:v1.3.2 graphql-engine \
  --database-url='postgresql://postgres:[email protected]:7432/postgres' serve \
  --enable-telemetry false --enable-console --query-plan-cache-size 0 --use-prepared-statements 'false'

# or for main:
$ cabal new-run -- exe:graphql-engine \
  --database-url='postgres://postgres:password@localhost:7432/postgres' \
  serve --server-port 9080 --enable-console --console-assets-dir=../console/static/dist \
--enable-telemetry false --query-plan-cache-size 0 --use-prepared-statements 'false' \
--enabled-apis "developer,graphql,metadata"

# now import the metadata in the console. you can also do this, at least running from main:
$ curl -XPOST -d '{"type":"replace_metadata","args":'$(cat metadata.json)'}' "http://localhost:9080/v1/metadata"

# run the tests
$ ./stress.py config/bursty.toml $(pidof graphql-engine)

single-loop configurations are provided as well, to test varying concurrency manually, using a total of 64 requests:

# 2 bursts, 32 requests each
$ ./stress.py config/single/2-32.toml $(pidof graphql-engine)

# 4 bursts, 16 requests each
$ ./stress.py config/single/4-16.toml $(pidof graphql-engine)

# 8 bursts, 8 requests each
$ ./stress.py config/single/8-8.toml $(pidof graphql-engine)