bin/fix-sequences

   1 #!/usr/bin/perl -Tw
   2
   3 use strict;
   4 use DBI;
   5 use DBIx::DBSchema 0.21;
   6 use DBIx::DBSchema::Table;
   7 use DBIx::DBSchema::Column;
   8 use DBIx::DBSchema::ColGroup::Unique;
   9 use DBIx::DBSchema::ColGroup::Index;
  10 use FS::UID qw(adminsuidsetup driver_name);
  11 use FS::Record qw(dbdef);
  12
  13 my $user = shift or die &usage;
  14 my $dbh = adminsuidsetup $user;
  15
  16 my $schema = dbdef();
  17
  18 #false laziness w/fs-setup
  19 my @tables = scalar(@ARGV)
  20                ? @ARGV
  21                : grep { ! /^h_/ } $schema->tables;
  22 foreach my $table ( @tables ) {
  23   my $tableobj = $schema->table($table)
  24     or die "unknown table $table (did you run dbdef-create?)\n";
  25
  26   my $primary_key = $tableobj->primary_key;
  27   next unless $primary_key;
  28
  29   my $col = $tableobj->column($primary_key);
  30
  31
  32   next unless uc($col->type) eq 'SERIAL'
  33               || ( driver_name eq 'Pg'
  34                      && defined($col->default)
  35                      && $col->default =~ /^nextval\(/i
  36                  )
  37               || ( driver_name eq 'mysql'
  38                      && defined($col->local)
  39                      && $col->local =~ /AUTO_INCREMENT/i
  40                  );
  41
  42   my $seq = "${table}_${primary_key}_seq";
  43   if ( driver_name eq 'Pg'
  44        && defined($col->default)
  45        && $col->default =~ /^nextval\('"(\w+_seq)"'::text\)$/
  46      ) {
  47     $seq = $1;
  48   }
  49
  50   warn "fixing sequence for $table\n";
  51
  52   $dbh->do( "SELECT setval( '$seq',
  53                             ( SELECT max($primary_key) FROM $table ) );" )
  54     or die $dbh->errstr;
  55
  56 }
  57
  58 $dbh->commit or die $dbh->errstr;
  59 $dbh->disconnect or die $dbh->errstr;
  60
  61 sub usage {
  62   die "Usage:\n  fix-sequences user [ table table ... ] \n";
  63 }
  64