aboutsummaryrefslogtreecommitdiff
path: root/src/pl/plpgsql/enable_plpgsql
blob: efd2cf1973d4e8030d5d66ca19802634f59181a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#!/bin/sh

# Enable the PL procedural language for PostgreSQL in one or more
# existing databases. 
#
# This script should be run by the PostgreSQL superuser

enable_database() {
	if ! psql -d $1 -qtc "select count(*) from pg_language where lanname='plpgsql'" >$TMPFIL2 2>&1
	then
		echo "Cannot connect to $1"
		exit 2
	fi
	if [ `cat $TMPFIL2` -eq 0 ]
	then
		if ! psql -d $1 <$sqlfile
		then
			echo "Failed to add PL to $1"
			exit 2
		fi
		echo "PL added to $1"
	else
		echo "PL is already enabled in $1"
	fi

}

# Execution starts here

TMPFILE=`mktemp /tmp/enable_pgpl.XXXXXX`
TMPFIL2=`mktemp /tmp/enable_pgpl.XXXXXX`
trap "rm $TMPFILE $TMPFIL2" EXIT

sqlfile=${PGLIB:=/usr/local/pgsql/lib}/mklang_pl.sql
if [ ! -f $sqlfile ]
then
	echo "Cannot find mklang_pl.sql"
	exit 2
fi

if [ -z "$1" ]
then
	echo "Syntax: $0 --all | database ..."
	exit 1
fi

if [ $1 = "--all" ]
then
	if ! psql -t -c "select datname from pg_database order by datname" >$TMPFILE
	then
		echo Cannot select databases
		exit 2
	fi
	for db in `cat $TMPFILE`
	do
		enable_database $db
	done
else
	while [ -n "$1" ]
	do
		db=$1
		enable_database $db
		shift
	done
fi