Files
pgcat/tests/sharding
Lev Kokotov b93303eb83 Live reloading entire config and bug fixes (#84)
* Support reloading the entire config (including sharding logic) without restart.

* Fix bug incorrectly handing error reporting when the shard is set incorrectly via SET SHARD TO command.
selected wrong shard and the connection keep reporting fatal #80.

* Fix total_received and avg_recv admin database statistics.

* Enabling the query parser by default.

* More tests.
2022-06-24 14:52:38 -07:00
..
2022-02-10 09:07:10 -08:00
2022-02-08 17:15:35 -08:00

Sharding tests

This helps us test the sharding algorithm we implemented.

Setup

We setup 3 Postgres DBs, shard0, shard1, and shard2. In each database, we create a partitioned table called data. The table is partitioned by hash, and each database will only have one partition, shard0 will satisfy modulus 3, remainder 0, shard1 will satisfy modulus 3, remainder 1, etc.

To set this up, you can just run:

psql -f query_routing_setup.sql

Run the tests

Start up PgCat by running cargo run --release in the root of the repo. In a different tab, run this:

psql -h 127.0.0.1 -p 6432 -f query_routing_test_insert.sql
psql -h 127.0.0.1 -p 6432 -f query_routing_test_select.sql

Note that no errors should take place. If our sharding logic was incorrect, we would get some errors about unsatisfiable partition bounds. We don't because the pooler picked the correct databases given the sharding keys.

Finally, you can validate the result again by running

psql -f query_routing_test_validate.sql

That's it!