SdbMigrate tool

sdbmigrate - easy-peasy tool for applying set of SQL migration on PostgreSQL or MySQL. Supports sharding out of the box.

Supported Python versions:

- 3.8.X
- 3.9.X
- 3.10.X

Deprecated Python versions(should work fine, buy support will be dropped soon):

- 2.7.X

Supported PostgreSQL versions:

- 10.X
- 11.X
- 12.X
- 13.X
- 14.X
- 15.X

Supported MySQL versions:

- 5.7.X
- 8.0.X

Main features

work both for PostgreSQL and MySQL
schema versions out of the box
transactional and non-transactional steps
sharded migration steps
dry-run for transactional steps
ability to apply stored procedures/functions

Installation

To install sdbmigrate migrate you also need to install proper database connection library. For Postgres it is "psycopg2", for MySQL it is "mysqlclient".

Use one of the following recepies.

Install for Postgres:

pip install sdbmigrate[postgres]

Install for Mysql:

pip install sdbmigrate[mysql]

Install for both Postgres and MySQL:

pip install sdbmigrate[postgres,mysql]

Getting started guide with sdbmigrate and PostgreSQL

Install sdbmigrate

pip install sdbmigrate[postgres]

or clone this repo.

Install PostgreSQL on your system, see https://www.postgresql.org/download/
Create PostgreSQL user, e.g. test/test:

sudo -u postgres createuser -p 5432 test --pwprompt

Create database for applying migrations, e.g. :

for i in `seq 1 3` ; do echo $i ; sudo -u postgres createdb -p 5432 test_db$i ; done

Create YAML-config for sdbmigrate :

# a total number of shards for sharded tables
shard_count: 16

# shard distribution mode:
# "auto" - sdbmigrate migrate distribute shard across DB servers before initial
#          migration and then continue to use this distribution for all
#          sharded migrations. shard_on_db/shard_count and databases info
#          is used for such process.
# "manual" - shard distribution is specified by shards params inside databases
#            section
shard_distribution_mode: auto

# amount of shard perf DB master, used with shard_distribution_mode: "auto"
shard_on_db: 8

# information about database masters and their connection info
databases:
    - name: test_db1
      host: 127.0.0.1
      port: 5432
      # supported DB types: ["postgres", "mysql"]
      type: postgres
      user: test
      password: test

    - name: test_db2
      host: 127.0.0.1
      port: 5436
      # supported DB types: ["postgres", "mysql"]
      type: postgres
      user: test
      password: test

Run sdbmigrate with test DB :

sdbmigrate.py -c sdbmigrate.yaml -d demo/test_migrations

See migrations on disk

$ ls -l demo/test_migrations
total 24
-rw-rw-r-- 1 dr dr  244 Jul 18 09:46 V0000__TRX_PLAIN__initial_types.sql
-rw-rw-r-- 1 dr dr  902 Jul 18 09:42 V0001__TRX_PLAIN__initial_tables.sql
-rw-rw-r-- 1 dr dr 3781 Jul 18 09:48 V0002__TRX_SHARD__initial_tables.sql
-rw-rw-r-- 1 dr dr 1257 Jul 18 09:50 V0003__TRX_PLAIN__initial_procedures.sql
-rw-rw-r-- 1 dr dr  133 Jul 23 08:08 V0004__NOTRX_SHARD__extra_indices.sql
-rw-rw-r-- 1 dr dr   76 Jul 23 08:15 V0005__NOTRX_SHARD__drop_indices.sql

$ cat demo/test_migrations/V0000__TRX_PLAIN__initial_types.sql
CREATE TYPE item_type AS ENUM ('ownership', 'access');
CREATE TYPE item_value_type AS ENUM ('durable', 'consumable');
CREATE TYPE item_amount_unit AS ENUM ('number', 'time_seconds');
CREATE TYPE operation_type AS ENUM ('deposit','withdrawal');

Inspect DB using psql or other tool:

$ sudo -u postgres psql -p 5432 test_db2
=# \d+
...

See more info about sdbmigrate internals in docs/internals.md

Running tests locally using Docker

cd src
make test

Running tests locally

Setup Postgres

Mac OS:

brew install postgres
brew services start postgres
createuser -s postgres
psql -U postgres

Create user and databases

CREATE USER test_behave WITH SUPERUSER PASSWORD 'test_behave';
CREATE DATABASE sdbmigrate1_behave OWNER test_behave;
CREATE DATABASE sdbmigrate2_behave OWNER test_behave;

Setup MySQL