X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FResultSet.pm;h=2b347ed3e473c2dc25abf51dcbf89facdbb28224;hb=c079518f7bbe43998aae31cead6127b3b85be92e;hp=7679e1138a475d7cfbe678c84d323a1daf5f361b;hpb=a7bf36a241a3fa7b15517058b060a683d8f60eb2;p=dbsrgits%2FDBIx-Class-Historic.git diff --git a/lib/DBIx/Class/ResultSet.pm b/lib/DBIx/Class/ResultSet.pm index 7679e11..2b347ed 100644 --- a/lib/DBIx/Class/ResultSet.pm +++ b/lib/DBIx/Class/ResultSet.pm @@ -196,44 +196,44 @@ call it as C. sub search { my $self = shift; - - my $rs; - if( @_ ) { - my $attrs = { %{$self->{attrs}} }; - my $having = delete $attrs->{having}; - $attrs = { %$attrs, %{ pop(@_) } } if @_ > 1 and ref $_[$#_] eq 'HASH'; - - my $where = (@_ - ? ((@_ == 1 || ref $_[0] eq "HASH") - ? shift - : ((@_ % 2) - ? $self->throw_exception( - "Odd number of arguments to search") - : {@_})) - : undef()); - if (defined $where) { - $attrs->{where} = (defined $attrs->{where} - ? { '-and' => - [ map { ref $_ eq 'ARRAY' ? [ -or => $_ ] : $_ } - $where, $attrs->{where} ] } - : $where); - } - - if (defined $having) { - $attrs->{having} = (defined $attrs->{having} - ? { '-and' => - [ map { ref $_ eq 'ARRAY' ? [ -or => $_ ] : $_ } - $having, $attrs->{having} ] } - : $having); - } + my $attrs = { %{$self->{attrs}} }; + my $having = delete $attrs->{having}; + $attrs = { %$attrs, %{ pop(@_) } } if @_ > 1 and ref $_[$#_] eq 'HASH'; + + my $where = (@_ + ? ((@_ == 1 || ref $_[0] eq "HASH") + ? shift + : ((@_ % 2) + ? $self->throw_exception( + "Odd number of arguments to search") + : {@_})) + : undef()); + if (defined $where) { + $attrs->{where} = (defined $attrs->{where} + ? { '-and' => + [ map { ref $_ eq 'ARRAY' ? [ -or => $_ ] : $_ } + $where, $attrs->{where} ] } + : $where); + } - $rs = (ref $self)->new($self->result_source, $attrs); + if (defined $having) { + $attrs->{having} = (defined $attrs->{having} + ? { '-and' => + [ map { ref $_ eq 'ARRAY' ? [ -or => $_ ] : $_ } + $having, $attrs->{having} ] } + : $having); } - else { - $rs = $self; - $rs->reset; + + my $rs = (ref $self)->new($self->result_source, $attrs); + + unless (@_) { # no search, effectively just a clone + my $rows = $self->get_cache; + if( @{$rows} ) { + $rs->set_cache($rows); + } } + return (wantarray ? $rs->all : $rs); } @@ -312,16 +312,19 @@ sub find { ? $self->result_source->unique_constraint_columns($attrs->{key}) : $self->result_source->primary_columns; - my %hash; + my $hash; if (ref $_[0] eq 'HASH') { - %hash = %{ $_[0] }; + $hash = { %{$_[0]} }; } elsif (@_ == @cols) { - @hash{@cols} = @_; + $hash = {}; + @{$hash}{@cols} = @_; } else { - # Hack for CDBI queries - %hash = @_; + $self->throw_exception( + "Arguments to find must be a hashref or match the number of columns in the " + . exists $attrs->{key} ? "$attrs->{key} unique constraint" : "primary key" + ); } # Check the hash we just parsed against our source's unique constraints @@ -332,21 +335,21 @@ sub find { "Can't find unless a primary key or unique constraint is defined" ) unless @constraint_names; - my @unique_hashes; + my @unique_queries; foreach my $name (@constraint_names) { my @unique_cols = $self->result_source->unique_constraint_columns($name); - my %unique_hash = $self->_unique_hash(\%hash, \@unique_cols); + my $unique_query = $self->_build_unique_query($hash, \@unique_cols); # Add the ResultSet's alias - foreach my $key (grep { ! m/\./ } keys %unique_hash) { - $unique_hash{"$self->{attrs}{alias}.$key"} = delete $unique_hash{$key}; + foreach my $key (grep { ! m/\./ } keys %$unique_query) { + $unique_query->{"$self->{attrs}{alias}.$key"} = delete $unique_query->{$key}; } - push @unique_hashes, \%unique_hash if %unique_hash; + push @unique_queries, $unique_query if %$unique_query; } # Handle cases where the ResultSet already defines the query - my $query = @unique_hashes ? \@unique_hashes : undef; + my $query = @unique_queries ? \@unique_queries : undef; # Run the query if (keys %$attrs) { @@ -360,26 +363,19 @@ sub find { } } -# _unique_hash +# _build_unique_query # -# Constrain the specified hash based on the specified column names. - -sub _unique_hash { - my ($self, $hash, $unique_cols) = @_; +# Constrain the specified query hash based on the specified column names. - # Ugh, CDBI lowercases column names - if (exists $INC{'DBIx/Class/CDBICompat/ColumnCase.pm'}) { - foreach my $key (keys %$hash) { - $hash->{lc $key} = delete $hash->{$key}; - } - } +sub _build_unique_query { + my ($self, $query, $unique_cols) = @_; - my %unique_hash = - map { $_ => $hash->{$_} } - grep { exists $hash->{$_} } + my %unique_query = + map { $_ => $query->{$_} } + grep { exists $query->{$_} } @$unique_cols; - return %unique_hash; + return \%unique_query; } =head2 search_related