54 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			54 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/**
 | 
						|
 * MXS-1160: LOAD DATA LOCAL INFILE with schemarouter
 | 
						|
 */
 | 
						|
 | 
						|
#include "testconnections.h"
 | 
						|
 | 
						|
#include <fstream>
 | 
						|
#include <sstream>
 | 
						|
#include <string>
 | 
						|
 | 
						|
int main(int argc, char** argv)
 | 
						|
{
 | 
						|
    // Create a CSV file
 | 
						|
    unlink("data.csv");
 | 
						|
    std::ofstream output("data.csv");
 | 
						|
    std::stringstream ss;
 | 
						|
 | 
						|
    for (int i = 0; i < 100; i++)
 | 
						|
    {
 | 
						|
        ss << i << "\n";
 | 
						|
    }
 | 
						|
 | 
						|
    output << ss.str();
 | 
						|
    output.close();
 | 
						|
 | 
						|
    TestConnections test(argc, argv);
 | 
						|
    test.repl->execute_query_all_nodes("DROP DATABASE db1");
 | 
						|
    test.repl->connect();
 | 
						|
    execute_query(test.repl->nodes[0], "CREATE DATABASE db1");
 | 
						|
    execute_query(test.repl->nodes[0], "CREATE TABLE db1.t1(id INT)");
 | 
						|
    test.connect_maxscale();
 | 
						|
 | 
						|
    test.tprintf("Loading local data file");
 | 
						|
 | 
						|
    test.try_query(test.conn_rwsplit, "LOAD DATA LOCAL INFILE 'data.csv' INTO TABLE db1.t1");
 | 
						|
 | 
						|
    test.tprintf("Verifying that data was loaded");
 | 
						|
 | 
						|
    long total = execute_query_count_rows(test.conn_rwsplit, "SELECT * FROM db1.t1");
 | 
						|
    test.add_result(total != 100, "Expected 100 rows, got %ld", total);
 | 
						|
 | 
						|
    test.tprintf("Dropping tables and databases");
 | 
						|
 | 
						|
    test.try_query(test.conn_rwsplit, "DROP TABLE db1.t1");
 | 
						|
    test.try_query(test.conn_rwsplit, "DROP DATABASE db1");
 | 
						|
 | 
						|
    test.close_maxscale_connections();
 | 
						|
 | 
						|
    // Remove the test data
 | 
						|
    unlink("data.csv");
 | 
						|
 | 
						|
    return test.global_result;
 | 
						|
}
 |