/[rserv]/bin/Replicate
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /bin/Replicate

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.4 by dpavlin, Tue Aug 5 10:01:42 2003 UTC revision 1.13 by dpavlin, Thu Oct 30 22:51:45 2003 UTC
# Line 1  Line 1 
1  #!/usr/bin/perl  #!/usr/bin/perl -w
2  # Replicate  # Replicate
3  # Vadim Mikheev, (c) 2000, PostgreSQL Inc.  # Vadim Mikheev, (c) 2000, PostgreSQL Inc.
4    
# Line 6  eval '(exit $?0)' && eval 'exec perl -S Line 6  eval '(exit $?0)' && eval 'exec perl -S
6      & eval 'exec perl -S $0 $argv:q'      & eval 'exec perl -S $0 $argv:q'
7      if 0;      if 0;
8    
9  use lib "/usr/share/postgresql/contrib";  BEGIN {
10            my $basedir = $0; $basedir =~ s#/[^/]+$##;
11            unshift(@INC, "$basedir/../share");
12    }
13    
14    use strict;
15  use IO::File;  use IO::File;
16  use Getopt::Long;  use Getopt::Long;
17  use RServ;  use RServ;
 use Sys::Hostname;  
18    
19  $| = 1;  $| = 1;
20    
21  $result = GetOptions("debug!", "verbose!", "help", "snapshot=s",  my ($debug,$verbose) = (0,0);
22                       "masterhost=s", "slavehost=s",  my ($help,$masterhost,$masterport,$masteruser,$masterpassword,
23                       "masteruser=s", "slaveuser=s",          $slavehost,$slaveport,$slaveuser,$slavepassword);
24                       "masterpassword=s", "slavepassword=s",  my $snapshot = ".__Snapshot";
25                       "masterport=i", "slaveport=i"  my $multiplemaster = 0;
26                       );  
27    my $result = GetOptions(
28  my $debug = $opt_debug || 0;          "debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
29  my $verbose = $opt_verbose || 0;          "masterhost=s" => \$masterhost, "masterport=i" => \$masterport,
30  my $snapshot = $opt_snapshot || ".__Snapshot";          "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            "multiplemaster" => \$multiplemaster,
35            );
36    
37    if (defined($help) || (scalar(@ARGV) < 2)) {
38        print "Usage: $0 [options] masterdb slavedb
39    Options:
40            --masterhost=hostname --masterport=port
41            --masteruser=username --masterpassword=string
42            --snapshot=snapshot
43            [--masterserver=master_number]
44    ";
45        exit ((scalar(@ARGV) < 2)? 1:0);
46    }
47    
48  $snapshot .= '.' . $$;  $snapshot .= '.' . $$;
49    
50  if (defined($opt_help) || (scalar(@ARGV) < 2)) {  $RServ::quiet = !$verbose;
51      print "Usage: $0 --snapshot=file masterdb slavedb\n";  if ($debug) {
52      print "\t--masterhost=name --masterport=port\n";          $RServ::quiet = 0;
53      print "\t--masteruser=name --masterpassword=string\n";          no warnings 'vars';
54      print "\t--slavehost=name --slaveport=port\n";          $RServ::debug = $debug;
     print "\t--slaveuser=name --slavepassword=string\n";  
     exit ((scalar(@ARGV) < 2)? 1:0);  
55  }  }
56    
57  my $master = $ARGV[0] || "master";  my $master = $ARGV[0] || "master";
58  my $slave = $ARGV[1] || "slave";  my $slave = $ARGV[1] || "slave";
59    my $tables = $#ARGV < 2 ? undef : { map {($_, undef)} @ARGV[2..$#ARGV] };
 $masterhost = $opt_masterhost || hostname;  
 $slavehost = $opt_slavehost || hostname;  
60    
61  # first, let's check if a instance is alrealy running  # first, let's check if a instance is alrealy running
62    
63  $str = $master.'@'.$masterhost.':'.$slave.'@'.$slavehost;  my $str = $master;
64    $str .= '@'.$masterhost if (defined($masterhost));
65    $str .= ':'.$slave;
66    $str .= '@'.$slavehost if (defined($slavehost));
67  my $fname = "/tmp/.lock_rserv-$str";  my $fname = "/tmp/.lock_rserv-$str";
68  if (-e $fname) {  if (-e $fname) {
69      print STDERR "rServ is already running for $str. Remove $fname to override this.\n";          open(PID,$fname) || die "can't open pid file '$fname'";
70      exit(1);          my $pid = <PID>; chomp($pid);
71            close(PID);
72            open(PS,"ps ax |") || die "can't start ps to check for running rServ with pid $pid\n";
73            my $is_running = 0;
74            while(<PS>) {
75                    chomp;
76                    $is_running = 1 if (/^\s*$pid\s/);
77            }
78            close(PS);
79            if ($is_running) {
80                    print STDERR "rServ [$pid] is already running for $str. Remove $fname to override this.\n";
81                    exit(1);
82            } else {
83                    print STDERR "found stale pid file, but no running rServ [$pid], overriding.\n";
84            }
85  }  }
86  open ARQ, ">$fname" || die "Cannot open $fname: $!\n";  open ARQ, ">$fname" || die "Cannot open $fname: $!\n";
87  print ARQ "$$\n";  print ARQ "$$\n";
88  close ARQ;  close ARQ;
89    
90    $SIG{__DIE__} = \&mydie;
91    sub mydie {
92            unlink $fname;
93            print @_;
94            exit 1;
95    }
96    
97    
98  my $minfo = "dbname=$master";  my $minfo = "dbname=$master";
99  $minfo = "$minfo host=$opt_masterhost" if (defined($opt_masterhost));  $minfo = "$minfo host=$masterhost" if (defined($masterhost));
100  $minfo = "$minfo user=$opt_masteruser" if (defined($opt_masteruser));  $minfo = "$minfo port=$masterport" if (defined($masterport));
101  $minfo = "$minfo password=$opt_masterpassword" if (defined($opt_masterpassword));  $minfo = "$minfo user=$masteruser" if (defined($masteruser));
102  $minfo = "$minfo port=$opt_masterport" if (defined($opt_masterport));  $minfo = "$minfo password=$masterpassword" if (defined($masterpassword));
103    
104  my $sinfo = "dbname=$slave";  my $sinfo = "dbname=$slave";
105  $sinfo = "$sinfo host=$opt_slavehost" if (defined($opt_slavehost));  $sinfo = "$sinfo host=$slavehost" if (defined($slavehost));
106  $sinfo = "$sinfo user=$opt_slaveuser" if (defined($opt_slaveuser));  $sinfo = "$sinfo port=$slaveport" if (defined($slaveport));
107  $sinfo = "$sinfo password=$opt_slavepassword" if (defined($opt_slavepassword));  $sinfo = "$sinfo user=$slaveuser" if (defined($slaveuser));
108  $sinfo = "$sinfo port=$opt_slaveport" if (defined($opt_slaveport));  $sinfo = "$sinfo password=$slavepassword" if (defined($slavepassword));
109    
110  print "Master connection is $minfo\n" if ($debug);  print "Master connection is $minfo\n" if ($debug);
111  print "Slave connection is $sinfo\n" if ($debug);  print "Slave connection is $sinfo\n" if ($debug);
112    
113  my $mconn = Pg::connectdb($minfo);  my $mconn = Pg::connectdb($minfo);
114  if ($mconn->status != PGRES_CONNECTION_OK) {  if ($mconn->status != Pg::PGRES_CONNECTION_OK) {
115      print STDERR "Failed opening $minfo\n";      print STDERR "Failed opening $minfo\n";
116      unlink $fname;      unlink $fname;
117      exit 1;      exit 1;
118  }  }
119  my $sconn = Pg::connectdb($sinfo);  my $sconn = Pg::connectdb($sinfo);
120  if ($sconn->status != PGRES_CONNECTION_OK) {  if ($sconn->status != Pg::PGRES_CONNECTION_OK) {
121      print STDERR "Failed opening $sinfo\n";      print STDERR "Failed opening $sinfo\n";
122      unlink $fname;      unlink $fname;
123      exit 1;      exit 1;
124  }  }
125    
126  my $slaveId = GetSlaveId($mconn, $slave, defined($opt_slavehost) ? $opt_slavehost : hostname);  $slavehost = 'localhost' if (! $slavehost);
127    $masterhost = 'localhost' if (! $masterhost);
128    
129    my $slaveId = GetServerId($mconn, $slave, $slavehost);
130    
131    die "\n>>>>>>>>>>>>> ERROR: Can't GetServerId for $sinfo\n" if (! defined($slaveId));
132    
133  if ($slaveId < 0) {  if ($slaveId < 0) {
134      unlink $fname;      unlink $fname;
135      die "\n>>>>>>>>>>>>> ERROR\n";      die "\n>>>>>>>>>>>>> ERROR: GetServerId returned $slaveId < 0\n";
136  }  }
137  SyncSync($mconn, $sconn);  SyncSync($mconn, $sconn);
138    
139    my $masterId = 0;
140    if ($multiplemaster) {
141            my $masterId = GetServerId($mconn, $master, $masterhost);
142    
143            die "\n>>>>>>>>>>>>> ERROR: Can't GetServerId for $sinfo\n" if (! defined($masterId));
144    
145            if ($masterId < 0) {
146                unlink $fname;
147                die "\n>>>>>>>>>>>>> ERROR: GetServerId returned $masterId < 0\n";
148            }
149            SyncSync($sconn, $mconn);
150    }
151    
152  my $outf = new IO::File;  my $outf = new IO::File;
153  open $outf, ">$snapshot";  open $outf, ">$snapshot";
154  print "\n>>>>>>>>>>>>> Prepare Snapshot\n\n" if ($verbose);  print "\n>>>>>>>>>>>>> Prepare Snapshot\n\n" if ($verbose);
155  $res = PrepareSnapshot($mconn, $sconn, $outf, $slaveId);  print "master id: $masterId slave id: $slaveId\n" if ($debug);
156    my $res = PrepareSnapshot($mconn, $sconn, $outf, $masterId, $slaveId, $tables);
157  close $outf;  close $outf;
158  if ($res < 0) {  if ($res < 0) {
159      unlink $fname;      unlink $fname;
# Line 108  if ($res == 0) Line 170  if ($res == 0)
170  my $inpf = new IO::File;  my $inpf = new IO::File;
171  open $inpf, "<$snapshot";  open $inpf, "<$snapshot";
172  print "\n>>>>>>>>>>>>> Apply Snapshot\n\n" if ($verbose);  print "\n>>>>>>>>>>>>> Apply Snapshot\n\n" if ($verbose);
173  $res = ApplySnapshot($sconn, $inpf);  $res = ApplySnapshot($sconn, $inpf, $tables);
174  close $inpf;  close $inpf;
175  if ($res < 0) {  if ($res < 0) {
176      unlink $fname;      unlink $fname;
# Line 133  sub SyncSync Line 195  sub SyncSync
195    
196      print "\n>>>>>>>>>>>>> Sync SyncID\n\n" if ($verbose);      print "\n>>>>>>>>>>>>> Sync SyncID\n\n" if ($verbose);
197      print "Get last SyncID from Slave DB\n" if ($verbose);      print "Get last SyncID from Slave DB\n" if ($verbose);
198      $syncid = GetSyncID($sconn);      my $syncid = GetSyncID($sconn);
199      if ($syncid > 0)      if ($syncid && $syncid > 0)
200      {      {
201          print "Last SyncID applied: $syncid\n" if ($verbose);          print "Last SyncID applied: $syncid\n" if ($verbose);
202          print "Sync SyncID\n" if ($verbose);          print "Sync SyncID\n" if ($verbose);
Legend:
Removed from v.1.4  
changed lines
  Added in v.1.13
  ViewVC Help
Powered by ViewVC 1.1.26