1
# Copyright (c) 2008, 2011 Oracle and/or its affiliates. All rights reserved.
2
# Use is subject to license terms.
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; version 2 of the License.
8
# This program is distributed in the hope that it will be useful, but
9
# WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11
# General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
18
package GenTest::Transform::Count;
21
@ISA = qw(GenTest GenTest::Transform);
27
use GenTest::Transform;
28
use GenTest::Constants;
31
# This Transform provides the following transformations
33
# SELECT COUNT(*) FROM ... -> SELECT * FROM ...
35
# SELECT ... FROM ... -> SELECT COUNT(*), ... FROM ...
37
# It avoids GROUP BY and any other aggregate functions because
38
# those are difficult to validate with a simple check such as
39
# TRANSFORM_OUTCOME_COUNT
43
my ($class, $orig_query) = @_;
45
return STATUS_WONT_HANDLE if $orig_query =~ m{GROUP\s+BY|LIMIT|HAVING}sio;
47
my ($select_list) = $orig_query =~ m{SELECT (.*?) FROM}sio;
49
if ($select_list =~ m{AVG|BIT|CONCAT|DISTINCT|GROUP|MAX|MIN|STD|SUM|VAR|STRAIGHT_JOIN|SQL_SMALL_RESULT}sio) {
50
return STATUS_WONT_HANDLE;
51
} elsif ($select_list =~ m{SELECT\s?\*}sio) {
52
# "SELECT *" was matched. Cannot have both * and COUNT(...) in SELECT list.
53
$orig_query =~ s{SELECT (.*?) FROM}{SELECT COUNT(*) FROM}sio;
54
} elsif ($select_list !~ m{COUNT}sio) {
55
$orig_query =~ s{SELECT (.*?) FROM}{SELECT COUNT(*) , $1 FROM}sio;
56
} elsif ($select_list =~ m{^\s*COUNT\(\s*\*\s*\)}sio) {
57
$orig_query =~ s{SELECT .*? FROM}{SELECT * FROM}sio;
59
return STATUS_WONT_HANDLE;
62
return $orig_query." /* TRANSFORM_OUTCOME_COUNT */";