X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI%2FMSSQL.pm;h=1b9ce7d31a21e04fde5d490fe492a97c074345c8;hb=8c41c3ce6f1db7b202d3465f84e0ba3e4d55e820;hp=8fbba248ba027f93387d3c825fe44e57833471ce;hpb=3559ae79e6819ac25ba7343d20156cc8a492e827;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 8fbba24..1b9ce7d 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI/MSSQL.pm @@ -6,11 +6,7 @@ use base 'DBIx::Class::Schema::Loader::DBI::Sybase::Common'; use Carp::Clan qw/^DBIx::Class/; use Class::C3; -__PACKAGE__->mk_group_accessors('simple', qw/ - case_sensitive_collation -/); - -our $VERSION = '0.05003'; +our $VERSION = '0.07000'; =head1 NAME @@ -27,36 +23,65 @@ L. See L and L for usage information. -=cut +=head1 CASE SENSITIVITY -sub _is_case_sensitive { - my $self = shift; +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. - return $self->case_sensitive_collation ? 1 : 0; -} +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; + $self->next::method(@_); - my $dbh = $self->schema->storage->dbh; + return if defined $self->preserve_case; - my ($collation_name) = $dbh->selectrow_array(<<'EOS'); -SELECT collation_name -FROM sys.databases -WHERE name = DB_NAME() -EOS - - my ($sensitivity) = $collation_name =~ /(C\w)_[A-z]+\z/; + my $dbh = $self->schema->storage->dbh; - $self->case_sensitive_collation($sensitivity eq 'CS' ? 1 : 0); -} + # 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; + } -sub _lc { - my ($self, $name) = @_; + my $case_sensitive = $collation_name =~ /_(?:CS|BIN2?)(?:_|\z)/; - return $self->case_sensitive_collation ? $name : lc($name); + $self->preserve_case($case_sensitive ? 1 : 0); } sub _tables_list { @@ -65,10 +90,10 @@ sub _tables_list { 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 }; @@ -126,10 +151,10 @@ 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; @@ -152,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; @@ -164,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 }; @@ -181,6 +275,10 @@ 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'; + } } }