/[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.2 by dpavlin, Tue Aug 5 09:20:08 2003 UTC revision 1.9 by dpavlin, Sun Aug 10 13:51:54 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;
18    use Sys::Hostname;
19    
20  $| = 1;  $| = 1;
21    
22  $result = GetOptions("debug!", "verbose!", "help", "snapshot=s",  my ($debug,$verbose) = (0,0);
23                       "masterhost=s", "slavehost=s", "host=s",  my ($help,$masterhost,$masterport,$masteruser,$masterpassword,
24                       "masteruser=s", "slaveuser=s", "user=s",          $slavehost,$slaveport,$slaveuser,$slavepassword);
25                       "masterpassword=s", "slavepassword=s", "password=s");  my $snapshot = ".__Snapshot";
26    
27  my $debug = $opt_debug || 0;  my $result = GetOptions(
28  my $verbose = $opt_verbose || 0;          "debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
29  my $snapshot = $opt_snapshot || "__Snapshot";          "masterhost=s" => \$masterhost, "masterport=i" => \$masterport,
30            "masteruser=s" => \$masteruser, "masterpassword=s" => \$masterpassword,
31  if (defined($opt_help) || (scalar(@ARGV) < 2)) {          "slavehost=s" => \$slavehost, "slaveport=i" => \$slaveport,
32      print "Usage: $0 --snapshot=file --host=name --user=name --password=string masterdb slavedb\n";          "slaveuser=s" => \$slaveuser, "slavepassword=s" => \$slavepassword,
33      print "\t--masterhost=name --masteruser=name --masterpassword=string\n";          "snapshot=s" => \$snapshot,
34      print "\t--slavehost=name --slaveuser=name --slavepassword=string\n";          );
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);      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";  my $master = $ARGV[0] || "master";
56  my $slave = $ARGV[1] || "slave";  my $slave = $ARGV[1] || "slave";
57  my $server = 0;  
58    # first, let's check if a instance is alrealy running
59    
60    my $str = $master;
61    $str .= '@'.$masterhost if (defined($masterhost));
62    $str .= ':'.$slave;
63    $str .= '@'.$slavehost if (defined($slavehost));
64    my $fname = "/tmp/.lock_rserv-$str";
65    if (-e $fname) {
66            open(PID,$fname) || die "can't open pid file '$fname'";
67            my $pid = <PID>; chomp($pid);
68            close(PID);
69            open(PS,"ps ax |") || die "can't start ps to check for running rServ with pid $pid\n";
70            my $is_running = 0;
71            while(<PS>) {
72                    chomp;
73                    $is_running = 1 if (/^\s*$pid\s/);
74            }
75            close(PS);
76            if ($is_running) {
77                    print STDERR "rServ [$pid] is already running for $str. Remove $fname to override this.\n";
78                    exit(1);
79            } else {
80                    print STDERR "found stale pid file, but no running rServ [$pid], overriding.\n";
81            }
82    }
83    open ARQ, ">$fname" || die "Cannot open $fname: $!\n";
84    print ARQ "$$\n";
85    close ARQ;
86    
87    $SIG{__DIE__} = \&mydie;
88    sub mydie {
89            unlink $fname;
90            print @_;
91            exit 1;
92    }
93    
94    
95  my $minfo = "dbname=$master";  my $minfo = "dbname=$master";
96  $minfo = "$minfo host=$opt_masterhost" if (defined($opt_masterhost));  $minfo = "$minfo host=$masterhost" if (defined($masterhost));
97  $minfo = "$minfo user=$opt_masteruser" if (defined($opt_masteruser));  $minfo = "$minfo port=$masterport" if (defined($masterport));
98  $minfo = "$minfo password=$opt_masterpassword" if (defined($opt_masterpassword));  $minfo = "$minfo user=$masteruser" if (defined($masteruser));
99    $minfo = "$minfo password=$masterpassword" if (defined($masterpassword));
100    
101  my $sinfo = "dbname=$slave";  my $sinfo = "dbname=$slave";
102  $sinfo = "$sinfo host=$opt_slavehost" if (defined($opt_slavehost));  $sinfo = "$sinfo host=$slavehost" if (defined($slavehost));
103  $sinfo = "$sinfo user=$opt_slaveuser" if (defined($opt_slaveuser));  $sinfo = "$sinfo port=$slaveport" if (defined($slaveport));
104  $sinfo = "$sinfo password=$opt_slavepassword" if (defined($opt_slavepassword));  $sinfo = "$sinfo user=$slaveuser" if (defined($slaveuser));
105    $sinfo = "$sinfo password=$slavepassword" if (defined($slavepassword));
106    
107  print "Master connection is $minfo\n" if ($debug);  print "Master connection is $minfo\n" if ($debug);
108  print "Slave connection is $sinfo\n" if ($debug);  print "Slave connection is $sinfo\n" if ($debug);
109    
110  my $mconn = Pg::connectdb($minfo);  my $mconn = Pg::connectdb($minfo);
111    if ($mconn->status != Pg::PGRES_CONNECTION_OK) {
112        print STDERR "Failed opening $minfo\n";
113        unlink $fname;
114        exit 1;
115    }
116  my $sconn = Pg::connectdb($sinfo);  my $sconn = Pg::connectdb($sinfo);
117    if ($sconn->status != Pg::PGRES_CONNECTION_OK) {
118        print STDERR "Failed opening $sinfo\n";
119        unlink $fname;
120        exit 1;
121    }
122    
123  $RServ::quiet = !$verbose;  $slavehost = hostname if (! $slavehost);
124    my $slaveId = GetSlaveId($mconn, $slave, $slavehost);
125    
126    die "\n>>>>>>>>>>>>> ERROR: Can't GetSlaveID for $sinfo\n" if (! $slaveId);
127    
128    if ($slaveId < 0) {
129        unlink $fname;
130        die "\n>>>>>>>>>>>>> ERROR\n";
131    }
132  SyncSync($mconn, $sconn);  SyncSync($mconn, $sconn);
133    
134  my $outf = new IO::File;  my $outf = new IO::File;
135  open $outf, ">$snapshot";  open $outf, ">$snapshot";
136  print "\n>>>>>>>>>>>>> Prepare Snapshot\n\n" if ($verbose);  print "\n>>>>>>>>>>>>> Prepare Snapshot\n\n" if ($verbose);
137  $res = PrepareSnapshot($mconn, $outf, $server);  my $res = PrepareSnapshot($mconn, $sconn, $outf, $slaveId);
138  close $outf;  close $outf;
139  die "\n>>>>>>>>>>>>> ERROR\n" if $res < 0;  if ($res < 0) {
140        unlink $fname;
141        die "\n>>>>>>>>>>>>> ERROR\n";
142    }
143  if ($res == 0)  if ($res == 0)
144  {  {
145          print "\n>>>>>>>>>>>>> DBases are sync-ed\n" if ($verbose);      print "\n>>>>>>>>>>>>> DBases are sync-ed\n" if ($verbose);
146          exit(0);      unlink $snapshot unless ($debug);
147        unlink $fname;
148        exit(0);
149  }  }
150    
151  my $inpf = new IO::File;  my $inpf = new IO::File;
# Line 70  open $inpf, "<$snapshot"; Line 153  open $inpf, "<$snapshot";
153  print "\n>>>>>>>>>>>>> Apply Snapshot\n\n" if ($verbose);  print "\n>>>>>>>>>>>>> Apply Snapshot\n\n" if ($verbose);
154  $res = ApplySnapshot($sconn, $inpf);  $res = ApplySnapshot($sconn, $inpf);
155  close $inpf;  close $inpf;
156  die "\n>>>>>>>>>>>>> ERROR\n" if $res < 0;  if ($res < 0) {
157        unlink $fname;
158        die "\n>>>>>>>>>>>>> ERROR\n";
159    }
160    
161  if ($res > 0)  if ($res > 0)
162  {  {
163          print "Snapshot applied\n" if ($verbose);      print "Snapshot applied\n" if ($verbose);
164          unlink $snapshot unless ($debug);      unlink $snapshot unless ($debug);
165          SyncSync($mconn, $sconn);      SyncSync($mconn, $sconn);
166  }  }
167    
168    unlink $fname;
169  exit(0);  exit(0);
170    
171  ###########################################################################  ###########################################################################
# Line 89  sub SyncSync Line 176  sub SyncSync
176    
177      print "\n>>>>>>>>>>>>> Sync SyncID\n\n" if ($verbose);      print "\n>>>>>>>>>>>>> Sync SyncID\n\n" if ($verbose);
178      print "Get last SyncID from Slave DB\n" if ($verbose);      print "Get last SyncID from Slave DB\n" if ($verbose);
179      $syncid = GetSyncID($sconn);      my $syncid = GetSyncID($sconn);
180      if ($syncid > 0)      if ($syncid && $syncid > 0)
181      {      {
182          print "Last SyncID applied: $syncid\n" if ($verbose);          print "Last SyncID applied: $syncid\n" if ($verbose);
183          print "Sync SyncID\n" if ($verbose);          print "Sync SyncID\n" if ($verbose);
184    
185          $res = SyncSyncID($mconn, $server, $syncid);          $res = SyncSyncID($mconn, $slaveId, $syncid);
186    
187          print "Succeeded\n" if (($res > 0) && ($verbose));          print "Succeeded\n" if (($res > 0) && ($verbose));
188      }      }
Legend:
Removed from v.1.2  
changed lines
  Added in v.1.9
  ViewVC Help
Powered by ViewVC 1.1.26