Docker啓動PostgreSQL時創建多個數據庫

1 前言

在文章《Docker啓動PostgreSQL並推薦幾款連接工具》中我們介紹如何通過Docker來啓動PostgreSQL，但只有一個數據庫，如果想要創建多個數據庫在同一個Docker容器上怎麼辦呢？

2 兩種方案

一種方案是把shell/sql腳本放入/docker-entrypoint-initdb.d/目錄中，讓容器啓動的時候自動執行創建；另一種是通過shell腳本指定創建，本質是一樣的。這裏只介紹第一種。

把shell腳本或sql腳本放入指定目錄，就會自動執行，兩種腳本都可以。

shell腳本例子如下：

#!/bin/bash

set -e
set -u

function create_user_and_database() {
	local database=$1
	echo "  Creating user and database '$database'"
	psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" <<-EOSQL
	    CREATE USER $database;
	    CREATE DATABASE $database;
	    GRANT ALL PRIVILEGES ON DATABASE $database TO $database;
EOSQL
}

if [ -n "$POSTGRES_MULTIPLE_DATABASES" ]; then
	echo "Multiple database creation requested: $POSTGRES_MULTIPLE_DATABASES"
	for db in $(echo $POSTGRES_MULTIPLE_DATABASES | tr ',' ' '); do
		create_user_and_database $db
	done
	echo "Multiple databases created"
fi

sql腳本例子如下：

CREATE USER pkslowuser;

CREATE DATABASE logdata;
GRANT ALL PRIVILEGES ON DATABASE logdata TO pkslowuser;

CREATE DATABASE orderdata;
GRANT ALL PRIVILEGES ON DATABASE orderdata TO pkslowuser;

CREATE DATABASE userdata;
GRANT ALL PRIVILEGES ON DATABASE userdata TO pkslowuser;

3 打包啓動

準備Dockerfile，把shell/sql腳本文件放入鏡像中去：

FROM postgres:10
COPY src/main/resources/create-multiple-postgresql-databases.sh /docker-entrypoint-initdb.d/
COPY src/main/resources/create-multiple-postgresql-databases.sql /docker-entrypoint-initdb.d/

啓動如下：

docker run -itd \
    --name pkslow-postgres \
    -e POSTGRES_MULTIPLE_DATABASES=db1,db2 \
    -e POSTGRES_USER=pkslow \
    -e POSTGRES_PASSWORD=pkslow \
    -p 5432:5432 \
    pkslow/postgresql-multiple-databases:1.0-SNAPSHOT

啓動成功後，就會創建如下數據庫：

db1,db2,
logdata,orderdata,userdata

4 總結

這是在開發測試階段使用的方案，實際上把數據庫放在容器中不是一個好的選擇。

代碼請查看：https://github.com/LarryDpk/pkslow-samples