X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FPK.pm;h=e420ac62497daf7b861e1bc51f14ad2a153f3f14;hb=656796f2088da66cc80f4eb127c39c923ef3c1dd;hp=fef88a08084e5c57f904f63b104d643e148f2515;hpb=4a07648ace2ace5b878c63aec52b7a30c1432b4d;p=dbsrgits%2FDBIx-Class.git diff --git a/lib/DBIx/Class/PK.pm b/lib/DBIx/Class/PK.pm index fef88a0..e420ac6 100644 --- a/lib/DBIx/Class/PK.pm +++ b/lib/DBIx/Class/PK.pm @@ -44,23 +44,23 @@ sub set_primary_key { $class->_primaries(\%pri); } -sub retrieve { +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 retrieve unless primary columns are defined" ) + $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) { - my $ret = ($class->retrieve_from_sql($class->_ident_cond, @vals, $attrs))[0]; + 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 retrieve unless all primary keys are specified" ) + $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 my $ret = ($class->search($query))[0]; @@ -72,7 +72,7 @@ sub discard_changes { my ($self) = @_; delete $self->{_dirty_columns}; return unless $self->in_database; # Don't reload if we aren't real! - my ($reload) = $self->retrieve($self->id); + my ($reload) = $self->find($self->id); unless ($reload) { # If we got deleted in the mean-time $self->in_database(0); return $self; @@ -91,6 +91,10 @@ sub id { return (wantarray ? @pk : $pk[0]); } +sub primary_columns { + return keys %{shift->_primaries}; +} + 1; =back