if --only filter is used, we will parse just normalization files for this database.
[webpac2] / run.pl
diff --git a/run.pl b/run.pl
index 036b1bb..576f4ca 100755 (executable)
--- a/run.pl
+++ b/run.pl
@@ -9,7 +9,7 @@ use lib './lib';
 use WebPAC::Common 0.02;
 use WebPAC::Parser 0.08;
 use WebPAC::Input 0.16;
-use WebPAC::Store 0.14;
+use WebPAC::Store 0.15;
 use WebPAC::Normalize 0.22;
 use WebPAC::Output::TT;
 use WebPAC::Validate 0.11;
@@ -133,7 +133,7 @@ GetOptions(
        "clean" => \$clean,
        "one=s" => \$only_filter,
        "only=s" => \$only_filter,
-       "config" => \$config_path,
+       "config=s" => \$config_path,
        "debug+" => \$debug,
        "stats" => \$stats,
        "validate=s" => \$validate_path,
@@ -255,10 +255,7 @@ foreach my $database ( sort keys %{ $config->databases } ) {
 
                } elsif ($use_indexer eq 'kinosearch') {
 
-                       # open KinoSearch
-                       require WebPAC::Output::KinoSearch;
-                       $indexer_config->{clean} = 1 unless (-e $indexer_config->{index_path});
-                       $indexer = new WebPAC::Output::KinoSearch( %{ $indexer_config } );
+                       die "no longer supported";
 
                } else {
                        $log->logdie("unknown use_indexer: $use_indexer");
@@ -298,36 +295,49 @@ foreach my $database ( sort keys %{ $config->databases } ) {
        #
        # now WebPAC::Store
        #
-       my $abs_path = abs_path($0);
-       $abs_path =~ s#/[^/]*$#/#;
+       my $store = new WebPAC::Store({
+               debug => $debug,
+       });
 
-       my $db_path = $config->webpac('db_path');
 
-       if ($clean) {
-               $log->info("creating new database '$database' in $db_path");
-               rmtree( $db_path ) || $log->warn("can't remove $db_path: $!");
-       } else {
-               $log->info("working on database '$database' in $db_path");
-       }
+       #
+       # prepare output
+       #
+       my @outputs = force_array( $db_config->{output}, sub {
+               $log->error("Database $database doesn't have any outputs defined. Do you want to remove it from configuration?" );
+       } );
 
-       my $store = new WebPAC::Store(
-               path => $db_path,
-               debug => $debug,
-       );
+       my @output_modules;
+
+       foreach my $output ( @outputs ) {
+
+#warn '## output = ',dump( $output );
+
+               my $module = $output->{module} || $log->logdie("need module in output section of $database");
+               $module = 'WebPAC::Output::' . $module unless $module =~ m/::/;
+       
+               $log->debug("loading output module $module");
+               eval "require $module";
+
+               # add database to arugemnts for output filter
+               $output->{database} = $database;
+
+               $log->debug("calling $module->new(",dump( $output ),")");
+               my $out = new $module->new( $output );
+               $out->init;
+
+               push @output_modules, $out;
+       }
 
 
        #
        # now, iterate through input formats
        #
 
-       my @inputs;
-       if (ref($db_config->{input}) eq 'ARRAY') {
-               @inputs = @{ $db_config->{input} };
-       } elsif ($db_config->{input}) {
-               push @inputs, $db_config->{input};
-       } else {
+
+       my @inputs = force_array( $db_config->{input}, sub {
                $log->info("database $database doesn't have inputs defined");
-       }
+       } );
 
        foreach my $input (@inputs) {
 
@@ -337,9 +347,16 @@ foreach my $database ( sort keys %{ $config->databases } ) {
 
                my $type = lc($input->{type});
 
-               die "I know only how to handle input types ", join(",", $config->webpac('inputs') ), " not '$type'!\n" unless (grep(/$type/, $config->webpac('inputs')));
+               # FIXME check if input module exists
+               my $input_module = $input->{module};
 
-               my $input_module = $config->webpac('inputs')->{$type};
+               if ( ! $input_module ) {
+                       if ( grep(/$type/, $config->webpac('inputs')) ) {
+                               $input_module = $config->webpac('inputs')->{$type};
+                       } else {
+                               $log->logdie("I know only how to handle input types ", join(",", $config->webpac('inputs') ), " not '$type'!" );
+                       }
+               }
 
                my @lookups = $parser->have_lookup_create($database, $input);
 
@@ -362,6 +379,7 @@ foreach my $database ( sort keys %{ $config->databases } ) {
                        stats => $stats,
                        modify_records => $input->{modify_records},
                        modify_file => $input->{modify_file},
+                       input_config => $input,
                );
                $log->logdie("can't create input using $input_module") unless ($input);
 
@@ -498,6 +516,16 @@ foreach my $database ( sort keys %{ $config->databases } ) {
                }
 
 
+               # setup input name for all output filters
+               foreach my $out ( @output_modules ) {
+                       if ( $out->can('input') ) {
+                               $out->input( $input_name );
+                       } else {
+                               $log->warn("output filter ",ref($out)," doesn't support input name");
+                       }
+               }
+
+
                foreach my $pos ( 0 ... $input_db->size ) {
 
                        my $row = $input_db->fetch || next;
@@ -506,7 +534,7 @@ foreach my $database ( sort keys %{ $config->databases } ) {
 
                        my $mfn = $row->{'000'}->[0];
 
-                       if (! $mfn || $mfn !~ m#^\d+$#) {
+                       if (! $mfn || $mfn !~ m{^\d+$}) {
                                $log->warn("record $pos doesn't have valid MFN but '$mfn', using $pos");
                                $mfn = $pos;
                                push @{ $row->{'000'} }, $pos;
@@ -522,7 +550,7 @@ foreach my $database ( sort keys %{ $config->databases } ) {
                                next;   # validation doesn't create any output
                        }
 
-                       my $ds = WebPAC::Normalize::data_structure(
+                       if ( my $ds = WebPAC::Normalize::data_structure(
                                row => $row,
                                rules => $rules,
                                lookup => $lookup_hash,
@@ -536,38 +564,47 @@ foreach my $database ( sort keys %{ $config->databases } ) {
                                                id => $mfn,
                                        );
                                },
-                       );
+                       ) ) { # if
 
-                       $log->debug("ds = ", sub { dump($ds) }) if ($ds);
-
-                       $store->save_ds(
-                               database => $database,
-                               input => $input_name,
-                               id => $mfn,
-                               ds => $ds,
-                       ) if ($ds && !$stats);
-
-                       $indexer->add(
-                               id => "${input_name}/${mfn}",
-                               ds => $ds,
-                               type => $config->get($indexer_config)->{type},
-                       ) if ($indexer && $ds);
-
-                       if ($marc) {
-                               my $i = 0;
-
-                               while (my $fields = WebPAC::Normalize::_get_marc_fields( fetch_next => 1 ) ) {
-                                       $marc->add(
-                                               id => $mfn . ( $i ? "/$i" : '' ),
-                                               fields => $fields,
-                                               leader => WebPAC::Normalize::_get_marc_leader(),
-                                               row => $row,
-                                       );
-                                       $i++;
+                               $log->debug("ds = ", sub { dump($ds) });
+
+                               $store->save_ds(
+                                       database => $database,
+                                       input => $input_name,
+                                       id => $mfn,
+                                       ds => $ds,
+                               ) if !$stats;
+
+                               $indexer->add(
+                                       id => "${input_name}/${mfn}",
+                                       ds => $ds,
+                                       type => $config->get($indexer_config)->{type},
+                               ) if $indexer;
+
+                               if ($marc) {
+                                       my $i = 0;
+
+                                       while (my $fields = WebPAC::Normalize::_get_marc_fields( fetch_next => 1 ) ) {
+                                               $marc->add(
+                                                       id => $mfn . ( $i ? "/$i" : '' ),
+                                                       fields => $fields,
+                                                       leader => WebPAC::Normalize::_get_marc_leader(),
+                                                       row => $row,
+                                               );
+                                               $i++;
+                                       }
+
+                                       $log->info("Created $i instances of MFN $mfn\n") if ($i > 1);
+                               }
+
+                               foreach my $out ( @output_modules ) {
+                                       $out->add( $mfn, $ds ) if $out->can('add');
                                }
 
-                               $log->info("Created $i instances of MFN $mfn\n") if ($i > 1);
+                       } else {
+                               $log->warn("record $pos didn't produce any output after normalization rules!");
                        }
+
                }
 
                if ($validate) {
@@ -598,6 +635,10 @@ foreach my $database ( sort keys %{ $config->databases } ) {
 
        eval { $indexer->finish } if ($indexer && $indexer->can('finish'));
 
+       foreach my $out ( @output_modules ) {
+               $out->finish if $out->can('finish');
+       }
+
        my $dt = time() - $start_t;
        $log->info("$total_rows records ", $indexer ? "indexed " : "",
                sprintf("in %.2f sec [%.2f rec/sec]",