104 lines
3.8 KiB
C++
104 lines
3.8 KiB
C++
/**
|
|
* @file prepared_statement.cpp Checks if prepared statement works via Maxscale
|
|
*
|
|
* - Create table t1 and fill it ith some data
|
|
* - via RWSplit:
|
|
* + PREPARE stmt FROM 'SELECT * FROM t1 WHERE fl=@x;';
|
|
* + SET @x = 3;")
|
|
* + EXECUTE stmt")
|
|
* + SET @x = 4;")
|
|
* + EXECUTE stmt")
|
|
* - check if Maxscale is alive
|
|
*/
|
|
|
|
#include "testconnections.h"
|
|
#include "sql_t1.h"
|
|
|
|
using namespace std;
|
|
|
|
void test_basic(TestConnections& test)
|
|
{
|
|
test.set_timeout(60);
|
|
int N = 4;
|
|
|
|
test.repl->connect();
|
|
test.maxscales->connect_maxscale(0);
|
|
|
|
create_t1(test.maxscales->conn_rwsplit[0]);
|
|
insert_into_t1(test.maxscales->conn_rwsplit[0], N);
|
|
|
|
test.set_timeout(20);
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "PREPARE stmt FROM 'SELECT * FROM t1 WHERE fl=@x;';");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "SET @x = 3;");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "EXECUTE stmt");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "SET @x = 4;");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "EXECUTE stmt");
|
|
|
|
test.check_maxscale_alive(0);
|
|
test.stop_timeout();
|
|
}
|
|
|
|
void test_routing(TestConnections& test)
|
|
{
|
|
test.set_timeout(60);
|
|
test.repl->connect();
|
|
int server_id = test.repl->get_server_id(0);
|
|
test.maxscales->connect_maxscale(0);
|
|
|
|
// Test that reads are routed to slaves
|
|
char buf[1024] = "-1";
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "PREPARE ps1 FROM 'SELECT @@server_id'");
|
|
test.add_result(find_field(test.maxscales->conn_rwsplit[0], "EXECUTE ps1", "@@server_id", buf),
|
|
"Execute should succeed");
|
|
int res = atoi(buf);
|
|
test.add_result(res == server_id, "Query should be routed to a slave (got %d, master is %d)", res, server_id);
|
|
|
|
|
|
// Test reads inside transactions are routed to master
|
|
strcpy(buf, "-1");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "BEGIN");
|
|
test.add_result(find_field(test.maxscales->conn_rwsplit[0], "EXECUTE ps1", "@@server_id", buf),
|
|
"Execute should succeed");
|
|
res = atoi(buf);
|
|
test.add_result(res != server_id, "Query should be routed to master inside a transaction (got %d, master is %d)", res, server_id);
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "COMMIT");
|
|
|
|
// Test reads inside read-only transactions are routed slaves
|
|
strcpy(buf, "-1");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "START TRANSACTION READ ONLY");
|
|
test.add_result(find_field(test.maxscales->conn_rwsplit[0], "EXECUTE ps1", "@@server_id", buf),
|
|
"Execute should succeed");
|
|
res = atoi(buf);
|
|
test.add_result(res == server_id, "Query should be routed to a slave inside a read-only transaction (got %d, master is %d)", res, server_id);
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "COMMIT");
|
|
|
|
// Test prepared statements that modify data
|
|
strcpy(buf, "-1");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "CREATE OR REPLACE TABLE test.t1 (id INT)");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "PREPARE ps2 FROM 'INSERT INTO test.t1 VALUES (?)'");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "SET @a = 1");
|
|
test.try_query(test.maxscales->conn_rwsplit[0], "EXECUTE ps2 USING @a");
|
|
test.repl->sync_slaves();
|
|
test.add_result(find_field(test.maxscales->conn_rwsplit[0], "SELECT id FROM test.t1", "id", buf),
|
|
"Read should succeed");
|
|
res = atoi(buf);
|
|
test.add_result(res != server_id, "Writes should be routed to the master (got %d, master is %d)", res, server_id);
|
|
|
|
// Cleanup
|
|
test.check_maxscale_alive(0);
|
|
test.stop_timeout();
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
TestConnections test(argc, argv);
|
|
|
|
test.tprintf("Running basic test");
|
|
test_basic(test);
|
|
|
|
test.tprintf("Running text PS routing test");
|
|
test_routing(test);
|
|
|
|
return test.global_result;
|
|
}
|