Docker啟動PostgreSQL時創(chuàng)建多個數(shù)據(jù)庫的解決方案

更新時間：2021年06月30日 08:44:29 作者：南瓜慢說

這篇文章主要介紹了Docker啟動PostgreSQL時創(chuàng)建多個數(shù)據(jù)庫的解決方案，重點給大家介紹如何把shell/sql腳本放入/docker-entrypoint-initdb.d/目錄中，讓容器啟動的時候自動執(zhí)行創(chuàng)建

1 前言

在文章《Docker啟動PostgreSQL并推薦幾款連接工具》中我們介紹如何通過Docker來啟動PostgreSQL，但只有一個數(shù)據(jù)庫，如果想要創(chuàng)建多個數(shù)據(jù)庫在同一個Docker容器上怎么辦呢？

2 兩種方案

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

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

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

啟動成功后，就會創(chuàng)建如下數(shù)據(jù)庫：