projects
/
freeside.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
RT# 79497 - removed test code
[freeside.git]
/
bin
/
fix-sequences
diff --git
a/bin/fix-sequences
b/bin/fix-sequences
index
f193e21
..
896f01c
100755
(executable)
--- a/
bin/fix-sequences
+++ b/
bin/fix-sequences
@@
-1,13
+1,9
@@
#!/usr/bin/perl -Tw
#!/usr/bin/perl -Tw
+# run dbdef-create first!
+
use strict;
use strict;
-use DBI;
-use DBIx::DBSchema 0.21;
-use DBIx::DBSchema::Table;
-use DBIx::DBSchema::Column;
-use DBIx::DBSchema::ColGroup::Unique;
-use DBIx::DBSchema::ColGroup::Index;
-use FS::UID qw(adminsuidsetup);
+use FS::UID qw(adminsuidsetup driver_name);
use FS::Record qw(dbdef);
my $user = shift or die &usage;
use FS::Record qw(dbdef);
my $user = shift or die &usage;
@@
-18,8
+14,10
@@
my $schema = dbdef();
#false laziness w/fs-setup
my @tables = scalar(@ARGV)
? @ARGV
#false laziness w/fs-setup
my @tables = scalar(@ARGV)
? @ARGV
- : grep { ! /^h_/ } $schema->tables;
+ #: grep { ! /^h_/ } $schema->tables;
+ : $schema->tables;
foreach my $table ( @tables ) {
foreach my $table ( @tables ) {
+
my $tableobj = $schema->table($table)
or die "unknown table $table (did you run dbdef-create?)\n";
my $tableobj = $schema->table($table)
or die "unknown table $table (did you run dbdef-create?)\n";
@@
-28,21
+26,42
@@
foreach my $table ( @tables ) {
my $col = $tableobj->column($primary_key);
my $col = $tableobj->column($primary_key);
+#warn $col->type;
+my $default = ref($col->default) ? ${ $col->default } : $col->default;
+#use Data::Dumper;
+#warn Dumper $col;
+
next unless uc($col->type) eq 'SERIAL'
|| ( driver_name eq 'Pg'
next unless uc($col->type) eq 'SERIAL'
|| ( driver_name eq 'Pg'
- && defined($
col->
default)
- && $
col->
default =~ /^nextval\(/i
+ && defined($default)
+ && $default =~ /^nextval\(/i
)
|| ( driver_name eq 'mysql'
&& defined($col->local)
&& $col->local =~ /AUTO_INCREMENT/i
);
)
|| ( driver_name eq 'mysql'
&& defined($col->local)
&& $col->local =~ /AUTO_INCREMENT/i
);
+ my $suffix = '_seq';
+ if ( $table =~ /^(objectcustomfields|objectcustomfieldvalues)$/ ) {
+ $suffix = '_s';
+ }
+
+ my $seq = "${table}_${primary_key}$suffix";
+ if ( driver_name eq 'Pg'
+ && defined($default)
+ && $default =~ /^nextval\('"(public\.)?(\w+_seq)"'::(text|regclass)\)$/
+ ) {
+ $seq = $2;
+ }
+
warn "fixing sequence for $table\n";
warn "fixing sequence for $table\n";
- $dbh->do( "SELECT setval( '${table}_${primary_key}_seq',
- ( SELECT max($primary_key) FROM $table ) );" )
- or die $dbh->errstr;
+
+ my $sql = "SELECT setval( '$seq',
+ ( SELECT max($primary_key) FROM $table ) );";
+
+ #warn $col->default. " $seq\n$sql\n";
+ $dbh->do( $sql ) or die $dbh->errstr;
}
}
@@
-50,6
+69,6
@@
$dbh->commit or die $dbh->errstr;
$dbh->disconnect or die $dbh->errstr;
sub usage {
$dbh->disconnect or die $dbh->errstr;
sub usage {
- die "Usage:\n
create-history-tabl
es user [ table table ... ] \n";
+ die "Usage:\n
fix-sequenc
es user [ table table ... ] \n";
}
}