Commit | Line | Data |
c0e7b4e5 |
1 | package # hide from PAUSE |
2 | DBIx::Class::CDBICompat::LazyLoading; |
fe5d862b |
3 | |
4 | use strict; |
5 | use warnings; |
6 | |
5e0eea35 |
7 | use base 'DBIx::Class'; |
8 | |
a9433341 |
9 | sub resultset_instance { |
10 | my $self = shift; |
367eaf50 |
11 | $self->next::method(@_) |
12 | ->search_rs(undef, { columns => [ $self->columns('Essential') ] }); |
fe5d862b |
13 | } |
14 | |
c0fcc63f |
15 | |
8273e845 |
16 | # Emulate that CDBI throws out all changed columns and reloads them on |
c0fcc63f |
17 | # request in case the database modifies the new value (say, via a trigger) |
18 | sub update { |
19 | my $self = shift; |
d4daee7b |
20 | |
c0fcc63f |
21 | my @dirty_columns = keys %{$self->{_dirty_columns}}; |
d4daee7b |
22 | |
c0fcc63f |
23 | my $ret = $self->next::method(@_); |
1d7e89b8 |
24 | $self->_clear_column_data(@dirty_columns); |
d4daee7b |
25 | |
c0fcc63f |
26 | return $ret; |
27 | } |
28 | |
29 | |
1d7e89b8 |
30 | # And again for create |
31 | sub create { |
32 | my $class = shift; |
33 | my($data) = @_; |
d4daee7b |
34 | |
1d7e89b8 |
35 | my @columns = keys %$data; |
d4daee7b |
36 | |
1d7e89b8 |
37 | my $obj = $class->next::method(@_); |
38 | return $obj unless defined $obj; |
d4daee7b |
39 | |
1d7e89b8 |
40 | my %primary_cols = map { $_ => 1 } $class->primary_columns; |
41 | my @data_cols = grep !$primary_cols{$_}, @columns; |
42 | $obj->_clear_column_data(@data_cols); |
43 | |
44 | return $obj; |
45 | } |
46 | |
47 | |
48 | sub _clear_column_data { |
49 | my $self = shift; |
d4daee7b |
50 | |
1d7e89b8 |
51 | delete $self->{_column_data}{$_} for @_; |
52 | delete $self->{_inflated_column}{$_} for @_; |
53 | } |
54 | |
55 | |
fe5d862b |
56 | sub get_column { |
57 | my ($self, $col) = @_; |
58 | if ((ref $self) && (!exists $self->{'_column_data'}{$col}) |
8d5134b0 |
59 | && $self->{'_in_storage'}) { |
fe5d862b |
60 | $self->_flesh(grep { exists $self->_column_groups->{$_}{$col} |
61 | && $_ ne 'All' } |
62 | keys %{ $self->_column_groups || {} }); |
63 | } |
147dd158 |
64 | $self->next::method(@_[1..$#_]); |
fe5d862b |
65 | } |
66 | |
e60dc79f |
67 | # CDBI does not explicitly declare auto increment columns, so |
68 | # we just clear out our primary columns before copying. |
69 | sub copy { |
70 | my($self, $changes) = @_; |
71 | |
72 | for my $col ($self->primary_columns) { |
73 | $changes->{$col} = undef unless exists $changes->{$col}; |
74 | } |
d4daee7b |
75 | |
e60dc79f |
76 | return $self->next::method($changes); |
77 | } |
78 | |
79 | sub discard_changes { |
80 | my($self) = shift; |
81 | |
82 | delete $self->{_column_data}{$_} for $self->is_changed; |
83 | delete $self->{_dirty_columns}; |
84 | delete $self->{_relationship_data}; |
85 | |
86 | return $self; |
87 | } |
88 | |
831ad24f |
89 | sub _ident_cond { |
90 | my ($class) = @_; |
91 | return join(" AND ", map { "$_ = ?" } $class->primary_columns); |
92 | } |
93 | |
fe5d862b |
94 | sub _flesh { |
95 | my ($self, @groups) = @_; |
b8e1e21f |
96 | @groups = ('All') unless @groups; |
fe5d862b |
97 | my %want; |
98 | $want{$_} = 1 for map { keys %{$self->_column_groups->{$_}} } @groups; |
99 | if (my @want = grep { !exists $self->{'_column_data'}{$_} } keys %want) { |
e5053694 |
100 | my $cursor = $self->result_source->schema->storage->select( |
b98e75f6 |
101 | $self->result_source->name, \@want, |
8b445e33 |
102 | \$self->_ident_cond, { bind => [ $self->_ident_values ] }); |
103 | #my $sth = $self->storage->select($self->_table_name, \@want, |
104 | # $self->ident_condition); |
105 | # Not sure why the first one works and this doesn't :( |
223b8fe3 |
106 | my @val = $cursor->next; |
e4d970ed |
107 | |
108 | return unless @val; # object must have been deleted from the database |
109 | |
fe5d862b |
110 | foreach my $w (@want) { |
111 | $self->{'_column_data'}{$w} = shift @val; |
112 | } |
113 | } |
114 | } |
115 | |
116 | 1; |