Skip to main content

DB migration tool for PostgreSQL/MySQL with sharding support

Project description

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:

- 2.7.X
- 3.8.X

Supported PostgreSQL versions:

- 9.6.X
- 10.X
- 11.X
- 13.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.

  1. Install for Postgres:
pip install sdbmigrate[postgres]
  1. Install for Mysql:
pip install sdbmigrate[mysql]
  1. Install for both Postgres and MySQL:
pip install sdbmigrate[postgres,mysql]

Getting started guide with sdbmigrate and PostgreSQL

  1. Install sdbmigrate
pip install sdbmigrate[postgres]

or clone this repo.

  1. Install PostgreSQL on your system, see https://www.postgresql.org/download/

  2. Create PostgreSQL user, e.g. test/test:

sudo -u postgres createuser -p 5432 test --pwprompt
  1. 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
  1. 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

  1. Run sdbmigrate with test DB :
sdbmigrate.py -c sdbmigrate.yaml -d demo/test_migrations
  1. 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');

  1. 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

  1. 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;
    
  2. Setup MySQL

    Mac OS:

    brew install mysql
    brew services start mysql
    mysql -h 127.0.0.1 -u root -p
    <enter>
    

    Create user and databases

    CREATE DATABASE IF NOT EXISTS sdbmigrate1_behave;
    CREATE DATABASE IF NOT EXISTS sdbmigrate2_behave;
    CREATE USER IF NOT EXISTS 'test_behave'@'%' IDENTIFIED BY 'test_behave';
    GRANT ALL PRIVILEGES ON sdbmigrate1_behave.* TO 'test_behave'@'%';
    GRANT ALL PRIVILEGES ON sdbmigrate2_behave.* TO 'test_behave'@'%';
    FLUSH PRIVILEGES;
    
  3. Setup python 2.7 and python 3.8 on your local machine.

    Mac OS:

    brew install python@2.7 python@3.8
    
  4. Install dev requirements.

    pip install -r src/requirements_dev.txt
    
  5. Run tests

    make -C src test_local
    

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sdbmigrate-1.0.5.tar.gz (21.5 kB view hashes)

Uploaded source

Built Distribution

sdbmigrate-1.0.5-py3-none-any.whl (41.0 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page