/[BackupPC]/trunk/bin/BackupPC_updatedb
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 /trunk/bin/BackupPC_updatedb

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 97 by dpavlin, Tue Aug 30 09:55:55 2005 UTC revision 242 by dpavlin, Fri Nov 18 19:40:47 2005 UTC
# Line 11  use Getopt::Std; Line 11  use Getopt::Std;
11  use Time::HiRes qw/time/;  use Time::HiRes qw/time/;
12  use File::Pid;  use File::Pid;
13  use POSIX qw/strftime/;  use POSIX qw/strftime/;
14    use BackupPC::SearchLib;
15    
16  use constant BPC_FTYPE_DIR => 5;  use constant BPC_FTYPE_DIR => 5;
17  use constant EST_SYNC_EVERY => 10000;  use constant EST_CHUNK => 100000;
18    
19  my $debug = 0;  my $debug = 0;
20  $|=1;  $|=1;
# Line 41  my $beenThere = {}; Line 42  my $beenThere = {};
42    
43  my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";  my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
44  my $user = $Conf{SearchUser} || '';  my $user = $Conf{SearchUser} || '';
 my $index_path = $Conf{HyperEstraierIndex};  
 $index_path = $TopDir . '/' . $index_path;  
 $index_path =~ s#//#/#g;  
45    
46    my $use_hest = $Conf{HyperEstraierIndex};
47    my ($index_path, $index_node_url) = BackupPC::SearchLib::getHyperEstraier_url($use_hest);
48    
49  my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });  my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });
50    
51  my %opt;  my %opt;
52    
53  if ( !getopts("cdm:v:i", \%opt ) ) {  if ( !getopts("cdm:v:ij", \%opt ) ) {
54          print STDERR <<EOF;          print STDERR <<EOF;
55  usage: $0 [-c|-d] [-m num] [-v|-v level] [-i]  usage: $0 [-c|-d] [-m num] [-v|-v level] [-i]
56    
# Line 59  Options: Line 59  Options:
59          -d      delete database before import          -d      delete database before import
60          -m num  import just num increments for one host          -m num  import just num increments for one host
61          -v num  set verbosity (debug) level (default $debug)          -v num  set verbosity (debug) level (default $debug)
62          -i      update HyperEstraier full text index          -i      update Hyper Estraier full text index
63            -j      update full text, don't check existing files
64    
65    Option -j is variation on -i. It will allow faster initial creation
66    of full-text index from existing database.
67    
68  EOF  EOF
69          exit 1;          exit 1;
70  }  }
# Line 85  sub curr_time { Line 90  sub curr_time {
90  }  }
91    
92  my $hest_db;  my $hest_db;
93    my $hest_node;
94    
95  sub signal {  sub signal {
96          my($sig) = @_;          my($sig) = @_;
# Line 103  sub hest_update { Line 109  sub hest_update {
109    
110          my ($host_id, $share_id, $num) = @_;          my ($host_id, $share_id, $num) = @_;
111    
112          print curr_time," updating HyperEstraier: select files";          my $skip_check = $opt{j} && print STDERR "Skipping check for existing files -- this should be used only with initital import\n";
   
         my $t = time();  
   
         my $where = '';  
         if ($host_id && $share_id && $num) {  
                 $where = qq{  
                 WHERE  
                         hosts.id = ? AND  
                         shares.id = ? AND  
                         files.backupnum = ?  
                 };  
         }  
   
         my $sth = $dbh->prepare(qq{  
                 SELECT  
                         files.id                        AS fid,  
                         hosts.name                      AS hname,  
                         shares.name                     AS sname,  
                         -- shares.share                 AS sharename,  
                         files.backupnum                 AS backupnum,  
                         -- files.name                   AS filename,  
                         files.path                      AS filepath,  
                         files.date                      AS date,  
                         files.type                      AS type,  
                         files.size                      AS size,  
                         files.shareid                   AS shareid,  
                         backups.date                    AS backup_date  
                 FROM files  
                         INNER JOIN shares       ON files.shareID=shares.ID  
                         INNER JOIN hosts        ON hosts.ID = shares.hostID  
                         INNER JOIN backups      ON backups.num = files.backupNum and backups.hostID = hosts.ID AND backups.shareID = shares.ID  
                 $where  
         });  
113    
114          $sth->execute(@_);          unless ($use_hest) {
115          my $results = $sth->rows;                  print STDERR "HyperEstraier support not enabled in configuration\n";
   
         if ($results == 0) {  
                 print " - no files, skipping\n";  
116                  return;                  return;
117          }          }
118    
119          my $dot = int($results / 15) || 1;          print curr_time," updating HyperEstraier:";
   
         print " $results ($dot/#)";  
120    
121          sub fmt_date {          my $t = time();
                 my $t = shift || return;  
                 my $iso = BackupPC::Lib::timeStamp($t);  
                 $iso =~ s/\s/T/;  
                 return $iso;  
         }  
   
         my $max = int($results / $dot);  
122    
123          print ", opening index $index_path...";          my $offset = 0;
124          use HyperEstraier;          my $added = 0;
         my $db = HyperEstraier::Database->new();  
125    
126  #       unless ($hest_db) {          print " opening index $use_hest";
127  #               print " open reader";          if ($index_path) {
128  #               $hest_db = HyperEstraier::Database->new();                  $hest_db = HyperEstraier::Database->new();
129  #                  $hest_db->open($TopDir . $index_path, $HyperEstraier::Database::DBWRITER | $HyperEstraier::Database::DBCREAT);
130  #       }                  print " directly";
131            } elsif ($index_node_url) {
132                    $hest_node ||= HyperEstraier::Node->new($index_node_url);
133                    $hest_node->set_auth('admin', 'admin');
134                    print " via node URL";
135            } else {
136                    die "don't know how to use HyperEstraier Index $use_hest";
137            }
138            print " increment is " . EST_CHUNK . " files:";
139    
140            my $results = 0;
141    
142          $db->open($index_path, $HyperEstraier::Database::DBWRITER | $HyperEstraier::Database::DBCREAT);          do {
143    
144          my $added = 0;                  my $where = '';
145                    my @data;
146                    if (defined($host_id) && defined($share_id) && defined($num)) {
147                            $where = qq{
148                            WHERE
149                                    hosts.id = ? AND
150                                    shares.id = ? AND
151                                    files.backupnum = ?
152                            };
153                            @data = ( $host_id, $share_id, $num );
154                    }
155    
156          while (my $row = $sth->fetchrow_hashref()) {                  my $limit = sprintf('LIMIT '.EST_CHUNK.' OFFSET %d', $offset);
157    
158                  my $fid = $row->{'fid'} || die "no fid?";                  my $sth = $dbh->prepare(qq{
159                  my $uri = 'file:///' . $fid;                          SELECT
160                                    files.id                        AS fid,
161                                    hosts.name                      AS hname,
162                                    shares.name                     AS sname,
163                                    -- shares.share                 AS sharename,
164                                    files.backupnum                 AS backupnum,
165                                    -- files.name                   AS filename,
166                                    files.path                      AS filepath,
167                                    files.date                      AS date,
168                                    files.type                      AS type,
169                                    files.size                      AS size,
170                                    files.shareid                   AS shareid,
171                                    backups.date                    AS backup_date
172                            FROM files
173                                    INNER JOIN shares       ON files.shareID=shares.ID
174                                    INNER JOIN hosts        ON hosts.ID = shares.hostID
175                                    INNER JOIN backups      ON backups.num = files.backupNum and backups.hostID = hosts.ID AND backups.shareID = shares.ID
176                            $where
177                            $limit
178                    });
179    
180                    $sth->execute(@data);
181                    $results = $sth->rows;
182    
183                    if ($results == 0) {
184                            print " - no new files\n";
185                            last;
186                    }
187    
188                  my $id = $db->uri_to_id($uri);                  sub fmt_date {
189                  next unless ($id == -1);                          my $t = shift || return;
190                            my $iso = BackupPC::Lib::timeStamp($t);
191                            $iso =~ s/\s/T/;
192                            return $iso;
193                    }
194    
195                  # create a document object                  while (my $row = $sth->fetchrow_hashref()) {
                 my $doc = HyperEstraier::Document->new;  
196    
197                  # add attributes to the document object                          my $fid = $row->{'fid'} || die "no fid?";
198                  $doc->add_attr('@uri', $uri);                          my $uri = 'file:///' . $fid;
199    
200                  foreach my $c (@{ $sth->{NAME} }) {                          unless ($skip_check) {
201                          $doc->add_attr($c, $row->{$c}) if ($row->{$c});                                  my $id = ($hest_db || $hest_node)->uri_to_id($uri);
202                                    next unless ($id == -1);
203                            }
204    
205                            # create a document object
206                            my $doc = HyperEstraier::Document->new;
207    
208                            # add attributes to the document object
209                            $doc->add_attr('@uri', $uri);
210    
211                            foreach my $c (@{ $sth->{NAME} }) {
212                                    $doc->add_attr($c, $row->{$c}) if (defined($row->{$c}));
213                            }
214    
215                            #$doc->add_attr('@cdate', fmt_date($row->{'date'}));
216    
217                            # add the body text to the document object
218                            my $path = $row->{'filepath'};
219                            $doc->add_text($path);
220                            $path =~ s/(.)/$1 /g;
221                            $doc->add_hidden_text($path);
222    
223                            print STDERR $doc->dump_draft,"\n" if ($debug > 1);
224    
225                            # register the document object to the database
226                            if ($hest_db) {
227                                    $hest_db->put_doc($doc, $HyperEstraier::Database::PDCLEAN);
228                            } elsif ($hest_node) {
229                                    $hest_node->put_doc($doc);
230                            } else {
231                                    die "not supported";
232                            }
233                            $added++;
234                  }                  }
235    
236                  #$doc->add_attr('@cdate', fmt_date($row->{'date'}));                  print " $added";
237                    $hest_db->sync() if ($index_path);
238    
239                  # add the body text to the document object                  $offset += EST_CHUNK;
                 my $path = $row->{'filepath'};  
                 $doc->add_text($path);  
                 $path =~ s/(.)/$1 /g;  
                 $doc->add_hidden_text($path);  
   
                 print STDERR $doc->dump_draft,"\n" if ($debug > 1);  
   
                 # register the document object to the database  
                 $db->put_doc($doc, $HyperEstraier::Database::PDCLEAN);  
   
                 $added++;  
                 if ($added % $dot == 0) {  
                         print "$max ";  
                         $max--;  
                 }  
240    
241                  if ($added % EST_SYNC_EVERY == 0) {          } while ($results == EST_CHUNK);
                         print "sync ";  
                         $db->sync();  
                 }  
242    
243            if ($index_path) {
244                    print ", close";
245                    $hest_db->close();
246          }          }
247    
         print "sync $added new files";  
         $db->sync();  
         print ", close";  
         $db->close();  
   
248          my $dur = (time() - $t) || 1;          my $dur = (time() - $t) || 1;
249          printf(" [%.2f/s new %.2f/s dur: %s]\n",          printf(" [%.2f/s dur: %s]\n",
                 ( $results / $dur ),  
250                  ( $added / $dur ),                  ( $added / $dur ),
251                  fmt_time($dur)                  fmt_time($dur)
252          );          );
# Line 235  sub hest_update { Line 256  sub hest_update {
256    
257    
258  ## update index ##  ## update index ##
259  if (($opt{i} || ($index_path && ! -e $index_path)) && !$opt{c}) {  if (($opt{i} || $opt{j} || ($index_path && ! -e $TopDir . $index_path)) && !$opt{c}) {
260          # update all          # update all
261          print "force update of HyperEstraier index ";          print "force update of HyperEstraier index ";
262          print "importing existing data" unless (-e $index_path);          print "importing existing data" unless (-e $TopDir . $index_path);
263          print "by -i flag" if ($opt{i});          print "by -i flag" if ($opt{i});
264            print "by -j flag" if ($opt{j});
265          print "\n";          print "\n";
266          hest_update();          hest_update();
267  }  }
# Line 248  if (($opt{i} || ($index_path && ! -e $in Line 270  if (($opt{i} || ($index_path && ! -e $in
270  if ($opt{c}) {  if ($opt{c}) {
271          sub do_index {          sub do_index {
272                  my $index = shift || return;                  my $index = shift || return;
273                  my ($table,$col,$unique) = split(/_/, $index);                  my ($table,$col,$unique) = split(/:/, $index);
274                  $unique ||= '';                  $unique ||= '';
275                  $index =~ s/,/_/g;                  $index =~ s/\W+/_/g;
276                    print "$index on $table($col)" . ( $unique ? "u" : "" ) . " ";
277                  $dbh->do(qq{ create $unique index $index on $table($col) });                  $dbh->do(qq{ create $unique index $index on $table($col) });
278          }          }
279    
280          print "creating tables...\n";          print "creating tables...\n";
281          
282          $dbh->do(qq{          $dbh->do( qq{
283                  create table hosts (                  create table hosts (
284                          ID      SERIAL          PRIMARY KEY,                          ID      SERIAL          PRIMARY KEY,
285                          name    VARCHAR(30)     NOT NULL,                          name    VARCHAR(30)     NOT NULL,
286                          IP      VARCHAR(15)                          IP      VARCHAR(15)
287                  );                              );            
288          });  
                 
         $dbh->do(qq{  
289                  create table shares (                  create table shares (
290                          ID      SERIAL          PRIMARY KEY,                          ID      SERIAL          PRIMARY KEY,
291                          hostID  INTEGER         NOT NULL references hosts(id),                          hostID  INTEGER         NOT NULL references hosts(id),
292                          name    VARCHAR(30)     NOT NULL,                          name    VARCHAR(30)     NOT NULL,
293                          share   VARCHAR(200)    NOT NULL,                          share   VARCHAR(200)    NOT NULL
                         localpath VARCHAR(200)        
294                  );                              );            
295          });  
296                            create table dvds (
297          $dbh->do(qq{                          ID      SERIAL          PRIMARY KEY,
298                            num     INTEGER         NOT NULL,
299                            name    VARCHAR(255)    NOT NULL,
300                            mjesto  VARCHAR(255)
301                    );
302    
303                  create table backups (                  create table backups (
304                            id      serial,
305                          hostID  INTEGER         NOT NULL references hosts(id),                          hostID  INTEGER         NOT NULL references hosts(id),
306                          num     INTEGER         NOT NULL,                          num     INTEGER         NOT NULL,
307                          date    integer         NOT NULL,                          date    integer         NOT NULL,
308                          type    CHAR(4)         not null,                          type    CHAR(4)         not null,
309                          shareID integer         not null references shares(id),                          shareID integer         not null references shares(id),
310                          size    integer         not null,                          size    bigint          not null,
311                          PRIMARY KEY(hostID, num, shareID)                          inc_size bigint         not null default -1,
312                            inc_deleted boolean     default false,
313                            parts   integer         not null default 1,
314                            PRIMARY KEY(id)
315                  );                              );            
         });  
316    
317          #do_index('backups_hostid,num_unique');                  create table files (
318                            ID              SERIAL,
319                            shareID         INTEGER NOT NULL references shares(id),
320                            backupNum       INTEGER NOT NULL,
321                            name            VARCHAR(255) NOT NULL,
322                            path            VARCHAR(255) NOT NULL,
323                            date            integer NOT NULL,
324                            type            INTEGER NOT NULL,
325                            size            bigint  NOT NULL,
326                            primary key(id)
327                    );
328    
329                    create table archive (
330                            id              serial,
331                            dvd_nr          int not null,
332                            total_size      bigint default -1,
333                            note            text,
334                            username        varchar(20) not null,
335                            date            timestamp default now(),
336                            primary key(id)
337                    );      
338    
339                    create table archive_backup (
340                            archive_id      int not null references archive(id) on delete cascade,
341                            backup_id       int not null references backups(id),
342                            primary key(archive_id, backup_id)
343                    );
344    
345          $dbh->do(qq{                  create table archive_burned (
346                  create table dvds (                          archive_id      int references archive(id),
347                          ID      SERIAL          PRIMARY KEY,                          date            timestamp default now(),
348                          num     INTEGER         NOT NULL,                          part            int not null default 1,
349                          name    VARCHAR(255)    NOT NULL,                          copy            int not null default 1,
350                          mjesto  VARCHAR(255)                          iso_size bigint default -1
351                  );                  );
         });  
352    
353          $dbh->do(qq{                      create table backup_parts (
354                  create table files (                          id serial,
355                          ID      SERIAL          PRIMARY KEY,                            backup_id int references backups(id),
356                          shareID INTEGER         NOT NULL references shares(id),                          part_nr int not null check (part_nr > 0),
357                          backupNum  INTEGER      NOT NULL,                          tar_size bigint not null check (tar_size > 0),
358                          name       VARCHAR(255) NOT NULL,                          size bigint not null check (size > 0),
359                          path       VARCHAR(255) NOT NULL,                          md5 text not null,
360                          date       integer      NOT NULL,                          items int not null check (items > 0),
361                          type       INTEGER      NOT NULL,                          date timestamp default now(),
362                          size       INTEGER      NOT NULL,                          primary key(id)
                         dvdid      INTEGER      references dvds(id)      
363                  );                  );
364          });          });
365    
366          print "creating indexes:";          print "creating indexes: ";
367    
368          foreach my $index (qw(          foreach my $index (qw(
369                  hosts_name                  hosts:name
370                  backups_hostID                  backups:hostID
371                  backups_num                  backups:num
372                  shares_hostID                  backups:shareID
373                  shares_name                  shares:hostID
374                  files_shareID                  shares:name
375                  files_path                  files:shareID
376                  files_name                  files:path
377                  files_date                  files:name
378                  files_size                  files:date
379                    files:size
380                    archive:dvd_nr
381                    archive_burned:archive_id
382                    backup_parts:backup_id,part_nr
383          )) {          )) {
                 print " $index";  
384                  do_index($index);                  do_index($index);
385          }          }
386    
387            print " creating sequence: ";
388            foreach my $seq (qw/dvd_nr/) {
389                    print "$seq ";
390                    $dbh->do( qq{ CREATE SEQUENCE $seq } );
391            }
392    
393    
394          print "...\n";          print "...\n";
395    
396          $dbh->commit;          $dbh->commit;
# Line 371  WHERE hostID=? AND num=? AND shareid=? Line 434  WHERE hostID=? AND num=? AND shareid=?
434    
435  $sth->{insert_backups} = $dbh->prepare(qq{  $sth->{insert_backups} = $dbh->prepare(qq{
436  INSERT INTO backups (hostID, num, date, type, shareid, size)  INSERT INTO backups (hostID, num, date, type, shareid, size)
437  VALUES (?,?,?,?,?,?)  VALUES (?,?,?,?,?,-1)
438    });
439    
440    $sth->{update_backups_size} = $dbh->prepare(qq{
441    UPDATE backups SET size = ?
442    WHERE hostID = ? and num = ? and date = ? and type =? and shareid = ?
443  });  });
444    
445  $sth->{insert_files} = $dbh->prepare(qq{  $sth->{insert_files} = $dbh->prepare(qq{
# Line 437  foreach my $host_key (keys %{$hosts}) { Line 505  foreach my $host_key (keys %{$hosts}) {
505                          # dump some log                          # dump some log
506                          print curr_time," ", $share;                          print curr_time," ", $share;
507    
                         my ($f, $nf, $d, $nd, $size) = recurseDir($bpc, $hostname, $files, $backupNum, $share, "", $shareID);  
   
508                          $sth->{insert_backups}->execute(                          $sth->{insert_backups}->execute(
509                                  $hostID,                                  $hostID,
510                                  $backupNum,                                  $backupNum,
511                                  $backup->{'endTime'},                                  $backup->{'endTime'},
512                                  $backup->{'type'},                                  substr($backup->{'type'},0,4),
513                                  $shareID,                                  $shareID,
514                            );
515    
516                            my ($f, $nf, $d, $nd, $size) = recurseDir($bpc, $hostname, $files, $backupNum, $share, "", $shareID);
517    
518                            $sth->{update_backups_size}->execute(
519                                  $size,                                  $size,
520                                    $hostID,
521                                    $backupNum,
522                                    $backup->{'endTime'},
523                                    substr($backup->{'type'},0,4),
524                                    $shareID,
525                          );                          );
526    
527                          print " commit";                          print " commit";
# Line 459  foreach my $host_key (keys %{$hosts}) { Line 535  foreach my $host_key (keys %{$hosts}) {
535                                  fmt_time($dur)                                  fmt_time($dur)
536                          );                          );
537    
538                          hest_update($hostID, $shareID, $backupNum);                          hest_update($hostID, $shareID, $backupNum) if ($nf + $nd > 0);
539                  }                  }
540    
541          }          }
# Line 488  sub getShareID() { Line 564  sub getShareID() {
564    
565          $sth->{insert_share} ||= $dbh->prepare(qq{          $sth->{insert_share} ||= $dbh->prepare(qq{
566                  INSERT INTO shares                  INSERT INTO shares
567                          (hostID,name,share,localpath)                          (hostID,name,share)
568                  VALUES (?,?,?,?)                  VALUES (?,?,?)
569          });          });
570    
571          my $drop_down = $hostname . '/' . $share;          my $drop_down = $hostname . '/' . $share;
572          $drop_down =~ s#//+#/#g;          $drop_down =~ s#//+#/#g;
573    
574          $sth->{insert_share}->execute($hostID,$share, $drop_down ,undef);          $sth->{insert_share}->execute($hostID,$share, $drop_down);
575          return $dbh->last_insert_id(undef,undef,'shares',undef);          return $dbh->last_insert_id(undef,undef,'shares',undef);
576  }  }
577    
# Line 567  sub recurseDir($$$$$$$$) { Line 643  sub recurseDir($$$$$$$$) {
643                                  $filesInBackup->{$path_key}->{'size'}                                  $filesInBackup->{$path_key}->{'size'}
644                          ));                          ));
645    
646                            # daylight saving time change offset for 1h
647                            my $dst_offset = 60 * 60;
648    
649                            my $key_dst_prev = join(" ", (
650                                    $shareID,
651                                    $dir,
652                                    $path_key,
653                                    $filesInBackup->{$path_key}->{'mtime'} - $dst_offset,
654                                    $filesInBackup->{$path_key}->{'size'}
655                            ));
656    
657                            my $key_dst_next = join(" ", (
658                                    $shareID,
659                                    $dir,
660                                    $path_key,
661                                    $filesInBackup->{$path_key}->{'mtime'} + $dst_offset,
662                                    $filesInBackup->{$path_key}->{'size'}
663                            ));
664    
665                          my $found;                          my $found;
666                          if (! defined($beenThere->{$key}) && ! ($found = found_in_db($key, @data)) ) {                          if (
667                                    ! defined($beenThere->{$key}) &&
668                                    ! defined($beenThere->{$key_dst_prev}) &&
669                                    ! defined($beenThere->{$key_dst_next}) &&
670                                    ! ($found = found_in_db($key, @data))
671                            ) {
672                                  print STDERR "# key: $key [", $beenThere->{$key},"]" if ($debug >= 2);                                  print STDERR "# key: $key [", $beenThere->{$key},"]" if ($debug >= 2);
673    
674                                  if ($filesInBackup->{$path_key}->{'type'} == BPC_FTYPE_DIR) {                                  if ($filesInBackup->{$path_key}->{'type'} == BPC_FTYPE_DIR) {
Legend:
Removed from v.97  
changed lines
  Added in v.242
  ViewVC Help
Powered by ViewVC 1.1.26