X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI%2FSybase.pm;h=56b0a35ef6fb3e527fb2e8ff88ee1f4b1c350c8b;hb=dc767cd32c6728d4d9c3504acd259c0b2f19da2b;hp=d5111e7377c1315057960ab46d3d971a0b0a6a38;hpb=c9373b79cecb2bde4c0f07f4068ca38b1646fe34;p=dbsrgits%2FDBIx-Class-Schema-Loader.git diff --git a/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm b/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm index d5111e7..56b0a35 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm @@ -2,14 +2,11 @@ package DBIx::Class::Schema::Loader::DBI::Sybase; use strict; use warnings; -use base qw/ - DBIx::Class::Schema::Loader::DBI - DBIx::Class::Schema::Loader::DBI::Sybase::Common -/; +use base 'DBIx::Class::Schema::Loader::DBI::Sybase::Common'; use Carp::Clan qw/^DBIx::Class/; use Class::C3; -our $VERSION = '0.04999_06'; +our $VERSION = '0.05001'; =head1 NAME @@ -30,6 +27,8 @@ See L. =cut +sub _is_case_sensitive { 1 } + sub _setup { my $self = shift; @@ -44,7 +43,8 @@ sub _rebless { my $dbh = $self->schema->storage->dbh; my $DBMS_VERSION = @{$dbh->selectrow_arrayref(qq{sp_server_info \@attribute_id=1})}[2]; if ($DBMS_VERSION =~ /^Microsoft /i) { - my $subclass = 'DBIx::Class::Schema::Loader::DBI::MSSQL'; + $DBMS_VERSION =~ s/\s/_/g; + my $subclass = "DBIx::Class::Schema::Loader::DBI::Sybase::$DBMS_VERSION"; if ($self->load_optional_class($subclass) && !$self->isa($subclass)) { bless $self, $subclass; $self->_rebless; @@ -56,7 +56,12 @@ sub _table_columns { my ($self, $table) = @_; my $dbh = $self->schema->storage->dbh; - my $columns = $dbh->selectcol_arrayref(qq{SELECT name FROM syscolumns WHERE id = (SELECT id FROM sysobjects WHERE name = '$table' AND type = 'U')}); + my $columns = $dbh->selectcol_arrayref(qq{ +SELECT c.name +FROM syscolumns c JOIN sysobjects o +ON c.id = o.id +WHERE o.name = @{[ $dbh->quote($table) ]} AND o.type = 'U' +}); return $columns; } @@ -65,7 +70,7 @@ sub _table_pk_info { my ($self, $table) = @_; my $dbh = $self->schema->storage->dbh; - my $sth = $dbh->prepare(qq{sp_pkeys '$table'}); + my $sth = $dbh->prepare(qq{sp_pkeys @{[ $dbh->quote($table) ]}}); $sth->execute; my @keydata; @@ -80,19 +85,45 @@ sub _table_pk_info { sub _table_fk_info { my ($self, $table) = @_; + # check if FK_NAME is supported + + my $dbh = $self->schema->storage->dbh; + local $dbh->{FetchHashKeyName} = 'NAME_lc'; + # hide "Object does not exist in this database." when trying to fetch fkeys + local $dbh->{syb_err_handler} = sub { return $_[0] == 17461 ? 0 : 1 }; + my $sth = $dbh->prepare(qq{sp_fkeys \@fktable_name = @{[ $dbh->quote($table) ]}}); + $sth->execute; + my $row = $sth->fetchrow_hashref; + + return unless $row; + + if (exists $row->{fk_name}) { + $sth->finish; + return $self->_table_fk_info_by_name($table); + } + + $sth->finish; + return $self->_table_fk_info_builder($table); +} + +sub _table_fk_info_by_name { + my ($self, $table) = @_; my ($local_cols, $remote_cols, $remote_table, @rels); + my $dbh = $self->schema->storage->dbh; + local $dbh->{FetchHashKeyName} = 'NAME_lc'; # hide "Object does not exist in this database." when trying to fetch fkeys - $dbh->{syb_err_handler} = sub { return 0 if $_[0] == 17461; }; - my $sth = $dbh->prepare(qq{sp_fkeys \@FKTABLE_NAME = '$table'}); + local $dbh->{syb_err_handler} = sub { return $_[0] == 17461 ? 0 : 1 }; + my $sth = $dbh->prepare(qq{sp_fkeys \@fktable_name = @{[ $dbh->quote($table) ]}}); $sth->execute; while (my $row = $sth->fetchrow_hashref) { - next unless $row->{FK_NAME}; - my $fk = $row->{FK_NAME}; - push @{$local_cols->{$fk}}, $row->{FKCOLUMN_NAME}; - push @{$remote_cols->{$fk}}, $row->{PKCOLUMN_NAME}; - $remote_table->{$fk} = $row->{PKTABLE_NAME}; + my $fk = $row->{fk_name}; + next unless defined $fk; + + push @{$local_cols->{$fk}}, $row->{fkcolumn_name}; + push @{$remote_cols->{$fk}}, $row->{pkcolumn_name}; + $remote_table->{$fk} = $row->{pktable_name}; } foreach my $fk (keys %$remote_table) { @@ -106,13 +137,74 @@ sub _table_fk_info { return \@rels; } -sub _table_uniq_info { +sub _table_fk_info_builder { my ($self, $table) = @_; my $dbh = $self->schema->storage->dbh; - my $sth = $dbh->prepare(qq{sp_helpconstraint \@objname='$table', \@nomsg='nomsg'}); + local $dbh->{FetchHashKeyName} = 'NAME_lc'; + # hide "Object does not exist in this database." when trying to fetch fkeys + local $dbh->{syb_err_handler} = sub { return 0 if $_[0] == 17461; }; + my $sth = $dbh->prepare(qq{sp_fkeys \@fktable_name = @{[ $dbh->quote($table) ]}}); $sth->execute; + my @fk_info; + while (my $row = $sth->fetchrow_hashref) { + (my $ksq = $row->{key_seq}) =~ s/\s+//g; + + my @keys = qw/pktable_name pkcolumn_name fktable_name fkcolumn_name/; + my %ds; + @ds{@keys} = @{$row}{@keys}; + $ds{key_seq} = $ksq; + + push @{ $fk_info[$ksq] }, \%ds; + } + + my $max_keys = $#fk_info; + my @rels; + for my $level (reverse 1 .. $max_keys) { + my @level_rels; + $level_rels[$level] = splice @fk_info, $level, 1; + my $count = @{ $level_rels[$level] }; + + for my $sub_level (reverse 1 .. $level-1) { + my $total = @{ $fk_info[$sub_level] }; + + $level_rels[$sub_level] = [ + splice @{ $fk_info[$sub_level] }, $total-$count, $count + ]; + } + + while (1) { + my @rel = map shift @$_, @level_rels[1..$level]; + + last unless defined $rel[0]; + + my @local_columns = map $_->{fkcolumn_name}, @rel; + my @remote_columns = map $_->{pkcolumn_name}, @rel; + my $remote_table = $rel[0]->{pktable_name}; + + push @rels, { + local_columns => \@local_columns, + remote_columns => \@remote_columns, + remote_table => $remote_table + }; + } + } + + return \@rels; +} + +sub _table_uniq_info { + my ($self, $table) = @_; + + local $SIG{__WARN__} = sub {}; + + my $dbh = $self->schema->storage->dbh; + local $dbh->{FetchHashKeyName} = 'NAME_lc'; + my $sth = $dbh->prepare(qq{sp_helpconstraint \@objname=@{[ $dbh->quote($table) ]}, \@nomsg='nomsg'}); + eval { $sth->execute }; + return if $@; + my $constraints; while (my $row = $sth->fetchrow_hashref) { if (exists $row->{constraint_type}) { @@ -137,6 +229,63 @@ sub _table_uniq_info { return \@uniqs; } +# get the correct data types, defaults and size +sub _columns_info_for { + my $self = shift; + my ($table) = @_; + my $result = $self->next::method(@_); + + my $dbh = $self->schema->storage->dbh; + my $sth = $dbh->prepare(qq{ +SELECT c.name name, t.name type, cm.text deflt, c.prec prec, c.scale scale, + c.length len +FROM syscolumns c +JOIN sysobjects o ON c.id = o.id +LEFT JOIN systypes t ON c.type = t.type AND c.usertype = t.usertype +LEFT JOIN syscomments cm + ON cm.id = CASE WHEN c.cdefault = 0 THEN c.computedcol ELSE c.cdefault END +WHERE o.name = @{[ $dbh->quote($table) ]} AND o.type = 'U' +}); + $sth->execute; + local $dbh->{FetchHashKeyName} = 'NAME_lc'; + my $info = $sth->fetchall_hashref('name'); + + while (my ($col, $res) = each %$result) { + my $data_type = $res->{data_type} = $info->{$col}{type}; + + if ($data_type && $data_type =~ /^timestamp\z/i) { + $res->{inflate_datetime} = 0; + } + + if (my $default = $info->{$col}{deflt}) { + if ($default =~ /^AS \s+ (\S+)/ix) { + my $function = $1; + $res->{default_value} = \$function; + } + elsif ($default =~ /^DEFAULT \s+ (\S+)/ix) { + my ($constant_default) = $1 =~ /^['"\[\]]?(.*?)['"\[\]]?\z/; + $res->{default_value} = $constant_default; + } + } + +# XXX we need to handle "binary precision" for FLOAT(X) +# (see: http://msdn.microsoft.com/en-us/library/aa258876(SQL.80).aspx ) + if (my $data_type = $res->{data_type}) { + if ($data_type =~ /^(?:text|unitext|image|bigint|int|integer|smallint|tinyint|real|double|double precision|float|date|time|datetime|smalldatetime|money|smallmoney|timestamp|bit)\z/i) { + delete $res->{size}; + } + elsif ($data_type =~ /^(?:numeric|decimal)\z/i) { + $res->{size} = [ $info->{$col}{prec}, $info->{$col}{scale} ]; + } + elsif ($data_type =~ /^(?:unichar|univarchar)\z/i) { + $res->{size} /= 2; + } + } + } + + return $result; +} + sub _extra_column_info { my ($self, $info) = @_; my %extra_info; @@ -144,7 +293,7 @@ sub _extra_column_info { my ($table, $column) = @$info{qw/TABLE_NAME COLUMN_NAME/}; my $dbh = $self->schema->storage->dbh; - my $sth = $dbh->prepare(qq{SELECT name FROM syscolumns WHERE id = (SELECT id FROM sysobjects WHERE name = '$table') AND (status & 0x80) = 0x80 AND name = '$column'}); + my $sth = $dbh->prepare(qq{SELECT name FROM syscolumns WHERE id = (SELECT id FROM sysobjects WHERE name = @{[ $dbh->quote($table) ]}) AND (status & 0x80) = 0x80 AND name = @{[ $dbh->quote($column) ]}}); $sth->execute(); if ($sth->fetchrow_array) { @@ -161,7 +310,12 @@ L =head1 AUTHOR -Justin Hunter C +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