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
'type' => 'varchar'
'null' => 'NOT NULL',
'length' => 64,
- 'default' => '
+ 'default' => '',
'local' => '',
} );
$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
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 {
}
}
+=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.
=cut
sub line {
- my($self,$dbh) = (shift, shift);
+ my($self, $dbh) = ( shift, _dbh(@_) );
+
+ my $driver = $dbh ? _load_driver($dbh) : '';
- my $created_dbh = 0;
- unless ( ref($dbh) || ! @_ ) {
- $dbh = DBI->connect( $dbh, @_ ) or die $DBI::errstr;
- my $gratuitous = $DBI::errstr; #surpress superfluous `used only once' error
- $created_dbh = 1;
- }
-
- my $driver = DBIx::DBSchema::_load_driver($dbh);
my %typemap;
%typemap = eval "\%DBIx::DBSchema::DBD::${driver}::typemap" if $driver;
my $type = defined( $typemap{uc($self->type)} )
my $null = $self->null;
my $default;
- if ( defined($self->default) && $self->default ne ''
+ if ( defined($self->default) && !ref($self->default) && $self->default ne ''
&& ref($dbh)
# false laziness: nicked from FS::Record::_quote
&& ( $self->default !~ /^\-?\d+(\.\d+)?$/
) {
$default = $dbh->quote($self->default);
} else {
- $default = $self->default;
+ $default = ref($self->default) ? ${$self->default} : $self->default;
}
#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 eq 'Pg' ) { #yucky Pg hack
+ } elsif ( $driver =~ /^(?:Pg|SQLite)$/ ) { #yucky Pg/SQLite hack
$null ||= "NOT NULL";
$null =~ s/^NULL$//;
}
- my @r = join(' ',
+ join(' ',
$self->name,
- $type.
- ( defined($self->length) && $self->length ? '('.$self->length.')' : '' ),
+ $type. ( ( defined($self->length) && $self->length )
+ ? '('.$self->length.')'
+ : ''
+ ),
$null,
( ( defined($default) && $default ne '' )
? 'DEFAULT '. $default
: ''
),
- ( ( $driver eq 'mysql' )
+ ( ( $driver eq 'mysql' && defined($self->local) )
? $self->local
: ''
),
);
- $dbh->disconnect if $created_dbh;
+
+}
+
+=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;
}
=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