git.sesse.net Git - wloh/blob - sync.pl

   1 #! /usr/bin/perl
   2 use DBI;
   3 use strict;
   4 use warnings;
   5 no warnings qw(once);
   6 use Encode;
   7 use Time::HiRes;
   8 use lib qw( include );
   9 require 'config.pm';
  10
  11 sub copy_escape {
  12         my $x = shift;
  13         if (!defined($x)) {
  14                 return '\N';
  15         }
  16         $x = Encode::decode('iso8859-1', $x);
  17         $x =~ s/([\\\r\n\t])/\\$1/g;
  18         return $x;
  19 }
  20
  21 sub sync_table {
  22         my ($mdbh, $sdbh, $table) = @_;
  23         my $start = [ Time::HiRes::gettimeofday ];
  24         my $data = $mdbh->selectall_arrayref("SELECT * FROM $table");
  25         my $elapsed = Time::HiRes::tv_interval($start);
  26         #printf "MySQL $table time used: $elapsed\n";
  27
  28         $start = [ Time::HiRes::gettimeofday ];
  29         $sdbh->do("DELETE FROM $table");
  30         $sdbh->do("COPY $table FROM STDIN");
  31
  32         for my $row (@$data) {
  33                 $sdbh->pg_putcopydata(join("\t", map { copy_escape($_) } (@$row)) . "\n");
  34         }
  35         $sdbh->pg_putcopyend();
  36         $elapsed = Time::HiRes::tv_interval($start);
  37         #printf "PostgreSQL $table time used: $elapsed\n";
  38 }
  39
  40 my $mdbh = DBI->connect($config::master_connstr, $config::master_username, $config::master_password)
  41         or die "PostgreSQL connect: " . $DBI::errstr;
  42 $mdbh->{AutoCommit} = 0;
  43 $mdbh->{RaiseError} = 1;
  44
  45 my $sdbh = DBI->connect($config::local_connstr, $config::local_username, $config::local_password)
  46         or die "PostgreSQL connect: " . $DBI::errstr;
  47 $sdbh->{AutoCommit} = 0;
  48 $sdbh->{RaiseError} = 1;
  49
  50 # For our own enjoyment.
  51 sync_table($mdbh, $sdbh, "Fotballbrukere");
  52 sync_table($mdbh, $sdbh, "Fotballdeltagere");
  53 sync_table($mdbh, $sdbh, "Fotballresultater");
  54 sync_table($mdbh, $sdbh, "Fotballresultater_2123");
  55 sync_table($mdbh, $sdbh, "Fotballserier");
  56
  57 # Just for backup.
  58 sync_table($mdbh, $sdbh, 'Fotballjuks');
  59 sync_table($mdbh, $sdbh, 'FotballKaffeklubb');
  60 sync_table($mdbh, $sdbh, 'Fotballkommentarer');
  61 sync_table($mdbh, $sdbh, 'Fotballkommentartema');
  62 sync_table($mdbh, $sdbh, 'FotballLogin');
  63 sync_table($mdbh, $sdbh, 'Fotballspraak');
  64 sync_table($mdbh, $sdbh, 'Fotballtoppti');
  65 sync_table($mdbh, $sdbh, 'Fotballvariabler');
  66
  67 $sdbh->do('DELETE FROM last_sync');
  68 $sdbh->do('INSERT INTO last_sync VALUES ( CURRENT_TIMESTAMP );');
  69
  70 $mdbh->rollback();
  71 $sdbh->commit();