1package # hide from PAUSE 2 DBIx::Class::CDBICompat::TempColumns; 3 4use strict; 5use warnings; 6use base qw/Class::Data::Inheritable/; 7 8use Carp; 9 10__PACKAGE__->mk_classdata('_temp_columns' => { }); 11 12sub _add_column_group { 13 my ($class, $group, @cols) = @_; 14 15 return $class->next::method($group, @cols) unless $group eq 'TEMP'; 16 17 my %new_cols = map { $_ => 1 } @cols; 18 my %tmp_cols = %{$class->_temp_columns}; 19 20 for my $existing_col ( grep $new_cols{$_}, $class->columns ) { 21 # Already been declared TEMP 22 next if $tmp_cols{$existing_col}; 23 24 carp "Declaring column $existing_col as TEMP but it already exists"; 25 } 26 27 $class->_register_column_group($group => @cols); 28 $class->mk_group_accessors('temp' => @cols); 29 30 $class->_temp_columns({ %tmp_cols, %new_cols }); 31} 32 33sub new { 34 my ($class, $attrs, @rest) = @_; 35 36 my $temp = $class->_extract_temp_data($attrs); 37 38 my $new = $class->next::method($attrs, @rest); 39 40 $new->set_temp($_, $temp->{$_}) for keys %$temp; 41 42 return $new; 43} 44 45sub _extract_temp_data { 46 my($self, $data) = @_; 47 48 my %temp; 49 foreach my $key (keys %$data) { 50 $temp{$key} = delete $data->{$key} if $self->_temp_columns->{$key}; 51 } 52 53 return \%temp; 54} 55 56sub find_column { 57 my ($class, $col, @rest) = @_; 58 return $col if $class->_temp_columns->{$col}; 59 return $class->next::method($col, @rest); 60} 61 62sub set { 63 my($self, %data) = @_; 64 65 my $temp_data = $self->_extract_temp_data(\%data); 66 67 $self->set_temp($_, $temp_data->{$_}) for keys %$temp_data; 68 69 return $self->next::method(%data); 70} 71 72sub get_temp { 73 my ($self, $column) = @_; 74 $self->throw_exception( "Can't fetch data as class method" ) unless ref $self; 75 $self->throw_exception( "No such TEMP column '${column}'" ) unless $self->_temp_columns->{$column} ; 76 return $self->{_temp_column_data}{$column} 77 if exists $self->{_temp_column_data}{$column}; 78 return undef; 79} 80 81sub set_temp { 82 my ($self, $column, $value) = @_; 83 $self->throw_exception( "No such TEMP column '${column}'" ) 84 unless $self->_temp_columns->{$column}; 85 $self->throw_exception( "set_temp called for ${column} without value" ) 86 if @_ < 3; 87 return $self->{_temp_column_data}{$column} = $value; 88} 89 90sub has_real_column { 91 return 1 if shift->has_column(shift); 92} 93 941; 95