+++ /dev/null
-#!/bin/bash
-#
-# priority: 900
-#
-# Bootstrap the database
-#
-# Mark Huang <mlhuang@cs.princeton.edu>
-# Copyright (C) 2006 The Trustees of Princeton University
-#
-
-# Source function library and configuration
-. /etc/plc.d/functions
-. /etc/planetlab/plc_config
-
-# Be verbose
-set -x
-
-# Export so that we do not have to specify -p to psql invocations
-export PGPORT=$PLC_DB_PORT
-
-# Install extensions
-function extend_db()
-{
- shopt -s nullglob
- for file in /usr/share/plc_api/extensions/*-up*; do
- script=${file##*/}
- name=${script%-up*}
- extension=${script##*.}
- version=$(psql -U $PLC_DB_USER --quiet --tuples-only --no-align -c \
- "SELECT version FROM plc_db_extensions WHERE name='$name' LIMIT 1" \
- $PLC_DB_NAME 2>/dev/null | awk 'BEGIN { ver=0 } /^[0-9]+$/ { ver=$1 } END { print ver }')
- if [ $version -eq 0 ]; then
- if [ "$extension" = "sql" ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- psql -U $PLC_DB_USER -f $file $PLC_DB_NAME
- elif [ -x $file ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- $file
- else
- dialog "\nWarning: extension $file not executable"
- fi
- check
- fi
- for file in /usr/share/plc_api/extensions/$name/migrations/[0-9]*-up-*; do
- script=${file##*/}
- index=${script%-up-*}
- extension=${script##*.}
- if [ $index -gt $version ] ; then
- if [ "$extension" = "sql" ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- psql -U $PLC_DB_USER -f $file $PLC_DB_NAME
- elif [ -x $file ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- $file
- else
- dialog "\nWarning: migration $file not executable"
- fi
- check
- fi
- done
- done
-}
-
-# Updates the database by applying all migration scripts in
-# /usr/share/plc_api/migrations/N-up-*, where N is greater than the
-# current subversion. At least one of the migration scripts with the
-# same N must update plc_db_version.subversion.
-function migrate_db()
-{
- subversion=$(psql -U $PLC_DB_USER --quiet --tuples-only --no-align -c \
- "SELECT subversion FROM plc_db_version LIMIT 1" \
- $PLC_DB_NAME 2>/dev/null || echo 0)
- shopt -s nullglob
- for file in /usr/share/plc_api/migrations/[0-9]*-up-* ; do
- script=$(basename $file)
- index=${script%-up*}
- extension=${script##*.}
- if [ $index -gt $subversion ] ; then
- if [ "$extension" = "sql" ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- psql -U $PLC_DB_USER -f $file $PLC_DB_NAME
- elif [ -x $file ] ; then
- dialog " - $script (dbdumped)"
- dump_planetlab_db "before-$script"
- $file
- else
- dialog "\nWarning: migration $file not executable"
- fi
- check
- fi
- done
-}
-
-function checkpoint_planetlab_db()
-{
- dumpfile=$1
- pg_dump -U $PLC_DB_USER $PLC_DB_NAME > $dumpfile
- check
-}
-
-function restore_planetlab_db()
-{
- dumpfile=$1
- if [ -n "$dumpfile" ] ; then
- [ -f "$dumpfile" ] && psql -a -U $PLC_DB_USER $PLC_DB_NAME < $dumpfile
- check
- fi
-}
-
-# use a single date of this script invocation for the dump_*_db functions.
-DATE=$(date +"%Y-%m-%d-%H-%M-%S")
-
-# Dumps the database - optional argument to specify filename suffix
-function dump_planetlab_db()
-{
- if [ -n "$1" ] ; then suffix="-$1" ; else suffix="" ; fi
- dumpfile=/var/lib/pgsql/backups/$(date +"${PLC_DB_NAME}.${DATE}${suffix}.sql")
- checkpoint_planetlab_db $dumpfile
-}
-
-function restore_drupal_db()
-{
- dumpfile=$1
- if [ -n "$dumpfile" ] ; then
- [ -f "$dumpfile" ] && psql -a -U $PLC_DB_USER drupal < $1
- check
- fi
-}
-
-function checkpoint_drupal_db()
-{
- dumpfile=$1
- pg_dump -U $PLC_DB_USER drupal > $dumpfile
- check
-}
-
-function dump_drupal_db()
-{
- dumpfile=/var/lib/pgsql/backups/$(date +"drupal.${DATE}.sql")
- checkpoint_drupal_db $dumpfile
- check
-}
-
-# Clean up old backups
-function clean_dumps()
-{
- find /var/lib/pgsql/backups '(' -name "$PLC_DB_NAME.*.sql" -o -name "drupal.*.sql" ')' -a -atime +15 | xargs rm -f
- check
-}
-
-[ $PLC_DB_ENABLED -ne 1 ] && exit 0
-case "$1" in
- start)
- MESSAGE=$"Bootstrapping the database"
- dialog "$MESSAGE"
-
- # Apply schema updates
- migrate_db
- extend_db
-
- # Update the maintenance account username. This can't be
- # done through the api-config script since it uses the
- # maintenance account to access the API. The maintenance
- # account should be person_id 1 since it is created by the
- # DB schema itself.
- psql -U $PLC_DB_USER -c "UPDATE persons SET email='$PLC_API_MAINTENANCE_USER' WHERE person_id=1" $PLC_DB_NAME
-
- # Update the Drupal site_name variable
- # also turn off drupal native user registration
- psql -U $PLC_DB_USER drupal <<EOF
-DELETE FROM variable WHERE name = 'site_name';
-INSERT INTO variable (name, value) VALUES ('site_name', 's:${#PLC_NAME}:"$PLC_NAME";');
-DELETE FROM variable WHERE name = 'user_register';
-INSERT INTO variable (name, value) VALUES ('user_register', 's:1:"0";');
-DELETE FROM cache;
-EOF
-
- # Bootstrap the DB
- db-config
- check
-
- result "$MESSAGE"
- ;;
-
- migrate)
- MESSAGE=$"Migrating the database"
- dialog "$MESSAGE"
-
- migrate_db
- result "$MESSAGE"
- ;;
-
- dump)
- MESSAGE=$"Dumping the databases in /var/lib/pgsql/backups"
- dialog "$MESSAGE"
-
- dump_planetlab_db
- dump_drupal_db
- result "$MESSAGE"
- ;;
-
- checkpoint)
- MESSAGE=$"Checkpointing the databases"
- checkpoint_planetlab_db $2
- checkpoint_drupal_db $3
- ;;
-
- restore)
- MESSAGE=$"Restoring the databases from checkpoint files"
- restore_planetlab_db $2
- restore_drupal_db $3
- ;;
-
- clean-dump)
- MESSAGE=$"Cleaning old database dumps"
- dialog "$MESSAGE"
-
- clean_dumps
- result "$MESSAGE"
- ;;
-
- stop)
- MESSAGE="Ignoring request to stop myplc databases"
- dialog "$MESSAGE"
- result ""
- ;;
-
- *)
- echo "Usage: $0 [start|migrate|dump|checkpoint|restore|clean-dump|stop]"
- exit 1
- ;;
-esac
-
-exit $ERRORS