Annotation of /trunk/bin/BackupPC_incPartsUpdate

#!/usr/local/bin/perl -w

use strict;
use lib "__INSTALLDIR__/lib";

use DBI;
use BackupPC::Lib;
use BackupPC::View;
use Data::Dumper;
use Time::HiRes qw/time/;
use POSIX qw/strftime/;
use BackupPC::SearchLib;
use Cwd qw/abs_path/;
use File::Which;

my $path = abs_path($0);
$path =~ s#/[^/]+$#/#;
my $tarIncCreate = $path .= 'BackupPC_tarIncCreate';

die "can't find $tarIncCreate: $!\n" unless (-x $tarIncCreate);

my $bin;
foreach my $c (qw/gzip split/) {
        $bin->{$c} = which($c) || die "$0 needs $c, install it\n";
}


my $debug = 0;
$|=1;

my $start_t = time();

my $t_fmt = '%Y-%m-%d %H:%M:%S';

my $hosts;
my $bpc = BackupPC::Lib->new || die;
my %Conf = $bpc->Conf();
my $TopDir = $bpc->TopDir();
my $beenThere = {};

my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
my $user = $Conf{SearchUser} || '';

my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });

my $tar_dir = $Conf{InstallDir}.'/'.$Conf{GzipTempDir};

die "problem with $tar_dir, check GzipTempDir in configuration\n" unless (-d $tar_dir && -w $tar_dir);

#---- subs ----

sub fmt_time {
        my $t = shift || return;
        my $out = "";
        my ($ss,$mm,$hh) = gmtime($t);
        $out .= "${hh}h" if ($hh);
        $out .= sprintf("%02d:%02d", $mm,$ss);
        return $out;
}

sub curr_time {
        return strftime($t_fmt,localtime());
}

#----- main

my $sth = $dbh->prepare( qq{
        
select
        backups.id as backup_id,
        hosts.name as host,
        shares.name as share,
        backups.num as num,
        inc_size,
        parts
from backups
        join shares on backups.hostid = shares.hostid
                and shares.id = backups.shareid
        join hosts on shares.hostid = hosts.id
where not inc_deleted
order by backups.date

} );

$sth->execute();

my $sth_inc_size = $dbh->prepare(qq{ update backups set inc_size = ?, parts = ? where id = ? });
my $sth_inc_deleted = $dbh->prepare(qq{ update backups set inc_deleted = ? where id = ? });

%BackupPC::SearchLib::Conf = %Conf;

while (my $row = $sth->fetchrow_hashref) {
        my $tar_file = BackupPC::SearchLib::getGzipName($row->{'host'}, $row->{'share'}, $row->{'num'});

        # this will return -1 if file doesn't exist
        my $size = BackupPC::SearchLib::get_tgz_size_by_name($tar_file);

        print curr_time, " ", $row->{'host'}, ":", $row->{'share'}, " #", $row->{'num'}, " -> $tar_file";

        my $t = time();

        # re-create archive?
        if ($row->{'inc_size'} == -1 || $size == -1 || $row->{'inc_size'} != $size) {
                my $cmd = qq{rm -Rf $tar_dir/$tar_file && $tarIncCreate -h "$row->{'host'}" -s "$row->{'share'}" -n $row->{'num'} | $bin->{'gzip'} $Conf{GzipLevel} > ${tar_dir}/${tar_file}.tmp};
                print STDERR "## $cmd\n" if ($debug);

                system($cmd) == 0 or die "failed: $?";

                rename "${tar_dir}/${$tar_file}.tmp", "$tar_dir/$tar_file" or die "can't rename $tar_dir/$tar_file: $!";

                $size = (stat( "$tar_dir/$tar_file" ))[7];
        }

        if ($size > 45) {

                my $max_size = $Conf{'MaxArchiveSize'} || die "problem with MaxArchieSize parametar";
                $max_size *= 1024;      # convert to bytes

                my $parts = int( ($size + $max_size - 1) / $max_size );

                if (-d "$tar_dir/$tar_file" && $parts != $row->{'parts'}) {
                        print " join";

                        my $in = my $out = "$tar_dir/$tar_file";
                        $out .= '.tmp';

                        # FIXME I should really order parts manually!
                        system("cat $in/part* > $out && rm -Rf $in && mv $out $in") == 0 or die "can't join $in: $?";
                }

                if ($size > $max_size && ! -d "$tar_dir/$tar_file") {
                        print " split/$parts";
                        my $in = my $out = "$tar_dir/$tar_file";
                        $out .= '.tmp';
                        rename $in, $out || die "can't rename $in: $!";
                        mkdir $in || die "can't mkdir $in: $!";

                        my $suffix_len = length("$parts");
                        system("$bin->{'split'} -d -b $max_size -a $suffix_len $out $in/part") == 0 or die "can't split $out: $?";
                        unlink $out || die "can't unlink $out: $!";
                }

                $sth_inc_size->execute($size, $parts, $row->{'backup_id'});
                $sth_inc_deleted->execute(0, $row->{'backup_id'});

                printf(" %1.2f MB", ($size / 1024 / 1024));

        } else {
                $sth_inc_deleted->execute(1, $row->{'backup_id'});
                unlink "$tar_dir/$tar_file" || die "can't delete $tar_dir/$tar_file: $!\n";
                print " EMPTY";
        }
        print ", dur: ",fmt_time(time() - $t), "\n";

        $dbh->commit;

}

undef $sth;
$dbh->disconnect;
1	dpavlin	157	#!/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	dpavlin	201	use File::Which;
15	dpavlin	157
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	dpavlin	201	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	dpavlin	162	my $debug = 0;
29	dpavlin	157	$\|=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	dpavlin	158	backups.id as backup_id,
71	dpavlin	157	hosts.name as host,
72			shares.name as share,
73	dpavlin	192	backups.num as num,
74	dpavlin	196	inc_size,
75			parts
76	dpavlin	157	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	dpavlin	192	where not inc_deleted
81	dpavlin	157	order by backups.date
82
83			} );
84
85			$sth->execute();
86
87	dpavlin	196	my $sth_inc_size = $dbh->prepare(qq{ update backups set inc_size = ?, parts = ? where id = ? });
88	dpavlin	163	my $sth_inc_deleted = $dbh->prepare(qq{ update backups set inc_deleted = ? where id = ? });
89	dpavlin	158
90	dpavlin	157	%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	dpavlin	192
95	dpavlin	194	# this will return -1 if file doesn't exist
96			my $size = BackupPC::SearchLib::get_tgz_size_by_name($tar_file);
97	dpavlin	192
98	dpavlin	182	print curr_time, " ", $row->{'host'}, ":", $row->{'share'}, " #", $row->{'num'}, " -> $tar_file";
99	dpavlin	157
100			my $t = time();
101
102	dpavlin	194	# re-create archive?
103			if ($row->{'inc_size'} == -1 \|\| $size == -1 \|\| $row->{'inc_size'} != $size) {
104	dpavlin	212	my $cmd = qq{rm -Rf $tar_dir/$tar_file && $tarIncCreate -h "$row->{'host'}" -s "$row->{'share'}" -n $row->{'num'} \| $bin->{'gzip'} $Conf{GzipLevel} > ${tar_dir}/${tar_file}.tmp};
105	dpavlin	194	print STDERR "## $cmd\n" if ($debug);
106	dpavlin	157
107	dpavlin	194	system($cmd) == 0 or die "failed: $?";
108	dpavlin	212
109			rename "${tar_dir}/${$tar_file}.tmp", "$tar_dir/$tar_file" or die "can't rename $tar_dir/$tar_file: $!";
110
111	dpavlin	194	$size = (stat( "$tar_dir/$tar_file" ))[7];
112			}
113	dpavlin	157
114	dpavlin	194	if ($size > 45) {
115	dpavlin	157
116	dpavlin	194	my $max_size = $Conf{'MaxArchiveSize'} \|\| die "problem with MaxArchieSize parametar";
117	dpavlin	196	$max_size *= 1024; # convert to bytes
118	dpavlin	158
119	dpavlin	196	my $parts = int( ($size + $max_size - 1) / $max_size );
120
121			if (-d "$tar_dir/$tar_file" && $parts != $row->{'parts'}) {
122			print " join";
123
124			my $in = my $out = "$tar_dir/$tar_file";
125			$out .= '.tmp';
126
127			# FIXME I should really order parts manually!
128			system("cat $in/part* > $out && rm -Rf $in && mv $out $in") == 0 or die "can't join $in: $?";
129			}
130
131	dpavlin	194	if ($size > $max_size && ! -d "$tar_dir/$tar_file") {
132	dpavlin	196	print " split/$parts";
133	dpavlin	194	my $in = my $out = "$tar_dir/$tar_file";
134			$out .= '.tmp';
135			rename $in, $out \|\| die "can't rename $in: $!";
136			mkdir $in \|\| die "can't mkdir $in: $!";
137	dpavlin	196
138			my $suffix_len = length("$parts");
139	dpavlin	201	system("$bin->{'split'} -d -b $max_size -a $suffix_len $out $in/part") == 0 or die "can't split $out: $?";
140	dpavlin	194	unlink $out \|\| die "can't unlink $out: $!";
141			}
142
143	dpavlin	196	$sth_inc_size->execute($size, $parts, $row->{'backup_id'});
144	dpavlin	168	$sth_inc_deleted->execute(0, $row->{'backup_id'});
145	dpavlin	194
146			printf(" %1.2f MB", ($size / 1024 / 1024));
147
148	dpavlin	163	} else {
149			$sth_inc_deleted->execute(1, $row->{'backup_id'});
150			unlink "$tar_dir/$tar_file" \|\| die "can't delete $tar_dir/$tar_file: $!\n";
151			print " EMPTY";
152			}
153	dpavlin	194	print ", dur: ",fmt_time(time() - $t), "\n";
154	dpavlin	158
155			$dbh->commit;
156
157	dpavlin	157	}
158
159			undef $sth;
160			$dbh->disconnect;