Commit | Line | Data |
65b76960 |
1 | package DBIx::Data::Collection::Set; |
2 | |
3 | use Moose; |
4 | use Method::Signatures::Simple; |
5 | use Data::Perl::Stream::Array; |
6 | |
7 | has _store => (is => 'ro', required => 1, init_arg => 'store'); |
8 | |
65b76960 |
9 | has _class => (is => 'ro', predicate => '_has_class'); |
10 | |
3a2e7c1c |
11 | has _set_over => (is => 'ro', required => 1, init_arg => 'set_over'); |
12 | |
13 | ## member cache (all members) |
14 | |
15 | has _member_cache => ( |
16 | is => 'rw', lazy_build => 1, |
17 | predicate => '_member_cache_built', |
18 | ); |
65b76960 |
19 | |
20 | method _build__member_cache { |
21 | my $stream = $self->_new_raw_stream; |
22 | my @cache; |
23 | while (my ($raw) = $stream->next) { |
3a2e7c1c |
24 | my $obj = do { |
25 | if (my ($obj) = $self->_key_cache_get_raw($raw)) { |
26 | $self->_merge($obj, $raw) |
27 | } else { |
28 | $self->_inflate($raw) |
29 | } |
30 | }; |
31 | push @cache, $obj; |
65b76960 |
32 | } |
33 | \@cache; |
34 | } |
35 | |
3a2e7c1c |
36 | method _add_to_member_cache ($to_add) { |
37 | return unless $self->_member_cache_built; |
38 | push @{$self->_member_cache}, $to_add; |
39 | } |
40 | |
41 | ## key cache - by primary/unique key |
42 | |
43 | has _key_cache => (is => 'ro', default => sub { {} }); |
44 | |
45 | method _add_to_key_cache ($to_add) { |
46 | $self->_key_cache->{$self->_object_to_id($to_add)} = $to_add; |
47 | return |
48 | } |
49 | |
50 | method _key_cache_has_raw ($raw) { |
51 | exists $self->_key_cache->{$self->_raw_to_id($raw)} |
52 | } |
53 | |
54 | method _key_cache_has_object ($obj) { |
55 | exists $self->_key_cache->{$self->_object_to_id($obj)} |
56 | } |
57 | |
58 | method _key_cache_get_raw ($raw) { |
59 | my $id = $self->_raw_to_id($raw); |
60 | exists $self->_key_cache->{$id} |
61 | ? ($self->_key_cache->{$id}) |
62 | : () |
63 | } |
64 | |
65 | method _key_cache_get_object ($obj) { |
66 | $self->_key_cache_get_raw($self->_deflate($obj)) |
67 | } |
68 | |
69 | ## loading data |
70 | |
65b76960 |
71 | method _new_raw_stream { |
72 | $self->_store->new_select_command([])->execute; |
73 | } |
74 | |
3a2e7c1c |
75 | ## thunking between the store representation and the set representation |
76 | # |
77 | # _inflate is raw data -> final repr |
78 | # _deflate is final repr -> raw data |
79 | # _merge takes final repr + raw data and updates the repr |
80 | # (this is used for pk-generated values and later lazy loading) |
81 | |
65b76960 |
82 | method _inflate ($raw) { |
3347c67e |
83 | bless($raw, $self->_class) if $self->_has_class; |
84 | $raw; |
65b76960 |
85 | } |
86 | |
3a2e7c1c |
87 | method _deflate ($obj) { |
88 | +{ %$obj } |
89 | } |
90 | |
91 | method _merge ($obj, $raw) { |
92 | @{$obj}{keys %$raw} = values %$raw; |
93 | $obj; |
94 | } |
95 | |
96 | ## methods to get ids |
97 | |
98 | method _raw_to_id ($raw) { |
99 | # XXX must escape this. or do something else. |
100 | join ';', map $raw->{$_}, @{$self->_set_over} |
101 | } |
102 | |
103 | method _object_to_id ($obj) { |
104 | $self->_raw_to_id($self->_deflate($obj)); |
105 | } |
106 | |
65b76960 |
107 | method flatten { |
108 | @{$self->_member_cache}; |
109 | } |
110 | |
111 | method as_stream { |
112 | Data::Perl::Stream::Array->new(array => $self->_member_cache); |
113 | } |
114 | |
3a2e7c1c |
115 | method add ($new) { |
116 | $self->_add_to_store($new); |
117 | $self->_add_to_caches($new); |
118 | $new; |
119 | } |
120 | |
121 | method _add_to_store ($new) { |
122 | my $new_raw = $self->_deflate($new); |
123 | $self->_merge($new, $self->_store->new_insert_command($new_raw)->execute); |
124 | $self->_add_to_caches($new); |
125 | return |
126 | } |
127 | |
128 | method _add_to_caches ($new) { |
129 | $self->_add_to_member_cache($new); |
130 | $self->_add_to_key_cache($new); |
131 | } |
132 | |
65b76960 |
133 | 1; |