56 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			56 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/**
 | 
						|
 * Test KILL QUERY functionality
 | 
						|
 */
 | 
						|
 | 
						|
#include <maxtest/testconnections.hh>
 | 
						|
 | 
						|
int main(int argc, char* argv[])
 | 
						|
{
 | 
						|
    TestConnections test(argc, argv);
 | 
						|
 | 
						|
    for (int i = 0; i < 3; i++)
 | 
						|
    {
 | 
						|
        auto a = test.maxscales->rwsplit();
 | 
						|
        auto b = test.maxscales->rwsplit();
 | 
						|
        test.expect(a.connect() && b.connect(), "Connections should work");
 | 
						|
 | 
						|
        auto id = a.thread_id();
 | 
						|
 | 
						|
        std::thread thr(
 | 
						|
            [&]() {
 | 
						|
                const char* query =
 | 
						|
                    "BEGIN NOT ATOMIC "
 | 
						|
                    "  DECLARE v1 INT DEFAULT 5; "
 | 
						|
                    "  CREATE OR REPLACE TABLE test.t1(id INT); "
 | 
						|
                    "  WHILE v1 <> 0 DO "
 | 
						|
                    "    INSERT INTO test.t1 VALUES (1); "
 | 
						|
                    "    SET v1 = (SELECT COUNT(*) FROM test.t1); "
 | 
						|
                    "  END WHILE;"
 | 
						|
                    "END";
 | 
						|
 | 
						|
                    // The ALTER should take longer than 15 seconds to complete so that a KILL is required to
 | 
						|
                // interrupt it.
 | 
						|
                test.expect(!a.query(query), "Query should fail");
 | 
						|
 | 
						|
                const char* expected = "Query execution was interrupted";
 | 
						|
                test.expect(strstr(a.error(), expected),
 | 
						|
                            "Alter should fail with '%s' but it failed with '%s'",
 | 
						|
                            expected, a.error());
 | 
						|
            });
 | 
						|
 | 
						|
        // Wait for a few seconds to make sure the other thread has started executing the query before killing it.
 | 
						|
        sleep(5);
 | 
						|
        test.expect(b.query("KILL QUERY " + std::to_string(id)), "KILL QUERY failed: %s", b.error());
 | 
						|
 | 
						|
        test.set_timeout(15);
 | 
						|
        thr.join();
 | 
						|
        test.stop_timeout();
 | 
						|
    }
 | 
						|
 | 
						|
    auto conn = test.maxscales->rwsplit();
 | 
						|
    conn.connect();
 | 
						|
    conn.query("DROP TABLE test.t1");
 | 
						|
 | 
						|
    return test.global_result;
 | 
						|
}
 |