summaryrefslogtreecommitdiffstats
path: root/run-upgrade-scripts
diff options
context:
space:
mode:
Diffstat (limited to '')
-rwxr-xr-xrun-upgrade-scripts114
1 files changed, 114 insertions, 0 deletions
diff --git a/run-upgrade-scripts b/run-upgrade-scripts
new file mode 100755
index 0000000..aae3e90
--- /dev/null
+++ b/run-upgrade-scripts
@@ -0,0 +1,114 @@
+#!/usr/bin/perl -w
+# Run all upgrade scripts.
+#
+# (C) 2005-2009 Martin Pitt <mpitt@debian.org>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+
+use strict;
+use PgCommon;
+
+error "Usage: $0 <version upgraded from>" if $#ARGV != 0;
+
+# Return the cluster's databases that match the given scope.
+# Arguments: <version> <cluster> <scope>
+sub dbs_from_scope {
+ my ($v, $c, $scope) = @_;
+ my @dbs = get_cluster_databases $v, $c;
+ unless (defined $dbs[0]) {
+ print ' Error: cluster is not running';
+ return ();
+ }
+
+ # filter out the postgres database
+ @dbs = grep { $_ ne 'postgres' } @dbs;
+
+ return @dbs if $scope eq 't0';
+ return grep { $_ ne 'template0' } @dbs if $scope eq 't1';
+ return grep { $_ ne 'template0' && $_ ne 'template1' } @dbs if $scope eq 'db';
+ return grep { $_ eq 'template1' } @dbs if $scope eq 'cluster';
+}
+
+# Arguments: <script> <scope> <version> <cluster> <database>
+sub call_sql {
+ if (my $child = fork) { # parent
+ waitpid $child, 0;
+ return;
+ }
+
+ # child code
+ my ($script, $scope, $version, $cluster, $db) = @_;
+ $cluster =~ s/'/''/g; # escape ' in cluster name
+ my $orig_euid = $>;
+ $< = $> = (stat (PgCommon::cluster_data_directory $version, $cluster))[4];
+
+ # temporarily enable connections
+ my $conallow = `psql --cluster '$version/$cluster' template1 -AXtqc "select datallowconn from pg_database where datname='$db'"`;
+ chomp $conallow;
+ if ($conallow eq 'f') {
+ system "psql --cluster '$version/$cluster' template1 -AXtqc \"update pg_database set datallowconn = 't' where datname='$db'\"";
+ }
+
+ my $out = `ON_ERROR_STOP=1 psql --cluster '$version/$cluster' -f '$script' '$db' 2>&1`;
+
+ # reset allowconn
+ if ($conallow eq 'f') {
+ system "psql --cluster '$version/$cluster' template1 -AXtqc \"update pg_database set datallowconn = 'f' where datname='$db'\"";
+ }
+ print "[FAIL]\n$out" if $?;
+
+ exit 0;
+}
+
+# Arguments: <script> <scope> <version> <cluster> <database>
+sub call_script {
+ my ($script, $scope, $version, $cluster, $db) = @_;
+ $cluster =~ s/'/\\'/g; # escape ' in cluster name
+ my $out = `$script '$version' '$cluster' '$db' 2>&1`;
+ print "[FAIL]\n$out" if $?;
+}
+
+my $upgraded_version = $ARGV[0];
+
+# determine path of upgrade scripts
+my @f = split(/\//, $0);
+pop @f;
+@f = ('.') if $#f == -1;
+push @f, 'upgrade-scripts';
+my $scriptpath = join ('/', @f);
+
+opendir S, $scriptpath or error "Could not open script path '$scriptpath'";
+for my $script (sort readdir S) {
+ my ($fname, $ext) = split /\./, $script;
+ my ($version, $name, $scope) = split /_/, $fname;
+ my $is_sql = (defined $ext && $ext eq 'sql');
+ next unless defined ($scope) && ($is_sql || -x "$scriptpath/$script");
+
+ next unless ($version eq 'all' || $upgraded_version <= $version);
+ print "Executing upgrade script $name...\n";
+
+ for my $v (get_versions) {
+ for my $c (get_version_clusters $v) {
+ print " cluster $v/$c:";
+ for my $db (dbs_from_scope $v, $c, $scope) {
+ print " $db";
+ if ($is_sql) {
+ call_sql "$scriptpath/$script", $scope, $v, $c, $db;
+ } else {
+ call_script "$scriptpath/$script", $scope, $v, $c, $db;
+ }
+ }
+ print "\n";
+ }
+ }
+
+}
+closedir S;