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);
13 my $user = shift or die &usage;
14 my $dbh = adminsuidsetup $user;
18 #false laziness w/fs-setup
19 my @tables = scalar(@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";
26 my $primary_key = $tableobj->primary_key;
27 next unless $primary_key;
29 my $col = $tableobj->column($primary_key);
31 next unless uc($col->type) eq 'SERIAL'
32 || ( driver_name eq 'Pg'
33 && defined($col->default)
34 && $col->default =~ /^nextval\(/i
36 || ( driver_name eq 'mysql'
37 && defined($col->local)
38 && $col->local =~ /AUTO_INCREMENT/i
41 warn "fixing sequence for $table\n";
43 $dbh->do( "SELECT setval( '${table}_${primary_key}_seq',
44 ( SELECT max($primary_key) FROM $table ) );" )
49 $dbh->commit or die $dbh->errstr;
50 $dbh->disconnect or die $dbh->errstr;
53 die "Usage:\n fix-sequences user [ table table ... ] \n";