X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI%2FMSSQL.pm;h=a2cc9166975aaaa9fbb83053f02221e8de575165;hb=6b1d4f76b756e4b4119153a1f1e8a7bd59ad4e87;hp=fa21dc4dc53bf799cfca12034890c3d26685503a;hpb=ea8d5d7c7145cf9eba916c857d4f67839fd64533;p=dbsrgits%2FDBIx-Class-Schema-Loader.git diff --git a/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm b/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm index fa21dc4..a2cc916 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm @@ -6,7 +6,7 @@ use base 'DBIx::Class::Schema::Loader::DBI::Sybase::Common'; use Carp::Clan qw/^DBIx::Class/; use Class::C3; -our $VERSION = '0.05003'; +our $VERSION = '0.07001'; =head1 NAME @@ -23,18 +23,77 @@ L. See L and L for usage information. +=head1 CASE SENSITIVITY + +Most MSSQL databases use C (case-insensitive) collation, for this reason +generated column names are lower-cased as this makes them easier to work with +in L. + +We attempt to detect the database collation at startup, and set the column +lowercasing behavior accordingly, as lower-cased column names do not work on +case-sensitive databases. + +To manually control case-sensitive mode, put: + + preserve_case => 1|0 + +in your Loader options. + +See L. + +B this option used to be called C, but has +been renamed to a more generic option. + =cut +sub _setup { + my $self = shift; + + $self->next::method(@_); + + return if defined $self->preserve_case; + + my $dbh = $self->schema->storage->dbh; + + # We use the sys.databases query for the general case, and fallback to + # databasepropertyex() if for some reason sys.databases is not available, + # which does not work over DBD::ODBC with unixODBC+FreeTDS. + # + # XXX why does databasepropertyex() not work over DBD::ODBC ? + # + # more on collations here: http://msdn.microsoft.com/en-us/library/ms143515.aspx + my ($collation_name) = + eval { $dbh->selectrow_array('SELECT collation_name FROM sys.databases WHERE name = DB_NAME()') } + || eval { $dbh->selectrow_array("SELECT CAST(databasepropertyex(DB_NAME(), 'Collation') AS VARCHAR)") }; + + if (not $collation_name) { + warn <<'EOF'; + +WARNING: MSSQL Collation detection failed. Defaulting to case-insensitive mode. +Override the 'preserve_case' attribute in your Loader options if needed. + +See 'preserve_case' in +perldoc DBIx::Class::Schema::Loader::Base +EOF + $self->preserve_case(0); + return; + } + + my $case_sensitive = $collation_name =~ /_(?:CS|BIN2?)(?:_|\z)/; + + $self->preserve_case($case_sensitive ? 1 : 0); +} + sub _tables_list { my ($self, $opts) = @_; my $dbh = $self->schema->storage->dbh; my $sth = $dbh->prepare(<<'EOF'); SELECT t.table_name -FROM information_schema.tables t -WHERE lower(t.table_schema) = ? +FROM INFORMATION_SCHEMA.TABLES t +WHERE t.table_schema = ? EOF - $sth->execute(lc $self->db_schema); + $sth->execute($self->db_schema); my @tables = map @$_, @{ $sth->fetchall_arrayref }; @@ -50,7 +109,7 @@ sub _table_pk_info { my @keydata; while (my $row = $sth->fetchrow_hashref) { - push @keydata, lc $row->{COLUMN_NAME}; + push @keydata, $self->_lc($row->{COLUMN_NAME}); } return \@keydata; @@ -68,8 +127,8 @@ sub _table_fk_info { while (my $row = eval { $sth->fetchrow_hashref }) { my $fk = $row->{FK_NAME}; - push @{$local_cols->{$fk}}, lc $row->{FKCOLUMN_NAME}; - push @{$remote_cols->{$fk}}, lc $row->{PKCOLUMN_NAME}; + push @{$local_cols->{$fk}}, $self->_lc($row->{FKCOLUMN_NAME}); + push @{$remote_cols->{$fk}}, $self->_lc($row->{PKCOLUMN_NAME}); $remote_table->{$fk} = $row->{PKTABLE_NAME}; } @@ -92,16 +151,16 @@ sub _table_uniq_info { 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 lower(ccu.table_name) = @{[ $dbh->quote(lc $table) ]} AND constraint_type = 'UNIQUE' ORDER BY kcu.ordinal_position +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 = @{[ $dbh->quote($table) ]} AND constraint_type = 'UNIQUE' ORDER BY kcu.ordinal_position }); $sth->execute; my $constraints; while (my $row = $sth->fetchrow_hashref) { my $name = $row->{constraint_name}; - my $col = lc $row->{column_name}; + my $col = $self->_lc($row->{column_name}); push @{$constraints->{$name}}, $col; } @@ -118,11 +177,12 @@ sub _columns_info_for { while (my ($col, $info) = each %$result) { my $dbh = $self->schema->storage->dbh; +# find identities my $sth = $dbh->prepare(qq{ SELECT column_name -FROM information_schema.columns -WHERE columnproperty(object_id(@{[ $dbh->quote(lc $table) ]}, 'U'), @{[ $dbh->quote(lc $col) ]}, 'IsIdentity') = 1 -AND lower(table_name) = @{[ $dbh->quote(lc $table) ]} AND lower(column_name) = @{[ $dbh->quote(lc $col) ]} +FROM INFORMATION_SCHEMA.COLUMNS +WHERE columnproperty(object_id(@{[ $dbh->quote($table) ]}, 'U'), @{[ $dbh->quote($col) ]}, 'IsIdentity') = 1 +AND table_name = @{[ $dbh->quote($table) ]} AND column_name = @{[ $dbh->quote($col) ]} }); if (eval { $sth->execute; $sth->fetchrow_array }) { $info->{is_auto_increment} = 1; @@ -130,11 +190,79 @@ AND lower(table_name) = @{[ $dbh->quote(lc $table) ]} AND lower(column_name) = @ delete $info->{size}; } +# fix types + if ($info->{data_type} eq 'int') { + $info->{data_type} = 'integer'; + } + elsif ($info->{data_type} eq 'timestamp') { + $info->{inflate_datetime} = 0; + } + elsif ($info->{data_type} =~ /^(?:numeric|decimal)\z/) { + if (ref($info->{size}) && $info->{size}[0] == 18 && $info->{size}[1] == 0) { + delete $info->{size}; + } + } + elsif ($info->{data_type} eq 'float') { + $info->{data_type} = 'double precision'; + } + elsif ($info->{data_type} =~ /^(?:small)?datetime\z/) { + # fixup for DBD::Sybase + if ($info->{default_value} && $info->{default_value} eq '3') { + delete $info->{default_value}; + } + } + elsif ($info->{data_type} eq 'datetimeoffset') { + $info->{size} = { + 26 => 0, + 28 => 1, + 29 => 2, + 30 => 3, + 31 => 4, + 32 => 5, + 33 => 6, + 34 => 7, + }->{$info->{size}}; + + delete $info->{size} if $info->{size} == 7; + } + elsif ($info->{data_type} eq 'datetime2') { + $info->{size} = { + 19 => 0, + 21 => 1, + 22 => 2, + 23 => 3, + 24 => 4, + 25 => 5, + 26 => 6, + 27 => 7, + }->{$info->{size}}; + + delete $info->{size} if $info->{size} == 7; + } + elsif ($info->{data_type} eq 'time') { + $info->{size} = { + 8 => 0, + 10 => 1, + 11 => 2, + 12 => 3, + 13 => 4, + 14 => 5, + 15 => 6, + 16 => 7, + }->{$info->{size}}; + + delete $info->{size} if $info->{size} == 7; + } + + if ($info->{data_type} !~ /^(?:n?char|n?varchar|binary|varbinary|numeric|decimal|float|datetime(?:2|offset)|time)\z/) { + delete $info->{size}; + } + # get default $sth = $dbh->prepare(qq{ SELECT column_default -FROM information_schema.columns -wHERE lower(table_name) = @{[ $dbh->quote(lc $table) ]} AND lower(column_name) = @{[ $dbh->quote(lc $col) ]} +FROM INFORMATION_SCHEMA.COLUMNS +wHERE table_name = @{[ $dbh->quote($table) ]} AND column_name = @{[ $dbh->quote($col) ]} }); my ($default) = eval { $sth->execute; $sth->fetchrow_array }; @@ -147,6 +275,13 @@ wHERE lower(table_name) = @{[ $dbh->quote(lc $table) ]} AND lower(column_name) = $info->{default_value} = $default =~ /^['(] (.*) [)']\z/x ? $1 : $default =~ /^\d/ ? $default : \$default; + + if ((eval { lc ${ $info->{default_value} } }||'') eq 'getdate()') { + ${ $info->{default_value} } = 'current_timestamp'; + + my $getdate = 'getdate()'; + $info->{original}{default_value} = \$getdate; + } } }