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);
32 next unless uc($col->type) eq 'SERIAL'
33 || ( driver_name eq 'Pg'
34 && defined($col->default)
35 && $col->default =~ /^nextval\(/i
37 || ( driver_name eq 'mysql'
38 && defined($col->local)
39 && $col->local =~ /AUTO_INCREMENT/i
42 my $seq = "${table}_${primary_key}_seq";
43 if ( driver_name eq 'Pg'
44 && defined($col->default)
45 && $col->default =~ /^nextval\('"(\w+_seq)"'::text\)$/
50 warn "fixing sequence for $table\n";
52 $dbh->do( "SELECT setval( '$seq',
53 ( SELECT max($primary_key) FROM $table ) );" )
58 $dbh->commit or die $dbh->errstr;
59 $dbh->disconnect or die $dbh->errstr;
62 die "Usage:\n fix-sequences user [ table table ... ] \n";