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

Legend:
Removed from v.1.1.1.1  
changed lines
  Added in v.1.18

  ViewVC Help
Powered by ViewVC 1.1.26