bin/BackupPC_incPartsUpdate

   1 #!/usr/local/bin/perl -w
   2
   3 use strict;
   4 use lib "__INSTALLDIR__/lib";
   5
   6 use DBI;
   7 use BackupPC::Lib;
   8 use BackupPC::View;
   9 use Data::Dumper;
  10 use Time::HiRes qw/time/;
  11 use POSIX qw/strftime/;
  12 use BackupPC::SearchLib;
  13 use Cwd qw/abs_path/;
  14 use File::Which;
  15
  16 my $path = abs_path($0);
  17 $path =~ s#/[^/]+$#/#;
  18 my $tarIncCreate = $path .= 'BackupPC_tarIncCreate';
  19
  20 die "can't find $tarIncCreate: $!\n" unless (-x $tarIncCreate);
  21
  22 my $bin;
  23 foreach my $c (qw/gzip split/) {
  24         $bin->{$c} = which($c) || die "$0 needs $c, install it\n";
  25 }
  26
  27
  28 my $debug = 0;
  29 $|=1;
  30
  31 my $start_t = time();
  32
  33 my $t_fmt = '%Y-%m-%d %H:%M:%S';
  34
  35 my $hosts;
  36 my $bpc = BackupPC::Lib->new || die;
  37 my %Conf = $bpc->Conf();
  38 my $TopDir = $bpc->TopDir();
  39 my $beenThere = {};
  40
  41 my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
  42 my $user = $Conf{SearchUser} || '';
  43
  44 my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });
  45
  46 my $tar_dir = $Conf{InstallDir}.'/'.$Conf{GzipTempDir};
  47
  48 die "problem with $tar_dir, check GzipTempDir in configuration\n" unless (-d $tar_dir && -w $tar_dir);
  49
  50 #---- subs ----
  51
  52 sub fmt_time {
  53         my $t = shift || return;
  54         my $out = "";
  55         my ($ss,$mm,$hh) = gmtime($t);
  56         $out .= "${hh}h" if ($hh);
  57         $out .= sprintf("%02d:%02d", $mm,$ss);
  58         return $out;
  59 }
  60
  61 sub curr_time {
  62         return strftime($t_fmt,localtime());
  63 }
  64
  65 #----- main
  66
  67 my $sth = $dbh->prepare( qq{
  68
  69 select
  70         backups.id as backup_id,
  71         hosts.name as host,
  72         shares.name as share,
  73         backups.num as num,
  74         inc_size,
  75         parts
  76 from backups
  77         join shares on backups.hostid = shares.hostid
  78                 and shares.id = backups.shareid
  79         join hosts on shares.hostid = hosts.id
  80 where not inc_deleted
  81 order by backups.date
  82
  83 } );
  84
  85 $sth->execute();
  86
  87 my $sth_inc_size = $dbh->prepare(qq{ update backups set inc_size = ?, parts = ? where id = ? });
  88 my $sth_inc_deleted = $dbh->prepare(qq{ update backups set inc_deleted = ? where id = ? });
  89
  90 %BackupPC::SearchLib::Conf = %Conf;
  91
  92 while (my $row = $sth->fetchrow_hashref) {
  93         my $tar_file = BackupPC::SearchLib::getGzipName($row->{'host'}, $row->{'share'}, $row->{'num'});
  94
  95         # this will return -1 if file doesn't exist
  96         my $size = BackupPC::SearchLib::get_tgz_size_by_name($tar_file);
  97
  98         print curr_time, " ", $row->{'host'}, ":", $row->{'share'}, " #", $row->{'num'}, " -> $tar_file";
  99
 100         my $t = time();
 101
 102         # re-create archive?
 103         if ($row->{'inc_size'} == -1 || $size == -1 || $row->{'inc_size'} != $size) {
 104                 my $cmd = qq{rm -Rf $tar_dir/$tar_file && $tarIncCreate -h "$row->{'host'}" -s "$row->{'share'}" -n $row->{'num'} | $bin->{'gzip'} -9 > $tar_dir/$tar_file};
 105                 print STDERR "## $cmd\n" if ($debug);
 106
 107                 system($cmd) == 0 or die "failed: $?";
 108
 109                 $size = (stat( "$tar_dir/$tar_file" ))[7];
 110         }
 111
 112         if ($size > 45) {
 113
 114                 my $max_size = $Conf{'MaxArchiveSize'} || die "problem with MaxArchieSize parametar";
 115                 $max_size *= 1024;      # convert to bytes
 116
 117                 my $parts = int( ($size + $max_size - 1) / $max_size );
 118
 119                 if (-d "$tar_dir/$tar_file" && $parts != $row->{'parts'}) {
 120                         print " join";
 121
 122                         my $in = my $out = "$tar_dir/$tar_file";
 123                         $out .= '.tmp';
 124
 125                         # FIXME I should really order parts manually!
 126                         system("cat $in/part* > $out && rm -Rf $in && mv $out $in") == 0 or die "can't join $in: $?";
 127                 }
 128
 129                 if ($size > $max_size && ! -d "$tar_dir/$tar_file") {
 130                         print " split/$parts";
 131                         my $in = my $out = "$tar_dir/$tar_file";
 132                         $out .= '.tmp';
 133                         rename $in, $out || die "can't rename $in: $!";
 134                         mkdir $in || die "can't mkdir $in: $!";
 135
 136                         my $suffix_len = length("$parts");
 137                         system("$bin->{'split'} -d -b $max_size -a $suffix_len $out $in/part") == 0 or die "can't split $out: $?";
 138                         unlink $out || die "can't unlink $out: $!";
 139                 }
 140
 141                 $sth_inc_size->execute($size, $parts, $row->{'backup_id'});
 142                 $sth_inc_deleted->execute(0, $row->{'backup_id'});
 143
 144                 printf(" %1.2f MB", ($size / 1024 / 1024));
 145
 146         } else {
 147                 $sth_inc_deleted->execute(1, $row->{'backup_id'});
 148                 unlink "$tar_dir/$tar_file" || die "can't delete $tar_dir/$tar_file: $!\n";
 149                 print " EMPTY";
 150         }
 151         print ", dur: ",fmt_time(time() - $t), "\n";
 152
 153         $dbh->commit;
 154
 155 }
 156
 157 undef $sth;
 158 $dbh->disconnect;