X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI%2FSybase.pm;h=3d48a83e32e1abe2a986e519f99855a39ba8cdc2;hb=dc379dc68c7a7d620e16ee91c0c57a0b875081bd;hp=4c7b7f3f5bcceda99e5c5f06ce4db62a5fdfa094;hpb=fe67d3434ad62ea06e1a483429cb07c744d6c70d;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 4c7b7f3..3d48a83 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI/Sybase.pm @@ -2,28 +2,20 @@ package DBIx::Class::Schema::Loader::DBI::Sybase; use strict; use warnings; -use base 'DBIx::Class::Schema::Loader::DBI'; +use base 'DBIx::Class::Schema::Loader::DBI::Sybase::Common'; use Carp::Clan qw/^DBIx::Class/; -use Class::C3; +use mro 'c3'; -our $VERSION = '0.04999_06'; +our $VERSION = '0.07010'; =head1 NAME -DBIx::Class::Schema::Loader::DBI::Sybase - DBIx::Class::Schema::Loader::DBI Sybase Implementation. - -=head1 SYNOPSIS - - package My::Schema; - use base qw/DBIx::Class::Schema::Loader/; - - __PACKAGE__->loader_options( debug => 1 ); - - 1; +DBIx::Class::Schema::Loader::DBI::Sybase - DBIx::Class::Schema::Loader::DBI +Sybase ASE Implementation. =head1 DESCRIPTION -See L. +See L and L. =cut @@ -31,7 +23,10 @@ sub _setup { my $self = shift; $self->next::method(@_); - $self->{db_schema} ||= 'dbo'; + + if (not defined $self->preserve_case) { + $self->preserve_case(1); + } } sub _rebless { @@ -40,19 +35,38 @@ 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'; - if ($self->load_optional_class($subclass)) { - bless $self, $subclass unless $self->isa($subclass); - $self->_rebless; + $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; } } } +sub _tables_list { + my ($self, $opts) = @_; + + my $dbh = $self->schema->storage->dbh; + + my $sth = $dbh->table_info(undef, $self->db_schema, undef, "'TABLE','VIEW'"); + + my @tables = grep $_ ne 'sysquerymetrics', + map $_->{table_name}, @{ $sth->fetchall_arrayref({ table_name => 1 }) }; + + return $self->_filter_tables(\@tables, $opts); +} + 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; } @@ -61,13 +75,13 @@ 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; while (my $row = $sth->fetchrow_hashref) { - push @keydata, lc $row->{column_name}; + push @keydata, $row->{column_name}; } return \@keydata; @@ -76,49 +90,127 @@ 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_by_sp_helpconstraint($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}}, lc $row->{FKCOLUMN_NAME}; - push @{$remote_cols->{$fk}}, lc $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) { - push @rels, { - local_columns => \@{$local_cols->{$fk}}, - remote_columns => \@{$remote_cols->{$fk}}, - remote_table => $remote_table->{$fk}, - }; + push @rels, { + local_columns => \@{$local_cols->{$fk}}, + remote_columns => \@{$remote_cols->{$fk}}, + remote_table => $remote_table->{$fk}, + }; } return \@rels; } -sub _table_uniq_info { +sub _table_fk_info_by_sp_helpconstraint { my ($self, $table) = @_; + my $warn_handler = $SIG{__WARN__} || sub { warn @_ }; + local $SIG{__WARN__} = sub { + $warn_handler->(@_) unless $_[0] =~ + /^\s*$|^Total Number of|^Details|^(?:--?|=|\+) Number|^Formula for/; + }; + my $dbh = $self->schema->storage->dbh; -# my $sth = $dbh->prepare(qq{SELECT CCU.CONSTRAINT_NAME, CCU.COLUMN_NAME FROM INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE CCU -# JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS TC ON (CCU.CONSTRAINT_NAME = TC.CONSTRAINT_NAME) -# JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE KCU ON (CCU.CONSTRAINT_NAME = KCU.CONSTRAINT_NAME AND CCU.COLUMN_NAME = KCU.COLUMN_NAME) -# WHERE CCU.TABLE_NAME = '$table' AND CONSTRAINT_TYPE = 'UNIQUE' ORDER BY KCU.ORDINAL_POSITION}); - my $sth = $dbh->prepare(qq{sp_helpconstraint \@objname='$table', \@nomsg='nomsg'}); + + local $dbh->{FetchHashKeyName} = 'NAME_lc'; + + my $sth = $dbh->prepare("sp_helpconstraint $table"); $sth->execute; + my $constraints = $sth->fetchall_arrayref({}); + + my @rels; + + foreach my $constraint (map $_->{definition}, @$constraints) { + my ($local_cols, $remote_table, $remote_cols) = $constraint =~ +/^$table FOREIGN KEY \(([^)]+)\) REFERENCES ([^(]+)\(([^)]+)\)/; + + next unless $local_cols; + + my @local_cols = split /,\s*/, $local_cols; + my @remote_cols = split /,\s*/, $remote_cols; + + push @rels, { + local_columns => \@local_cols, + remote_columns => \@remote_cols, + remote_table => $remote_table, + }; + } + + return \@rels; +} + +sub _table_uniq_info { + no warnings 'uninitialized'; # for presumably XS weirdness with null operations + my ($self, $table) = @_; + + local $SIG{__WARN__} = sub { warn @_ + unless $_[0] =~ /^Formula for Calculation:|^(?:--?|\+|=) Number of (?:self )?references|^Total Number of Referential Constraints|^Details:|^\s*$/ }; + + 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) { - my $type = $row->{constraint_type} || ''; - if ($type =~ /^unique/i) { - my $name = lc $row->{constraint_name}; - push @{$constraints->{$name}}, ( split /,/, lc $row->{constraint_keys} ); + if (exists $row->{constraint_type}) { + my $type = $row->{constraint_type} || ''; + if ($type =~ /^unique/i) { + my $name = $row->{constraint_name}; + push @{$constraints->{$name}}, + ( split /,/, $row->{constraint_keys} ); + } + } else { + my $def = $row->{definition} || next; + next unless $def =~ /^unique/i; + my $name = $row->{name}; + my ($keys) = $def =~ /\((.*)\)/; + $keys =~ s/\s*//g; + my @keys = split /,/ => $keys; + push @{$constraints->{$name}}, @keys; } } @@ -126,14 +218,92 @@ 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, bt.name base_type, ut.name user_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 bt ON c.type = bt.type +LEFT JOIN systypes ut ON c.usertype = ut.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}{user_type} || $info->{$col}{base_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; + + if ($function =~ /^getdate\b/) { + $res->{inflate_datetime} = 1; + } + + delete $res->{size}; + $res->{data_type} = undef; + } + elsif ($default =~ /^DEFAULT \s+ (\S+)/ix) { + my ($constant_default) = $1 =~ /^['"\[\]]?(.*?)['"\[\]]?\z/; + $res->{default_value} = $constant_default; + } + } + + if (my $data_type = $res->{data_type}) { + if ($data_type eq 'int') { + $data_type = $res->{data_type} = 'integer'; + } + elsif ($data_type eq 'decimal') { + $data_type = $res->{data_type} = 'numeric'; + } + + if ($data_type =~ /^(?:text|unitext|image|bigint|integer|smallint|tinyint|real|double|double precision|float|date|time|datetime|smalldatetime|money|smallmoney|timestamp|bit)\z/i) { + delete $res->{size}; + } + elsif ($data_type eq 'numeric') { + my ($prec, $scale) = @{$info->{$col}}{qw/prec scale/}; + + if ($prec == 18 && $scale == 0) { + delete $res->{size}; + } + else { + $res->{size} = [ $prec, $scale ]; + } + } + elsif ($data_type =~ /^(?:unichar|univarchar)\z/i) { + $res->{size} /= 2; + } + } + + if ($data_type eq 'float') { + $res->{data_type} = $info->{$col}{len} <= 4 ? 'real' : 'double precision'; + } + } + + return $result; +} + sub _extra_column_info { - my ($self, $info) = @_; + my ($self, $table, $column, $info, $dbi_info) = @_; my %extra_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) { @@ -145,13 +315,20 @@ sub _extra_column_info { =head1 SEE ALSO +L, L, L, 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 1; +# vim:et sts=4 sw=4 tw=0: