1 package # hide from PAUSE
2 DBIx::Class::CDBICompat::ColumnGroups;
7 use base qw/DBIx::Class::Row/;
9 __PACKAGE__->mk_classdata('_column_groups' => { });
13 my $class = ref $proto || $proto;
14 my $group = shift || "All";
15 $class->_init_result_source_instance();
17 $class->_add_column_group($group => @_) if @_;
18 return $class->all_columns if $group eq "All";
19 return $class->primary_column if $group eq "Primary";
20 return keys %{$class->_column_groups->{$group}};
23 sub _add_column_group {
24 my ($class, $group, @cols) = @_;
25 $class->add_columns(@cols);
26 $class->_register_column_group($group => @cols);
29 sub _register_column_group {
30 my ($class, $group, @cols) = @_;
32 my $groups = { %{$class->_column_groups} };
34 if ($group eq 'Primary') {
35 $class->set_primary_key(@cols);
36 $groups->{'Essential'}{$_} ||= 1 for @cols;
39 if ($group eq 'All') {
40 unless (exists $class->_column_groups->{'Primary'}) {
41 $groups->{'Primary'}{$cols[0]} = 1;
42 $class->set_primary_key($cols[0]);
44 unless (exists $class->_column_groups->{'Essential'}) {
45 $groups->{'Essential'}{$cols[0]} = 1;
49 $groups->{$group}{$_} ||= 1 for @cols;
51 $class->_column_groups($groups);
54 sub all_columns { return shift->result_source_instance->columns; }
58 my @pri = $class->primary_columns;
59 return wantarray ? @pri : $pri[0];
63 my ($class, $col) = @_;
64 return $col if $class->has_column($col);
69 my $grouper = { class => $class };
70 return bless($grouper, 'DBIx::Class::CDBICompat::ColumnGroups::GrouperShim');
74 my ($class, @col) = @_;
75 return map { $class->find_column($_) } @col;
78 package DBIx::Class::CDBICompat::ColumnGroups::GrouperShim;
81 my ($self, @cols) = @_;
83 foreach my $col (@cols) {
84 foreach my $group (keys %{$self->{class}->_column_groups}) {
85 $groups{$group} = 1 if $self->{class}->_column_groups->{$group}->{$col};