components/sqldb/pgsql-conf

#!/bin/sh

#  Licensed to the Apache Software Foundation (ASF) under one
#  or more contributor license agreements.  See the NOTICE file
#  distributed with this work for additional information
#  regarding copyright ownership.  The ASF licenses this file
#  to you under the Apache License, Version 2.0 (the
#  "License"); you may not use this file except in compliance
#  with the License.  You may obtain a copy of the License at
#  
#    http://www.apache.org/licenses/LICENSE-2.0
#    
#  Unless required by applicable law or agreed to in writing,
#  software distributed under the License is distributed on an
#  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#  KIND, either express or implied.  See the License for the
#  specific language governing permissions and limitations
#  under the License.

# Configure a postgresql master server
here=`echo "import os; print os.path.realpath('$0')" | python`; here=`dirname $here`
mkdir -p $1
root=`echo "import os; print os.path.realpath('$1')" | python`

addr=$2
if [ "$addr" = "" ]; then
    ip="*"
    port="5432"
else
    ip=`$here/../../modules/http/httpd-addr ip $addr`
    if [ "$ip" = "" ]; then
        ip="*"
    fi
    port=`$here/../../modules/http/httpd-addr port $addr`
fi

pgsql_prefix=`cat $here/pgsql.prefix`
mkdir -p $root/sqldb/data
chmod 700 $root/sqldb/data
mkdir -p $root/sqldb/archive

if [ -f "$root/sqldb/log.conf" ]; then
    pgsql_log=`cat $root/sqldb/log.conf`
else
    mkdir -p $root/logs
    pgsql_log="cat >>$root/logs/postgresql"
fi
mkdir -p $root/sqldb
echo $pgsql_log >$root/sqldb/logger

if [ ! -f $root/sqldb/data/postgresql.conf ]; then
    $pgsql_prefix/bin/pg_ctl init -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger
    cp $root/sqldb/data/postgresql.conf $root/sqldb/data/postgresql-init.conf
    cp $root/sqldb/data/pg_hba.conf $root/sqldb/data/pg_hba-init.conf
fi

# Generate server configuration
cp $root/sqldb/data/postgresql-init.conf $root/sqldb/data/postgresql.conf
cat >>$root/sqldb/data/postgresql.conf <<EOF

# Generated by: pgsql-conf $*

# Setup logging
log_min_messages = INFO
log_min_error_statement = INFO
log_min_duration_statement = 0
log_checkpoints = on
log_connections = on
log_disconnections = on
log_duration = on
log_lock_waits = on
log_statement = all

# Listen
listen_addresses = '$ip'
port = $port

# Setup archival
archive_mode = on
archive_command = 'cp %p $root/sqldb/archive/%f'

# Setup hot standby with streaming replication
wal_level = hot_standby
max_wal_senders = 5
wal_keep_segments = 32

EOF

# Generate client auth configuration
cp $root/sqldb/data/pg_hba-init.conf $root/sqldb/data/pg_hba.conf
cat >>$root/sqldb/data/pg_hba.conf <<EOF

# Generated by: pgsql-conf $*
# TYPE  DATABASE        USER            CIDR-ADDRESS            METHOD
host    all             all             samenet                 trust
host    replication     all             samenet                 trust

EOF

# Create the db
($pgsql_prefix/bin/pg_ctl start -W -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger
sti=0
while [ $sti -ne 30 ]; do
    st=`$pgsql_prefix/bin/pg_ctl status -D $root/sqldb/data | grep 'server is running'`
    if [ "$st" != "" ]; then
        break
    fi
    sleep 1
    sti=$((sti+1))
done

$pgsql_prefix/bin/createdb -h localhost -p $port db 2>&1 | sh $root/sqldb/logger
$pgsql_prefix/bin/pg_ctl stop -w -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger

# Generate database backup script
mkdir -p $root/sqldb/scripts
cat >$root/sqldb/scripts/backup <<EOF
#!/bin/sh
$here/pgsql-backup $root localhost $port
EOF
chmod 700 $root/sqldb/scripts/backup

# Configure HTTPD to serve backup and archive files
if [ -f "$root/conf/httpd.conf" ]; then
    cat >>$root/conf/httpd.conf <<EOF
# Generated by: pgsql-conf $*

# Serve PostgreSQL backup and WAL archive files
ScriptAlias /pgsql-backup "$root/sqldb/scripts/backup"
Alias /pgsql-archive "$root/sqldb/archive"

EOF

fi