/[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 246 by dpavlin, Fri Dec 9 12:49:55 2005 UTC revision 303 by dpavlin, Sat Jan 28 16:45:46 2006 UTC
# Line 12  use Time::HiRes qw/time/; Line 12  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;  use BackupPC::SearchLib;
15    use Cwd qw/abs_path/;
16    
17  use constant BPC_FTYPE_DIR => 5;  use constant BPC_FTYPE_DIR => 5;
18  use constant EST_CHUNK => 100000;  use constant EST_CHUNK => 100000;
# Line 24  $|=1; Line 25  $|=1;
25    
26  my $start_t = time();  my $start_t = time();
27    
28  my $pidfile = new File::Pid;  my $pid_path = abs_path($0);
29    $pid_path =~ s/\W+/_/g;
30    
31    my $pidfile = new File::Pid({
32            file => "/tmp/$pid_path",
33    });
34    
35  if (my $pid = $pidfile->running ) {  if (my $pid = $pidfile->running ) {
36          die "$0 already running: $pid\n";          die "$0 already running: $pid\n";
# Line 32  if (my $pid = $pidfile->running ) { Line 38  if (my $pid = $pidfile->running ) {
38          $pidfile->remove;          $pidfile->remove;
39          $pidfile = new File::Pid;          $pidfile = new File::Pid;
40  }  }
 $pidfile->write;  
41  print STDERR "$0 using pid ",$pidfile->pid," file ",$pidfile->file,"\n";  print STDERR "$0 using pid ",$pidfile->pid," file ",$pidfile->file,"\n";
42    $pidfile->write;
43    
44  my $t_fmt = '%Y-%m-%d %H:%M:%S';  my $t_fmt = '%Y-%m-%d %H:%M:%S';
45    
# Line 46  my $beenThere = {}; Line 52  my $beenThere = {};
52  my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";  my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
53  my $user = $Conf{SearchUser} || '';  my $user = $Conf{SearchUser} || '';
54    
55  my $use_hest = $Conf{HyperEstraierIndex};  my $index_node_url = $Conf{HyperEstraierIndex};
 my ($index_path, $index_node_url) = BackupPC::SearchLib::getHyperEstraier_url($use_hest);  
56    
57  my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });  my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 0 });
58    
# Line 81  if ($opt{v}) { Line 86  if ($opt{v}) {
86          $debug = $opt{v};          $debug = $opt{v};
87  } elsif ($opt{f}) {  } elsif ($opt{f}) {
88          print "WARNING: disabling full-text index update. You need to re-run $0 -j !\n";          print "WARNING: disabling full-text index update. You need to re-run $0 -j !\n";
89          ($use_hest, $index_path, $index_node_url) = (undef, undef, undef);          $index_node_url = undef;
90  }  }
91    
92  #---- subs ----  #---- subs ----
# Line 121  sub hest_update { Line 126  sub hest_update {
126    
127          my $skip_check = $opt{j} && print STDERR "Skipping check for existing files -- this should be used only with initital import\n";          my $skip_check = $opt{j} && print STDERR "Skipping check for existing files -- this should be used only with initital import\n";
128    
129          unless ($use_hest) {          unless (defined($index_node_url)) {
130                  print STDERR "HyperEstraier support not enabled in configuration\n";                  print STDERR "HyperEstraier support not enabled in configuration\n";
131                    $index_node_url = 0;
132                  return;                  return;
133          }          }
134    
135          print curr_time," updating HyperEstraier:";          print curr_time," updating Hyper Estraier:";
136    
137          my $t = time();          my $t = time();
138    
139          my $offset = 0;          my $offset = 0;
140          my $added = 0;          my $added = 0;
141    
142          print " opening index $use_hest";          print " opening index $index_node_url";
143          if ($index_path) {          if ($index_node_url) {
144                  $hest_db = HyperEstraier::Database->new();                  $hest_node ||= Search::Estraier::Node->new($index_node_url);
                 $hest_db->open($TopDir . $index_path, $HyperEstraier::Database::DBWRITER | $HyperEstraier::Database::DBCREAT);  
                 print " directly";  
         } elsif ($index_node_url) {  
                 $hest_node ||= HyperEstraier::Node->new($index_node_url);  
145                  $hest_node->set_auth('admin', 'admin');                  $hest_node->set_auth('admin', 'admin');
146                  print " via node URL";                  print " via node URL";
147          } else {          } else {
148                  die "don't know how to use HyperEstraier Index $use_hest";                  die "don't know how to use Hyper Estraier Index $index_node_url";
149          }          }
         print " increment is " . EST_CHUNK . " files:";  
150    
151          my $results = 0;          my $results = 0;
152    
# Line 193  sub hest_update { Line 194  sub hest_update {
194                  if ($results == 0) {                  if ($results == 0) {
195                          print " - no new files\n";                          print " - no new files\n";
196                          last;                          last;
197                    } else {
198                            print " - $results files: ";
199                  }                  }
200    
201                  sub fmt_date {                  sub fmt_date {
# Line 213  sub hest_update { Line 216  sub hest_update {
216                          }                          }
217    
218                          # create a document object                          # create a document object
219                          my $doc = HyperEstraier::Document->new;                          my $doc = Search::Estraier::Document->new;
220    
221                          # add attributes to the document object                          # add attributes to the document object
222                          $doc->add_attr('@uri', $uri);                          $doc->add_attr('@uri', $uri);
# Line 233  sub hest_update { Line 236  sub hest_update {
236                          print STDERR $doc->dump_draft,"\n" if ($debug > 1);                          print STDERR $doc->dump_draft,"\n" if ($debug > 1);
237    
238                          # register the document object to the database                          # register the document object to the database
239                          if ($hest_db) {                          if ($hest_node) {
                                 $hest_db->put_doc($doc, $HyperEstraier::Database::PDCLEAN);  
                         } elsif ($hest_node) {  
240                                  $hest_node->put_doc($doc);                                  $hest_node->put_doc($doc);
241                          } else {                          } else {
242                                  die "not supported";                                  die "not supported";
# Line 244  sub hest_update { Line 245  sub hest_update {
245                  }                  }
246    
247                  print " $added";                  print " $added";
                 $hest_db->sync() if ($index_path);  
248    
249                  $offset += EST_CHUNK;                  $offset += EST_CHUNK;
250    
251          } while ($results == EST_CHUNK);          } while ($results == EST_CHUNK);
252    
         if ($index_path) {  
                 print ", close";  
                 $hest_db->close();  
         }  
   
253          my $dur = (time() - $t) || 1;          my $dur = (time() - $t) || 1;
254          printf(" [%.2f/s dur: %s]\n",          printf(" [%.2f/s dur: %s]\n",
255                  ( $added / $dur ),                  ( $added / $dur ),
# Line 266  sub hest_update { Line 261  sub hest_update {
261    
262    
263  ## update index ##  ## update index ##
264  if (($opt{i} || $opt{j} || ($index_path && ! -e $TopDir . $index_path)) && !$opt{c}) {  if ( ( $opt{i} || $opt{j} ) && !$opt{c} ) {
265          # update all          # update all
266          print "force update of HyperEstraier index ";          print "force update of Hyper Estraier index ";
         print "importing existing data" unless (-e $TopDir . $index_path);  
267          print "by -i flag" if ($opt{i});          print "by -i flag" if ($opt{i});
268          print "by -j flag" if ($opt{j});          print "by -j flag" if ($opt{j});
269          print "\n";          print "\n";
# Line 458  INSERT INTO files Line 452  INSERT INTO files
452          VALUES (?,?,?,?,?,?,?)          VALUES (?,?,?,?,?,?,?)
453  });  });
454    
455  foreach my $host_key (keys %{$hosts}) {  my @hosts = keys %{$hosts};
456    my $host_nr = 0;
457    
458    foreach my $host_key (@hosts) {
459    
460          my $hostname = $hosts->{$host_key}->{'host'} || die "can't find host for $host_key";          my $hostname = $hosts->{$host_key}->{'host'} || die "can't find host for $host_key";
461    
# Line 473  foreach my $host_key (keys %{$hosts}) { Line 470  foreach my $host_key (keys %{$hosts}) {
470                  $hostID = $dbh->last_insert_id(undef,undef,'hosts',undef);                  $hostID = $dbh->last_insert_id(undef,undef,'hosts',undef);
471          }          }
472    
473          print "host ".$hosts->{$host_key}->{'host'}.": ";          $host_nr++;
474            print "host ", $hosts->{$host_key}->{'host'}, " [",
475                    $host_nr, "/", ($#hosts + 1), "]: ";
476    
477          # get backups for a host          # get backups for a host
478          my @backups = $bpc->BackupInfoRead($hostname);          my @backups = $bpc->BackupInfoRead($hostname);
# Line 525  foreach my $host_key (keys %{$hosts}) { Line 524  foreach my $host_key (keys %{$hosts}) {
524    
525                          my ($f, $nf, $d, $nd, $size) = recurseDir($bpc, $hostname, $files, $backupNum, $share, "", $shareID);                          my ($f, $nf, $d, $nd, $size) = recurseDir($bpc, $hostname, $files, $backupNum, $share, "", $shareID);
526    
527                          $sth->{update_backups_size}->execute(                          eval {
528                                  $size,                                  $sth->{update_backups_size}->execute(
529                                  $hostID,                                          $size,
530                                  $backupNum,                                          $hostID,
531                                  $backup->{'endTime'},                                          $backupNum,
532                                  substr($backup->{'type'},0,4),                                          $backup->{'endTime'},
533                                  $shareID,                                          substr($backup->{'type'},0,4),
534                          );                                          $shareID,
535                                    );
536                          print " commit";                                  print " commit";
537                          $dbh->commit();                                  $dbh->commit();
538                            };
539                            if ($@) {
540                                    print " rollback";
541                                    $dbh->rollback();
542                            }
543    
544                          my $dur = (time() - $t) || 1;                          my $dur = (time() - $t) || 1;
545                          printf(" %d/%d files %d/%d dirs %0.2f MB [%.2f/s dur: %s]\n",                          printf(" %d/%d files %d/%d dirs %0.2f MB [%.2f/s dur: %s]\n",

Legend:
Removed from v.246  
changed lines
  Added in v.303

  ViewVC Help
Powered by ViewVC 1.1.26