1 package DBIx::Class::PK;
6 use base qw/Class::Data::Inheritable DBIx::Class::SQL/;
8 __PACKAGE__->mk_classdata('_primaries' => {});
12 return join(" AND ", map { "$_ = ?" } keys %{$class->_primaries});
17 return (map { $self->{_column_data}{$_} } keys %{$self->_primaries});
21 my ($class, @cols) = @_;
23 $pri{$_} = {} for @cols;
24 $class->_primaries(\%pri);
28 my ($class, @vals) = @_;
29 my @pk = keys %{$class->_primaries};
30 die "Can't retrieve unless primary columns are defined" unless @pk;
32 if (ref $vals[0] eq 'HASH') {
34 } elsif (@pk == @vals) {
35 my $ret = ($class->retrieve_from_sql($class->_ident_cond, @vals))[0];
36 #warn "$class: ".join(', ', %{$ret->{_column_data}});
41 die "Can't retrieve unless all primary keys are specified"
42 unless (keys %$query >= @pk); # If we check 'em we run afoul of uc/lc
43 # column names etc. Not sure what to do yet
44 my $ret = ($class->search($query))[0];
45 #warn "$class: ".join(', ', %{$ret->{_column_data}});
51 delete $self->{_dirty_columns};
52 $_[0] = $self->retrieve($self->id);
57 die "Can't call id() as a class method" unless ref $self;
58 my @pk = $self->_ident_values;
59 return (wantarray ? @pk : $pk[0]);