X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FPK.pm;h=cf8a194cc99f7a1fb51b82594c4cdf680d734163;hb=8c60e1a7be16f71332b2efa55a3ee211e8384bce;hp=ddd37b9f0c9bcdd9186d800f391bb1b615dbe570;hpb=1edd17220a3f0fa2768084572d8ca57cfc2a2fcc;p=dbsrgits%2FDBIx-Class.git diff --git a/lib/DBIx/Class/PK.pm b/lib/DBIx/Class/PK.pm index ddd37b9..cf8a194 100644 --- a/lib/DBIx/Class/PK.pm +++ b/lib/DBIx/Class/PK.pm @@ -2,13 +2,10 @@ package DBIx::Class::PK; use strict; use warnings; -use Tie::IxHash; use base qw/DBIx::Class::Row/; -__PACKAGE__->mk_classdata('_primaries' => {}); - -=head1 NAME +=head1 NAME DBIx::Class::PK - Primary Key class @@ -16,140 +13,88 @@ DBIx::Class::PK - Primary Key class =head1 DESCRIPTION -This class represents methods handling primary keys -and depending on them. +This class contains methods for handling primary keys and methods +depending on them. =head1 METHODS -=over 4 - -=cut - -sub _ident_cond { - my ($class) = @_; - return join(" AND ", map { "$_ = ?" } keys %{$class->_primaries}); -} - -sub _ident_values { - my ($self) = @_; - return (map { $self->{_column_data}{$_} } keys %{$self->_primaries}); -} - -=item set_primary_key <@cols> - -define one or more columns as primary key for this class - =cut -sub set_primary_key { - my ($class, @cols) = @_; - my %pri; - tie %pri, 'Tie::IxHash', map { $_ => {} } @cols; - $class->_primaries(\%pri); -} - -=item find +=head2 id -Finds columns based on the primary key(s). +Returns the primary key(s) for a row. Can't be called as +a class method. =cut -sub find { - my ($class, @vals) = @_; - my $attrs = (@vals > 1 && ref $vals[$#vals] eq 'HASH' ? pop(@vals) : {}); - my @pk = keys %{$class->_primaries}; - $class->throw( "Can't find unless primary columns are defined" ) - unless @pk; - my $query; - if (ref $vals[0] eq 'HASH') { - $query = $vals[0]; - } elsif (@pk == @vals) { - $query = {}; - @{$query}{@pk} = @vals; - #my $ret = ($class->search_literal($class->_ident_cond, @vals, $attrs))[0]; - #warn "$class: ".join(', ', %{$ret->{_column_data}}); - #return $ret; - } else { - $query = {@vals}; - } - $class->throw( "Can't find unless all primary keys are specified" ) - unless (keys %$query >= @pk); # If we check 'em we run afoul of uc/lc - # column names etc. Not sure what to do yet - #return $class->search($query)->next; - my @cols = $class->_select_columns; - my @row = $class->storage->select_single($class->_table_name, \@cols, $query); - return (@row ? $class->_row_to_object(\@cols, \@row) : ()); +sub id { + my ($self) = @_; + $self->throw_exception( "Can't call id() as a class method" ) + unless ref $self; + my @pk = $self->_ident_values; + return (wantarray ? @pk : $pk[0]); } -=item discard_changes - -Roll back changes that hasn't been comitted to the database. - -=cut - -sub discard_changes { +sub _ident_values { my ($self) = @_; - delete $self->{_dirty_columns}; - return unless $self->in_storage; # Don't reload if we aren't real! - my ($reload) = $self->find($self->id); - unless ($reload) { # If we got deleted in the mean-time - $self->in_storage(0); - return $self; - } - delete @{$self}{keys %$self}; - @{$self}{keys %$reload} = values %$reload; - return $self; + return (map { $self->{_column_data}{$_} } $self->primary_columns); } -=item id +=head2 ID -returns the primary key(s) for the current row. Can't be called as -a class method. +Returns a unique id string identifying a row object by primary key. +Used by L and +L. -=cut +=over -sub id { - my ($self) = @_; - $self->throw( "Can't call id() as a class method" ) unless ref $self; - my @pk = $self->_ident_values; - return (wantarray ? @pk : $pk[0]); -} +=item WARNING -=item primary_columns +The default C<_create_ID> method used by this function orders the returned +values by the alphabetical order of the primary column names, B +the L method, which follows the same order in which columns were fed +to L. -read-only accessor which returns a list of primary keys. +=back =cut -sub primary_columns { - return keys %{shift->_primaries}; -} - sub ID { my ($self) = @_; - $self->throw( "Can't call ID() as a class method" ) unless ref $self; + $self->throw_exception( "Can't call ID() as a class method" ) + unless ref $self; return undef unless $self->in_storage; - return $self->_create_ID(map { $_ => $self->{_column_data}{$_} } keys %{$self->_primaries}); + return $self->_create_ID(map { $_ => $self->{_column_data}{$_} } + $self->primary_columns); } sub _create_ID { - my ($class,%vals) = @_; + my ($self,%vals) = @_; return undef unless 0 == grep { !defined } values %vals; - $class = ref $class || $class; - return join '|', $class, map { $_ . '=' . $vals{$_} } sort keys %vals; + return join '|', ref $self || $self, $self->result_source->name, + map { $_ . '=' . $vals{$_} } sort keys %vals; } +=head2 ident_condition + + my $cond = $result_source->ident_condition(); + + my $cond = $result_source->ident_condition('alias'); + +Produces a condition hash to locate a row based on the primary key(s). + +=cut + sub ident_condition { - my ($self) = @_; + my ($self, $alias) = @_; my %cond; - $cond{$_} = $self->get_column($_) for $self->primary_columns; + my $prefix = defined $alias ? $alias.'.' : ''; + $cond{$prefix.$_} = $self->get_column($_) for $self->primary_columns; return \%cond; } 1; -=back - =head1 AUTHORS Matt S. Trout