X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FResultSet.pm;h=8f4a5ba7997d9d37300b68cbb06ed4e7ef312e69;hb=a28009914a3e447af53737d503b1953160d146c9;hp=111e0381cee0fd5e7b1f5c4efac0bf337cb3cf96;hpb=55cd67557d46ab8957c24d5e77f85f94687d42ff;p=dbsrgits%2FDBIx-Class-Historic.git diff --git a/lib/DBIx/Class/ResultSet.pm b/lib/DBIx/Class/ResultSet.pm index 111e038..8f4a5ba 100644 --- a/lib/DBIx/Class/ResultSet.pm +++ b/lib/DBIx/Class/ResultSet.pm @@ -34,7 +34,7 @@ In the examples below, the following table classes are used: package MyApp::Schema::Artist; use base qw/DBIx::Class/; - __PACKAGE__->load_components(qw/Core/) + __PACKAGE__->load_components(qw/Core/); __PACKAGE__->table('artist'); __PACKAGE__->add_columns(qw/artistid name/); __PACKAGE__->set_primary_key('artistid'); @@ -91,8 +91,6 @@ sub new { if ($attrs->{page}) { $attrs->{rows} ||= 10; - $attrs->{offset} ||= 0; - $attrs->{offset} += ($attrs->{rows} * ($attrs->{page} - 1)); } $attrs->{alias} ||= 'me'; @@ -937,7 +935,9 @@ sub count { my $count = $self->_count; return 0 unless $count; - $count -= $self->{attrs}{offset} if $self->{attrs}{offset}; + # need to take offset from resolved attrs + + $count -= $self->{_attrs}{offset} if $self->{_attrs}{offset}; $count = $self->{attrs}{rows} if $self->{attrs}{rows} and $self->{attrs}{rows} < $count; return $count; @@ -1244,7 +1244,7 @@ sub delete_all { =over 4 -=item Arguments: $source_name, \@data; +=item Arguments: \@data; =back @@ -1280,18 +1280,18 @@ Example: Assuming an Artist Class that has many CDs Classes relating: ## Array Context Example my ($ArtistOne, $ArtistTwo, $ArtistThree) = $Artist_rs->populate([ { name => "Artist One"}, - { name => "Artist Two"}, - { name => "Artist Three", cds=> [ - { title => "First CD", year => 2007}, - { title => "Second CD", year => 2008}, - ]} + { name => "Artist Two"}, + { name => "Artist Three", cds=> [ + { title => "First CD", year => 2007}, + { title => "Second CD", year => 2008}, + ]} ]); print $ArtistOne->name; ## response is 'Artist One' print $ArtistThree->cds->count ## reponse is '2' =cut -use Data::Dump qw/dump/; + sub populate { my ($self, $data) = @_; @@ -1304,35 +1304,35 @@ sub populate { } else { my ($first, @rest) = @$data; - my @names = grep {!ref $first->{$_}} keys %$first; + my @names = grep {!ref $first->{$_}} keys %$first; my @rels = grep { $self->result_source->has_relationship($_) } keys %$first; - my @pks = $self->result_source->primary_columns; - - ## do the belongs_to relationships - foreach my $index (0..$#{@$data}) - { - foreach my $rel (@rels) - { - next unless $data->[$index]->{$rel} && ref $data->[$index]->{$rel} eq "HASH"; - - my $result = $self->related_resultset($rel)->create($data->[$index]->{$rel}); - - my ($reverse) = keys %{$self->result_source->reverse_relationship_info($rel)}; - - my $related = $result->result_source->resolve_condition( - - $result->result_source->relationship_info($reverse)->{cond}, - $self, - $result, - ); - - delete $data->[$index]->{$rel}; - $data->[$index] = {%{$data->[$index]}, %$related}; - - push @names, keys %$related if $index == 0; - } - } - + my @pks = $self->result_source->primary_columns; + + ## do the belongs_to relationships + foreach my $index (0..$#$data) { + if( grep { !defined $data->[$index]->{$_} } @pks ) { + my @ret = $self->populate($data); + return; + } + + foreach my $rel (@rels) { + next unless $data->[$index]->{$rel} && ref $data->[$index]->{$rel} eq "HASH"; + my $result = $self->related_resultset($rel)->create($data->[$index]->{$rel}); + my ($reverse) = keys %{$self->result_source->reverse_relationship_info($rel)}; + my $related = $result->result_source->resolve_condition( + $result->result_source->relationship_info($reverse)->{cond}, + $self, + $result, + ); + + delete $data->[$index]->{$rel}; + $data->[$index] = {%{$data->[$index]}, %$related}; + + push @names, keys %$related if $index == 0; + } + } + + ## do bulk insert on current row my @values = map { [ map { defined $_ ? $_ : $self->throw_exception("Undefined value for column!") @@ -1345,15 +1345,17 @@ sub populate { \@values, ); - ## do the has_many relationships + ## do the has_many relationships foreach my $item (@$data) { foreach my $rel (@rels) { next unless $item->{$rel} && ref $item->{$rel} eq "ARRAY"; - my $parent = $self->find(map {{$_=>$item->{$_}} } @pks) || next; + my $parent = $self->find(map {{$_=>$item->{$_}} } @pks) + || $self->throw_exception('Cannot find the relating object.'); + my $child = $parent->$rel; - + my $related = $child->result_source->resolve_condition( $parent->result_source->relationship_info($rel)->{cond}, $child, @@ -1893,6 +1895,11 @@ sub _resolved_attrs { } $attrs->{collapse} = $collapse; + if ($attrs->{page}) { + $attrs->{offset} ||= 0; + $attrs->{offset} += ($attrs->{rows} * ($attrs->{page} - 1)); + } + return $self->{_attrs} = $attrs; }