| 1 |
#!/usr/bin/perl -w |
| 2 |
# Replicate |
| 3 |
# Vadim Mikheev, (c) 2000, PostgreSQL Inc. |
| 4 |
|
| 5 |
eval '(exit $?0)' && eval 'exec perl -S $0 ${1+"$@"}' |
| 6 |
& eval 'exec perl -S $0 $argv:q' |
| 7 |
if 0; |
| 8 |
|
| 9 |
BEGIN { |
| 10 |
my $basedir = $0; $basedir =~ s#/[^/]+$##; |
| 11 |
unshift(@INC, "$basedir/../share"); |
| 12 |
} |
| 13 |
|
| 14 |
use strict; |
| 15 |
use IO::File; |
| 16 |
use Getopt::Long; |
| 17 |
use RServ; |
| 18 |
use Sys::Hostname; |
| 19 |
|
| 20 |
$| = 1; |
| 21 |
|
| 22 |
my ($debug,$verbose) = (0,0); |
| 23 |
my ($help,$masterhost,$masterport,$masteruser,$masterpassword, |
| 24 |
$slavehost,$slaveport,$slaveuser,$slavepassword); |
| 25 |
my $snapshot = ".__Snapshot"; |
| 26 |
|
| 27 |
my $result = GetOptions( |
| 28 |
"debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help, |
| 29 |
"masterhost=s" => \$masterhost, "masterport=i" => \$masterport, |
| 30 |
"masteruser=s" => \$masteruser, "masterpassword=s" => \$masterpassword, |
| 31 |
"slavehost=s" => \$slavehost, "slaveport=i" => \$slaveport, |
| 32 |
"slaveuser=s" => \$slaveuser, "slavepassword=s" => \$slavepassword, |
| 33 |
"snapshot=s" => \$snapshot, |
| 34 |
); |
| 35 |
|
| 36 |
if (defined($help) || (scalar(@ARGV) < 2)) { |
| 37 |
print "Usage: $0 [options] masterdb slavedb |
| 38 |
Options: |
| 39 |
--masterhost=hostname --masterport=port |
| 40 |
--masteruser=username --masterpassword=string |
| 41 |
--snapshot=snapshot |
| 42 |
"; |
| 43 |
exit ((scalar(@ARGV) < 2)? 1:0); |
| 44 |
} |
| 45 |
|
| 46 |
$snapshot .= '.' . $$; |
| 47 |
|
| 48 |
$RServ::quiet = !$verbose; |
| 49 |
if ($debug) { |
| 50 |
$RServ::quiet = 0; |
| 51 |
no warnings 'vars'; |
| 52 |
$RServ::debug = $debug; |
| 53 |
} |
| 54 |
|
| 55 |
my $master = $ARGV[0] || "master"; |
| 56 |
my $slave = $ARGV[1] || "slave"; |
| 57 |
my $tables = $#ARGV < 2 ? undef : { map {($_, undef)} @ARGV[2..$#ARGV] }; |
| 58 |
|
| 59 |
# first, let's check if a instance is alrealy running |
| 60 |
|
| 61 |
my $str = $master; |
| 62 |
$str .= '@'.$masterhost if (defined($masterhost)); |
| 63 |
$str .= ':'.$slave; |
| 64 |
$str .= '@'.$slavehost if (defined($slavehost)); |
| 65 |
my $fname = "/tmp/.lock_rserv-$str"; |
| 66 |
if (-e $fname) { |
| 67 |
open(PID,$fname) || die "can't open pid file '$fname'"; |
| 68 |
my $pid = <PID>; chomp($pid); |
| 69 |
close(PID); |
| 70 |
open(PS,"ps ax |") || die "can't start ps to check for running rServ with pid $pid\n"; |
| 71 |
my $is_running = 0; |
| 72 |
while(<PS>) { |
| 73 |
chomp; |
| 74 |
$is_running = 1 if (/^\s*$pid\s/); |
| 75 |
} |
| 76 |
close(PS); |
| 77 |
if ($is_running) { |
| 78 |
print STDERR "rServ [$pid] is already running for $str. Remove $fname to override this.\n"; |
| 79 |
exit(1); |
| 80 |
} else { |
| 81 |
print STDERR "found stale pid file, but no running rServ [$pid], overriding.\n"; |
| 82 |
} |
| 83 |
} |
| 84 |
open ARQ, ">$fname" || die "Cannot open $fname: $!\n"; |
| 85 |
print ARQ "$$\n"; |
| 86 |
close ARQ; |
| 87 |
|
| 88 |
$SIG{__DIE__} = \&mydie; |
| 89 |
sub mydie { |
| 90 |
unlink $fname; |
| 91 |
print @_; |
| 92 |
exit 1; |
| 93 |
} |
| 94 |
|
| 95 |
|
| 96 |
my $minfo = "dbname=$master"; |
| 97 |
$minfo = "$minfo host=$masterhost" if (defined($masterhost)); |
| 98 |
$minfo = "$minfo port=$masterport" if (defined($masterport)); |
| 99 |
$minfo = "$minfo user=$masteruser" if (defined($masteruser)); |
| 100 |
$minfo = "$minfo password=$masterpassword" if (defined($masterpassword)); |
| 101 |
|
| 102 |
my $sinfo = "dbname=$slave"; |
| 103 |
$sinfo = "$sinfo host=$slavehost" if (defined($slavehost)); |
| 104 |
$sinfo = "$sinfo port=$slaveport" if (defined($slaveport)); |
| 105 |
$sinfo = "$sinfo user=$slaveuser" if (defined($slaveuser)); |
| 106 |
$sinfo = "$sinfo password=$slavepassword" if (defined($slavepassword)); |
| 107 |
|
| 108 |
print "Master connection is $minfo\n" if ($debug); |
| 109 |
print "Slave connection is $sinfo\n" if ($debug); |
| 110 |
|
| 111 |
my $mconn = Pg::connectdb($minfo); |
| 112 |
if ($mconn->status != Pg::PGRES_CONNECTION_OK) { |
| 113 |
print STDERR "Failed opening $minfo\n"; |
| 114 |
unlink $fname; |
| 115 |
exit 1; |
| 116 |
} |
| 117 |
my $sconn = Pg::connectdb($sinfo); |
| 118 |
if ($sconn->status != Pg::PGRES_CONNECTION_OK) { |
| 119 |
print STDERR "Failed opening $sinfo\n"; |
| 120 |
unlink $fname; |
| 121 |
exit 1; |
| 122 |
} |
| 123 |
|
| 124 |
$slavehost = hostname if (! $slavehost); |
| 125 |
my $slaveId = GetSlaveId($mconn, $slave, $slavehost); |
| 126 |
|
| 127 |
die "\n>>>>>>>>>>>>> ERROR: Can't GetSlaveID for $sinfo\n" if (! $slaveId); |
| 128 |
|
| 129 |
if ($slaveId < 0) { |
| 130 |
unlink $fname; |
| 131 |
die "\n>>>>>>>>>>>>> ERROR\n"; |
| 132 |
} |
| 133 |
SyncSync($mconn, $sconn); |
| 134 |
|
| 135 |
my $outf = new IO::File; |
| 136 |
open $outf, ">$snapshot"; |
| 137 |
print "\n>>>>>>>>>>>>> Prepare Snapshot\n\n" if ($verbose); |
| 138 |
my $res = PrepareSnapshot($mconn, $sconn, $outf, $slaveId, $tables); |
| 139 |
close $outf; |
| 140 |
if ($res < 0) { |
| 141 |
unlink $fname; |
| 142 |
die "\n>>>>>>>>>>>>> ERROR\n"; |
| 143 |
} |
| 144 |
if ($res == 0) |
| 145 |
{ |
| 146 |
print "\n>>>>>>>>>>>>> DBases are sync-ed\n" if ($verbose); |
| 147 |
unlink $snapshot unless ($debug); |
| 148 |
unlink $fname; |
| 149 |
exit(0); |
| 150 |
} |
| 151 |
|
| 152 |
my $inpf = new IO::File; |
| 153 |
open $inpf, "<$snapshot"; |
| 154 |
print "\n>>>>>>>>>>>>> Apply Snapshot\n\n" if ($verbose); |
| 155 |
$res = ApplySnapshot($sconn, $inpf, $tables); |
| 156 |
close $inpf; |
| 157 |
if ($res < 0) { |
| 158 |
unlink $fname; |
| 159 |
die "\n>>>>>>>>>>>>> ERROR\n"; |
| 160 |
} |
| 161 |
|
| 162 |
if ($res > 0) |
| 163 |
{ |
| 164 |
print "Snapshot applied\n" if ($verbose); |
| 165 |
unlink $snapshot unless ($debug); |
| 166 |
SyncSync($mconn, $sconn); |
| 167 |
} |
| 168 |
|
| 169 |
unlink $fname; |
| 170 |
exit(0); |
| 171 |
|
| 172 |
########################################################################### |
| 173 |
|
| 174 |
sub SyncSync |
| 175 |
{ |
| 176 |
($mconn, $sconn) = @_; |
| 177 |
|
| 178 |
print "\n>>>>>>>>>>>>> Sync SyncID\n\n" if ($verbose); |
| 179 |
print "Get last SyncID from Slave DB\n" if ($verbose); |
| 180 |
my $syncid = GetSyncID($sconn); |
| 181 |
if ($syncid && $syncid > 0) |
| 182 |
{ |
| 183 |
print "Last SyncID applied: $syncid\n" if ($verbose); |
| 184 |
print "Sync SyncID\n" if ($verbose); |
| 185 |
|
| 186 |
$res = SyncSyncID($mconn, $slaveId, $syncid); |
| 187 |
|
| 188 |
print "Succeeded\n" if (($res > 0) && ($verbose)); |
| 189 |
} |
| 190 |
} |
Parent Directory
| 
Revision Log