Commit | Line | Data |
4c2b30d6 |
1 | use strict; |
2 | use warnings; |
3 | |
4 | use Test::More; |
52864fbd |
5 | use Test::Deep; |
4c2b30d6 |
6 | use lib qw(t/lib); |
a5a7bb73 |
7 | use DBICTest ':DiffSQL'; |
4c2b30d6 |
8 | |
9 | my $schema = DBICTest->init_schema(); |
4c2b30d6 |
10 | |
11 | my $cdrs = $schema->resultset('CD')->search({ 'me.artist' => { '!=', 2 }}); |
12 | |
13 | my $cd_data = { map { |
14 | $_->cdid => { |
15 | siblings => $cdrs->search ({ artist => $_->get_column('artist') })->count - 1, |
16 | track_titles => [ map { $_->title } ($_->tracks->all) ], |
17 | }, |
18 | } ( $cdrs->all ) }; |
19 | |
4c2b30d6 |
20 | my $c_rs = $cdrs->search ({}, { |
21 | prefetch => 'tracks', |
22 | '+columns' => { sibling_count => $cdrs->search( |
23 | { |
24 | 'siblings.artist' => { -ident => 'me.artist' }, |
0e773352 |
25 | 'siblings.cdid' => { '!=' => ['-and', { -ident => 'me.cdid' }, 23414] }, |
4c2b30d6 |
26 | }, { alias => 'siblings' }, |
27 | )->count_rs->as_query, |
28 | }, |
29 | }); |
30 | |
31 | is_same_sql_bind( |
32 | $c_rs->as_query, |
33 | '( |
34 | SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track, |
35 | (SELECT COUNT( * ) |
36 | FROM cd siblings |
8d005ad9 |
37 | WHERE me.artist != ? |
38 | AND siblings.artist = me.artist |
4c2b30d6 |
39 | AND siblings.cdid != me.cdid |
40 | AND siblings.cdid != ? |
4c2b30d6 |
41 | ), |
42 | tracks.trackid, tracks.cd, tracks.position, tracks.title, tracks.last_updated_on, tracks.last_updated_at |
43 | FROM cd me |
44 | LEFT JOIN track tracks |
45 | ON tracks.cd = me.cdid |
46 | WHERE me.artist != ? |
4c2b30d6 |
47 | )', |
48 | [ |
36fd7f07 |
49 | |
50 | # subselect |
0e773352 |
51 | [ { sqlt_datatype => 'integer', dbic_colname => 'me.artist' } |
52 | => 2 ], |
36fd7f07 |
53 | |
8d005ad9 |
54 | [ { sqlt_datatype => 'integer', dbic_colname => 'siblings.cdid' } |
55 | => 23414 ], |
56 | |
36fd7f07 |
57 | # outher WHERE |
0e773352 |
58 | [ { sqlt_datatype => 'integer', dbic_colname => 'me.artist' } |
59 | => 2 ], |
4c2b30d6 |
60 | ], |
61 | 'Expected SQL on correlated realiased subquery' |
62 | ); |
63 | |
49eeb48d |
64 | $schema->is_executed_querycount( sub { |
65 | cmp_deeply ( |
66 | { map |
67 | { $_->cdid => { |
68 | track_titles => [ map { $_->title } ($_->tracks->all) ], |
69 | siblings => $_->get_column ('sibling_count'), |
70 | } } |
71 | $c_rs->all |
72 | }, |
73 | $cd_data, |
74 | 'Proper information retrieved from correlated subquery' |
75 | ); |
76 | }, 1, 'Only 1 query fired to retrieve everything'); |
4c2b30d6 |
77 | |
36fd7f07 |
78 | # now add an unbalanced select/as pair |
79 | $c_rs = $c_rs->search ({}, { |
80 | '+select' => $cdrs->search( |
81 | { 'siblings.artist' => { -ident => 'me.artist' } }, |
82 | { alias => 'siblings', columns => [ |
83 | { first_year => { min => 'year' }}, |
84 | { last_year => { max => 'year' }}, |
85 | ]}, |
86 | )->as_query, |
87 | '+as' => [qw/active_from active_to/], |
88 | }); |
89 | |
36fd7f07 |
90 | is_same_sql_bind( |
91 | $c_rs->as_query, |
92 | '( |
93 | SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track, |
94 | (SELECT COUNT( * ) |
95 | FROM cd siblings |
8d005ad9 |
96 | WHERE me.artist != ? |
97 | AND siblings.artist = me.artist |
36fd7f07 |
98 | AND siblings.cdid != me.cdid |
99 | AND siblings.cdid != ? |
36fd7f07 |
100 | ), |
101 | (SELECT MIN( year ), MAX( year ) |
102 | FROM cd siblings |
8d005ad9 |
103 | WHERE me.artist != ? |
104 | AND siblings.artist = me.artist |
36fd7f07 |
105 | ), |
f7f53a89 |
106 | tracks.trackid, tracks.cd, tracks.position, tracks.title, tracks.last_updated_on, tracks.last_updated_at |
36fd7f07 |
107 | FROM cd me |
108 | LEFT JOIN track tracks |
109 | ON tracks.cd = me.cdid |
110 | WHERE me.artist != ? |
36fd7f07 |
111 | )', |
112 | [ |
113 | |
114 | # first subselect |
0e773352 |
115 | [ { sqlt_datatype => 'integer', dbic_colname => 'me.artist' } |
116 | => 2 ], |
36fd7f07 |
117 | |
8d005ad9 |
118 | [ { sqlt_datatype => 'integer', dbic_colname => 'siblings.cdid' } |
119 | => 23414 ], |
120 | |
36fd7f07 |
121 | # second subselect |
0e773352 |
122 | [ { sqlt_datatype => 'integer', dbic_colname => 'me.artist' } |
123 | => 2 ], |
36fd7f07 |
124 | |
36fd7f07 |
125 | # outher WHERE |
0e773352 |
126 | [ { sqlt_datatype => 'integer', dbic_colname => 'me.artist' } |
127 | => 2 ], |
36fd7f07 |
128 | ], |
129 | 'Expected SQL on correlated realiased subquery' |
130 | ); |
131 | |
0dadd60d |
132 | # test for subselect identifier leakage |
133 | # NOTE - the hodge-podge mix of literal and regular identifuers is *deliberate* |
134 | for my $quote_names (0,1) { |
135 | my $schema = DBICTest->init_schema( quote_names => $quote_names ); |
136 | |
137 | my ($ql, $qr) = $schema->storage->sql_maker->_quote_chars; |
138 | |
139 | my $art_rs = $schema->resultset('Artist')->search ({}, { |
140 | order_by => 'me.artistid', |
141 | prefetch => 'cds', |
142 | rows => 2, |
143 | }); |
144 | |
145 | my $inner_lim_bindtype = { sqlt_datatype => 'integer' }; |
146 | |
147 | for my $inner_relchain (qw( cds_unordered cds ) ) { |
148 | |
149 | my $stupid_latest_competition_release_query = $schema->resultset('Artist')->search( |
150 | { 'competition.artistid' => { '!=', { -ident => 'me.artistid' } } }, |
151 | { alias => 'competition' }, |
152 | )->search_related( $inner_relchain, {}, { |
153 | rows => 1, order_by => 'year', columns => { year => \'year' }, distinct => 1 |
154 | })->get_column(\'year')->max_rs; |
155 | |
156 | my $final_query = $art_rs->search( {}, { |
157 | '+columns' => { max_competition_release => \[ |
158 | @${ $stupid_latest_competition_release_query->as_query } |
159 | ]}, |
160 | }); |
161 | |
c6a7ffb3 |
162 | # we are using cds_unordered explicitly above - do the sorting manually |
163 | my @results = sort { $a->{artistid} <=> $b->{artistid} } @{$final_query->all_hri}; |
164 | @$_ = sort { $a->{cdid} <=> $b->{cdid} } @$_ for map { $_->{cds} } @results; |
165 | |
0dadd60d |
166 | is_deeply ( |
c6a7ffb3 |
167 | \@results, |
0dadd60d |
168 | [ |
169 | { artistid => 1, charfield => undef, max_competition_release => 1998, name => "Caterwauler McCrae", rank => 13, cds => [ |
0dadd60d |
170 | { artist => 1, cdid => 1, genreid => 1, single_track => undef, title => "Spoonful of bees", year => 1999 }, |
c6a7ffb3 |
171 | { artist => 1, cdid => 2, genreid => undef, single_track => undef, title => "Forkful of bees", year => 2001 }, |
172 | { artist => 1, cdid => 3, genreid => undef, single_track => undef, title => "Caterwaulin' Blues", year => 1997 }, |
0dadd60d |
173 | ] }, |
174 | { artistid => 2, charfield => undef, max_competition_release => 1997, name => "Random Boy Band", rank => 13, cds => [ |
175 | { artist => 2, cdid => 4, genreid => undef, single_track => undef, title => "Generic Manufactured Singles", year => 2001 }, |
176 | ] }, |
177 | ], |
178 | "Expected result from weird query", |
179 | ); |
180 | |
181 | # the decomposition to sql/bind is *deliberate* in both instances |
182 | # we want to ensure this keeps working for lietral sql, even when |
183 | # as_query switches to return an overloaded dq node |
184 | my ($sql, @bind) = @${ $final_query->as_query }; |
185 | |
186 | my $correlated_sql = qq{ ( |
187 | SELECT MAX( year ) |
188 | FROM ( |
189 | SELECT year |
190 | FROM ${ql}artist${qr} ${ql}competition${qr} |
191 | JOIN cd ${ql}${inner_relchain}${qr} |
192 | ON ${ql}${inner_relchain}${qr}.${ql}artist${qr} = ${ql}competition${qr}.${ql}artistid${qr} |
193 | WHERE ${ql}competition${qr}.${ql}artistid${qr} != ${ql}me${qr}.${ql}artistid${qr} |
194 | GROUP BY year |
195 | ORDER BY MIN( ${ql}year${qr} ) |
196 | LIMIT ? |
197 | ) ${ql}${inner_relchain}${qr} |
198 | )}; |
199 | |
200 | is_same_sql_bind( |
201 | $sql, |
202 | \@bind, |
203 | qq{ ( |
204 | SELECT ${ql}me${qr}.${ql}artistid${qr}, ${ql}me${qr}.${ql}name${qr}, ${ql}me${qr}.${ql}rank${qr}, ${ql}me${qr}.${ql}charfield${qr}, |
205 | $correlated_sql, |
206 | ${ql}cds${qr}.${ql}cdid${qr}, ${ql}cds${qr}.${ql}artist${qr}, ${ql}cds${qr}.${ql}title${qr}, ${ql}cds${qr}.${ql}year${qr}, ${ql}cds${qr}.${ql}genreid${qr}, ${ql}cds${qr}.${ql}single_track${qr} |
207 | FROM ( |
208 | SELECT ${ql}me${qr}.${ql}artistid${qr}, ${ql}me${qr}.${ql}name${qr}, ${ql}me${qr}.${ql}rank${qr}, ${ql}me${qr}.${ql}charfield${qr}, |
209 | $correlated_sql |
210 | FROM ${ql}artist${qr} ${ql}me${qr} |
211 | ORDER BY ${ql}me${qr}.${ql}artistid${qr} |
212 | LIMIT ? |
213 | ) ${ql}me${qr} |
214 | LEFT JOIN cd ${ql}cds${qr} |
215 | ON ${ql}cds${qr}.${ql}artist${qr} = ${ql}me${qr}.${ql}artistid${qr} |
216 | ORDER BY ${ql}me${qr}.${ql}artistid${qr} |
217 | ) }, |
218 | [ |
219 | [ $inner_lim_bindtype |
220 | => 1 ], |
221 | [ $inner_lim_bindtype |
222 | => 1 ], |
223 | [ { sqlt_datatype => 'integer' } |
224 | => 2 ], |
225 | ], |
226 | "No leakage of correlated subquery identifiers (quote_names => $quote_names, inner alias '$inner_relchain')" |
227 | ); |
228 | } |
229 | } |
230 | |
4c2b30d6 |
231 | done_testing; |