1 package DBIx::Class::ResultSetManager;
3 use base 'DBIx::Class';
6 __PACKAGE__->mk_classdata($_) for qw/ _attr_cache base_resultset_class custom_resultset_class_suffix /;
7 __PACKAGE__->_attr_cache({});
8 __PACKAGE__->base_resultset_class('DBIx::Class::ResultSet');
9 __PACKAGE__->custom_resultset_class_suffix('::_resultset');
12 my ($self,@rest) = @_;
13 $self->next::method(@rest);
14 $self->_register_attributes;
15 $self->_register_resultset_class;
18 sub load_resultset_components {
19 my ($self,@comp) = @_;
20 my $resultset_class = $self->_setup_resultset_class;
21 $resultset_class->load_components(@comp);
24 sub MODIFY_CODE_ATTRIBUTES {
25 my ($class,$code,@attrs) = @_;
26 $class->_attr_cache({ %{$class->_attr_cache}, $code => [@attrs] });
30 sub _register_attributes {
32 my $cache = $self->_attr_cache;
33 foreach my $meth (@{Class::Inspector->methods($self) || []}) {
34 my $attrs = $cache->{$self->can($meth)};
36 if ($attrs->[0] eq 'resultset') {
38 my $resultset_class = $self->_setup_resultset_class;
39 *{"$resultset_class\::$meth"} = *{"$self\::$meth"};
40 undef *{"$self\::$meth"};
43 $self->_attr_cache(undef);
46 sub _setup_resultset_class {
48 my $resultset_class = $self . $self->custom_resultset_class_suffix;
50 unless (@{"$resultset_class\::ISA"}) {
51 @{"$resultset_class\::ISA"} = ($self->base_resultset_class);
53 return $resultset_class;
56 sub _register_resultset_class {
58 my $resultset_class = $self . $self->custom_resultset_class_suffix;
60 if (@{"$resultset_class\::ISA"}) {
61 $self->result_source_instance->resultset_class($resultset_class);
63 $self->result_source_instance->resultset_class($self->base_resultset_class);