r8502@llin: dpavlin | 2005-10-13 19:05:29 +0200
[BackupPC.git] / 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
15 my $path = abs_path($0);
16 $path =~ s#/[^/]+$#/#;
17 my $tarIncCreate = $path .= 'BackupPC_tarIncCreate';
18
19 die "can't find $tarIncCreate: $!\n" unless (-x $tarIncCreate);
20
21 my $debug = 0;
22 $|=1;
23
24 my $start_t = time();
25
26 my $t_fmt = '%Y-%m-%d %H:%M:%S';
27
28 my $hosts;
29 my $bpc = BackupPC::Lib->new || die;
30 my %Conf = $bpc->Conf();
31 my $TopDir = $bpc->TopDir();
32 my $beenThere = {};
33
34 my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
35 my $user = $Conf{SearchUser} || '';
36
37 my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });
38
39 my $tar_dir = $Conf{InstallDir}.'/'.$Conf{GzipTempDir};
40
41 die "problem with $tar_dir, check GzipTempDir in configuration\n" unless (-d $tar_dir && -w $tar_dir);
42
43 #---- subs ----
44
45 sub fmt_time {
46         my $t = shift || return;
47         my $out = "";
48         my ($ss,$mm,$hh) = gmtime($t);
49         $out .= "${hh}h" if ($hh);
50         $out .= sprintf("%02d:%02d", $mm,$ss);
51         return $out;
52 }
53
54 sub curr_time {
55         return strftime($t_fmt,localtime());
56 }
57
58 #----- main
59
60 my $sth = $dbh->prepare( qq{
61         
62 select
63         backups.id as backup_id,
64         hosts.name as host,
65         shares.name as share,
66         backups.num as num,
67         inc_size
68 from backups
69         join shares on backups.hostid = shares.hostid
70                 and shares.id = backups.shareid
71         join hosts on shares.hostid = hosts.id
72 where not inc_deleted
73 order by backups.date
74
75 } );
76
77 $sth->execute();
78
79 my $sth_inc_size = $dbh->prepare(qq{ update backups set inc_size = ? where id = ? });
80 my $sth_inc_deleted = $dbh->prepare(qq{ update backups set inc_deleted = ? where id = ? });
81
82 %BackupPC::SearchLib::Conf = %Conf;
83
84 while (my $row = $sth->fetchrow_hashref) {
85         my $tar_file = BackupPC::SearchLib::getGzipName($row->{'host'}, $row->{'share'}, $row->{'num'});
86
87         # this will return -1 if file doesn't exist
88         my $size = BackupPC::SearchLib::get_tgz_size_by_name($tar_file);
89
90         print curr_time, " ", $row->{'host'}, ":", $row->{'share'}, " #", $row->{'num'}, " -> $tar_file";
91
92         my $t = time();
93
94         # re-create archive?
95         if ($row->{'inc_size'} == -1 || $size == -1 || $row->{'inc_size'} != $size) {
96                 my $cmd = qq{$tarIncCreate -h "$row->{'host'}" -s "$row->{'share'}" -n $row->{'num'} | gzip -9 > $tar_dir/$tar_file};
97                 print STDERR "## $cmd\n" if ($debug);
98
99                 system($cmd) == 0 or die "failed: $?";
100         
101                 $size = (stat( "$tar_dir/$tar_file" ))[7];
102         }
103
104         if ($size > 45) {
105
106                 my $max_size = $Conf{'MaxArchiveSize'} || die "problem with MaxArchieSize parametar";
107
108                 if ($size > $max_size && ! -d "$tar_dir/$tar_file") {
109                         print " split";
110                         my $in = my $out = "$tar_dir/$tar_file";
111                         $out .= '.tmp';
112                         rename $in, $out || die "can't rename $in: $!";
113                         mkdir $in || die "can't mkdir $in: $!";
114                         system("split -d -b $max_size $out $in/part") == 0 or die "can't split $out: $!";
115                         unlink $out || die "can't unlink $out: $!";
116                 }
117
118                 $sth_inc_size->execute($size, $row->{'backup_id'});
119                 $sth_inc_deleted->execute(0, $row->{'backup_id'});
120
121                 printf(" %1.2f MB", ($size / 1024 / 1024));
122
123         } else {
124                 $sth_inc_deleted->execute(1, $row->{'backup_id'});
125                 unlink "$tar_dir/$tar_file" || die "can't delete $tar_dir/$tar_file: $!\n";
126                 print " EMPTY";
127         }
128         print ", dur: ",fmt_time(time() - $t), "\n";
129
130         $dbh->commit;
131
132 }
133
134 undef $sth;
135 $dbh->disconnect;