89 lines
2.9 KiB
C++
89 lines
2.9 KiB
C++
/**
|
|
* @file mxs365.cpp Load data with LOAD DATA LOCAL INFILE
|
|
*
|
|
* 1. Create a 50Mb test file
|
|
* 2. Load and read it through MaxScale
|
|
*/
|
|
|
|
|
|
#include "testconnections.h"
|
|
|
|
void create_data_file(char* filename, size_t size)
|
|
{
|
|
int fd, i = 0;
|
|
snprintf(filename, size, "local_infile_%u", i++);
|
|
while ((fd = open(filename, O_CREAT | O_RDWR | O_EXCL, 0755)) == -1)
|
|
{
|
|
snprintf(filename, size, "local_infile_%u", i++);
|
|
}
|
|
|
|
const size_t maxsize = 1024 * 1024 * 50;
|
|
size_t filesize = 0;
|
|
i = 0;
|
|
|
|
while (filesize < maxsize)
|
|
{
|
|
char buffer[1024];
|
|
sprintf(buffer, "%d,'%x','%x'\n", i, i << 10 + i, i << 5 + i);
|
|
int written = write(fd, buffer, strlen(buffer));
|
|
if (written <= 0)
|
|
{
|
|
break;
|
|
}
|
|
i++;
|
|
filesize += written;
|
|
}
|
|
|
|
close(fd);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
|
|
TestConnections * test = new TestConnections(argc, argv);
|
|
char filename[1024];
|
|
test->tprintf("Generation file to load\n");
|
|
test->set_timeout(30);
|
|
create_data_file(filename, sizeof (filename));
|
|
|
|
/** Set max packet size and create test table */
|
|
test->set_timeout(20);
|
|
test->tprintf("Connect to Maxscale\n");
|
|
test->maxscales->connect_maxscale(0);
|
|
test->tprintf("Setting max_allowed_packet, creating table\n");
|
|
test->add_result(execute_query(test->maxscales->conn_rwsplit[0],
|
|
"set global max_allowed_packet=(1048576 * 60)"),
|
|
"Setting max_allowed_packet failed.");
|
|
test->add_result(execute_query(test->maxscales->conn_rwsplit[0],
|
|
"DROP TABLE IF EXISTS test.dump"),
|
|
"Dropping table failed.");
|
|
test->add_result(execute_query(test->maxscales->conn_rwsplit[0],
|
|
"CREATE TABLE test.dump(a int, b varchar(80), c varchar(80))"),
|
|
"Creating table failed.");
|
|
test->tprintf("Closing connection to Maxscale\n");
|
|
test->maxscales->close_maxscale_connections(0);
|
|
|
|
/** Reconnect, load the data and then read it */
|
|
test->tprintf("Re-connect to Maxscale\n");
|
|
test->set_timeout(20);
|
|
test->maxscales->connect_maxscale(0);
|
|
char query[1024];
|
|
snprintf(query, sizeof (filename),
|
|
"LOAD DATA LOCAL INFILE '%s' INTO TABLE test.dump FIELDS TERMINATED BY ','",
|
|
filename);
|
|
test->tprintf("Loading data\n");
|
|
test->set_timeout(100);
|
|
test->add_result(execute_query(test->maxscales->conn_rwsplit[0], query), "Loading data failed.");
|
|
test->tprintf("Reading data\n");
|
|
test->set_timeout(100);
|
|
test->add_result(execute_query(test->maxscales->conn_rwsplit[0], "SELECT * FROM test.dump"),
|
|
"Reading data failed.");
|
|
test->maxscales->close_maxscale_connections(0);
|
|
test->tprintf("Cecking if Maxscale alive\n");
|
|
test->check_maxscale_alive(0);
|
|
int rval = test->global_result;
|
|
delete test;
|
|
unlink(filename);
|
|
return rval;
|
|
}
|