1 package DBIx::Class::CDBICompat::ColumnGroups;
7 use base qw/Class::Data::Inheritable/;
9 __PACKAGE__->mk_classdata('_column_groups' => { });
13 my $class = ref $proto || $proto;
14 my $group = shift || "All";
15 $class->_add_column_group($group => @_) if @_;
16 return $class->all_columns if $group eq "All";
17 return $class->primary_column if $group eq "Primary";
18 return keys %{$class->_column_groups->{$group}};
21 sub _add_column_group {
22 my ($class, $group, @cols) = @_;
23 $class->_register_column_group($group => @cols);
24 $class->add_columns(@cols);
27 sub _register_column_group {
28 my ($class, $group, @cols) = @_;
30 my $groups = { %{$class->_column_groups} };
32 if ($group eq 'Primary') {
33 $class->set_primary_key(@cols);
34 $groups->{'Essential'}{$_} ||= {} for @cols;
37 if ($group eq 'All') {
38 unless (exists $class->_column_groups->{'Primary'}) {
39 $groups->{'Primary'}{$cols[0]} = {};
40 $class->set_primary_key($cols[0]);
42 unless (exists $class->_column_groups->{'Essential'}) {
43 #$class->_register_column_group('Essential' => $cols[0]);
44 $groups->{'Essential'}{$cols[0]} = {};
45 #$groups->{'Essential'}{$_} ||= {} for keys %{ $class->_primaries || {} };
49 $groups->{$group}{$_} ||= {} for @cols;
50 #if ($group eq 'Essential') {
51 # $groups->{$group}{$_} ||= {} for keys %{ $class->_primaries || {} };
54 $class->_column_groups($groups);
57 sub all_columns { return keys %{$_[0]->_columns}; }
61 my @pri = keys %{$class->_primaries};
62 return wantarray ? @pri : $pri[0];
66 my ($class, $col) = @_;
67 return $col if $class->_columns->{$col};
72 return bless({ class => $class}, 'DBIx::Class::CDBICompat::ColumnGroups::GrouperShim');
76 my ($class, @col) = @_;
77 return map { $class->find_column($_) } @col;
80 package DBIx::Class::CDBICompat::ColumnGroups::GrouperShim;
83 my ($self, @cols) = @_;
85 foreach my $col (@cols) {
86 foreach my $group (keys %{$self->{class}->_column_groups}) {
87 $groups{$group} = 1 if $self->{class}->_column_groups->{$group}->{$col};