1
# Copyright (C) 2008-2009 Sun Microsystems, Inc. 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::ExecuteAsUpdateDelete;
21
@ISA = qw(GenTest GenTest::Transform);
27
use GenTest::Transform;
28
use GenTest::Constants;
32
my ($class, $original_query, $executor, $original_result) = @_;
34
return STATUS_WONT_HANDLE if $original_query !~ m{^\s*SELECT}sio
35
|| $original_query =~ m{LIMIT}sio
36
|| $original_query =~ m{(AVG|STD|STDDEV_POP|STDDEV_SAMP|STDDEV|SUM|VAR_POP|VAR_SAMP|VARIANCE)\s*\(}sio;
37
return STATUS_WONT_HANDLE if $original_result->rows() == 0;
38
return STATUS_WONT_HANDLE if $#{$original_result->data()->[0]} != 0; # Only single-column resultsets
40
# This transformation can not work if the result set contains NULLs
41
foreach my $orig_row (@{$original_result->data()}) {
42
foreach my $orig_col (@$orig_row) {
43
return STATUS_WONT_HANDLE if $orig_col eq '';
47
my $table_name = 'transforms.where_updatedelete_'.$$;
48
my $col_name = $original_result->columnNames()->[0];
50
return STATUS_WONT_HANDLE if $col_name =~ m{`}sgio;
53
"DROP TABLE IF EXISTS $table_name",
54
"CREATE TABLE $table_name $original_query",
56
# If the result set has more than 1 row, we can not use it in the SET clause
57
( $original_result->rows() == 1 ?
58
"UPDATE $table_name SET `$col_name` = ( $original_query ) + 9999 WHERE `$col_name` NOT IN ( $original_query ) " :
59
"UPDATE $table_name SET `$col_name` = $col_name + 9999 WHERE `$col_name` NOT IN ( $original_query ) "
62
# The queries above should not have updated any rows. Sometimes ROW_COUNT() returns -1
63
"SELECT IF(ROW_COUNT() = 0 OR ROW_COUNT() = -1, 1, 0) /* TRANSFORM_OUTCOME_SINGLE_INTEGER_ONE */",
64
"SELECT * FROM $table_name /* TRANSFORM_OUTCOME_UNORDERED_MATCH */",
66
( $original_result->rows() == 1 ?
67
"UPDATE $table_name SET `$col_name` = ( $original_query ) WHERE `$col_name` IN ( $original_query ) " :
68
"UPDATE $table_name SET `$col_name` = $col_name WHERE `$col_name` IN ( $original_query ) "
71
# The queries above should have updated all rows
72
"SELECT IF(ROW_COUNT() = ".$original_result->rows()." OR ROW_COUNT() = -1, 1, 0) /* TRANSFORM_OUTCOME_SINGLE_INTEGER_ONE */",
73
"SELECT * FROM $table_name /* TRANSFORM_OUTCOME_UNORDERED_MATCH */",
75
# All rows should end up deleted
76
"DELETE FROM $table_name WHERE `$col_name` IN ( $original_query ) ",
77
"SELECT IF(ROW_COUNT() = ".$original_result->rows()." OR ROW_COUNT() = -1, 1, 0) /* TRANSFORM_OUTCOME_SINGLE_INTEGER_ONE */",
78
"SELECT * FROM $table_name /* TRANSFORM_OUTCOME_EMPTY_RESULT */",
79
"DROP TABLE IF EXISTS $table_name",