File: rpl_kill_query.inc

package info (click to toggle)
mysql-8.0 8.0.43-3
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 1,273,924 kB
  • sloc: cpp: 4,684,605; ansic: 412,450; pascal: 108,398; java: 83,641; perl: 30,221; cs: 27,067; sql: 26,594; sh: 24,181; python: 21,816; yacc: 17,169; php: 11,522; xml: 7,388; javascript: 7,076; makefile: 2,194; lex: 1,075; awk: 670; asm: 520; objc: 183; ruby: 97; lisp: 86
file content (129 lines) | stat: -rw-r--r-- 4,238 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# ==== Purpose ====
#
# This is an auxiliary file used by rpl_kill_query.test.
#
# The purpose is to generate a statement that updates a
# non-transactional table, and kill the statement in the middle,
# so that it gets half-executed.
#
# ==== Usage ====
#
# --let $database= DATABASE
# --let $table= TABLE
# --source include/rpl_kill_query.inc
#
# Parameters:
#   $database
#     The current database will be set to $database before executing the
#     killed statement.
#
#   $table
#     The statement will modify this table.
#
# Assumptions:
# - $database should *not* be created before executing this script.
#
# ==== Implementation ====
#
# Apparently, not all statements can be killed in the middle; a simple
# INSERT does not seem to work.  One case that can be killed is an
# INSERT with a trigger; then there is a check for being killed in the
# trigger invocation.
#
# So our approach is to put a BEFORE INSERT trigger on the table.  The
# trigger will execute fast for the first row, and go to sleep for the
# second row. This allows us to kill the query in the middle of the
# sleep. Then, we make sure to wait until *after* the first row is
# inserted, and *before* the sleep is finished.  That ensures that at
# least one row is inserted in the trigger table, and not both rows
# are inserted in the main table, so the statement is half-executed,
# as required.
#
# To sleep on the second row, we use a SLEEP() inside a condition that
# is triggered only for the second row.
#
# To indicate to other threads that the first row has been inserted,
# we do SET @@GLOBAL.BINLOG_FORMAT = ROW before going to sleep.  (It
# does not seem to be enough to wait for rows to be inserted in the
# main table or trigger table, probably since locks prevent other
# threads from seeing the partial insert even in a non-transactional
# table.)  This will not affect execution since the global values is
# only used as default for new connections.
#
# As a sanity-check, we verify that after the query is killed, at most
# one row has been inserted in the main table, and at least one row
# has been inserted in the second table.


--let $_rkq_current_database_old= `SELECT DATABASE()`

--let $assert_cond= @@GLOBAL.BINLOG_FORMAT = "STATEMENT"
--let $assert_text= Test only works if binlog_format is statement
--source include/assert.inc

# Get connection id
--source include/rpl_connection_master.inc
--let $connection_id= `SELECT CONNECTION_ID()`

# Create database, table, and trigger table
eval CREATE DATABASE $database;
eval USE $database;
eval CREATE TABLE $table (a INT) ENGINE = MyISAM;
eval CREATE TABLE trigger_table (a INT) ENGINE = MyISAM;

# This master-only trigger will sleep for N seconds when the value N
# is inserted to a row.
SET SQL_LOG_BIN=0;
--delimiter |
eval
CREATE TRIGGER trig1 BEFORE INSERT ON $table FOR EACH ROW
BEGIN
  IF NEW.a != 0 THEN
    SET @@GLOBAL.BINLOG_FORMAT = 'row';
    DO SLEEP(1000000);
  END IF;
  INSERT INTO trigger_table VALUES (1);
END|
--delimiter ;
SET SQL_LOG_BIN=1;

# Sleep 1000000 seconds before inserting the second row.
--send
eval INSERT INTO $table VALUES (0), (1);

# Wait until the sleep for the second row happens.
--source include/rpl_connection_master1.inc

--let $wait_condition= SELECT @@GLOBAL.BINLOG_FORMAT = 'ROW'
--let $show_rpl_debug_info= 1
--source include/wait_condition.inc

# Kill query.
--replace_result $connection_id <CONNECTION_ID>
eval KILL QUERY $connection_id;

# Get the error message.
--source include/rpl_connection_master.inc
--error ER_QUERY_INTERRUPTED
reap;

SET @@GLOBAL.BINLOG_FORMAT = 'STATEMENT';

# Check that at most one row was inserted in main table.
--let $assert_text= At most one row should be inserted in $database.$table
--let $assert_cond= COUNT(*) <= 1 FROM $database.$table
--source include/assert.inc

# Check that at least one row was inserted in trigger table.
--let $assert_text= At least one row should be inserted in $database.trigger_table
--let $assert_cond= COUNT(*) >= 1 FROM $database.trigger_table
--source include/assert.inc

# Clean up.
SET SQL_LOG_BIN=0;
eval DROP TRIGGER trig1;
SET SQL_LOG_BIN=1;
DROP TABLE trigger_table;
eval DROP TABLE $table;
eval DROP DATABASE $database;
--eval USE $_rkq_current_database_old