X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI%2Fmysql.pm;h=cb83e7a16ea3ef2d105357b2af910336df3e6f54;hb=e00d61acf4f772083500a04783a8bce56dc322bf;hp=219f06b5d6eb9e3757b397ac8ce5f4e4a2608944;hpb=78b7ccaaaff070d6053dda2843007369d7501662;p=dbsrgits%2FDBIx-Class-Schema-Loader.git diff --git a/lib/DBIx/Class/Schema/Loader/DBI/mysql.pm b/lib/DBIx/Class/Schema/Loader/DBI/mysql.pm index 219f06b..cb83e7a 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI/mysql.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI/mysql.pm @@ -3,55 +3,56 @@ package DBIx::Class::Schema::Loader::DBI::mysql; use strict; use warnings; use base 'DBIx::Class::Schema::Loader::DBI'; -use Carp::Clan qw/^DBIx::Class/; -use Class::C3; +use mro 'c3'; +use List::Util 'first'; +use namespace::clean; -our $VERSION = '0.04999_01'; +our $VERSION = '0.07010'; =head1 NAME DBIx::Class::Schema::Loader::DBI::mysql - DBIx::Class::Schema::Loader::DBI mysql Implementation. -=head1 SYNOPSIS +=head1 DESCRIPTION - package My::Schema; - use base qw/DBIx::Class::Schema::Loader/; +See L and L. - __PACKAGE__->loader_options( debug => 1 ); +=cut - 1; +sub _setup { + my $self = shift; -=head1 DESCRIPTION + $self->schema->storage->sql_maker->quote_char("`"); + $self->schema->storage->sql_maker->name_sep("."); -See L. + $self->next::method(@_); -=cut + if (not defined $self->preserve_case) { + $self->preserve_case(0); + } +} -# had to override here because MySQL apparently -# doesn't support '%' syntax. Perhaps the other -# drivers support this syntax also, but I didn't -# want to risk breaking some esoteric DBD::foo version -# in a maint release... sub _tables_list { - my $self = shift; - - my $dbh = $self->schema->storage->dbh; - my @tables = $dbh->tables(undef, $self->db_schema, undef, undef); - s/\Q$self->{_quoter}\E//g for @tables; - s/^.*\Q$self->{_namesep}\E// for @tables; + my ($self, $opts) = @_; - return @tables; + return $self->next::method($opts, undef, undef); } sub _table_fk_info { my ($self, $table) = @_; my $dbh = $self->schema->storage->dbh; - my $table_def_ref = $dbh->selectrow_arrayref("SHOW CREATE TABLE `$table`") - or croak ("Cannot get table definition for $table"); - my $table_def = $table_def_ref->[1] || ''; - - my (@reldata) = ($table_def =~ /CONSTRAINT `.*` FOREIGN KEY \(`(.*)`\) REFERENCES `(.*)` \(`(.*)`\)/ig); + + my $table_def_ref = eval { $dbh->selectrow_arrayref("SHOW CREATE TABLE `$table`") }; + my $table_def = $table_def_ref->[1]; + + return [] if not $table_def; + + my $qt = qr/["`]/; + + my (@reldata) = ($table_def =~ + /CONSTRAINT $qt.*$qt FOREIGN KEY \($qt(.*)$qt\) REFERENCES $qt(.*)$qt \($qt(.*)$qt\)/ig + ); my @rels; while (scalar @reldata > 0) { @@ -59,16 +60,18 @@ sub _table_fk_info { my $f_table = shift @reldata; my $f_cols = shift @reldata; - my @cols = map { s/\Q$self->{_quoter}\E//; lc $_ } - split(/\s*,\s*/, $cols); + my @cols = map { s/(?: \Q$self->{_quoter}\E | $qt )//x; $self->_lc($_) } + split(/$qt?\s*$qt?,$qt?\s*$qt?/, $cols); - my @f_cols = map { s/\Q$self->{_quoter}\E//; lc $_ } - split(/\s*,\s*/, $f_cols); + my @f_cols = map { s/(?: \Q$self->{_quoter}\E | $qt )//x; $self->_lc($_) } + split(/$qt?\s*$qt?,$qt?\s*$qt?/, $f_cols); + + my $remote_table = first { $_ =~ /^${f_table}\z/i } $self->_tables_list; push(@rels, { local_columns => \@cols, remote_columns => \@f_cols, - remote_table => $f_table + remote_table => $remote_table, }); } @@ -83,12 +86,12 @@ sub _mysql_table_get_keys { if(!exists($self->{_cache}->{_mysql_keys}->{$table})) { my %keydata; my $dbh = $self->schema->storage->dbh; - my $sth = $dbh->prepare("SHOW INDEX FROM `$table`"); + my $sth = $dbh->prepare('SHOW INDEX FROM '.$self->_table_as_sql($table)); $sth->execute; while(my $row = $sth->fetchrow_hashref) { next if $row->{Non_unique}; push(@{$keydata{$row->{Key_name}}}, - [ $row->{Seq_in_index}, lc $row->{Column_name} ] + [ $row->{Seq_in_index}, $self->_lc($row->{Column_name}) ] ); } foreach my $keyname (keys %keydata) { @@ -121,10 +124,121 @@ sub _table_uniq_info { return \@uniqs; } -sub _column_is_auto_increment { - my ($self, $info) = @_; +sub _columns_info_for { + my $self = shift; + my ($table) = @_; + + my $result = $self->next::method(@_); + + my $dbh = $self->schema->storage->dbh; + + while (my ($col, $info) = each %$result) { + if ($info->{data_type} eq 'int') { + $info->{data_type} = 'integer'; + } + elsif ($info->{data_type} eq 'double') { + $info->{data_type} = 'double precision'; + } + my $data_type = $info->{data_type}; + + delete $info->{size} if $data_type !~ /^(?: (?:var)?(?:char(?:acter)?|binary) | bit | year)\z/ix; + + # information_schema is available in 5.0+ + my ($precision, $scale, $column_type, $default) = eval { $dbh->selectrow_array(<<'EOF', {}, $table, $col) }; +SELECT numeric_precision, numeric_scale, column_type, column_default +FROM information_schema.columns +WHERE table_name = ? AND column_name = ? +EOF + my $has_information_schema = not $@; + + $column_type = '' if not defined $column_type; + + if ($data_type eq 'bit' && (not exists $info->{size})) { + $info->{size} = $precision if defined $precision; + } + elsif ($data_type =~ /^(?:float|double precision|decimal)\z/i) { + if (defined $precision && defined $scale) { + if ($precision == 10 && $scale == 0) { + delete $info->{size}; + } + else { + $info->{size} = [$precision,$scale]; + } + } + } + elsif ($data_type eq 'year') { + if ($column_type =~ /\(2\)/) { + $info->{size} = 2; + } + elsif ($column_type =~ /\(4\)/ || $info->{size} == 4) { + delete $info->{size}; + } + } + elsif ($data_type =~ /^(?:date(?:time)?|timestamp)\z/) { + if (not (defined $self->datetime_undef_if_invalid && $self->datetime_undef_if_invalid == 0)) { + $info->{datetime_undef_if_invalid} = 1; + } + } + elsif ($data_type =~ /^(?:enum|set)\z/ && $has_information_schema + && $column_type =~ /^(?:enum|set)\(/) { + + delete $info->{extra}{list}; + + while ($column_type =~ /'((?:[^']* (?:''|\\')* [^']*)* [^\\'])',?/xg) { + my $el = $1; + $el =~ s/''/'/g; + push @{ $info->{extra}{list} }, $el; + } + } + + # Sometimes apparently there's a bug where default_value gets set to '' + # for things that don't actually have or support that default (like ints.) + if (exists $info->{default_value} && $info->{default_value} eq '') { + if ($has_information_schema) { + if (not defined $default) { + delete $info->{default_value}; + } + } + else { # just check if it's a char/text type, otherwise remove + delete $info->{default_value} unless $data_type =~ /char|text/i; + } + } + } + + return $result; +} + +sub _extra_column_info { + no warnings 'uninitialized'; + my ($self, $table, $col, $info, $dbi_info) = @_; + my %extra_info; - return $info->{mysql_is_auto_increment}; + if ($dbi_info->{mysql_is_auto_increment}) { + $extra_info{is_auto_increment} = 1 + } + if ($dbi_info->{mysql_type_name} =~ /\bunsigned\b/i) { + $extra_info{extra}{unsigned} = 1; + } + if ($dbi_info->{mysql_values}) { + $extra_info{extra}{list} = $dbi_info->{mysql_values}; + } + if ( lc($dbi_info->{COLUMN_DEF}) eq 'current_timestamp' + && lc($dbi_info->{mysql_type_name}) eq 'timestamp') { + + my $current_timestamp = 'current_timestamp'; + $extra_info{default_value} = \$current_timestamp; + } + + return \%extra_info; +} + +sub _dbh_column_info { + my $self = shift; + + local $SIG{__WARN__} = sub { warn @_ + unless $_[0] =~ /^column_info: unrecognized column type/ }; + + $self->next::method(@_); } =head1 SEE ALSO @@ -132,6 +246,16 @@ sub _column_is_auto_increment { L, L, L +=head1 AUTHOR + +See L and L. + +=head1 LICENSE + +This library is free software; you can redistribute it and/or modify it under +the same terms as Perl itself. + =cut 1; +# vim:et sw=4 sts=4 tw=0: