1 package DBIx::Class::CDBICompat::ColumnGroups;
7 use base qw/Class::Data::Inheritable/;
9 __PACKAGE__->mk_classdata('_column_groups' => { });
12 shift->_table_name(@_);
17 my $class = ref $proto || $proto;
18 my $group = shift || "All";
19 $class->_set_column_group($group => @_) if @_;
20 return $class->all_columns if $group eq "All";
21 return $class->primary_column if $group eq "Primary";
22 return keys %{$class->_column_groups->{$group}};
25 sub _set_column_group {
26 my ($class, $group, @cols) = @_;
27 $class->_register_column_group($group => @cols);
28 $class->_register_columns(@cols);
29 $class->_mk_column_accessors(@cols);
32 sub _register_column_group {
33 my ($class, $group, @cols) = @_;
34 if ($group eq 'Primary') {
36 $pri{$_} = {} for @cols;
37 $class->_primaries(\%pri);
40 my $groups = { %{$class->_column_groups} };
42 if ($group eq 'All') {
43 unless ($class->_column_groups->{'Primary'}) {
44 $groups->{'Primary'}{$cols[0]} = {};
45 $class->_primaries({ $cols[0] => {} });
47 unless ($class->_column_groups->{'Essential'}) {
48 $groups->{'Essential'}{$cols[0]} = {};
52 $groups->{$group}{$_} ||= {} for @cols;
53 $class->_column_groups($groups);
56 sub all_columns { return keys %{$_[0]->_columns}; }
60 my @pri = keys %{$class->_primaries};
61 return wantarray ? @pri : $pri[0];
65 my ($class, $col) = @_;
66 return $col if $class->_columns->{$col};
71 return bless({ class => $class}, 'DBIx::Class::CDBICompat::ColumnGroups::GrouperShim');
75 my ($class, @col) = @_;
76 return map { $class->find_column($_) } @col;
79 package DBIx::Class::CDBICompat::ColumnGroups::GrouperShim;
82 my ($self, @cols) = @_;
84 foreach my $col (@cols) {
85 foreach my $group (keys %{$self->{class}->_column_groups}) {
86 $groups{$group} = 1 if $self->{class}->_column_groups->{$group}->{$col};