lib/DBIx/Class/ResultSetManager.pm

   1 package DBIx::Class::ResultSetManager;
   2 use strict;
   3 use base 'DBIx::Class';
   4 use Class::Inspector;
   5
   6 __PACKAGE__->mk_classdata($_) for qw/ _attr_cache custom_resultset_class_suffix /;
   7 __PACKAGE__->_attr_cache({});
   8 __PACKAGE__->custom_resultset_class_suffix('::_rs');
   9
  10 sub base_resultset_class {
  11     my ($self,$class) = @_;
  12     $self->result_source_instance->resultset_class($class);
  13 }
  14
  15 sub table {
  16     my ($self,@rest) = @_;
  17     $self->next::method(@rest);
  18     $self->_register_attributes;
  19 }
  20
  21 sub load_resultset_components {
  22     my ($self,@comp) = @_;
  23     my $resultset_class = $self->_setup_resultset_class;
  24     $resultset_class->load_components(@comp);
  25 }
  26
  27 sub MODIFY_CODE_ATTRIBUTES {
  28     my ($class,$code,@attrs) = @_;
  29     $class->_attr_cache({ %{$class->_attr_cache}, $code => [@attrs] });
  30     return ();
  31 }
  32
  33 sub _register_attributes {
  34     my $self = shift;
  35     my $cache = $self->_attr_cache;
  36     foreach my $meth (@{Class::Inspector->methods($self) || []}) {
  37         my $attrs = $cache->{$self->can($meth)};
  38         next unless $attrs;
  39         if ($attrs->[0] eq 'resultset') {
  40             no strict 'refs';
  41             my $resultset_class = $self->_setup_resultset_class;
  42             *{"$resultset_class\::$meth"} = *{"$self\::$meth"};
  43             undef *{"$self\::$meth"};
  44         }
  45     }
  46     $self->_attr_cache(undef);
  47 }
  48
  49 sub _setup_resultset_class {
  50     my $self = shift;
  51     my $resultset_class = $self . $self->custom_resultset_class_suffix;
  52     no strict 'refs';
  53     unless (@{"$resultset_class\::ISA"}) {
  54         @{"$resultset_class\::ISA"} = ($self->result_source_instance->resultset_class);
  55         $self->result_source_instance->resultset_class($resultset_class);
  56     }
  57     return $resultset_class;
  58 }
  59
  60 1;