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