DEV Community

Franck Pachot for YugabyteDB

Posted on • Updated on

docker-compose.yaml to start YugabyteDB with POSTGRES_USER POSTGRES_PASSWORD POSTGRES_DB env

The Docker image provided by Yugabyte for YugabyteDB doesn't create and start a database because it can be used for different purposes: start one yb-master or yb-tserver, or both with yugabyted. An operational database is composed of multiple nodes. However, for development or automated tests, you just want to start one container with a defined database, user and password.

YugabyteDB is PostgreSQL compatible, which means that you can replace PostgreSQL by YugabyteDB without changing the application. The PostgreSQL image allows to create a database and set user and password with the following environment variables: POSTGRES_USER POSTGRES_PASSWORD POSTGRES_DB and you probably want the same for YugabyteDB. And, why not, with the same environment variables.

Here is an example based on knex.js test suite

This docker-compose.yml starts PostgreSQL with a database named knex_test and user/password testuser/knextest and starts another container to check when this connection is available.

Here is how I add YugabyteDB in the same way:

version: '3'

    command: |
     bash -c "
     # create database and user as soon as database is up
     until [ -z "POSTGRES_USER" ] || PGPASSWORD=yugabyte bin/ysqlsh -v ON_ERROR_STOP=1 \\
      -c \"create database $${POSTGRES_DB:-$${POSTGRES_USER}} \" \\
      -c \"create user $${POSTGRES_USER} password '$${POSTGRES_PASSWORD}' \" \\
       echo \"Waiting for YugabyteDB to be up for creating user $${POSTGRES_USER}.\" ; sleep 5
      done &
     # start YugabyteDB
     bin/yugabyted start --listen --daemon=false --tserver_flags='ysql_enable_auth=true'
      - POSTGRES_USER=testuser
      - POSTGRES_PASSWORD=knextest
      - POSTGRES_DB=knex_test
      - 25433:5433
      - 27000:7000

    image: postgres:13-alpine
      - yugabytedb
      - yugabytedb
      - bash
      - -c
      - 'until PGPASSWORD=knextest /usr/local/bin/psql postgres://testuser@yugabytedb:5433/knex_test -c "SELECT 1"; do sleep 5; done'

Enter fullscreen mode Exit fullscreen mode

The YugabyteDB database is started with bin/yugabyted start --daemon=false --tserver_flags='ysql_enable_auth=true' and a background loop tries to connect to create the database and the user according to the environment variables.

The default port for YugabyteBD is 5433 which makes it possible to start it along with PostgreSQL on its default port. The port 7000 is the graphical console.

If you want a more complex docker-compose.yaml with full control on yb-master/yb-tserver cluster topology, here is how I generate it for my labs:

Update on June 2022

The following command is probably better, using the --initial_scripts_dir:

    command: |
     bash -c '
     mkdir -p /var/ybinit
     echo "create database $${POSTGRES_DB:-$${POSTGRES_USER}};             " > /var/ybinit/01-db.sql
     echo "create user $${POSTGRES_USER} with password '\''$${POSTGRES_PASSWORD}'\'';" > /var/ybinit/02-usr.sql
     # start YugabyteDB
     bin/yugabyted start --listen --daemon=false --initial_scripts_dir=/var/ybinit --tserver_flags=ysql_enable_auth=true
Enter fullscreen mode Exit fullscreen mode

and another example with Nakama docker-compose if the waityugabyte doesn't work for you:

Top comments (0)