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   next unless uc($col->type) eq 'SERIAL'
  32               || ( driver_name eq 'Pg'
  33                      && defined($col->default)
  34                      && $col->default =~ /^nextval\(/i
  35                  )
  36               || ( driver_name eq 'mysql'
  37                      && defined($col->local)
  38                      && $col->local =~ /AUTO_INCREMENT/i
  39                  );
  40
  41   warn "fixing sequence for $table\n";
  42
  43   $dbh->do( "SELECT setval( '${table}_${primary_key}_seq',
  44                             ( SELECT max($primary_key) FROM $table ) );" )
  45     or die $dbh->errstr;
  46
  47 }
  48
  49 $dbh->commit or die $dbh->errstr;
  50 $dbh->disconnect or die $dbh->errstr;
  51
  52 sub usage {
  53   die "Usage:\n  fix-sequences user [ table table ... ] \n";
  54 }
  55