package DBIx::DBSchema::Column;
use strict;
-use vars qw(@ISA);
+use vars qw(@ISA $VERSION);
#use Carp;
#use Exporter;
+use DBIx::DBSchema::_util qw(_load_driver _dbh);
#@ISA = qw(Exporter);
@ISA = qw();
+$VERSION = '0.08';
+
=head1 NAME
DBIx::DBSchema::Column - Column objects
use DBIx::DBSchema::Column;
- $column = new DBIx::DBSchema::Column ( $name, $sql_type, '' );
- $column = new DBIx::DBSchema::Column ( $name, $sql_type, 'NULL' );
- $column = new DBIx::DBSchema::Column ( $name, $sql_type, '', $length );
- $column = new DBIx::DBSchema::Column ( $name, $sql_type, 'NULL', $length );
- $column = new DBIx::DBSchema::Column ( $name, $sql_type, 'NULL', $length, $local );
+ #named params with a hashref (preferred)
+ $column = new DBIx::DBSchema::Column ( {
+ 'name' => 'column_name',
+ 'type' => 'varchar'
+ 'null' => 'NOT NULL',
+ 'length' => 64,
+ 'default' => '',
+ 'local' => '',
+ } );
+
+ #list
+ $column = new DBIx::DBSchema::Column ( $name, $sql_type, $nullability, $length, $default, $local );
$name = $column->name;
$column->name( 'name' );
$sql_type = $column->type;
- $column->sql_type( 'sql_type' );
+ $column->type( 'sql_type' );
$null = $column->null;
$column->null( 'NULL' );
$column->length( '10' );
$column->length( '8,2' );
+ $default = $column->default;
+ $column->default( 'Roo' );
+
$sql_line = $column->line;
$sql_line = $column->line($datasrc);
+ $sql_add_column = $column->sql_add_column;
+ $sql_add_column = $column->sql_add_column($datasrc);
+
=head1 DESCRIPTION
DBIx::DBSchema::Column objects represent columns in tables (see
=over 4
-=item new [ NAME [ , SQL_TYPE [ , NULL [ , LENGTH [ , LOCAL ] ] ] ] ]
+=item new HASHREF
-Creates a new DBIx::DBSchema::Column object. NAME is the name of the column.
-SQL_TYPE is the SQL data type. NULL is the nullability of the column (the
-empty string is equivalent to `NOT NULL'). LENGTH is the SQL length of the
-column. LOCAL is reserved for database-specific information.
+=item new [ name [ , type [ , null [ , length [ , default [ , local ] ] ] ] ] ]
+
+Creates a new DBIx::DBSchema::Column object. Takes a hashref of named
+parameters, or a list. B<name> is the name of the column. B<type> is the SQL
+data type. B<null> is the nullability of the column (intrepreted using Perl's
+rules for truth, with one exception: `NOT NULL' is false). B<length> is the
+SQL length of the column. B<default> is the default value of the column.
+B<local> is reserved for database-specific information.
+
+Note: If you pass a scalar reference as the B<default> rather than a scalar value, it will be dereferenced and quoting will be forced off. This can be used to pass SQL functions such as C<$now()> or explicit empty strings as C<''> as
+defaults.
=cut
sub new {
- my($proto,$name,$type,$null,$length,$local)=@_;
+ my $proto = shift;
+ my $class = ref($proto) || $proto;
- #croak "Illegal name: $name" if grep $name eq $_, @reserved_words;
+ my $self;
+ if ( ref($_[0]) ) {
+ $self = shift;
+ } else {
+ $self = { map { $_ => shift } qw(name type null length default local) };
+ }
- $null =~ s/^NOT NULL$//i;
- $null = 'NULL' if $null;
+ #croak "Illegal name: ". $self->{'name'}
+ # if grep $self->{'name'} eq $_, @reserved_words;
- my $class = ref($proto) || $proto;
- my $self = {
- 'name' => $name,
- 'type' => $type,
- 'null' => $null,
- 'length' => $length,
- 'local' => $local,
- };
+ $self->{'null'} =~ s/^NOT NULL$//i;
+ $self->{'null'} = 'NULL' if $self->{'null'};
bless ($self, $class);
}
}
+=item default [ LOCAL ]
+
+Returns or sets the default value.
+
+=cut
+
+sub default {
+ my($self,$value)=@_;
+ if ( defined($value) ) {
+ $self->{'default'} = $value;
+ } else {
+ $self->{'default'};
+ }
+}
+
+
=item local [ LOCAL ]
Returns or sets the database-specific field.
}
}
-=item line [ $datasrc ]
+=item table_obj [ TABLE_OBJ ]
+
+Returns or sets the table object (see L<DBIx::DBSchema::Table>). Typically
+set internally when a column object is added to a table object.
+
+=cut
+
+sub table_obj {
+ my($self,$value)=@_;
+ if ( defined($value) ) {
+ $self->{'table_obj'} = $value;
+ } else {
+ $self->{'table_obj'};
+ }
+}
+
+=item table_name
+
+Returns the table name, or the empty string if this column has not yet been
+assigned to a table.
+
+=cut
+
+sub table_name {
+ my $self = shift;
+ $self->{'table_obj'} ? $self->{'table_obj'}->name : '';
+}
+
+=item line [ DATABASE_HANDLE | DATA_SOURCE [ USERNAME PASSWORD [ ATTR ] ] ]
Returns an SQL column definition.
-If passed a DBI data source such as `DBI:mysql:database' or
+The data source can be specified by passing an open DBI database handle, or by
+passing the DBI data source name, username and password.
+
+Although the username and password are optional, it is best to call this method
+with a database handle or data source including a valid username and password -
+a DBI connection will be opened and the quoting and type mapping will be more
+reliable.
+
+If passed a DBI data source (or handle) such as `DBI:mysql:database' or
`DBI:Pg:dbname=database', will use syntax specific to that database engine.
Currently supported databases are MySQL and PostgreSQL. Non-standard syntax
for other engines (if applicable) may also be supported in the future.
=cut
sub line {
- my($self,$datasrc)=@_;
- my($null)=$self->null;
- if ( $datasrc =~ /^dbi:mysql:/i ) { #yucky mysql hack
- $null ||= "NOT NULL"
+ my($self, $dbh) = ( shift, _dbh(@_) );
+
+ my $driver = $dbh ? _load_driver($dbh) : '';
+
+ my %typemap;
+ %typemap = eval "\%DBIx::DBSchema::DBD::${driver}::typemap" if $driver;
+ my $type = defined( $typemap{uc($self->type)} )
+ ? $typemap{uc($self->type)}
+ : $self->type;
+
+ my $null = $self->null;
+
+ my $default;
+ if ( defined($self->default) && !ref($self->default) && $self->default ne ''
+ && ref($dbh)
+ # false laziness: nicked from FS::Record::_quote
+ && ( $self->default !~ /^\-?\d+(\.\d+)?$/
+ || $type =~ /(char|binary|blob|text)$/i
+ )
+ ) {
+ $default = $dbh->quote($self->default);
+ } else {
+ $default = ref($self->default) ? ${$self->default} : $self->default;
}
- if ( $datasrc =~ /^dbi:pg/i ) { #yucky Pg hack
+
+ #this should be a callback into the driver
+ if ( $driver eq 'mysql' ) { #yucky mysql hack
+ $null ||= "NOT NULL";
+ $self->local('AUTO_INCREMENT') if uc($self->type) eq 'SERIAL';
+ } elsif ( $driver =~ /^(?:Pg|SQLite)$/ ) { #yucky Pg/SQLite hack
$null ||= "NOT NULL";
$null =~ s/^NULL$//;
}
+
join(' ',
$self->name,
- $self->type. ( $self->length ? '('.$self->length.')' : '' ),
+ $type. ( ( defined($self->length) && $self->length )
+ ? '('.$self->length.')'
+ : ''
+ ),
$null,
- ( ( $datasrc =~ /^dbi:mysql:/i )
+ ( ( defined($default) && $default ne '' )
+ ? 'DEFAULT '. $default
+ : ''
+ ),
+ ( ( $driver eq 'mysql' && defined($self->local) )
? $self->local
: ''
),
);
+
+}
+
+=item sql_add_column [ DBH ]
+
+Returns a list of SQL statements to add this column to an existing table. (To
+create a new table, see L<DBIx::DBSchema::Table/sql_create_table> instead.)
+
+The data source can be specified by passing an open DBI database handle, or by
+passing the DBI data source name, username and password.
+
+Although the username and password are optional, it is best to call this method
+with a database handle or data source including a valid username and password -
+a DBI connection will be opened and the quoting and type mapping will be more
+reliable.
+
+If passed a DBI data source (or handle) such as `DBI:Pg:dbname=database', will
+use PostgreSQL-specific syntax. Non-standard syntax for other engines (if
+applicable) may also be supported in the future.
+
+=cut
+
+sub sql_add_column {
+ my($self, $dbh) = ( shift, _dbh(@_) );
+
+ die "$self: this column is not assigned to a table"
+ unless $self->table_name;
+
+ my $driver = $dbh ? _load_driver($dbh) : '';
+
+ my @after_add = ();
+
+ my $real_type = '';
+ if ( $driver eq 'Pg' && $self->type eq 'serial' ) {
+ $real_type = 'serial';
+ $self->type('int');
+
+ push @after_add, sub {
+ my($table, $column) = @_;
+
+ #needs more work for old Pg
+
+ my $nextval;
+ if ( $dbh->{'pg_server_version'} > 70300 ) {
+ $nextval = "nextval('public.${table}_${column}_seq'::text)";
+ } else {
+ $nextval = "nextval('${table}_${column}_seq'::text)";
+ }
+
+ (
+ "ALTER TABLE $table ALTER COLUMN $column SET DEFAULT $nextval",
+ "CREATE SEQUENCE ${table}_${column}_seq",
+ "UPDATE $table SET $column = $nextval WHERE $column IS NULL",
+ #"ALTER TABLE $table ALTER $column SET NOT NULL",
+ );
+
+ };
+
+ }
+
+ my $real_null = undef;
+ if ( $driver eq 'Pg' && ! $self->null ) {
+ $real_null = $self->null;
+ $self->null('NULL');
+
+ #if ( $dbh->{'pg_server_version'} > 70300 ) { #this seemed to work on 7.3
+ if ( $dbh->{'pg_server_version'} > 70400 ) { #after all...
+
+ push @after_add, sub {
+ my($table, $column) = @_;
+ "ALTER TABLE $table ALTER $column SET NOT NULL";
+ };
+
+ } else {
+
+ push @after_add, sub {
+ my($table, $column) = @_;
+ "UPDATE pg_attribute SET attnotnull = TRUE ".
+ " WHERE attname = '$column' ".
+ " AND attrelid = ( SELECT oid FROM pg_class WHERE relname = '$table' )";
+ };
+
+ }
+
+ }
+
+ my @r = ();
+ my $table = $self->table_name;
+ my $column = $self->name;
+
+ push @r, "ALTER TABLE $table ADD COLUMN ". $self->line($dbh);
+
+ push @r, &{$_}($table, $column) foreach @after_add;
+
+ push @r, "ALTER TABLE $table ADD PRIMARY KEY ( ".
+ $self->table_obj->primary_key. " )"
+ if $self->name eq $self->table_obj->primary_key;
+
+ $self->type($real_type) if $real_type;
+ $self->null($real_null) if defined $real_null;
+
+ @r;
+
+}
+
+=item sql_alter_column PROTOTYPE_COLUMN [ DATABASE_HANDLE | DATA_SOURCE [ USERNAME PASSWORD [ ATTR ] ] ]
+
+Returns a list of SQL statements to alter this column so that it is identical
+to the provided prototype column, also a DBIx::DBSchema::Column object.
+
+ #Optionally, the data source can be specified by passing an open DBI database
+ #handle, or by passing the DBI data source name, username and password.
+ #
+ #If passed a DBI data source (or handle) such as `DBI:Pg:dbname=database', will
+ #use PostgreSQL-specific syntax. Non-standard syntax for other engines (if
+ #applicable) may also be supported in the future.
+ #
+ #If not passed a data source (or handle), or if there is no driver for the
+ #specified database, will attempt to use generic SQL syntax.
+
+
+Or should, someday. Right now it knows how to change NOT NULL into NULL and
+vice-versa.
+
+=cut
+
+sub sql_alter_column {
+ my( $self, $new, $dbh ) = ( shift, shift, _dbh(@_) );
+
+ my $table = $self->table_name;
+ die "$self: this column is not assigned to a table"
+ unless $table;
+
+ my $name = $self->name;
+
+ my $driver = $dbh ? _load_driver($dbh) : '';
+
+ my @r = ();
+
+ # change the name...
+
+ # change the type...
+
+ # change nullability from NOT NULL to NULL
+ if ( ! $self->null && $new->null ) {
+
+ if ( $driver eq 'Pg' && $dbh->{'pg_server_version'} < 70300 ) {
+ push @r, "UPDATE pg_attribute SET attnotnull = FALSE
+ WHERE attname = '$name'
+ AND attrelid = ( SELECT oid FROM pg_class
+ WHERE relname = '$table'
+ )";
+ } else {
+ push @r, "ALTER TABLE $table ALTER COLUMN $name DROP NOT NULL";
+ }
+ }
+
+ # change nullability from NULL to NOT NULL...
+ # this one could be more complicated, need to set a DEFAULT value and update
+ # the table first...
+ if ( $self->null && ! $new->null ) {
+
+ if ( $driver eq 'Pg' && $dbh->{'pg_server_version'} < 70300 ) {
+ push @r, "UPDATE pg_attribute SET attnotnull = TRUE
+ WHERE attname = '$name'
+ AND attrelid = ( SELECT oid FROM pg_class
+ WHERE relname = '$table'
+ )";
+ } else {
+ push @r, "ALTER TABLE $table ALTER COLUMN $name SET NOT NULL";
+ }
+ }
+
+ # change other stuff...
+
+ @r;
+
}
=back
=head1 COPYRIGHT
-Copyright (c) 2000 Ivan Kohler
-Copyright (c) 2000 Mail Abuse Prevention System LLC
+Copyright (c) 2000-2006 Ivan Kohler
All rights reserved.
This program is free software; you can redistribute it and/or modify it under
the same terms as Perl itself.
=head1 BUGS
-line() has database-specific foo that probably ought to be abstracted into
-the DBIx::DBSchema:DBD:: modules.
+Better documentation is needed for sql_add_column
+
+line() and sql_add_column() hav database-specific foo that should be abstracted
+into the DBIx::DBSchema:DBD:: modules.
=head1 SEE ALSO
projects / DBIx-DBSchema.git / blobdiff