X-Git-Url: http://git.equinoxoli.org/?p=migration-tools.git;a=blobdiff_plain;f=mig-bin%2Fmig-stage;fp=mig-bin%2Fmig-stage;h=0000000000000000000000000000000000000000;hp=6e7faf59e407cefb2fe68e87733607086b45f020;hb=155eb9eac077ca803f75d1295e584e7012e1b883;hpb=69588457ab8f70fbb77af29cc0653933d24ed2ac diff --git a/mig-bin/mig-stage b/mig-bin/mig-stage deleted file mode 100755 index 6e7faf5..0000000 --- a/mig-bin/mig-stage +++ /dev/null @@ -1,128 +0,0 @@ -#!/usr/bin/perl -w -############################################################################### -=pod - -=head1 NAME - -mig-stage - -Load the SQL-converted version of the specified file into the migration schema. - -Extra arguments are passed to the underlying call to psql - -If the tracked file was previously staged with a different table, drop that -table. - - -=head1 SYNOPSIS - -B [other arguments...] - -=cut - -############################################################################### - -use strict; -use Switch; -use Env qw( - HOME PGHOST PGPORT PGUSER PGDATABASE MIGSCHEMA - MIGBASEWORKDIR MIGBASEGITDIR MIGGITDIR MIGWORKDIR -); -use Pod::Usage; -use DBI; -use Cwd 'abs_path'; -use FindBin; -my $mig_bin = "$FindBin::Bin/"; -use lib "$FindBin::Bin/"; -use Mig; - -pod2usage(-verbose => 2) if ! $ARGV[0] || $ARGV[0] eq '--help'; - -Mig::die_if_no_env_migschema(); -Mig::die_if_mig_tracking_table_does_not_exist(); - -my $file = abs_path($ARGV[0]); -if ($file =~ /^$MIGBASEWORKDIR/) { - stage_csv(@ARGV); -} else { - print "File falls outside of MIGWORKDIR ($MIGWORKDIR): $file\n"; -} - -exit 0; - -############################################################################### - -sub stage_csv { - my $file = abs_path(shift); - my @args = @_; - - my $tracked_file_id = Mig::check_for_tracked_file($file); - if ($tracked_file_id) { - my $data = Mig::status_this_file($file); - - if (! $data->{'utf8_filename'}) { - die "mig-iconv or mig-skip-iconv needed for UTF8 version of file: $file\n"; - } - - if (! $data->{'clean_filename'}) { - die "mig-clean or mig-skip-clean needed for .clean version of file: $file\n"; - } - - if (! $data->{'stage_sql_filename'}) { - die "mig-convert needed for .stage.sql version of file: $file\n"; - } - - my $stage_sql_filename = $data->{'stage_sql_filename'}; - if (! -e $stage_sql_filename) { - die "missing file: $stage_sql_filename\n"; - } - - my $schema_table = `grep 'CREATE UNLOGGED TABLE' $stage_sql_filename | cut -f4 -d\\ | head -1`; - chomp $schema_table; - my ($schema,$table) = split /\./, $schema_table; - - if (defined $data->{'staged_table'} && $data->{'staged_table'} ne $table) { - my $dbh2 = Mig::db_connect(); - print "dropping previously staged table: $MIGSCHEMA.$data->{staged_table}\n"; - my $rv2 = $dbh2->do(" - DROP TABLE $MIGSCHEMA.$data->{staged_table}; - ") || die "Error dropping table $data->{staged_table}: $!\n"; - print "changing references to old tables\n"; - my $rv3 = $dbh2->do(" - UPDATE $MIGSCHEMA.tracked_column - SET staged_table = " . $dbh2->quote($table) . " - WHERE staged_table = " . $dbh2->quote($data->{staged_table}) . " - ") || die "Error changing references to $data->{staged_table}: $!\n"; - my $rv4 = $dbh2->do(" - UPDATE $MIGSCHEMA.tracked_column - SET target_table = " . $dbh2->quote($table) . " - WHERE target_table = " . $dbh2->quote($data->{staged_table}) . " - ") || die "Error changing references to $data->{staged_table}: $!\n"; - Mig::db_disconnect($dbh2); - } - - print "running staging SQL: $stage_sql_filename\n"; - - system('psql', @args, '-f', $stage_sql_filename); - - if ($schema ne $MIGSCHEMA) { - die "Schema mismatch: env => $MIGSCHEMA sql => $schema\n"; - } - if (! Mig::check_db_migschema_for_specific_table($table)) { - die "Missing staged table: $schema_table\n"; - } else { - print "table staged: $schema_table\n"; - } - - my $dbh = Mig::db_connect(); - my $rv = $dbh->do(" - UPDATE $MIGSCHEMA.tracked_file - SET staged_table = " . $dbh->quote($table) . " - WHERE base_filename = " . $dbh->quote($file) . " - ; - ") || die "Error updating table $MIGSCHEMA.tracked_file: $!\n"; - Mig::db_disconnect($dbh); - } else { - print "File not currently tracked: $file\n"; - } -}