You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
proxysql/test/tap/tests/test_backend_conn_ping-t.cpp

729 lines
28 KiB

/**
* @file test_backend_conn_ping-t.cpp
* @brief Checks that backend connections are getting properly kept alive via ping by ProxySQL.
* @details For achieving this, the test performs the following actions:
* 1. Create an specific number of backend connections in batches in the target hostgroups via
* 'create_new_connection' and 'hostgroup' query annotations.
* NOTE: The number of created connections per batch and in total shouldn't exceed ProxySQL
* maximum processing rate. Determined by 'mysql-ping_interval_server_msec'.
* 2. Check periodically that the number of current connections reflected in both the target MySQL server
* ('information_schema.processlist') and in ProxySQL ('stats_mysql_free_connections') doesn't change
* over time. I.e. connections are not getting destroyed due to not being kept alive.
* 3. Perform a query per each created connection exahusting the backend connections while checking for
* any error reported by the client due to broken connections.
*
* Backend Connection Pinging Algorithm:
* ====================================
*
* The algorithm used for pinging backend connections is simple:
*
* 1. Pinging is done for 'idle_session' in batches of SESSIONS_FOR_CONNECTIONS_HANDLER.
* 2. The pinging interval is determined by 'ping_interval_server_msec'.
* 3. Oldest sessions are always selected first for pinging.
*
* With this approach, it should be possible to ping a very large number of backend connections before
* losing them due to inactivity (wait_timeout), this will be defined by:
*
* ```
* (NUM_THREADS * SESSIONS_FOR_CONNECTIONS_HANDLER)*floor((MYSQL_wait_timeout-1)/ping_interval_server_msec)
* ```
*
* This means that, for example, for a config like: `num_threads=4,wait_timeout=60,batch_size=64`, we can
* expect: `15104` connections. Of course, this suggests that this algorithm wont have any issues holding a
* healthy connection pool of any size when using real `wait_timeout` values.
*
* Algorithm - Special Cases:
* =========================
*
* This algorithm has a expected behavior that might seems unexpected when pushing the limits of the
* connections that are being maintained; like we can accidentally do in this test. If the number of
* connections being maintained exceeds the number that can be pinged within the `MySQL_wait_timeout`
* interval, this will cause a cascading effect, that will result in extra connections timing out. The
* number of connections from the connection pool that will timeout will be proportional to the fraction of
* the batch (NUM_THREADS*SESSIONS_FOR_CONNECTIONS_HANDLER) filled with connections that exceeded the
* interval processing capacity.
*
* If for example, the number of connections exceeded the ones that can be maintained by 50% of the batching
* interval, for the case of 4 threads, this will be 128 connections, then 50% of the connection pool will
* be lost due to this cascading effect. Elaborating a little bit further, when the number of connections
* being maintained exceeds the number of connections that can be pinged, the connections exceeding the
* interval capabilities will timeout, since these connections will be the oldest, they will be the first to
* be selected for the next interval. This will shift all subsequent batching intervals by this number of
* connections, since we are operating at maximum capacity (maximum number exceeded), all the intervals were
* selecting connections that should be pinged, otherwise will timeout. This is why the previously mentioned
* shift will result in the timeout of these connections, causing the mentioned cascading effect.
*
* This is expected behavior, and **it's not an issue**. The number of connections ProxySQL can ping is
* mainly determined by the interval that defines `wait_timeout` in the MySQL server and
* `ping_interval_server_msec`, which default value is `1000`. With reasonable values for these two
* parameters this number is very high, even with extremely low values for `wait_timeout`, like
* `num_threads=4,wait_timeout=60,batch_size=64`, a large number of connections(`15104`) would still be
* maintained without issues. This is what makes this issue an artificial one.
*
* IMPORTANT: In case of failure see note below on 'wait_timeout' definition.
*/
#include <string>
#include <vector>
#include <map>
#include <utility>
#include <unistd.h>
#include <string.h>
#include <sys/resource.h>
#include "mysql.h"
#include "tap.h"
#include "utils.h"
#include "json.hpp"
using std::vector;
using std::string;
using std::pair;
using srv_cfg = vector<pair<string,int>>;
#define SESSIONS_FOR_CONNECTIONS_HANDLER 64
int max_pinged_conns(uint32_t num_threads, uint32_t ping_interval_server_msec, uint32_t wait_timeout) {
uint32_t ping_proc_batches = floor(wait_timeout/float(ping_interval_server_msec/1000.0));
return (num_threads * SESSIONS_FOR_CONNECTIONS_HANDLER) * ping_proc_batches;
}
int change_mysql_cfg(
const CommandLine& cl, const string& host, const string& port, const srv_cfg& new_srv_cfg, srv_cfg& out_old_srv_cfg
) {
int res = EXIT_SUCCESS;
MYSQL* my_conn = mysql_init(NULL);
if (!my_conn) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
return EXIT_FAILURE;
}
diag("Connecting to %s:%s with user %s", host.c_str(), port.c_str(), cl.mysql_username);
if (!mysql_real_connect(my_conn, host.c_str(), cl.mysql_username, cl.mysql_password, NULL, std::stol(port.c_str()), NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
res = EXIT_FAILURE;
}
if (res == EXIT_SUCCESS) {
srv_cfg old_server_config {};
for (const pair<string,int>& config_var : new_srv_cfg) {
string query = "SELECT @@" + config_var.first;
diag("Line:%d : Running: %s", __LINE__, query.c_str());
res = mysql_query(my_conn, query.c_str());
if (res != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
res = EXIT_FAILURE;
break;
}
MYSQL_RES* my_res = mysql_store_result(my_conn);
if (my_res == nullptr) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
res = EXIT_FAILURE;
break;
}
MYSQL_ROW row = mysql_fetch_row(my_res);
if (row == nullptr || row[0] == nullptr) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
res = EXIT_FAILURE;
break;
} else {
diag("Line:%d : Returned: %s = %s", __LINE__, config_var.first.c_str(), row[0]);
old_server_config.push_back({ config_var.first, std::stol(row[0]) });
}
mysql_free_result(my_res);
query = string { "SET GLOBAL " + config_var.first + "=" + std::to_string(config_var.second) };
diag("Line:%d : Setting on %s:%s : %s", __LINE__ , host.c_str(), port.c_str(), query.c_str());
mysql_query(my_conn, query.c_str());
if (res != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(my_conn));
res = EXIT_FAILURE;
break;
}
}
if (res == EXIT_SUCCESS) {
out_old_srv_cfg = old_server_config;
}
}
mysql_close(my_conn);
return res;
}
int create_new_backend_conn(const CommandLine& cl, int tg_hg, vector<MYSQL*>& mysql_conns) {
MYSQL* conn = mysql_init(NULL);
if (!conn) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(conn));
return EXIT_FAILURE;
}
if (!mysql_real_connect(conn, cl.host, cl.username, cl.password, "backend_conn_ping_test", cl.port, NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(conn));
return EXIT_FAILURE;
}
const string query { "DO /* ;hostgroup=" + std::to_string(tg_hg) + ";create_new_connection=1 */ 1" };
MYSQL_QUERY(conn, query.c_str());
mysql_conns.push_back(conn);
return EXIT_SUCCESS;
}
int get_query_result(MYSQL* mysql, const string& query, uint64_t& out_val) {
int rc = mysql_query(mysql, query.c_str());
if (rc != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(mysql));
return EXIT_FAILURE;
}
MYSQL_RES* myres = mysql_store_result(mysql);
if (myres == nullptr) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(mysql));
return EXIT_FAILURE;
}
MYSQL_ROW row = mysql_fetch_row(myres);
if (row == nullptr || row[0] == nullptr) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, "mysql_fetch_row() failed");
return EXIT_FAILURE;
}
out_val = std::stol(row[0]);
mysql_free_result(myres);
return EXIT_SUCCESS;
}
struct test_params_t {
double init_batch_size;
double batch_size;
double its;
double delay_s;
};
using svr_addr = pair<string,uint32_t>;
int check_backend_conns(
const CommandLine& cl, const test_params_t& test_params, uint32_t hg, const vector<svr_addr>& svrs_addrs
) {
vector<MYSQL*> mysql_conns {};
int res = EXIT_SUCCESS;
diag("Line:%d : Creating %f connections on hg %d", __LINE__ , test_params.init_batch_size, hg);
for (uint32_t i = 0; i < test_params.init_batch_size; i++) {
int c_res = create_new_backend_conn(cl, hg, mysql_conns);
if (c_res != EXIT_SUCCESS) { return EXIT_FAILURE; }
}
usleep(test_params.delay_s * (test_params.init_batch_size / test_params.batch_size) * 1000 * 1000);
// 1. Create server connections to monitor
for (uint32_t i = 0; i < test_params.its; i++) {
diag("Line:%d : Creating %f connections on hg %d , iteration %d", __LINE__ , test_params.batch_size, hg, i);
for (uint32_t j = 0; j < test_params.batch_size; j++) {
int c_res = create_new_backend_conn(cl, hg, mysql_conns);
if (c_res != EXIT_SUCCESS) { return EXIT_FAILURE; }
}
usleep(test_params.delay_s * 1000 * 1000);
}
// 2. Check that the connections remain steady for a period of time
MYSQL* admin = mysql_init(NULL);
{
if (!mysql_real_connect(admin, cl.host, cl.admin_username, cl.admin_password, NULL, cl.admin_port, NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin));
return EXIT_FAILURE;
}
sleep(5);
uint64_t exp_conn_count = test_params.init_batch_size + test_params.batch_size * test_params.its;
int q_res = EXIT_SUCCESS;
uint64_t act_mysql_conn_count = 0;
uint64_t act_proxy_free_conn_count = 0;
uint64_t act_proxy_used_conn_count = 0;
uint32_t intv = 5;
uint32_t total_wait_time = 40;
uint32_t total_checks = total_wait_time / intv;
for (uint32_t check_num = 0; check_num < total_checks; check_num++) {
// Reset the previous value
act_mysql_conn_count = 0;
const string mysql_query_string {
"SELECT count(*) FROM INFORMATION_SCHEMA.PROCESSLIST WHERE"
" USER=\"" + string { cl.username } + "\""
//" USER=\"" + string { cl.username } + "\" and DB=\"backend_conn_ping_test\""
//" COMMAND=\"Sleep\" and USER=\"" + string { cl.username } + "\" and DB=\"backend_conn_ping_test\""
};
diag("Line:%d : Running: %s", __LINE__ , mysql_query_string.c_str());
vector<MYSQL*> svrs_conns {};
// Recreate the connections at each interval
for (const auto& svr_addr : svrs_addrs) {
const char* c_svr_addr { svr_addr.first.c_str() };
MYSQL* mysql = mysql_init(NULL);
if (!mysql_real_connect(mysql, c_svr_addr, cl.mysql_username, cl.mysql_password, NULL, svr_addr.second, NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(mysql));
res = EXIT_FAILURE;
goto cleanup;
}
svrs_conns.push_back(mysql);
}
for (MYSQL* mysql : svrs_conns) {
uint64_t tmp_mysql_conn_count = 0;
int q_res = get_query_result(mysql, mysql_query_string, tmp_mysql_conn_count);
if (q_res != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, "get_query_result() failed");
break;
}
act_mysql_conn_count += tmp_mysql_conn_count;
}
for (MYSQL* mysql : svrs_conns) {
mysql_close(mysql);
}
const string srv_ports {
std::accumulate(std::begin(svrs_addrs), std::end(svrs_addrs), string {},
[](const string& s1, const svr_addr& addr) -> string {
if (s1.empty()) {
return "'" + std::to_string(addr.second) + "'";
} else {
return s1 + ", '" + std::to_string(addr.second) + "'";
}
}
)
};
const string proxy_query {
"SELECT COUNT(*) FROM stats_mysql_free_connections WHERE"
" user='" + string { cl.username } + "'"
" AND hostgroup='" + std::to_string(hg) + "'"
" AND schema='backend_conn_ping_test'"
" AND srv_port IN (" + srv_ports + ")"
};
diag("Line:%d : Running: %s", __LINE__ , proxy_query.c_str());
q_res = get_query_result(admin, proxy_query, act_proxy_free_conn_count);
if (q_res != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, "get_query_result() failed");
break;
}
{
string query = "SELECT hostgroup , srv_host , srv_port , status , ConnUsed , ConnFree , ConnOK , ConnERR FROM stats_mysql_connection_pool";
diag("Line:%d : Running: %s", __LINE__ , query.c_str());
MYSQL_QUERY(admin, query.c_str());
MYSQL_RES* result = mysql_store_result(admin);
while (MYSQL_ROW row = mysql_fetch_row(result)) {
diag("%s , %s , %s , %s , %s , %s , %s , %s", row[0], row[1], row[2], row[3], row[4], row[5], row[6], row[7]);
}
mysql_free_result(result);
}
const string proxy_used_query {
"SELECT ConnUsed from stats_mysql_connection_pool WHERE hostgroup=" + std::to_string(hg)
};
q_res = get_query_result(admin, proxy_query, act_proxy_used_conn_count);
diag(
"Waiting and checking backend connections...:"
" { CheckNum: %d, ExpConns: %ld, ActMySQLConns: %ld, ActProxyFreeConns: %ld, ActProxyUsedConns: %ld }",
check_num, exp_conn_count, act_mysql_conn_count, act_proxy_free_conn_count, act_proxy_used_conn_count
);
diag("check_num = %u", check_num);
diag("exp_conn_count = %lu", exp_conn_count);
diag("act_mysql_conn_count = %lu", act_mysql_conn_count);
diag("act_proxy_free_conn_count = %lu", act_proxy_free_conn_count);
diag("act_proxy_used_conn_count = %lu", act_proxy_used_conn_count);
if (intv) {
diag("Line:%d : Sleeping %d" , __LINE__ , intv);
sleep(intv);
}
}
ok(
q_res == EXIT_SUCCESS && act_mysql_conn_count >= ((float) exp_conn_count * 0.95) // allow 5% margin of error
&&
((act_proxy_free_conn_count + act_proxy_used_conn_count) >= exp_conn_count),
"Created server connections should be properly maintained (pinged) by ProxySQL:"
" { ExpConns: %ld, ActMySQLConns: %ld, ActProxyConns: %ld }",
exp_conn_count, act_mysql_conn_count, act_proxy_free_conn_count
);
}
// 3. Check that no client side errors take place when exhausting backend connections
{
uint32_t broken_conns = 0;
diag("Line:%d : Running DO 1 on %lu connections", __LINE__, mysql_conns.size());
for (MYSQL* conn : mysql_conns) {
int rc = mysql_query(conn, string {"/* ;hostgroup=" + std::to_string(hg) + "*/ BEGIN"}.c_str());
if (rc != EXIT_SUCCESS) {
diag("'mysql_query' failed with error: '%s'", mysql_error(conn));
broken_conns += 1;
}
rc = mysql_query(conn, "DO 1");
if (rc != EXIT_SUCCESS) {
diag("'mysql_query' failed with error: '%s'", mysql_error(conn));
broken_conns += 1;
}
}
ok(broken_conns == 0, "Found '%d' client side errors due to broken connections.", broken_conns);
}
cleanup:
mysql_close(admin);
for (MYSQL* mysql : mysql_conns) {
mysql_close(mysql);
}
return res;
}
int wait_target_backend_conns(MYSQL* admin, uint32_t tg_backend_conns, uint32_t timeout) {
uint32_t waited = 0;
while (waited < timeout) {
uint64_t cur_conn_num = 0;
int q_res = get_query_result(admin, "SELECT SUM(ConnFree + ConnUsed) FROM stats_mysql_connection_pool", cur_conn_num);
if (q_res != EXIT_SUCCESS) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, "'get_query_result() failed.");
return -1;
}
if (cur_conn_num == tg_backend_conns) {
diag("tg_backend_conns: %d, cur_conn_num: %ld", tg_backend_conns, cur_conn_num);
break;
} else {
waited += 1;
diag("tg_backend_conns: %d, cur_conn_num: %ld , not matching after %u checks", tg_backend_conns, cur_conn_num, waited);
sleep(1);
}
}
return waited < timeout ? 0 : -2;
}
int main(int, char**) {
CommandLine cl;
plan(4);
if (cl.getEnv()) {
diag("Failed to get the required environmental variables.");
return EXIT_FAILURE;
}
struct rlimit limits { 0, 0 };
getrlimit(RLIMIT_NOFILE, &limits);
limits.rlim_cur = 10000;
setrlimit(RLIMIT_NOFILE, &limits);
MYSQL* proxy_mysql = mysql_init(NULL);
// Initialize connections
if (!proxy_mysql) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxy_mysql));
return exit_status();
}
if (!mysql_real_connect(proxy_mysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxy_mysql));
return exit_status();
}
// Create a new 'db' for connection filtering
MYSQL_QUERY(proxy_mysql, "CREATE DATABASE IF NOT EXISTS backend_conn_ping_test");
// Close no longer required connection
mysql_close(proxy_mysql);
MYSQL* admin = mysql_init(NULL);
if (!admin) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin));
return exit_status();
}
if (!mysql_real_connect(admin, cl.host, cl.admin_username, cl.admin_password, NULL, cl.admin_port, NULL, 0)) {
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin));
return exit_status();
}
double b_0 = 256;
double b = 128;
double freq = 1000;
double rate = 64 / (freq / 1000);
const char q_mysql_threads[] { "SELECT @@mysql-threads" };
ext_val_t<int32_t> ext_threads { mysql_query_ext_val(admin, q_mysql_threads, int32_t(0)) };
if (ext_threads.err != EXIT_SUCCESS) {
const string err { get_ext_val_err(admin, ext_threads) };
diag("Failed query query:`%s`, err:`%s`", q_mysql_threads, err.c_str());
return EXIT_FAILURE;
}
// IMPORTANT-NOTE:
// Variable 'ping_interval_server_msec' isn't relevant because ProxySQL isn't under load. The relevant
// variable passes to be `mysql-poll_timeout`; with each timeout threads will process idle sessions. If we
// wished to extend this test with a dummy load on ProxySQL, this should be uncommented.
// ===================================================================================================
// const char q_ping_intv[] { "SELECT @@mysql-ping_interval_server_msec" };
const char q_ping_intv[] { "SELECT @@mysql-poll_timeout" };
// ===================================================================================================
ext_val_t<int32_t> ext_ping_intv { mysql_query_ext_val(admin, q_ping_intv, int32_t(0)) };
if (ext_ping_intv.err != EXIT_SUCCESS) {
const string err { get_ext_val_err(admin, ext_ping_intv) };
diag("Failed query query:`%s`, err:`%s`", q_ping_intv, err.c_str());
return EXIT_FAILURE;
}
// IMPORTANT: Previously, this test gave ProxySQL only '1s' margin in the computation of the 'max
// connections' that can be maintained for the selected 'wait_timeout' interval. While this computation
// was correct, this could lead to failures due to small network stalls or slowdowns. Meaning that
// ProxySQL acknowledges the ping for the connection **before** MySQL actual response to it, i.e, the last
// time the connection was used by ProxySQL can diverge slightly from the time in which MySQL acknowledges
// the keep-alive. With only '1s' margin, this could result in lost connections for these slowdown
// scenarios.
//
// There are two main reasons to relax this margin. First, this is an artificial scenario created for this
// test, with a way too low 'wait_timetout', i.e, the real number of connections that ProxySQL can be kept
// alive with the current pinging algorithm, using a default or sensible 'wait_timeout' for most loads, is
// too large to even be practical. Secondly, testing this maximum number of connections that can be kept
// alive isn't the goal of this test, but to ensure correctness in the selection of connections to ping
// from the connection pool, which can be done with much smaller and less error prone numbers.
//
// Due to this, we are going to give a full extra grace interval in the computation for the maximum number
// of connections that could be kept alive. This should result in creating `(max conns) - (conns pinged in
// one interval)` connections. This should give enough room for the test not to fail even in the
// previously described scenarios. If further failures are detected, this **should be investigated** as
// could mean that something is not working as expected, either in the setup or the pinging algorithm
// itself.
int wait_timeout = 10;
uint32_t grace_timeout = wait_timeout - (ext_ping_intv.val / 1000) - 1;
uint32_t max_conns = max_pinged_conns(ext_threads.val, ext_ping_intv.val, grace_timeout);
uint32_t server_max_conns = max_conns + 1000;
double its = floor((max_conns - b_0) / b);
double conn_creation_intv = b_0/rate + (b/rate)*its;
double delay_s = conn_creation_intv / its;
diag("ProxySQL config mysql_threads:%d, ping_intv:%d", ext_threads.val, ext_ping_intv.val);
diag("Selected test params b_0:%lf, b:%lf, freq:%lf, rate:%lf", b_0, b, freq, rate);
diag("Computed test params max_conns=%d, intv=%lf, its=%lf", max_conns, conn_creation_intv, its);
// Cleanup previous backend connections
diag("Cleaning up previous backend connections...");
MYSQL_QUERY(admin, "UPDATE mysql_servers SET max_connections=0");
MYSQL_QUERY(admin, "LOAD MYSQL SERVERS TO RUNTIME");
// Wait for backend connection cleanup
int w_res = wait_target_backend_conns(admin, 0, 10);
if (w_res != EXIT_SUCCESS) {
if (w_res == -2) {
const char* err_msg = "'wait_target_backend_conns()' timed out";
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, err_msg);
}
return exit_status();
}
diag("Setting mysql_servers config...");
{
string query { "UPDATE mysql_servers SET max_connections=" + std::to_string(server_max_conns) };
diag("Running: %s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
query = "LOAD MYSQL SERVERS TO RUNTIME";
diag("Running: %s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
}
diag("Setting ProxySQL config...");
{
// Set the backend connections ping frequency
string query = string { "SET mysql-ping_interval_server_msec=" + std::to_string(freq) };
diag("%s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
// Make sure no connection cleanup takes place
query = "SET mysql-free_connections_pct=100";
diag("%s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
// Don't retry on failure
query = "SET mysql-query_retries_on_failure=0";
diag("%s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
query = "SET mysql-max_connections=" + std::to_string(server_max_conns*3);
diag("%s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
// Set a higher max_connection number for the servers
query = "LOAD MYSQL VARIABLES TO RUNTIME";
diag("%s", query.c_str());
MYSQL_QUERY(admin, query.c_str());
}
// Configure MySQL infra servers with: 'wait_timeout' and 'max_connections'
vector<pair<mysql_res_row, srv_cfg>> servers_old_configs {};
diag("Configure 'MYSQL' infra servers...");
{
MYSQL_QUERY(admin, "SELECT DISTINCT hostname, port FROM mysql_servers WHERE hostgroup_id IN (0,1)");
MYSQL_RES* my_servers_res = mysql_store_result(admin);
vector<mysql_res_row> servers_rows = extract_mysql_rows(my_servers_res);
mysql_free_result(my_servers_res);
if (servers_rows.empty()) {
fprintf(stderr, "File %s, line %d, Error: Invalid result returned from 'mysql_servers'\n", __FILE__, __LINE__);
return exit_status();
}
srv_cfg new_srv_cfg { { "wait_timeout", wait_timeout }, { "max_connections", 5000 } };
for (const mysql_res_row& srv_row : servers_rows) {
srv_cfg old_srv_cfg {};
diag("Line:%d : %s:%s", __LINE__ , srv_row[0].c_str(), srv_row[1].c_str());
int cfg_res = change_mysql_cfg(cl, srv_row[0], srv_row[1], new_srv_cfg, old_srv_cfg);
if (cfg_res != EXIT_SUCCESS) {
return exit_status();
} else {
servers_old_configs.push_back({ srv_row, old_srv_cfg });
}
}
usleep(500 * 1000);
}
test_params_t test_params { b_0, b, its, delay_s };
vector<svr_addr> s_server_test;
vector<svr_addr> m_server_test;
const string docker_mode = getenv("DOCKER_MODE");
if (docker_mode.find("dns") == docker_mode.size() - 3) {
s_server_test.assign({ { "mysql1.infra-mysql57", 3306 } });
m_server_test.assign({
{ "mysql1.infra-mysql57", 3306 },
{ "mysql2.infra-mysql57", 3306 },
{ "mysql3.infra-mysql57", 3306 }
});
} else {
s_server_test.assign({ { "127.0.0.1", 13306 } });
m_server_test.assign({ { "127.0.0.1", 13306 }, { "127.0.0.1", 13307 }, { "127.0.0.1", 13308 } });
}
for (const svr_addr& svr : s_server_test) {
diag("Line:%d : s_server_test: %s:%d", __LINE__, svr.first.c_str(), svr.second);
}
for (const svr_addr& svr : m_server_test) {
diag("Line:%d : m_server_test: %s:%d", __LINE__, svr.first.c_str(), svr.second);
}
diag("Performing 'check_backend_conns()' for servers: '%s'", nlohmann::json(s_server_test).dump().c_str());
int s_server_rc = check_backend_conns(cl, test_params, 0, s_server_test);
if (s_server_rc == EXIT_SUCCESS) {
diag("Cleaning up previous backend connections...");
string query = "UPDATE mysql_servers SET max_connections=0";
diag("Line:%d : Running: %s", __LINE__ , query.c_str());
MYSQL_QUERY(admin, query.c_str());
query = "LOAD MYSQL SERVERS TO RUNTIME";
diag("Line:%d : Running: %s", __LINE__ , query.c_str());
MYSQL_QUERY(admin, query.c_str());
int w_res = wait_target_backend_conns(admin, 0, 10);
if (w_res != EXIT_SUCCESS) {
string err_msg {};
if (w_res == -2) {
err_msg = "'wait_target_backend_conns()' timed out";
} else {
err_msg = "'wait_target_backend_conns()' failed";
}
fprintf(stderr, "File %s, line %d, Error: \"%s\"\n", __FILE__, __LINE__, err_msg.c_str());
}
diag("Reconfiguring ProxySQL 'mysql_servers' after connection cleanup");
query = "UPDATE mysql_servers SET max_connections=" + std::to_string(server_max_conns);
diag("Line:%d : Running: %s", __LINE__ , query.c_str());
MYSQL_QUERY(admin, query.c_str());
query = "LOAD MYSQL SERVERS TO RUNTIME";
diag("Line:%d : Running: %s", __LINE__ , query.c_str());
MYSQL_QUERY(admin, query.c_str());
if (w_res == EXIT_SUCCESS) {
diag("Performing 'check_backend_conns()' for servers: '%s'", nlohmann::json(m_server_test).dump().c_str());
int m_server_rc = check_backend_conns(cl, test_params, 1, m_server_test);
if (m_server_rc == EXIT_FAILURE) {
diag("'check_backend_conns()' failed for servers: '%s'", nlohmann::json(s_server_test).dump().c_str());
}
}
} else {
diag("'check_backend_conns()' failed for servers: '%s'", nlohmann::json(m_server_test).dump().c_str());
}
// NOTE-TODO: We are required to recover the connection pool to an usable status for subsequent tests.
// This could be done giving sensible values to `mysql-free_connections_pct`, for keeping always a good
// number of connections below the max server threshold, and `mysql-ping_interval_server_msec`, ensuring
// that after the test is finished, those idle connections would be release till reaching
// `free_connections_pct`. These conditions are not ensured right now.
{
diag("Connection pool cleanup before test exit");
MYSQL_QUERY_T(admin, "UPDATE mysql_servers SET max_connections=0");
MYSQL_QUERY_T(admin, "LOAD MYSQL SERVERS TO RUNTIME");
diag("Waiting for connection cleanup on ProxySQL...");
w_res = wait_target_backend_conns(admin, 0, 10);
if (w_res != EXIT_SUCCESS) {
string err_msg {};
if (w_res == -2) {
err_msg = "'wait_target_backend_conns()' timed out";
} else {
err_msg = "'wait_target_backend_conns()' failed";
}
fprintf(stderr, "File %s, line %d, Error: \"%s\"\n", __FILE__, __LINE__, err_msg.c_str());
}
}
diag("Restoring previous 'MySQL' servers infra config...");
{
for (const auto& server_old_config : servers_old_configs) {
const mysql_res_row& res_row = server_old_config.first;
const srv_cfg& old_srv_config = server_old_config.second;
srv_cfg _tmp_conf {};
int cfg_res = change_mysql_cfg(cl, res_row[0], res_row[1], old_srv_config, _tmp_conf);
if (cfg_res != EXIT_SUCCESS) {
return EXIT_FAILURE;
}
}
}
mysql_close(admin);
return exit_status();
}