#include "../deps/json/json.hpp" using json = nlohmann::json; #define PROXYJSON //#define __CLASS_STANDARD_MYSQL_THREAD_H #include #include #include "proxysql_utils.h" #include "MySQL_HostGroups_Manager.h" #include "prometheus_helpers.h" #define MYSQL_THREAD_IMPLEMENTATION #include "proxysql.h" #include "cpp.h" #include "MySQL_Thread.h" #include #include #include "re2/re2.h" #include "re2/regexp.h" #include "MySQL_Data_Stream.h" #include "MySQL_Query_Processor.h" #include "StatCounters.h" #include "MySQL_PreparedStatement.h" #include "MySQL_Logger.hpp" #include using std::vector; using std::function; #ifdef DEBUG static MySQL_Session *sess_stopat; #endif #ifdef epoll_create1 #define EPOLL_CREATE epoll_create1(0) #else #define EPOLL_CREATE epoll_create(1) #endif #define PROXYSQL_LISTEN_LEN 1024 #define MIN_THREADS_FOR_MAINTENANCE 8 /** * @brief Helper macro to stringify a macro argument. * * This macro takes a single argument 'x' and converts it into a string literal. * It is a helper macro used by the STRINGIFY macro. * * @param x The macro argument to be converted into a string. * @return The string representation of the macro argument. */ #define STRINGIFY_HELPER(x) #x /** * @brief Macro to stringify a macro argument. * * This macro takes a single argument 'x' and converts it into a string literal * using the STRINGIFY_HELPER macro. * * @param x The macro argument to be converted into a string. * @return The string representation of the macro argument. */ #define STRINGIFY(x) STRINGIFY_HELPER(x) /** * @brief Refreshes a boolean variable from the MySQL thread. * * This macro updates the value of a boolean variable named 'name' from * the MySQL thread. It retrieves the value using the 'get_variable_int' function * from the 'GloMTH' object and assigns it to the variable 'mysql_thread___name'. * The retrieved integer value is cast to a boolean before assignment. * * @param name The name of the boolean variable to be refreshed. */ #define REFRESH_VARIABLE_BOOL(name) \ mysql_thread___ ## name = (bool)GloMTH->get_variable_int((char *)STRINGIFY(name)) /** * @brief Refreshes an integer variable from the MySQL thread. * * This macro updates the value of an integer variable named 'name' from * the MySQL thread. It retrieves the value using the 'get_variable_int' function * from the 'GloMTH' object and assigns it to the variable 'mysql_thread___name'. * * @param name The name of the integer variable to be refreshed. */ #define REFRESH_VARIABLE_INT(name) \ mysql_thread___ ## name = GloMTH->get_variable_int((char *)STRINGIFY(name)) /** * @brief Refreshes a character variable from the MySQL thread. * * This macro updates the value of a character variable named 'name' from * the MySQL thread. It retrieves the value using the 'get_variable_string' function * from the 'GloMTH' object and assigns it to the variable 'mysql_thread___name'. * If the variable 'mysql_thread___name' was previously allocated memory, * it frees that memory before assigning the new value. * * @param name The name of the character variable to be refreshed. */ #define REFRESH_VARIABLE_CHAR(name) \ do { \ if (mysql_thread___ ## name) free(mysql_thread___ ## name); \ mysql_thread___ ## name = GloMTH->get_variable_string((char *)STRINGIFY(name)); \ } while (0) extern MySQL_Query_Processor* GloMyQPro; extern MySQL_Authentication *GloMyAuth; extern MySQL_Threads_Handler *GloMTH; extern MySQL_Monitor *GloMyMon; extern MySQL_Logger *GloMyLogger; typedef struct mythr_st_vars { enum MySQL_Thread_status_variable v_idx; p_th_counter::metric m_idx; char * name; uint32_t conv; } mythr_st_vars_t; typedef struct mythr_g_st_vars { enum MySQL_Thread_status_variable v_idx; p_th_gauge::metric m_idx; char * name; uint32_t conv; } mythr_g_st_vars_t; // Note: the order here is not important. mythr_st_vars_t MySQL_Thread_status_variables_counter_array[] { { st_var_backend_stmt_prepare, p_th_counter::com_backend_stmt_prepare, (char *)"Com_backend_stmt_prepare" }, { st_var_backend_stmt_execute, p_th_counter::com_backend_stmt_execute, (char *)"Com_backend_stmt_execute" }, { st_var_backend_stmt_close, p_th_counter::com_backend_stmt_close, (char *)"Com_backend_stmt_close" }, { st_var_frontend_stmt_prepare, p_th_counter::com_frontend_stmt_prepare, (char *)"Com_frontend_stmt_prepare" }, { st_var_frontend_stmt_execute, p_th_counter::com_frontend_stmt_execute, (char *)"Com_frontend_stmt_execute" }, { st_var_frontend_stmt_close, p_th_counter::com_frontend_stmt_close, (char *)"Com_frontend_stmt_close" }, { st_var_queries, p_th_counter::questions, (char *)"Questions" }, { st_var_queries_slow, p_th_counter::slow_queries, (char *)"Slow_queries" }, { st_var_queries_gtid, p_th_counter::gtid_consistent_queries, (char *)"GTID_consistent_queries" }, { st_var_gtid_session_collected,p_th_counter::gtid_session_collected, (char *)"GTID_session_collected" }, { st_var_queries_backends_bytes_recv, p_th_counter::queries_backends_bytes_recv, (char *)"Queries_backends_bytes_recv" }, { st_var_queries_backends_bytes_sent, p_th_counter::queries_backends_bytes_sent, (char *)"Queries_backends_bytes_sent" }, { st_var_queries_frontends_bytes_recv, p_th_counter::queries_frontends_bytes_recv, (char *)"Queries_frontends_bytes_recv" }, { st_var_queries_frontends_bytes_sent, p_th_counter::queries_frontends_bytes_sent, (char *)"Queries_frontends_bytes_sent" }, { st_var_query_processor_time , p_th_counter::query_processor_time_nsec, (char *)"Query_Processor_time_nsec", 1000*1000*1000 }, { st_var_backend_query_time , p_th_counter::backend_query_time_nsec, (char *)"Backend_query_time_nsec", 1000*1000*1000 }, { st_var_ConnPool_get_conn_latency_awareness , p_th_counter::connpool_get_conn_latency_awareness, (char *)"ConnPool_get_conn_latency_awareness" }, { st_var_ConnPool_get_conn_immediate, p_th_counter::connpool_get_conn_immediate, (char *)"ConnPool_get_conn_immediate" }, { st_var_ConnPool_get_conn_success, p_th_counter::connpool_get_conn_success, (char *)"ConnPool_get_conn_success" }, { st_var_ConnPool_get_conn_failure, p_th_counter::connpool_get_conn_failure, (char *)"ConnPool_get_conn_failure" }, { st_var_killed_connections, p_th_counter::mysql_killed_backend_connections, (char *)"mysql_killed_backend_connections" }, { st_var_killed_queries, p_th_counter::mysql_killed_backend_queries, (char *)"mysql_killed_backend_queries" }, { st_var_hostgroup_locked_set_cmds, p_th_counter::hostgroup_locked_set_cmds, (char *)"hostgroup_locked_set_cmds" }, { st_var_hostgroup_locked_queries, p_th_counter::hostgroup_locked_queries, (char *)"hostgroup_locked_queries" }, { st_var_unexpected_com_quit, p_th_counter::mysql_unexpected_frontend_com_quit,(char *)"mysql_unexpected_frontend_com_quit" }, { st_var_unexpected_com_ping, p_th_counter::mysql_unexpected_frontend_com_ping,(char *)"mysql_unexpected_frontend_com_ping" }, { st_var_unexpected_packet, p_th_counter::mysql_unexpected_frontend_packets,(char *)"mysql_unexpected_frontend_packets" }, { st_var_queries_with_max_lag_ms__total_wait_time_us , p_th_counter::queries_with_max_lag_ms__total_wait_time_us, (char *)"queries_with_max_lag_ms__total_wait_time_us" }, { st_var_queries_with_max_lag_ms__delayed , p_th_counter::queries_with_max_lag_ms__delayed, (char *)"queries_with_max_lag_ms__delayed" }, { st_var_queries_with_max_lag_ms, p_th_counter::queries_with_max_lag_ms, (char *)"queries_with_max_lag_ms" }, { st_var_backend_lagging_during_query,p_th_counter::backend_lagging_during_query, (char *)"backend_lagging_during_query" }, { st_var_backend_offline_during_query,p_th_counter::backend_offline_during_query, (char *)"backend_offline_during_query" }, { st_var_aws_aurora_replicas_skipped_during_query , p_th_counter::aws_aurora_replicas_skipped_during_query, (char *)"get_aws_aurora_replicas_skipped_during_query" }, { st_var_automatic_detected_sqli, p_th_counter::automatic_detected_sql_injection, (char *)"automatic_detected_sql_injection" }, { st_var_mysql_whitelisted_sqli_fingerprint,p_th_counter::mysql_whitelisted_sqli_fingerprint, (char *)"mysql_whitelisted_sqli_fingerprint" }, { st_var_ai_detected_anomalies, p_th_counter::ai_detected_anomalies, (char *)"ai_detected_anomalies" }, { st_var_ai_blocked_queries, p_th_counter::ai_blocked_queries, (char *)"ai_blocked_queries" }, { st_var_max_connect_timeout_err, p_th_counter::max_connect_timeouts, (char *)"max_connect_timeouts" }, { st_var_generated_pkt_err, p_th_counter::generated_error_packets, (char *)"generated_error_packets" }, { st_var_client_host_error_killed_connections, p_th_counter::client_host_error_killed_connections, (char *)"client_host_error_killed_connections" }, { st_var_set_wait_timeout_commands, p_th_counter::mysql_set_wait_timeout_commands, (char *)"mysql_set_wait_timeout_commands" }, { st_var_timeout_terminated_connections, p_th_counter::mysql_timeout_terminated_connections, (char *)"mysql_timeout_terminated_connections" }, }; mythr_g_st_vars_t MySQL_Thread_status_variables_gauge_array[] { { st_var_hostgroup_locked, p_th_gauge::client_connections_hostgroup_locked, (char *)"Client_Connections_hostgroup_locked" } }; extern mysql_variable_st mysql_tracked_variables[]; #ifdef __cplusplus extern "C" { #endif /* __cplusplus */ #ifdef __cplusplus } #endif /* __cplusplus */ #ifdef DEBUG #define DEB "_DEBUG" #else #define DEB "" #endif /* DEBUG */ #define MYSQL_THREAD_VERSION "0.2.0902" DEB #define DEFAULT_NUM_THREADS 4 #define DEFAULT_STACK_SIZE 1024*1024 #define SESSIONS_FOR_CONNECTIONS_HANDLER 64 __thread unsigned int __thread_MySQL_Thread_Variables_version; volatile static unsigned int __global_MySQL_Thread_Variables_version; MySQL_Listeners_Manager::MySQL_Listeners_Manager() { ifaces=new PtrArray(); } MySQL_Listeners_Manager::~MySQL_Listeners_Manager() { while (ifaces->len) { iface_info *ifi=(iface_info *)ifaces->remove_index_fast(0); shutdown(ifi->fd,SHUT_RDWR); close(ifi->fd); if (ifi->port==0) { unlink(ifi->address); } delete ifi; } delete ifaces; ifaces=NULL; } int MySQL_Listeners_Manager::add(const char *iface, unsigned int num_threads, int **perthrsocks) { for (unsigned int i=0; ilen; i++) { iface_info *ifi=(iface_info *)ifaces->index(i); if (strcmp(ifi->iface,iface)==0) { return -1; } } char *address=NULL; char *port=NULL; int s = -1; char *h = NULL; bool is_ipv6 = false; if (*(char *)iface == '[') { is_ipv6 = true; char *p = strchr((char *)iface, ']'); if (p == NULL) { proxy_error("Invalid IPv6 address: %s\n", iface); return -1; } h = (char *)++iface; // remove first '[' *p = '\0'; iface = p++; // remove last ']' address = h; port = ++p; // remove ':' } else { c_split_2(iface, ":" , &address, &port); } #ifdef SO_REUSEPORT if (GloVars.global.reuseport==false) { s = ( atoi(port) ? listen_on_port(address, atoi(port), PROXYSQL_LISTEN_LEN) : listen_on_unix(address, PROXYSQL_LISTEN_LEN)); } else { if (atoi(port)==0) { s = listen_on_unix(address, PROXYSQL_LISTEN_LEN); } else { // for TCP we will use SO_REUSEPORT int *l_perthrsocks=(int *)malloc(sizeof(int)*num_threads); unsigned int i; for (i=0;iadd(ifi); l_perthrsocks[i]=s; } *perthrsocks=l_perthrsocks; s=0; } } #else s = ( atoi(port) ? listen_on_port(address, atoi(port), PROXYSQL_LISTEN_LEN) : listen_on_unix(address, PROXYSQL_LISTEN_LEN)); #endif /* SO_REUSEPORT */ if (s==-1) { if (is_ipv6 == false) { free(address); free(port); } return s; } if (s>0) { ioctl_FIONBIO(s,1); iface_info *ifi=new iface_info((char *)iface, address, atoi(port), s); ifaces->add(ifi); } if (is_ipv6 == false) { free(address); free(port); } return s; } int MySQL_Listeners_Manager::find_idx(const char *iface) { for (unsigned int i=0; ilen; i++) { iface_info *ifi=(iface_info *)ifaces->index(i); if (strcmp(ifi->iface,iface)==0) { return i; } } return -1; } iface_info * MySQL_Listeners_Manager::find_iface_from_fd(int fd) { for (unsigned int i=0; ilen; i++) { iface_info *ifi=(iface_info *)ifaces->index(i); if (ifi->fd==fd) { return ifi; } } return NULL; } int MySQL_Listeners_Manager::find_idx(const char *address, int port) { for (unsigned int i=0; ilen; i++) { iface_info *ifi=(iface_info *)ifaces->index(i); if (strcmp(ifi->address,address)==0 && ifi->port==port) { return i; } } return -1; } int MySQL_Listeners_Manager::get_fd(unsigned int idx) { iface_info *ifi=(iface_info *)ifaces->index(idx); return ifi->fd; } void MySQL_Listeners_Manager::del(unsigned int idx) { iface_info *ifi=(iface_info *)ifaces->remove_index_fast(idx); if (ifi->port==0) { unlink(ifi->address); } delete ifi; } static char * mysql_thread_variables_names[]= { (char *)"shun_on_failures", (char *)"shun_recovery_time_sec", (char *)"unshun_algorithm", (char *)"query_retries_on_failure", (char *)"client_host_cache_size", (char *)"client_host_error_counts", (char *)"connect_retries_on_failure", (char *)"connect_retries_delay", (char *)"connection_delay_multiplex_ms", (char *)"connection_max_age_ms", (char *)"connect_timeout_client", (char *)"connect_timeout_server", (char *)"connect_timeout_server_max", (char *)"enable_client_deprecate_eof", (char *)"enable_server_deprecate_eof", (char *)"enable_load_data_local_infile", (char *)"eventslog_filename", (char *)"eventslog_filesize", (char *)"eventslog_buffer_history_size", (char *)"eventslog_table_memory_size", (char *)"eventslog_buffer_max_query_length", (char *)"eventslog_default_log", (char *)"eventslog_format", (char *)"eventslog_stmt_parameters", (char *)"auditlog_filename", (char *)"auditlog_filesize", //(char *)"default_charset", // removed in 2.0.13 . Obsoleted previously using MySQL_Variables instead (char *)"handle_unknown_charset", (char *)"free_connections_pct", (char *)"connection_warming", #ifdef IDLE_THREADS (char *)"session_idle_ms", #endif // IDLE_THREADS (char *)"have_ssl", (char *)"have_compress", (char *)"interfaces", (char *)"log_mysql_warnings_enabled", (char *)"monitor_enabled", (char *)"monitor_history", (char *)"monitor_connect_interval", (char *)"monitor_connect_timeout", (char *)"monitor_ping_interval", (char *)"monitor_ping_max_failures", (char *)"monitor_ping_timeout", (char *)"monitor_aws_rds_topology_discovery_interval", (char *)"monitor_read_only_interval", (char *)"monitor_read_only_timeout", (char *)"monitor_read_only_max_timeout_count", (char *)"monitor_replication_lag_group_by_host", (char *)"monitor_replication_lag_interval", (char *)"monitor_replication_lag_timeout", (char *)"monitor_replication_lag_count", (char *)"monitor_groupreplication_healthcheck_interval", (char *)"monitor_groupreplication_healthcheck_timeout", (char *)"monitor_groupreplication_healthcheck_max_timeout_count", (char *)"monitor_groupreplication_max_transactions_behind_count", (char *)"monitor_groupreplication_max_transactions_behind_for_read_only", (char *)"monitor_galera_healthcheck_interval", (char *)"monitor_galera_healthcheck_timeout", (char *)"monitor_galera_healthcheck_max_timeout_count", (char *)"monitor_username", (char *)"monitor_password", (char *)"monitor_replication_lag_use_percona_heartbeat", (char *)"monitor_query_interval", (char *)"monitor_query_timeout", (char *)"monitor_slave_lag_when_null", (char *)"monitor_threads_min", (char *)"monitor_threads_max", (char *)"monitor_threads_queue_maxsize", (char *)"monitor_local_dns_cache_ttl", (char *)"monitor_local_dns_cache_refresh_interval", (char *)"monitor_local_dns_resolver_queue_maxsize", (char *)"monitor_wait_timeout", (char *)"monitor_writer_is_also_reader", (char *)"max_allowed_packet", (char *)"tcp_keepalive_time", (char *)"use_tcp_keepalive", (char *)"automatic_detect_sqli", (char *)"firewall_whitelist_enabled", (char *)"firewall_whitelist_errormsg", (char *)"throttle_connections_per_sec_to_hostgroup", (char *)"max_transaction_idle_time", (char *)"max_transaction_time", (char *)"multiplexing", (char *)"log_unhealthy_connections", (char *)"enforce_autocommit_on_reads", (char *)"autocommit_false_not_reusable", (char *)"autocommit_false_is_transaction", (char *)"verbose_query_error", (char *)"hostgroup_manager_verbose", (char *)"binlog_reader_connect_retry_msec", (char *)"threshold_query_length", (char *)"threshold_resultset_size", (char *)"query_digests_max_digest_length", (char *)"query_digests_max_query_length", (char *)"query_digests_grouping_limit", (char *)"query_digests_groups_grouping_limit", (char *)"query_rules_fast_routing_algorithm", (char *)"wait_timeout", (char *)"throttle_max_bytes_per_second_to_client", (char *)"throttle_ratio_server_to_client", (char *)"max_connections", (char *)"max_stmts_per_connection", (char *)"max_stmts_cache", (char *)"mirror_max_concurrency", (char *)"mirror_max_queue_length", (char *)"default_max_latency_ms", (char *)"default_query_delay", (char *)"default_query_timeout", (char *)"query_processor_iterations", (char *)"query_processor_regex", (char *)"set_query_lock_on_hostgroup", (char *)"set_parser_algorithm", (char *)"reset_connection_algorithm", (char *)"auto_increment_delay_multiplex", (char *)"auto_increment_delay_multiplex_timeout_ms", (char *)"long_query_time", (char *)"query_cache_size_MB", (char *)"query_cache_soft_ttl_pct", (char *)"query_cache_handle_warnings", (char *)"ping_interval_server_msec", (char *)"ping_timeout_server", (char *)"default_schema", (char *)"poll_timeout", (char *)"poll_timeout_on_failure", (char *)"server_capabilities", (char *)"server_version", (char *)"keep_multiplexing_variables", (char *)"default_authentication_plugin", (char *)"kill_backend_connection_when_disconnect", (char *)"client_session_track_gtid", (char *)"sessions_sort", #ifdef IDLE_THREADS (char *)"session_idle_show_processlist", #endif // IDLE_THREADS (char *)"show_processlist_extended", (char *)"processlist_max_query_length", (char *)"commands_stats", (char *)"query_digests", (char *)"query_digests_lowercase", (char *)"query_digests_replace_null", (char *)"query_digests_no_digits", (char *)"query_digests_normalize_digest_text", (char *)"query_digests_track_hostname", (char *)"query_digests_keep_comment", (char *)"parse_failure_logs_digest", (char *)"servers_stats", (char *)"default_reconnect", #ifdef DEBUG (char *)"session_debug", #endif /* DEBUG */ (char *)"ssl_p2s_ca", (char *)"ssl_p2s_capath", (char *)"ssl_p2s_cert", (char *)"ssl_p2s_key", (char *)"ssl_p2s_cipher", (char *)"ssl_p2s_crl", (char *)"ssl_p2s_crlpath", (char *)"stacksize", (char *)"threads", (char *)"init_connect", (char *)"ldap_user_variable", (char *)"add_ldap_user_comment", (char *)"default_session_track_gtids", (char *)"connpoll_reset_queue_length", (char *)"min_num_servers_lantency_awareness", (char *)"aurora_max_lag_ms_only_read_from_replicas", (char *)"stats_time_backend_query", (char *)"stats_time_query_processor", (char *)"query_cache_stores_empty_result", (char *)"data_packets_history_size", (char *)"handle_warnings", (char *)"evaluate_replication_lag_on_servers_load", (char *)"proxy_protocol_networks", (char *)"protocol_compression_level", (char *)"ignore_min_gtid_annotations", (char *)"fast_forward_grace_close_ms", NULL }; using metric_name = std::string; using metric_help = std::string; using metric_tags = std::map; using th_counter_tuple = std::tuple< p_th_counter::metric, metric_name, metric_help, metric_tags >; using th_gauge_tuple = std::tuple< p_th_gauge::metric, metric_name, metric_help, metric_tags >; using th_counter_vector = std::vector; using th_gauge_vector = std::vector; /** * @brief Metrics map holding the metrics for the MySQL_Thread module. * * @note Many metrics in this map, share a common "id name", because * they differ only by label, because of this, HELP is shared between * them. For better visual identification of this groups they are * separated using a line separator comment. */ const std::tuple th_metrics_map = std::make_tuple( th_counter_vector { // ==================================================================== std::make_tuple ( p_th_counter::queries_backends_bytes_sent, "proxysql_queries_backends_bytes_total", "Total number of bytes (sent|received) in backend connections.", metric_tags { { "traffic_flow", "sent" } } ), std::make_tuple ( p_th_counter::queries_backends_bytes_recv, "proxysql_queries_backends_bytes_total", "Total number of bytes (sent|received) in backend connections.", metric_tags { { "traffic_flow", "received" } } ), // ==================================================================== // ==================================================================== std::make_tuple ( p_th_counter::queries_frontends_bytes_sent, "proxysql_queries_frontends_bytes_total", "Total number of bytes (sent|received) in frontend connections.", metric_tags { { "traffic_flow", "sent" } } ), std::make_tuple ( p_th_counter::queries_frontends_bytes_recv, "proxysql_queries_frontends_bytes_total", "Total number of bytes (sent|received) in frontend connections.", metric_tags { { "traffic_flow", "received" } } ), // ==================================================================== std::make_tuple ( p_th_counter::query_processor_time_nsec, "proxysql_query_processor_time_seconds_total", "The time spent inside the \"Query Processor\" to determine what action needs to be taken with the query (internal module).", metric_tags {} ), std::make_tuple ( p_th_counter::backend_query_time_nsec, "proxysql_backend_query_time_seconds_total", "Time spent making network calls to communicate with the backends.", metric_tags {} ), // ==================================================================== std::make_tuple ( p_th_counter::com_backend_stmt_prepare, "proxysql_com_backend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by ProxySQL against the backends.", metric_tags { { "op", "prepare" } } ), std::make_tuple ( p_th_counter::com_backend_stmt_execute, "proxysql_com_backend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by ProxySQL against the backends.", metric_tags { { "op", "execute" } } ), std::make_tuple ( p_th_counter::com_backend_stmt_close, "proxysql_com_backend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by ProxySQL against the backends.", metric_tags { { "op", "close" } } ), // ==================================================================== // ==================================================================== std::make_tuple ( p_th_counter::com_frontend_stmt_prepare, "proxysql_com_frontend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by clients.", metric_tags { { "op", "prepare" } } ), std::make_tuple ( p_th_counter::com_frontend_stmt_execute, "proxysql_com_frontend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by clients.", metric_tags { { "op", "execute" } } ), std::make_tuple ( p_th_counter::com_frontend_stmt_close, "proxysql_com_frontend_stmt_total", "Represents the number of statements (PREPARE|EXECUTE|CLOSE) executed by clients.", metric_tags { { "op", "close" } } ), // ==================================================================== std::make_tuple ( p_th_counter::questions, "proxysql_questions_total", "The total number of client requests / statements executed.", metric_tags {} ), std::make_tuple ( p_th_counter::slow_queries, "proxysql_slow_queries_total", "The total number of queries with an execution time greater than \"mysql-long_query_time\" milliseconds.", metric_tags {} ), std::make_tuple ( p_th_counter::gtid_consistent_queries, "proxysql_gtid_consistent_queries_total", "Total queries with GTID consistent read.", metric_tags {} ), std::make_tuple ( p_th_counter::gtid_session_collected, "proxysql_gtid_session_collected_total", "Total queries with GTID session state.", metric_tags {} ), // ==================================================================== std::make_tuple ( p_th_counter::connpool_get_conn_latency_awareness, "proxysql_connpool_get_conn_success_latency_awareness_total", "The connection was picked using the latency awareness algorithm.", metric_tags {} ), std::make_tuple ( p_th_counter::connpool_get_conn_immediate, "proxysql_connpool_get_conn_success_immediate_total", "The connection is provided from per-thread cache.", metric_tags {} ), std::make_tuple ( p_th_counter::connpool_get_conn_success, "proxysql_connpool_get_conn_success_total", "The session is able to get a connection, either from per-thread cache or connection pool.", metric_tags {} ), std::make_tuple ( p_th_counter::connpool_get_conn_failure, "proxysql_connpool_get_conn_failure_total", "The connection pool cannot provide any connection.", metric_tags {} ), // ==================================================================== std::make_tuple ( p_th_counter::generated_error_packets, "proxysql_generated_error_packets_total", "Total generated error packets.", metric_tags {} ), std::make_tuple ( p_th_counter::max_connect_timeouts, "proxysql_max_connect_timeouts_total", "Maximum connection timeout reached when trying to connect to backend sever.", metric_tags {} ), std::make_tuple ( p_th_counter::backend_lagging_during_query, "proxysql_backend_lagging_during_query_total", "Query failed because server was shunned due to lag.", metric_tags {} ), std::make_tuple ( p_th_counter::backend_offline_during_query, "proxysql_backend_offline_during_query_total", "Query failed because server was offline.", metric_tags {} ), std::make_tuple ( p_th_counter::queries_with_max_lag_ms, "proxysql_queries_with_max_lag_total", "Received queries that have a 'max_lag' attribute.", metric_tags {} ), std::make_tuple ( p_th_counter::queries_with_max_lag_ms__delayed, "proxysql_queries_with_max_lag__delayed_total", "Query delayed because no connection was selected due to 'max_lag' annotation.", metric_tags {} ), std::make_tuple ( p_th_counter::queries_with_max_lag_ms__total_wait_time_us, "proxysql_queries_with_max_lag__total_wait_time_total", "Total waited time due to connection selection because of 'max_lag' annotation.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_unexpected_frontend_com_ping, "proxysql_mysql_unexpected_frontend_com_ping_total", "Unexpected 'COM_PING' received from the client.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_unexpected_frontend_com_quit, "proxysql_mysql_unexpected_frontend_com_quit_total", "Unexpected 'COM_QUIT' received from the client.", metric_tags {} ), std::make_tuple ( p_th_counter::hostgroup_locked_set_cmds, "proxysql_hostgroup_locked_set_cmds_total", "Total number of connections that have been locked in a hostgroup.", metric_tags {} ), std::make_tuple ( p_th_counter::hostgroup_locked_queries, "proxysql_hostgroup_locked_queries_total", "Query blocked because connection is locked into some hostgroup but is trying to reach other.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_unexpected_frontend_packets, "proxysql_mysql_unexpected_frontend_packets_total", "Unexpected packet received from client.", metric_tags {} ), std::make_tuple ( p_th_counter::aws_aurora_replicas_skipped_during_query, "proxysql_aws_aurora_replicas_skipped_during_query_total", "Replicas skipped due to current lag being higher than 'max_lag' annotation.", metric_tags {} ), std::make_tuple ( p_th_counter::automatic_detected_sql_injection, "proxysql_automatic_detected_sql_injection_total", "Blocked a detected 'sql injection' attempt.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_whitelisted_sqli_fingerprint, "proxysql_mysql_whitelisted_sqli_fingerprint_total", "Detected a whitelisted 'sql injection' fingerprint.", metric_tags {} ), std::make_tuple ( p_th_counter::ai_detected_anomalies, "proxysql_ai_detected_anomalies_total", "AI Anomaly Detection detected anomalous query behavior.", metric_tags {} ), std::make_tuple ( p_th_counter::ai_blocked_queries, "proxysql_ai_blocked_queries_total", "AI Anomaly Detection blocked a query.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_killed_backend_connections, "proxysql_mysql_killed_backend_connections_total", "Number of backend connection killed.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_killed_backend_queries, "proxysql_mysql_killed_backend_queries_total", "Killed backend queries.", metric_tags {} ), std::make_tuple ( p_th_counter::client_host_error_killed_connections, "proxysql_client_host_error_killed_connections", "Killed client connections because address exceeded 'client_host_error_counts'.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_set_wait_timeout_commands, "proxysql_mysql_set_wait_timeout_commands_total", "Number of SET wait_timeout commands received from clients.", metric_tags {} ), std::make_tuple ( p_th_counter::mysql_timeout_terminated_connections, "proxysql_mysql_timeout_terminated_connections_total", "Number of client connections terminated due to wait_timeout.", metric_tags {} ) }, th_gauge_vector { std::make_tuple ( p_th_gauge::active_transactions, "proxysql_active_transactions", "Provides a count of how many client connection are currently processing a transaction.", metric_tags {} ), std::make_tuple ( p_th_gauge::client_connections_non_idle, "proxysql_client_connections_non_idle", "Number of client connections that are currently handled by the main worker threads.", metric_tags {} ), std::make_tuple ( p_th_gauge::client_connections_hostgroup_locked, "proxysql_client_connections_hostgroup_locked", "Number of client connection locked to a specific hostgroup.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_backend_buffers_bytes, "proxysql_mysql_backend_buffers_bytes", "Buffers related to backend connections if \"fast_forward\" is used (0 means fast_forward is not used).", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_frontend_buffers_bytes, "proxysql_mysql_frontend_buffers_bytes", "Buffers related to frontend connections (read/write buffers and other queues).", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_session_internal_bytes, "proxysql_mysql_session_internal_bytes", "Other memory used by ProxySQL to handle MySQL Sessions.", metric_tags {} ), std::make_tuple ( p_th_gauge::mirror_concurrency, "proxysql_mirror_concurrency", "Mirror current concurrency", metric_tags {} ), std::make_tuple ( p_th_gauge::mirror_queue_lengths, "proxysql_mirror_queue_lengths", "Mirror queue length", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_thread_workers, "proxysql_mysql_thread_workers", "Number of MySQL Thread workers i.e. 'mysql-threads'", metric_tags {} ), // global_variables std::make_tuple ( p_th_gauge::mysql_wait_timeout, "proxysql_mysql_wait_timeout", "If a proxy session has been idle for more than this threshold, the proxy will kill the session.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_max_connections, "proxysql_mysql_max_connections", "The maximum number of client connections that the proxy can handle.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_enabled, "proxysql_mysql_monitor_enabled", "Enables or disables MySQL Monitor.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_ping_interval, "proxysql_mysql_monitor_ping_interval", "How frequently a ping check is performed, in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_ping_timeout, "proxysql_mysql_monitor_ping_timeout_seconds", "Ping timeout in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_ping_max_failures, "proxysql_mysql_monitor_ping_max_failures", "Reached maximum ping attempts from monitor.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_aws_rds_topology_discovery_interval, "proxysql_mysql_monitor_aws_rds_topology_discovery_interval", "How frequently a topology discovery is performed, e.g. a value of 500 means one topology discovery every 500 read-only checks ", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_read_only_interval, "proxysql_mysql_monitor_read_only_interval_seconds", "How frequently a read only check is performed, in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_read_only_timeout, "proxysql_mysql_monitor_read_only_timeout_seconds", "Read only check timeout in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_writer_is_also_reader, "proxysql_mysql_monitor_writer_is_also_reader", "Encodes different behaviors for nodes depending on their 'READ_ONLY' flag value.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_replication_lag_group_by_host, "proxysql_monitor_replication_lag_group_by_host", "Encodes different replication lag check if the same server is in multiple hostgroups.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_replication_lag_interval, "proxysql_mysql_monitor_replication_lag_interval_seconds", "How frequently a replication lag check is performed, in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_replication_lag_timeout, "proxysql_mysql_monitor_replication_lag_timeout_seconds", "Replication lag check timeout in seconds.", metric_tags {} ), std::make_tuple ( p_th_gauge::mysql_monitor_history, "proxysql_mysql_monitor_history_timeout_seconds", "The duration for which the events for the checks made by the Monitor module are kept, in seconds.", metric_tags {} ) } ); MySQL_Threads_Handler::MySQL_Threads_Handler() { #ifdef DEBUG if (glovars.has_debug==false) { #else if (glovars.has_debug==true) { #endif /* DEBUG */ // LCOV_EXCL_START perror("Incompatible debugging version"); exit(EXIT_FAILURE); // LCOV_EXCL_STOP } num_threads=0; mysql_threads=NULL; #ifdef IDLE_THREADS mysql_threads_idles=NULL; #endif // IDLE_THREADS stacksize=0; shutdown_=0; bootstrapping_listeners = true; pthread_rwlock_init(&rwlock,NULL); pthread_attr_init(&attr); // Zero initialize all variables memset(&variables, 0, sizeof(variables)); variables.shun_on_failures=5; variables.shun_recovery_time_sec=10; variables.unshun_algorithm=0; variables.query_retries_on_failure=1; variables.client_host_cache_size=0; variables.client_host_error_counts=0; variables.handle_warnings=1; variables.evaluate_replication_lag_on_servers_load=1; variables.connect_retries_on_failure=10; variables.connection_delay_multiplex_ms=0; variables.connection_max_age_ms=0; variables.connect_timeout_client=10000; variables.connect_timeout_server=1000; variables.connect_timeout_server_max=10000; variables.free_connections_pct=10; variables.connect_retries_delay=1; variables.monitor_enabled=true; variables.monitor_history=7200000; // changed in 2.6.0 : was 600000 variables.monitor_connect_interval=120000; variables.monitor_connect_timeout=600; variables.monitor_ping_interval=8000; variables.monitor_ping_max_failures=3; variables.monitor_ping_timeout=1000; variables.monitor_aws_rds_topology_discovery_interval=0; variables.monitor_read_only_interval=1000; variables.monitor_read_only_timeout=800; variables.monitor_read_only_max_timeout_count=3; variables.monitor_replication_lag_group_by_host=false; variables.monitor_replication_lag_interval=10000; variables.monitor_replication_lag_timeout=1000; variables.monitor_replication_lag_count=1; variables.monitor_groupreplication_healthcheck_interval=5000; variables.monitor_groupreplication_healthcheck_timeout=800; variables.monitor_groupreplication_healthcheck_max_timeout_count=3; variables.monitor_groupreplication_max_transactions_behind_count=3; variables.monitor_groupreplication_max_transactions_behind_for_read_only=1; variables.monitor_galera_healthcheck_interval=5000; variables.monitor_galera_healthcheck_timeout=800; variables.monitor_galera_healthcheck_max_timeout_count=3; variables.monitor_query_interval=60000; variables.monitor_query_timeout=100; variables.monitor_slave_lag_when_null=60; variables.monitor_threads_min = 8; variables.monitor_threads_max = 128; variables.monitor_threads_queue_maxsize = 128; variables.monitor_local_dns_cache_ttl = 300000; variables.monitor_local_dns_cache_refresh_interval = 60000; variables.monitor_local_dns_resolver_queue_maxsize = 128; variables.monitor_username=strdup((char *)"monitor"); variables.monitor_password=strdup((char *)"monitor"); variables.monitor_replication_lag_use_percona_heartbeat=strdup((char *)""); variables.monitor_wait_timeout=true; variables.monitor_writer_is_also_reader=true; variables.max_allowed_packet=64*1024*1024; variables.automatic_detect_sqli=false; variables.firewall_whitelist_enabled=false; variables.firewall_whitelist_errormsg = strdup((char *)"Firewall blocked this query"); variables.use_tcp_keepalive=true; // changed in 2.6.0 , was false variables.tcp_keepalive_time=120; // changed in 2.6.0 , was 0 variables.throttle_connections_per_sec_to_hostgroup=1000000; variables.max_transaction_idle_time=4*3600*1000; variables.max_transaction_time=4*3600*1000; variables.hostgroup_manager_verbose=1; variables.binlog_reader_connect_retry_msec=3000; variables.threshold_query_length=512*1024; variables.threshold_resultset_size=4*1024*1024; variables.query_digests_max_digest_length=2*1024; variables.query_digests_max_query_length=65000; // legacy default variables.query_rules_fast_routing_algorithm=1; variables.wait_timeout=8*3600*1000; variables.throttle_max_bytes_per_second_to_client=0; variables.throttle_ratio_server_to_client=0; variables.connection_warming=false; variables.max_connections=10*1000; variables.max_stmts_per_connection=20; variables.max_stmts_cache=10000; variables.mirror_max_concurrency=16; variables.mirror_max_queue_length=32000; variables.default_max_latency_ms=1*1000; // by default, the maximum allowed latency for a host is 1000ms variables.default_query_delay=0; variables.default_query_timeout=24*3600*1000; variables.query_processor_iterations=0; variables.query_processor_regex=1; variables.set_query_lock_on_hostgroup=1; variables.set_parser_algorithm=2; // before 2.6.0 this was 1 variables.reset_connection_algorithm=2; variables.auto_increment_delay_multiplex=5; variables.auto_increment_delay_multiplex_timeout_ms=10000; variables.long_query_time=1000; variables.query_cache_size_MB=256; variables.query_cache_soft_ttl_pct=0; variables.query_cache_handle_warnings=0; variables.init_connect=NULL; variables.ldap_user_variable=NULL; variables.add_ldap_user_comment=NULL; for (int i=0; i(th_metrics_map, this->status_variables.p_counter_array); init_prometheus_gauge_array(th_metrics_map, this->status_variables.p_gauge_array); // Init client_host_cache mutex pthread_mutex_init(&mutex_client_host_cache, NULL); } unsigned int MySQL_Threads_Handler::get_global_version() { return __sync_fetch_and_add(&__global_MySQL_Thread_Variables_version,0); } int MySQL_Threads_Handler::listener_add(const char *address, int port) { char *s=(char *)malloc(strlen(address)+32); sprintf(s,"%s:%d",address,port); int ret=listener_add((const char *)s); free(s); return ret; } int MySQL_Threads_Handler::listener_add(const char *iface) { int rc; int *perthrsocks=NULL;; rc=MLM->add(iface, num_threads, &perthrsocks); if (rc>-1) { unsigned int i; if (perthrsocks==NULL) { for (i=0;imypolls.pending_listener_add,0,rc)) { usleep(10); // pause a bit } } } else { for (i=0;imypolls.pending_listener_add,0,perthrsocks[i])) { usleep(10); // pause a bit } } free(perthrsocks); } } return rc; } int MySQL_Threads_Handler::listener_del(const char *iface) { int idx; while ((idx=MLM->find_idx(iface)) >= 0) { unsigned int i; int fd=MLM->get_fd(idx); for (i=0;imypolls.pending_listener_del,0,-1)); else #endif while(!__sync_bool_compare_and_swap(&thr->mypolls.pending_listener_del,0,fd)); } for (i=0;imypolls.pending_listener_del,0)) { // Since 'listeners_stop' is performed in 'maintenance_loops' by the // workers this active-wait is likely to take some time. usleep(std::min(std::max(mysql_thread___poll_timeout/20, 10000), 40000)); } } MLM->del(idx); #ifdef SO_REUSEPORT if (GloVars.global.reuseport) { continue; } #endif shutdown(fd,SHUT_RDWR); close(fd); } return 0; } void MySQL_Threads_Handler::wrlock() { pthread_rwlock_wrlock(&rwlock); } void MySQL_Threads_Handler::wrunlock() { pthread_rwlock_unlock(&rwlock); } void MySQL_Threads_Handler::commit() { __sync_add_and_fetch(&__global_MySQL_Thread_Variables_version,1); proxy_debug(PROXY_DEBUG_MYSQL_SERVER, 1, "Increasing version number to %d - all threads will notice this and refresh their variables\n", __global_MySQL_Thread_Variables_version); } /** * Retrieves the string value of a specified global variable. * * This method searches for a global variable by name and returns its current * string value. It's used to access configuration settings that are stored * as strings within the MySQL Threads Handler. If the variable is not found, * or if it is not a string type, a default or NULL value may be returned. * * @param name The name of the variable to retrieve. * @return A pointer to the string value of the variable, or NULL if the * variable does not exist or is not of string type. */ char * MySQL_Threads_Handler::get_variable_string(char *name) { if (!strncmp(name,"monitor_",8)) { if (!strcmp(name,"monitor_username")) return strdup(variables.monitor_username); if (!strcmp(name,"monitor_password")) return strdup(variables.monitor_password); if (!strcmp(name,"monitor_replication_lag_use_percona_heartbeat")) return strdup(variables.monitor_replication_lag_use_percona_heartbeat); } if (!strncmp(name,"ssl_",4)) { if (!strcmp(name,"ssl_p2s_ca")) { if (variables.ssl_p2s_ca==NULL || strlen(variables.ssl_p2s_ca)==0) { return NULL; } else { return strdup(variables.ssl_p2s_ca); } } if (!strcmp(name,"ssl_p2s_cert")) { if (variables.ssl_p2s_cert==NULL || strlen(variables.ssl_p2s_cert)==0) { return NULL; } else { return strdup(variables.ssl_p2s_cert); } } if (!strcmp(name,"ssl_p2s_capath")) { if (variables.ssl_p2s_capath==NULL || strlen(variables.ssl_p2s_capath)==0) { return NULL; } else { return strdup(variables.ssl_p2s_capath); } } if (!strcmp(name,"ssl_p2s_key")) { if (variables.ssl_p2s_key==NULL || strlen(variables.ssl_p2s_key)==0) { return NULL; } else { return strdup(variables.ssl_p2s_key); } } if (!strcmp(name,"ssl_p2s_cipher")) { if (variables.ssl_p2s_cipher==NULL || strlen(variables.ssl_p2s_cipher)==0) { return NULL; } else { return strdup(variables.ssl_p2s_cipher); } } if (!strcmp(name,"ssl_p2s_crl")) { if (variables.ssl_p2s_crl==NULL || strlen(variables.ssl_p2s_crl)==0) { return NULL; } else { return strdup(variables.ssl_p2s_crl); } } if (!strcmp(name,"ssl_p2s_crlpath")) { if (variables.ssl_p2s_crlpath==NULL || strlen(variables.ssl_p2s_crlpath)==0) { return NULL; } else { return strdup(variables.ssl_p2s_crlpath); } } } if (!strcmp(name,"firewall_whitelist_errormsg")) { if (variables.firewall_whitelist_errormsg==NULL || strlen(variables.firewall_whitelist_errormsg)==0) { return NULL; } else { return strdup(variables.firewall_whitelist_errormsg); } } if (!strcmp(name,"init_connect")) { if (variables.init_connect==NULL || strlen(variables.init_connect)==0) { return NULL; } else { return strdup(variables.init_connect); } } if (!strcmp(name,"ldap_user_variable")) { if (variables.ldap_user_variable==NULL || strlen(variables.ldap_user_variable)==0) { return NULL; } else { return strdup(variables.ldap_user_variable); } } if (!strcmp(name,"add_ldap_user_comment")) { if (variables.add_ldap_user_comment==NULL || strlen(variables.add_ldap_user_comment)==0) { return NULL; } else { return strdup(variables.add_ldap_user_comment); } } if (!strncmp(name,"default_",8)) { for (int i=0; i>::const_iterator it = VariablesPointers_int.find(nameS); if (it != VariablesPointers_int.end()) { int * v = std::get<0>(it->second); return *v; } } { // bool variable std::unordered_map>::const_iterator it = VariablesPointers_bool.find(nameS); if (it != VariablesPointers_bool.end()) { bool * v = std::get<0>(it->second); int a = (int)*v; return a; } } //VALGRIND_DISABLE_ERROR_REPORTING; if (!strcmp(name,"stacksize")) return ( stacksize ? stacksize : DEFAULT_STACK_SIZE); // LCOV_EXCL_START proxy_error("Not existing variable: %s\n", name); assert(0); return 0; // LCOV_EXCL_STOP //VALGRIND_ENABLE_ERROR_REPORTING; } /** * Retrieves the value of a specified configuration variable. * * This method looks up a global configuration variable by its name and returns its value as a char pointer. * It's designed to access various types of configuration settings within the MySQL Threads Handler context. * The function can return values for variables of different types, but the caller must ensure proper type handling. * If the variable is not found, a NULL pointer is returned. * * @param name The name of the variable to retrieve. * @return The value of the variable as a char pointer, or NULL if the variable does not exist. */ char * MySQL_Threads_Handler::get_variable(char *name) { // this is the public function, accessible from admin //VALGRIND_DISABLE_ERROR_REPORTING; #define INTBUFSIZE 4096 char intbuf[INTBUFSIZE]; // convert name to string, and lowercase std::string nameS = string(name); std::transform(nameS.begin(), nameS.end(), nameS.begin(), [](unsigned char c){ return std::tolower(c); }); { // integer variable std::unordered_map>::const_iterator it = VariablesPointers_int.find(nameS); if (it != VariablesPointers_int.end()) { int * v = std::get<0>(it->second); sprintf(intbuf,"%d", *v); return strdup(intbuf); } } { // bool variable std::unordered_map>::const_iterator it = VariablesPointers_bool.find(nameS); if (it != VariablesPointers_bool.end()) { bool * v = std::get<0>(it->second); return strdup((*v ? "true" : "false")); } } if (!strcasecmp(name,"firewall_whitelist_errormsg")) { if (variables.firewall_whitelist_errormsg==NULL || strlen(variables.firewall_whitelist_errormsg)==0) { return NULL; } else { return strdup(variables.firewall_whitelist_errormsg); } } if (!strcasecmp(name,"init_connect")) { if (variables.init_connect==NULL || strlen(variables.init_connect)==0) { return NULL; } else { return strdup(variables.init_connect); } } if (!strcasecmp(name,"ldap_user_variable")) { if (variables.ldap_user_variable==NULL || strlen(variables.ldap_user_variable)==0) { return NULL; } else { return strdup(variables.ldap_user_variable); } } if (!strcasecmp(name,"add_ldap_user_comment")) { if (variables.add_ldap_user_comment==NULL || strlen(variables.add_ldap_user_comment)==0) { return NULL; } else { return strdup(variables.add_ldap_user_comment); } } if (!strcasecmp(name,"default_session_track_gtids")) { if (variables.default_session_track_gtids==NULL) { variables.default_session_track_gtids=strdup((char *)MYSQL_DEFAULT_SESSION_TRACK_GTIDS); } return strdup(variables.default_session_track_gtids); } if (strlen(name) > 8) { if (strncmp(name, "default_", 8) == 0) { for (unsigned int i = 0; i < SQL_NAME_LAST_LOW_WM ; i++) { if (mysql_tracked_variables[i].is_global_variable) { size_t var_len = strlen(mysql_tracked_variables[i].internal_variable_name); if (strlen(name) == (var_len+8)) { if (!strncmp(name+8, mysql_tracked_variables[i].internal_variable_name, var_len)) { return strdup(variables.default_variables[i]); } } } } } } if (!strcasecmp(name,"firewall_whitelist_errormsg")) return strdup(variables.firewall_whitelist_errormsg); if (!strcasecmp(name,"server_version")) return strdup(variables.server_version); if (!strcasecmp(name,"auditlog_filename")) return strdup(variables.auditlog_filename); if (!strcasecmp(name,"eventslog_filename")) return strdup(variables.eventslog_filename); if (!strcasecmp(name,"default_schema")) return strdup(variables.default_schema); if (!strcasecmp(name,"keep_multiplexing_variables")) return strdup(variables.keep_multiplexing_variables); if (!strcasecmp(name,"default_authentication_plugin")) return strdup(variables.default_authentication_plugin); if (!strcasecmp(name,"proxy_protocol_networks")) return strdup(variables.proxy_protocol_networks); if (!strcasecmp(name, "interfaces")) { return strdup((strlen(variables.interfaces) == 0) ? "0.0.0.0:6033" : variables.interfaces); } if (!strcasecmp(name,"server_capabilities")) { // FIXME : make it human readable sprintf(intbuf,"%d",variables.server_capabilities); return strdup(intbuf); } // SSL variables if (!strncasecmp(name,"ssl_",4)) { if (!strcasecmp(name,"ssl_p2s_ca")) { if (variables.ssl_p2s_ca==NULL || strlen(variables.ssl_p2s_ca)==0) { return NULL; } else { return strdup(variables.ssl_p2s_ca); } } if (!strcasecmp(name,"ssl_p2s_capath")) { if (variables.ssl_p2s_capath==NULL || strlen(variables.ssl_p2s_capath)==0) { return NULL; } else { return strdup(variables.ssl_p2s_capath); } } if (!strcasecmp(name,"ssl_p2s_cert")) { if (variables.ssl_p2s_cert==NULL || strlen(variables.ssl_p2s_cert)==0) { return NULL; } else { return strdup(variables.ssl_p2s_cert); } } if (!strcasecmp(name,"ssl_p2s_key")) { if (variables.ssl_p2s_key==NULL || strlen(variables.ssl_p2s_key)==0) { return NULL; } else { return strdup(variables.ssl_p2s_key); } } if (!strcasecmp(name,"ssl_p2s_cipher")) { if (variables.ssl_p2s_cipher==NULL || strlen(variables.ssl_p2s_cipher)==0) { return NULL; } else { return strdup(variables.ssl_p2s_cipher); } } if (!strcasecmp(name,"ssl_p2s_crl")) { if (variables.ssl_p2s_crl==NULL || strlen(variables.ssl_p2s_crl)==0) { return NULL; } else { return strdup(variables.ssl_p2s_crl); } } if (!strcasecmp(name,"ssl_p2s_crlpath")) { if (variables.ssl_p2s_crlpath==NULL || strlen(variables.ssl_p2s_crlpath)==0) { return NULL; } else { return strdup(variables.ssl_p2s_crlpath); } } } // monitor variables if (!strncasecmp(name,"monitor_",8)) { if (!strcasecmp(name,"monitor_username")) return strdup(variables.monitor_username); if (!strcasecmp(name,"monitor_password")) return strdup(variables.monitor_password); if (!strcasecmp(name,"monitor_replication_lag_use_percona_heartbeat")) return strdup(variables.monitor_replication_lag_use_percona_heartbeat); } if (!strcasecmp(name,"threads")) { sprintf(intbuf,"%d", (num_threads ? num_threads : DEFAULT_NUM_THREADS)); return strdup(intbuf); } if (!strcasecmp(name,"stacksize")) { sprintf(intbuf,"%d", (int)(stacksize ? stacksize : DEFAULT_STACK_SIZE)); return strdup(intbuf); } return NULL; //VALGRIND_ENABLE_ERROR_REPORTING; } /** * Sets the value of a specified global variable. * * This method allows for the modification of global variables within the MySQL Threads Handler * by specifying the variable's name and the desired new value. It supports changing the values * of variables across various types, though the input is accepted as a string. Proper conversion * to the variable's actual type is performed internally. This function is key for dynamic configuration * updates and runtime adjustments of the proxy's behavior. * * @param name The name of the variable to set. * @param value The new value for the variable, passed as a const char pointer. * @return True if the variable was successfully updated, false otherwise. */ bool MySQL_Threads_Handler::set_variable(char *name, const char *value) { // this is the public function, accessible from admin if (!value) return false; size_t vallen=strlen(value); // convert name to string, and lowercase std::string nameS = string(name); std::transform(nameS.begin(), nameS.end(), nameS.begin(), [](unsigned char c){ return std::tolower(c); }); { // integer variable ? std::unordered_map>::const_iterator it = VariablesPointers_int.find(nameS); if (it != VariablesPointers_int.end()) { // Log warnings for variables with possibly wrong values if (nameS == "auto_increment_delay_multiplex_timeout_ms") { int intv = atoi(value); if (intv <= 60) { proxy_warning("'mysql-auto_increment_delay_multiplex_timeout_ms' is set to a low value: %ums. Remember value is in 'ms'\n", intv); } } if (nameS == "query_rules_fast_routing_algorithm") { if (GloMyQPro) { int intv = atoi(value); if (intv >= std::get<1>(it->second) && intv <= std::get<2>(it->second)) { GloMyQPro->wrlock(); GloMyQPro->query_rules_fast_routing_algorithm = intv; GloMyQPro->wrunlock(); } } } bool special_variable = std::get<3>(it->second); // if special_variable is true, min and max values are ignored, and more input validation is needed if (special_variable == false) { int intv=atoi(value); if (intv >= std::get<1>(it->second) && intv <= std::get<2>(it->second)) { int * v = std::get<0>(it->second); *v = intv; return true; } return false; } else { // we need to perform input validation } } } { // boolean variable ? std::unordered_map>::const_iterator it = VariablesPointers_bool.find(nameS); if (it != VariablesPointers_bool.end()) { bool special_variable = std::get<1>(it->second); // if special_variable is true, more input validation is needed if (special_variable == false) { bool * v = std::get<0>(it->second); if (strcasecmp(value,"true")==0 || strcasecmp(value,"1")==0) { *v = true; return true; } if (strcasecmp(value,"false")==0 || strcasecmp(value,"0")==0) { *v = false; return true; } return false; } else { // we need to perform input validation } } } // monitor variables if (!strncasecmp(name,"monitor_",8)) { if (!strcasecmp(name,"monitor_username")) { if (vallen) { free(variables.monitor_username); variables.monitor_username=strdup(value); return true; } else { return false; } } if (!strcasecmp(name,"monitor_password")) { free(variables.monitor_password); variables.monitor_password=strdup(value); return true; } if (!strcasecmp(name,"monitor_replication_lag_use_percona_heartbeat")) { if (vallen==0) { // empty string free(variables.monitor_replication_lag_use_percona_heartbeat); variables.monitor_replication_lag_use_percona_heartbeat=strdup((value)); return true; } else { re2::RE2::Options *opt2=new re2::RE2::Options(RE2::Quiet); opt2->set_case_sensitive(false); char *patt = (char *)"`?([a-z\\d_]+)`?\\.`?([a-z\\d_]+)`?"; RE2 *re = new RE2(patt, *opt2); bool rc=false; rc = RE2::FullMatch(value,*re); delete re; delete opt2; if(rc) { free(variables.monitor_replication_lag_use_percona_heartbeat); variables.monitor_replication_lag_use_percona_heartbeat=strdup(value); return true; } else { proxy_error("%s is an invalid value for %s, not matching regex \"%s\"\n", value, name, patt); } } return false; } } if (!strcasecmp(name,"binlog_reader_connect_retry_msec")) { int intv=atoi(value); if (intv >= 200 && intv <= 120000) { __sync_lock_test_and_set(&variables.binlog_reader_connect_retry_msec,intv); return true; } else { return false; } } if (!strcasecmp(name,"wait_timeout")) { int intv=atoi(value); if (intv >= 0 && intv <= 20*24*3600*1000) { variables.wait_timeout=intv; if (variables.wait_timeout < 5000) { proxy_warning("mysql-wait_timeout is set to a low value: %ums\n", variables.wait_timeout); } return true; } else { return false; } } if (!strcasecmp(name,"eventslog_format")) { int intv=atoi(value); if (intv >= 1 && intv <= 2) { if (variables.eventslog_format!=intv) { // if we are switching format, we need to switch file too if (GloMyLogger) { proxy_info("Switching query logging format from %d to %d\n", variables.eventslog_format , intv); GloMyLogger->flush_log(); } variables.eventslog_format=intv; } return true; } else { return false; } } if (!strcasecmp(name,"default_schema")) { if (vallen) { free(variables.default_schema); variables.default_schema=strdup(value); return true; } else { return false; } } if (!strcasecmp(name,"interfaces")) { if (vallen && strlen(variables.interfaces)==0) { free(variables.interfaces); variables.interfaces=strdup(value); return true; } else { if (vallen && strcmp(value,variables.interfaces)==0) { return true; } else { return false; } } } if (!strcasecmp(name,"server_version")) { if (vallen) { free(variables.server_version); if (strcmp(value,(const char *)"5.1.30")==0) { // per issue #632 , the default 5.1.30 is replaced with 5.5.30 variables.server_version=strdup((char *)"5.5.30"); } else { variables.server_version=strdup(value); } return true; } else { return false; } } if (!strcasecmp(name,"init_connect")) { if (variables.init_connect) free(variables.init_connect); variables.init_connect=NULL; if (vallen) { if (strcmp(value,"(null)")) variables.init_connect=strdup(value); } return true; } if (!strcasecmp(name,"firewall_whitelist_errormsg")) { if (variables.firewall_whitelist_errormsg) free(variables.firewall_whitelist_errormsg); variables.firewall_whitelist_errormsg=NULL; if (vallen) { if (strcmp(value,"(null)")) variables.firewall_whitelist_errormsg=strdup(value); } return true; } if (!strcasecmp(name,"ldap_user_variable")) { if (variables.ldap_user_variable) free(variables.ldap_user_variable); variables.ldap_user_variable=NULL; if (vallen) { if (strcmp(value,"(null)")) variables.ldap_user_variable=strdup(value); } return true; } if (!strcasecmp(name,"add_ldap_user_comment")) { if (variables.add_ldap_user_comment) free(variables.add_ldap_user_comment); variables.add_ldap_user_comment=NULL; if (vallen) { if (strcmp(value,"(null)")) variables.add_ldap_user_comment=strdup(value); } return true; } if (!strcasecmp(name,"default_session_track_gtids")) { if (variables.default_session_track_gtids) free(variables.default_session_track_gtids); variables.default_session_track_gtids=NULL; if (vallen) { // we only accept 2 value for session_track_gtids = OFF or OWN_GTID if (strcasecmp(value,(char *)"OFF") == 0) { // for convention, we stored the value as uppercase variables.default_session_track_gtids=strdup((char *)"OFF"); return true; } else if (strcasecmp(value,(char *)"OWN_GTID") == 0) { // for convention, we stored the value as uppercase variables.default_session_track_gtids=strdup((char *)"OWN_GTID"); return true; } } return false; // we couldn't set it to a valid value. It will be reset to default } if (!strncmp(name,"default_",8)) { for (int i=0; i 10) { // Note that: // - some capabilities are changed at runtime while performing the handshake with the client // - even if we support 32 bits capabilities, many of them do not have any real meaning for proxysql (not supported) variables.server_capabilities=intv; return true; } else { return false; } } if (!strcasecmp(name,"stacksize")) { int intv=atoi(value); if (intv >= 256*1024 && intv <= 4*1024*1024) { stacksize=intv; return true; } else { return false; } } if (!strcasecmp(name,"threads")) { unsigned int intv=atoi(value); if ((num_threads==0 || num_threads==intv || mysql_threads==NULL) && intv > 0 && intv < 256) { num_threads=intv; this->status_variables.p_gauge_array[p_th_gauge::mysql_thread_workers]->Set(intv); return true; } else { return false; } } if (!strcasecmp(name,"have_compress")) { if (strcasecmp(value,"true")==0 || strcasecmp(value,"1")==0) { variables.have_compress=true; variables.server_capabilities |= CLIENT_COMPRESS; return true; } if (strcasecmp(value,"false")==0 || strcasecmp(value,"0")==0) { variables.have_compress=false; variables.server_capabilities &= ~CLIENT_COMPRESS; return true; } return false; } if (!strcasecmp(name,"have_ssl")) { if (strcasecmp(value,"true")==0 || strcasecmp(value,"1")==0) { variables.have_ssl=true; variables.server_capabilities |= CLIENT_SSL; return true; } if (strcasecmp(value,"false")==0 || strcasecmp(value,"0")==0) { variables.have_ssl=false; variables.server_capabilities &= ~CLIENT_SSL; return true; } return false; } if (!strcasecmp(name,"forward_autocommit")) { if (strcasecmp(value,"true")==0 || strcasecmp(value,"1")==0) { proxy_error("Variable mysql-forward_autocommit is deprecated. See issue #3253\n"); return false; } return false; } if (!strcasecmp(name,"data_packets_history_size")) { int intv=atoi(value); if (intv >= 0 && intv < INT_MAX) { variables.data_packets_history_size = intv; GloVars.global.data_packets_history_size = intv; return true; } else { return false; } } return false; } /** * Retrieves a list of all global configuration variables' names. * * This method returns an array of strings, each representing the name of a global configuration * variable managed by the MySQL Threads Handler. It's utilized to enumerate all available configuration * settings, facilitating dynamic inspection or modification of the proxy's configuration. The list * includes variables of all types and categories, providing a comprehensive overview of the proxy's * configurable parameters. * * @return A null-terminated array of char pointers, where each entry is the name of a global variable. * The last element of the array is NULL to indicate the end of the list. */ char ** MySQL_Threads_Handler::get_variables_list() { // initialize VariablesPointers_bool // it is safe to do it here because get_variables_list() is the first function called during start time if (VariablesPointers_bool.size() == 0) { VariablesPointers_bool["autocommit_false_is_transaction"] = make_tuple(&variables.autocommit_false_is_transaction, false); VariablesPointers_bool["autocommit_false_not_reusable"] = make_tuple(&variables.autocommit_false_not_reusable, false); VariablesPointers_bool["automatic_detect_sqli"] = make_tuple(&variables.automatic_detect_sqli, false); VariablesPointers_bool["client_session_track_gtid"] = make_tuple(&variables.client_session_track_gtid, false); VariablesPointers_bool["commands_stats"] = make_tuple(&variables.commands_stats, false); VariablesPointers_bool["connection_warming"] = make_tuple(&variables.connection_warming, false); VariablesPointers_bool["default_reconnect"] = make_tuple(&variables.default_reconnect, false); VariablesPointers_bool["enable_client_deprecate_eof"] = make_tuple(&variables.enable_client_deprecate_eof, false); VariablesPointers_bool["enable_server_deprecate_eof"] = make_tuple(&variables.enable_server_deprecate_eof, false); VariablesPointers_bool["enable_load_data_local_infile"] = make_tuple(&variables.enable_load_data_local_infile, false); VariablesPointers_bool["enforce_autocommit_on_reads"] = make_tuple(&variables.enforce_autocommit_on_reads, false); VariablesPointers_bool["firewall_whitelist_enabled"] = make_tuple(&variables.firewall_whitelist_enabled, false); VariablesPointers_bool["kill_backend_connection_when_disconnect"] = make_tuple(&variables.kill_backend_connection_when_disconnect, false); VariablesPointers_bool["log_mysql_warnings_enabled"] = make_tuple(&variables.log_mysql_warnings_enabled, false); VariablesPointers_bool["log_unhealthy_connections"] = make_tuple(&variables.log_unhealthy_connections, false); VariablesPointers_bool["monitor_enabled"] = make_tuple(&variables.monitor_enabled, false); VariablesPointers_bool["monitor_replication_lag_group_by_host"] = make_tuple(&variables.monitor_replication_lag_group_by_host, false); VariablesPointers_bool["monitor_wait_timeout"] = make_tuple(&variables.monitor_wait_timeout, false); VariablesPointers_bool["monitor_writer_is_also_reader"] = make_tuple(&variables.monitor_writer_is_also_reader, false); VariablesPointers_bool["multiplexing"] = make_tuple(&variables.multiplexing, false); VariablesPointers_bool["query_cache_stores_empty_result"] = make_tuple(&variables.query_cache_stores_empty_result, false); VariablesPointers_bool["query_digests"] = make_tuple(&variables.query_digests, false); VariablesPointers_bool["query_digests_lowercase"] = make_tuple(&variables.query_digests_lowercase, false); VariablesPointers_bool["query_digests_replace_null"] = make_tuple(&variables.query_digests_replace_null, false); VariablesPointers_bool["query_digests_no_digits"] = make_tuple(&variables.query_digests_no_digits, false); VariablesPointers_bool["query_digests_normalize_digest_text"] = make_tuple(&variables.query_digests_normalize_digest_text, false); VariablesPointers_bool["query_digests_track_hostname"] = make_tuple(&variables.query_digests_track_hostname, false); VariablesPointers_bool["query_digests_keep_comment"] = make_tuple(&variables.query_digests_keep_comment, false); VariablesPointers_bool["parse_failure_logs_digest"] = make_tuple(&variables.parse_failure_logs_digest, false); VariablesPointers_bool["servers_stats"] = make_tuple(&variables.servers_stats, false); VariablesPointers_bool["sessions_sort"] = make_tuple(&variables.sessions_sort, false); VariablesPointers_bool["stats_time_backend_query"] = make_tuple(&variables.stats_time_backend_query, false); VariablesPointers_bool["stats_time_query_processor"] = make_tuple(&variables.stats_time_query_processor, false); VariablesPointers_bool["use_tcp_keepalive"] = make_tuple(&variables.use_tcp_keepalive, false); VariablesPointers_bool["verbose_query_error"] = make_tuple(&variables.verbose_query_error, false); VariablesPointers_bool["ignore_min_gtid_annotations"] = make_tuple(&variables.ignore_min_gtid_annotations, false); #ifdef IDLE_THREADS VariablesPointers_bool["session_idle_show_processlist"] = make_tuple(&variables.session_idle_show_processlist, false); #endif // IDLE_THREADS #ifdef DEBUG VariablesPointers_bool["session_debug"] = make_tuple(&variables.session_debug, false); #endif /* DEBUG */ // variables with special variable == true // the input validation for these variables MUST be EXPLICIT VariablesPointers_bool["have_compress"] = make_tuple(&variables.have_compress, true); VariablesPointers_bool["have_ssl"] = make_tuple(&variables.have_ssl, true); } // initialize VariablesPointers_int // it is safe to do it here because get_variables_list() is the first function called during start time if (VariablesPointers_int.size() == 0) { // Monitor variables VariablesPointers_int["monitor_history"] = make_tuple(&variables.monitor_history, 1000, 7*24*3600*1000, false); VariablesPointers_int["monitor_connect_interval"] = make_tuple(&variables.monitor_connect_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_connect_timeout"] = make_tuple(&variables.monitor_connect_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_ping_interval"] = make_tuple(&variables.monitor_ping_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_ping_timeout"] = make_tuple(&variables.monitor_ping_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_ping_max_failures"] = make_tuple(&variables.monitor_ping_max_failures, 1, 1000*1000, false); VariablesPointers_int["monitor_aws_rds_topology_discovery_interval"] = make_tuple(&variables.monitor_aws_rds_topology_discovery_interval, 0, 100000, false); VariablesPointers_int["monitor_read_only_interval"] = make_tuple(&variables.monitor_read_only_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_read_only_timeout"] = make_tuple(&variables.monitor_read_only_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_read_only_max_timeout_count"] = make_tuple(&variables.monitor_read_only_max_timeout_count, 1, 1000*1000, false); VariablesPointers_int["monitor_replication_lag_interval"] = make_tuple(&variables.monitor_replication_lag_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_replication_lag_timeout"] = make_tuple(&variables.monitor_replication_lag_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_replication_lag_count"] = make_tuple(&variables.monitor_replication_lag_count, 1, 10, false); VariablesPointers_int["monitor_groupreplication_healthcheck_interval"] = make_tuple(&variables.monitor_groupreplication_healthcheck_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_groupreplication_healthcheck_timeout"] = make_tuple(&variables.monitor_groupreplication_healthcheck_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_groupreplication_healthcheck_max_timeout_count"] = make_tuple(&variables.monitor_groupreplication_healthcheck_max_timeout_count, 1, 10, false); VariablesPointers_int["monitor_groupreplication_max_transactions_behind_count"] = make_tuple(&variables.monitor_groupreplication_max_transactions_behind_count, 1, 10, false); VariablesPointers_int["monitor_groupreplication_max_transactions_behind_for_read_only"] = make_tuple(&variables.monitor_groupreplication_max_transactions_behind_for_read_only, 0, 2, false); VariablesPointers_int["monitor_galera_healthcheck_interval"] = make_tuple(&variables.monitor_galera_healthcheck_interval, 50, 7*24*3600*1000, false); VariablesPointers_int["monitor_galera_healthcheck_timeout"] = make_tuple(&variables.monitor_galera_healthcheck_timeout, 50, 600*1000, false); VariablesPointers_int["monitor_galera_healthcheck_max_timeout_count"] = make_tuple(&variables.monitor_galera_healthcheck_max_timeout_count, 1, 10, false); VariablesPointers_int["monitor_query_interval"] = make_tuple(&variables.monitor_query_interval, 100, 7*24*3600*1000, false); VariablesPointers_int["monitor_query_timeout"] = make_tuple(&variables.monitor_query_timeout, 100, 600*1000, false); VariablesPointers_int["monitor_threads_min"] = make_tuple(&variables.monitor_threads_min, 2, 256, false); VariablesPointers_int["monitor_threads_max"] = make_tuple(&variables.monitor_threads_max, 4, 1024, false); VariablesPointers_int["monitor_slave_lag_when_null"] = make_tuple(&variables.monitor_slave_lag_when_null, 0, 604800, false); VariablesPointers_int["monitor_threads_queue_maxsize"] = make_tuple(&variables.monitor_threads_queue_maxsize, 16, 1024, false); VariablesPointers_int["monitor_local_dns_cache_ttl"] = make_tuple(&variables.monitor_local_dns_cache_ttl, 0, 7*24*3600*1000, false); VariablesPointers_int["monitor_local_dns_cache_refresh_interval"] = make_tuple(&variables.monitor_local_dns_cache_refresh_interval, 0, 7*24*3600*1000, false); VariablesPointers_int["monitor_local_dns_resolver_queue_maxsize"] = make_tuple(&variables.monitor_local_dns_resolver_queue_maxsize, 16, 1024, false); // mirroring VariablesPointers_int["mirror_max_concurrency"] = make_tuple(&variables.mirror_max_concurrency, 1, 8*1024, false); VariablesPointers_int["mirror_max_queue_length"] = make_tuple(&variables.mirror_max_queue_length, 0, 1024*1024, false); // query processor and query digest VariablesPointers_int["auto_increment_delay_multiplex"] = make_tuple(&variables.auto_increment_delay_multiplex, 0, 1000000, false); VariablesPointers_int["auto_increment_delay_multiplex_timeout_ms"] = make_tuple(&variables.auto_increment_delay_multiplex_timeout_ms, 0, 3600*1000, false); VariablesPointers_int["default_query_delay"] = make_tuple(&variables.default_query_delay, 0, 3600*1000, false); VariablesPointers_int["default_query_timeout"] = make_tuple(&variables.default_query_timeout, 1000,20*24*3600*1000, false); VariablesPointers_int["query_digests_grouping_limit"] = make_tuple(&variables.query_digests_grouping_limit, 1, 2089, false); VariablesPointers_int["query_digests_groups_grouping_limit"] = make_tuple(&variables.query_digests_groups_grouping_limit, 0, 2089, false); VariablesPointers_int["query_digests_max_digest_length"] = make_tuple(&variables.query_digests_max_digest_length, 16, 1*1024*1024, false); VariablesPointers_int["query_digests_max_query_length"] = make_tuple(&variables.query_digests_max_query_length, 16, 1*1024*1024, false); VariablesPointers_int["query_rules_fast_routing_algorithm"] = make_tuple(&variables.query_rules_fast_routing_algorithm, 1, 2, false); VariablesPointers_int["query_processor_iterations"] = make_tuple(&variables.query_processor_iterations, 0, 1000*1000, false); VariablesPointers_int["query_processor_regex"] = make_tuple(&variables.query_processor_regex, 1, 2, false); VariablesPointers_int["query_retries_on_failure"] = make_tuple(&variables.query_retries_on_failure, 0, 1000, false); VariablesPointers_int["set_query_lock_on_hostgroup"] = make_tuple(&variables.set_query_lock_on_hostgroup, 0, 1, false); VariablesPointers_int["set_parser_algorithm"] = make_tuple(&variables.set_parser_algorithm, 1, 2, false); // throttle VariablesPointers_int["throttle_connections_per_sec_to_hostgroup"] = make_tuple(&variables.throttle_connections_per_sec_to_hostgroup, 1, 100*1000*1000, false); VariablesPointers_int["throttle_max_bytes_per_second_to_client"] = make_tuple(&variables.throttle_max_bytes_per_second_to_client, 0, 2147483647, false); VariablesPointers_int["throttle_ratio_server_to_client"] = make_tuple(&variables.throttle_ratio_server_to_client, 0, 100, false); // backend management VariablesPointers_int["connpoll_reset_queue_length"] = make_tuple(&variables.connpoll_reset_queue_length, 0, 10000, false); VariablesPointers_int["default_max_latency_ms"] = make_tuple(&variables.default_max_latency_ms, 0, 20*24*3600*1000, false); VariablesPointers_int["free_connections_pct"] = make_tuple(&variables.free_connections_pct, 0, 100, false); VariablesPointers_int["poll_timeout"] = make_tuple(&variables.poll_timeout, 10, 20000, false); VariablesPointers_int["poll_timeout_on_failure"] = make_tuple(&variables.poll_timeout_on_failure, 10, 20000, false); VariablesPointers_int["reset_connection_algorithm"] = make_tuple(&variables.reset_connection_algorithm, 1, 2, false); VariablesPointers_int["shun_on_failures"] = make_tuple(&variables.shun_on_failures, 0, 10000000, false); VariablesPointers_int["shun_recovery_time_sec"] = make_tuple(&variables.shun_recovery_time_sec, 0, 3600*24*365, false); VariablesPointers_int["unshun_algorithm"] = make_tuple(&variables.unshun_algorithm, 0, 1, false); VariablesPointers_int["hostgroup_manager_verbose"] = make_tuple(&variables.hostgroup_manager_verbose, 0, 3, false); VariablesPointers_int["tcp_keepalive_time"] = make_tuple(&variables.tcp_keepalive_time, 0, 7200, false); VariablesPointers_int["min_num_servers_lantency_awareness"] = make_tuple(&variables.min_num_servers_lantency_awareness, 0, 10000, false); VariablesPointers_int["aurora_max_lag_ms_only_read_from_replicas"] = make_tuple(&variables.aurora_max_lag_ms_only_read_from_replicas, 0, 100, false); // connection management VariablesPointers_int["connect_retries_on_failure"] = make_tuple(&variables.connect_retries_on_failure, 0, 1000, false); VariablesPointers_int["connect_retries_delay"] = make_tuple(&variables.connect_retries_delay, 0, 10000, false); VariablesPointers_int["connect_timeout_client"] = make_tuple(&variables.connect_timeout_client, 500, 3600*1000, false); VariablesPointers_int["connect_timeout_server"] = make_tuple(&variables.connect_timeout_server, 10, 120*1000, false); VariablesPointers_int["connect_timeout_server_max"] = make_tuple(&variables.connect_timeout_server_max, 10, 3600*1000, false); VariablesPointers_int["connection_delay_multiplex_ms"] = make_tuple(&variables.connection_delay_multiplex_ms, 0, 300*1000, false); VariablesPointers_int["connection_max_age_ms"] = make_tuple(&variables.connection_max_age_ms, 0, 3600*24*1000, false); VariablesPointers_int["handle_unknown_charset"] = make_tuple(&variables.handle_unknown_charset, 0, HANDLE_UNKNOWN_CHARSET__MAX_HANDLE_VALUE, false); VariablesPointers_int["ping_interval_server_msec"] = make_tuple(&variables.ping_interval_server_msec, 1000, 7*24*3600*1000, false); VariablesPointers_int["ping_timeout_server"] = make_tuple(&variables.ping_timeout_server, 10, 600*1000, false); VariablesPointers_int["fast_forward_grace_close_ms"] = make_tuple(&variables.fast_forward_grace_close_ms, 0, 3600*1000, false); VariablesPointers_int["client_host_cache_size"] = make_tuple(&variables.client_host_cache_size, 0, 1024*1024, false); VariablesPointers_int["client_host_error_counts"] = make_tuple(&variables.client_host_error_counts, 0, 1024*1024, false); VariablesPointers_int["handle_warnings"] = make_tuple(&variables.handle_warnings, 0, 1, false); VariablesPointers_int["evaluate_replication_lag_on_servers_load"] = make_tuple(&variables.evaluate_replication_lag_on_servers_load, 0, 1, false); VariablesPointers_int["protocol_compression_level"] = make_tuple(&variables.protocol_compression_level, -1, 9, false); // logs VariablesPointers_int["auditlog_filesize"] = make_tuple(&variables.auditlog_filesize, 1024*1024, 1*1024*1024*1024, false); VariablesPointers_int["eventslog_filesize"] = make_tuple(&variables.eventslog_filesize, 1024*1024, 1*1024*1024*1024, false); VariablesPointers_int["eventslog_buffer_history_size"] = make_tuple(&variables.eventslog_buffer_history_size, 0, 8*1024*1024, false); VariablesPointers_int["eventslog_table_memory_size"] = make_tuple(&variables.eventslog_table_memory_size, 0, 8*1024*1024, false); VariablesPointers_int["eventslog_buffer_max_query_length"] = make_tuple(&variables.eventslog_buffer_max_query_length, 128, 32*1024*1024, false); VariablesPointers_int["eventslog_default_log"] = make_tuple(&variables.eventslog_default_log, 0, 1, false); VariablesPointers_int["eventslog_stmt_parameters"] = make_tuple(&variables.eventslog_stmt_parameters, 0, 1, false); // various VariablesPointers_int["long_query_time"] = make_tuple(&variables.long_query_time, 0, 20*24*3600*1000, false); VariablesPointers_int["max_allowed_packet"] = make_tuple(&variables.max_allowed_packet, 8192, 1024*1024*1024, false); VariablesPointers_int["max_connections"] = make_tuple(&variables.max_connections, 1, 1000*1000, false); VariablesPointers_int["max_stmts_per_connection"] = make_tuple(&variables.max_stmts_per_connection, 1, 1024, false); VariablesPointers_int["max_stmts_cache"] = make_tuple(&variables.max_stmts_cache, 128, 1024*1024, false); VariablesPointers_int["max_transaction_idle_time"] = make_tuple(&variables.max_transaction_idle_time, 1000, 20*24*3600*1000, false); VariablesPointers_int["max_transaction_time"] = make_tuple(&variables.max_transaction_time, 1000, 20*24*3600*1000, false); VariablesPointers_int["query_cache_size_mb"] = make_tuple(&variables.query_cache_size_MB, 0, 1024*10240, false); VariablesPointers_int["query_cache_soft_ttl_pct"] = make_tuple(&variables.query_cache_soft_ttl_pct, 0, 100, false); VariablesPointers_int["query_cache_handle_warnings"] = make_tuple(&variables.query_cache_handle_warnings, 0, 1, false); #ifdef IDLE_THREADS VariablesPointers_int["session_idle_ms"] = make_tuple(&variables.session_idle_ms, 1, 3600*1000, false); #endif // IDLE_THREADS VariablesPointers_int["show_processlist_extended"] = make_tuple(&variables.show_processlist_extended, 0, 2, false); VariablesPointers_int["processlist_max_query_length"] = make_tuple( &variables.processlist_max_query_length, PROCESSLIST_MAX_QUERY_LEN_MIN, PROCESSLIST_MAX_QUERY_LEN_MAX, false ); VariablesPointers_int["threshold_query_length"] = make_tuple(&variables.threshold_query_length, 1024, 1*1024*1024*1024, false); VariablesPointers_int["threshold_resultset_size"] = make_tuple(&variables.threshold_resultset_size, 1024, 1*1024*1024*1024, false); // variables with special variable == true // the input validation for these variables MUST be EXPLICIT VariablesPointers_int["binlog_reader_connect_retry_msec"] = make_tuple(&variables.binlog_reader_connect_retry_msec, 0, 0, true); VariablesPointers_int["eventslog_format"] = make_tuple(&variables.eventslog_format, 0, 0, true); VariablesPointers_int["wait_timeout"] = make_tuple(&variables.wait_timeout, 0, 0, true); VariablesPointers_int["data_packets_history_size"] = make_tuple(&variables.data_packets_history_size, 0, 0, true); } const size_t l=sizeof(mysql_thread_variables_names)/sizeof(char *); unsigned int i; size_t ltv = 0; for (i=0; i < SQL_NAME_LAST_LOW_WM ; i++) { if (mysql_tracked_variables[i].is_global_variable) ltv++; } char **ret=(char **)malloc(sizeof(char *)*(l+ltv)); // not adding + 1 because mysql_thread_variables_names is already NULL terminated size_t fv = 0; for (i=0; i < SQL_NAME_LAST_LOW_WM ; i++) { if (mysql_tracked_variables[i].is_global_variable) { char * m = (char *)malloc(strlen(mysql_tracked_variables[i].internal_variable_name)+1+strlen((char *)"default_")); sprintf(m,"default_%s", mysql_tracked_variables[i].internal_variable_name); ret[fv] = m; fv++; } } // this is an extra check. assert(fv==ltv); for (i=ltv;i 8) { if (strncmp(name, "default_", 8) == 0) { for (unsigned int i = 0; i < SQL_NAME_LAST_LOW_WM ; i++) { if (mysql_tracked_variables[i].is_global_variable) { size_t var_len = strlen(mysql_tracked_variables[i].internal_variable_name); if (strlen(name) == (var_len+8)) { if (!strncmp(name+8, mysql_tracked_variables[i].internal_variable_name, var_len)) { return true; } } } } } } size_t no_vars = sizeof(mysql_thread_variables_names) / sizeof(char *); for (unsigned int i = 0; i < no_vars-1 ; ++i) { size_t var_len = strlen(mysql_thread_variables_names[i]); if (strlen(name) == var_len && !strncmp(name, mysql_thread_variables_names[i], var_len)) { return true; } } return false; } void MySQL_Threads_Handler::print_version() { fprintf(stderr,"Standard MySQL Threads Handler rev. %s -- %s -- %s\n", MYSQL_THREAD_VERSION, __FILE__, __TIMESTAMP__); } /** * @brief Initializes the MySQL Threads Handler with the given number of threads and stack size. * * This function initializes the MySQL Threads Handler with the specified number of threads and * stack size. If the stack size is not provided (or is 0), it defaults to DEFAULT_STACK_SIZE. * If the number of threads is not provided (or is 0), it defaults to DEFAULT_NUM_THREADS. * After initialization, it sets the stack size using pthread_attr_setstacksize, allocates memory * for the MySQL threads, and initializes status variables accordingly. * * @param num The number of threads to initialize. If 0, defaults to DEFAULT_NUM_THREADS. * @param stack The size of the stack for each thread. If 0, defaults to DEFAULT_STACK_SIZE. */ void MySQL_Threads_Handler::init(unsigned int num, size_t stack) { if (stack) { stacksize=stack; } else { if (stacksize==0) stacksize=DEFAULT_STACK_SIZE; } if (num) { num_threads=num; this->status_variables.p_gauge_array[p_th_gauge::mysql_thread_workers]->Set(num); } else { if (num_threads==0) { num_threads=DEFAULT_NUM_THREADS; //default this->status_variables.p_gauge_array[p_th_gauge::mysql_thread_workers]->Set(DEFAULT_NUM_THREADS); } } int rc=pthread_attr_setstacksize(&attr, stacksize); assert(rc==0); mysql_threads=(proxysql_mysql_thread_t *)calloc(num_threads,sizeof(proxysql_mysql_thread_t)); #ifdef IDLE_THREADS if (GloVars.global.idle_threads) mysql_threads_idles=(proxysql_mysql_thread_t *)calloc(num_threads,sizeof(proxysql_mysql_thread_t)); #endif // IDLE_THREADS } /** * @brief Creates a new MySQL thread and starts its execution. * * This function creates a new MySQL thread with the specified thread number, start routine, * and whether it's an idle thread or not. If idles is set to false, a regular MySQL thread * is created using pthread_create. If idles is true and idle_threads are enabled, an idle * MySQL thread is created. After creating the thread, it returns NULL. * * @param tn The thread number to assign to the created thread. * @param start_routine The function the new thread should start executing. * @param idles A boolean indicating whether the created thread is an idle thread or not. * @return A pointer to the created MySQL thread. */ proxysql_mysql_thread_t * MySQL_Threads_Handler::create_thread(unsigned int tn, void *(*start_routine) (void *), bool idles) { if (idles==false) { if (pthread_create(&mysql_threads[tn].thread_id, &attr, start_routine , &mysql_threads[tn]) != 0 ) { // LCOV_EXCL_START proxy_error("Thread creation\n"); assert(0); // LCOV_EXCL_STOP } #if defined(__linux__) || defined(__FreeBSD__) if (GloVars.set_thread_name == true) { char thr_name[16]; snprintf(thr_name, sizeof(thr_name), "MySQLWorker%d", tn); pthread_setname_np(mysql_threads[tn].thread_id, thr_name); } #endif // defined(__linux__) || defined(__FreeBSD__) #ifdef IDLE_THREADS } else { if (GloVars.global.idle_threads) { if (pthread_create(&mysql_threads_idles[tn].thread_id, &attr, start_routine , &mysql_threads_idles[tn]) != 0) { // LCOV_EXCL_START proxy_error("Thread creation\n"); assert(0); // LCOV_EXCL_STOP } #if defined(__linux__) || defined(__FreeBSD__) if (GloVars.set_thread_name == true) { char thr_name[16]; snprintf(thr_name, sizeof(thr_name), "MySQLIdle%d", tn); pthread_setname_np(mysql_threads[tn].thread_id, thr_name); } } #endif // defined(__linux__) || defined(__FreeBSD__) #endif // IDLE_THREADS } return NULL; } void MySQL_Threads_Handler::shutdown_threads() { unsigned int i; shutdown_=1; if (mysql_threads) { for (i=0; ithread_mutex); mysql_threads[i].worker->shutdown=1; pthread_mutex_unlock(&mysql_threads[i].worker->thread_mutex); } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { for (i=0; ithread_mutex); mysql_threads_idles[i].worker->shutdown=1; pthread_mutex_unlock(&mysql_threads[i].worker->thread_mutex); } } } #endif /* IDLE_THREADS */ signal_all_threads(1); for (i=0; iget_variable((char *)"interfaces"); if (strlen(_tmp)==0) { //GloMTH->set_variable((char *)"interfaces", (char *)"0.0.0.0:6033;/tmp/proxysql.sock"); // set default GloMTH->set_variable((char *)"interfaces", (char *)"0.0.0.0:6033"); // changed. See isseu #1104 } free(_tmp); tokenizer_t tok; tokenizer( &tok, variables.interfaces, ";", TOKENIZER_NO_EMPTIES ); const char* token; for (token = tokenize( &tok ); token; token = tokenize( &tok )) { listener_add((char *)token); } free_tokenizer( &tok ); // no more listeners to add bootstrapping_listeners = false; } void MySQL_Threads_Handler::stop_listeners() { if (variables.interfaces==NULL || strlen(variables.interfaces)==0) return; tokenizer_t tok; tokenizer( &tok, variables.interfaces, ";", TOKENIZER_NO_EMPTIES ); const char* token; for (token = tokenize( &tok ); token; token = tokenize( &tok )) { listener_del((char *)token); } free_tokenizer( &tok ); } MySQL_Client_Host_Cache_Entry MySQL_Threads_Handler::find_client_host_cache(struct sockaddr* client_sockaddr) { MySQL_Client_Host_Cache_Entry entry { 0, 0 }; // Client_sockaddr **shouldn't** ever by 'NULL', no matter the // 'session_type' in from which this function is called. Because // `MySQL_Session::client_myds::client_addr` should **always** be // initialized before `handler` is called. assert(client_sockaddr != NULL); if (client_sockaddr->sa_family != AF_INET && client_sockaddr->sa_family != AF_INET6) { return entry; } std::string client_addr = get_client_addr(client_sockaddr); if (client_addr == "127.0.0.1") { return entry; } pthread_mutex_lock(&mutex_client_host_cache); auto found_entry = client_host_cache.find(client_addr); if (found_entry != client_host_cache.end()) { entry = found_entry->second; } pthread_mutex_unlock(&mutex_client_host_cache); return entry; } /** * @brief Number of columns for representing a 'MySQL_Client_Host_Cache_Entry' * in a 'SQLite3_result'. */ const int CLIENT_HOST_CACHE_COLUMNS = 3; /** * @brief Helper function that converts a given client address and a * 'MySQL_Client_Host_Cache_Entry', into a row for a 'SQLite3_result' for * table 'STATS_SQLITE_TABLE_MYSQL_CLIENT_HOST_CACHE'. * * @param address The client address to be added to the resulset row. * @param entry The 'MySQL_Client_Host_Cache_Entry' to be added to the resulset * row. * * @return A pointer array holding the values for each of the columns of the * row. It should be freed through helper function 'free_client_host_cache_row'. */ char** client_host_cache_entry_row( const std::string address, const MySQL_Client_Host_Cache_Entry& entry ) { // INET6_ADDRSTRLEN length should be enough for holding any member: // { address: MAX INET6_ADDRSTRLEN, updated_at: uint64_t, error_count: uint32_t } char buff[INET6_ADDRSTRLEN]; char** row = static_cast(malloc(sizeof(char*)*CLIENT_HOST_CACHE_COLUMNS)); time_t __now = time(NULL); unsigned long long curtime = monotonic_time(); time_t last_updated = __now - curtime/1000000 + entry.updated_at/1000000; row[0]=strdup(address.c_str()); sprintf(buff, "%u", entry.error_count); row[1]=strdup(buff); sprintf(buff, "%lu", last_updated); row[2]=strdup(buff); return row; } /** * @brief Helper function to free the row returned by * 'client_host_cache_entry_row'. * * @param row The pointer array holding the row values to be freed. */ static void free_client_host_cache_row(char** row) { for (int i = 0; i < CLIENT_HOST_CACHE_COLUMNS; i++) { free(row[i]); } free(row); } SQLite3_result* MySQL_Threads_Handler::get_client_host_cache(bool reset) { SQLite3_result *result = new SQLite3_result(CLIENT_HOST_CACHE_COLUMNS); pthread_mutex_lock(&mutex_client_host_cache); result->add_column_definition(SQLITE_TEXT,"client_address"); result->add_column_definition(SQLITE_TEXT,"error_count"); result->add_column_definition(SQLITE_TEXT,"last_updated"); for (const auto& cache_entry : client_host_cache) { char** row = client_host_cache_entry_row(cache_entry.first, cache_entry.second); result->add_row(row); free_client_host_cache_row(row); } if (reset) { client_host_cache.clear(); } pthread_mutex_unlock(&mutex_client_host_cache); return result; } void MySQL_Threads_Handler::update_client_host_cache(struct sockaddr* client_sockaddr, bool error) { // Client_sockaddr **shouldn't** ever by 'NULL', no matter the // 'session_type' in from which this function is called. Because // `MySQL_Session::client_myds::client_addr` should **always** be // initialized before `handler` is called. assert(client_sockaddr != NULL); if (client_sockaddr->sa_family != AF_INET && client_sockaddr->sa_family != AF_INET6) { return; } std::string client_addr = get_client_addr(client_sockaddr); if (client_addr == "127.0.0.1") { return; } if (error) { pthread_mutex_lock(&mutex_client_host_cache); // If the cache is full, find the oldest entry on it, and update/remove it. if ( mysql_thread___client_host_cache_size && client_host_cache.size() >= static_cast(mysql_thread___client_host_cache_size) ) { auto older_elem = std::min_element( client_host_cache.begin(), client_host_cache.end(), [] (const std::pair& f_entry, const std::pair& s_entry) { return f_entry.second.updated_at < s_entry.second.updated_at; } ); if (older_elem != client_host_cache.end()) { if (older_elem->first != client_addr) { client_host_cache.erase(older_elem); } } } // Find the entry for the client, and update/insert it. auto cache_entry = client_host_cache.find(client_addr); if (cache_entry != client_host_cache.end()) { cache_entry->second.error_count += 1; cache_entry->second.updated_at = monotonic_time(); } else { // Notice than the value of 'mysql_thread___client_host_cache_size' can // change at runtime. Due to this, we should only insert when the size of the // cache is smaller than this value, otherwise we could end in situations in // which cache doesn't shrink after it's size is reduced at runtime. if (client_host_cache.size() < static_cast(mysql_thread___client_host_cache_size)) { MySQL_Client_Host_Cache_Entry new_entry { monotonic_time(), 1 }; client_host_cache.insert({client_addr, new_entry}); } } pthread_mutex_unlock(&mutex_client_host_cache); } else { pthread_mutex_lock(&mutex_client_host_cache); client_host_cache.erase(client_addr); pthread_mutex_unlock(&mutex_client_host_cache); } } /** * @brief Flushes the client host cache. * * This function locks the mutex associated with the client host cache, clears the cache, * and then unlocks the mutex. It is used to remove all entries from the client host cache. * * @note This function assumes that the mutex_client_host_cache has been initialized * and is accessible within the MySQL Threads Handler. */ void MySQL_Threads_Handler::flush_client_host_cache() { pthread_mutex_lock(&mutex_client_host_cache); client_host_cache.clear(); pthread_mutex_unlock(&mutex_client_host_cache); } MySQL_Threads_Handler::~MySQL_Threads_Handler() { if (variables.monitor_username) { free(variables.monitor_username); variables.monitor_username=NULL; } if (variables.monitor_password) { free(variables.monitor_password); variables.monitor_password=NULL; } if (variables.monitor_replication_lag_use_percona_heartbeat) { free(variables.monitor_replication_lag_use_percona_heartbeat); variables.monitor_replication_lag_use_percona_heartbeat=NULL; } if (variables.default_schema) free(variables.default_schema); if (variables.interfaces) free(variables.interfaces); if (variables.server_version) free(variables.server_version); if (variables.keep_multiplexing_variables) free(variables.keep_multiplexing_variables); if (variables.default_authentication_plugin) free(variables.default_authentication_plugin); if (variables.proxy_protocol_networks) free(variables.proxy_protocol_networks); if (variables.firewall_whitelist_errormsg) free(variables.firewall_whitelist_errormsg); if (variables.init_connect) free(variables.init_connect); if (variables.ldap_user_variable) free(variables.ldap_user_variable); if (variables.add_ldap_user_comment) free(variables.add_ldap_user_comment); if (variables.default_session_track_gtids) free(variables.default_session_track_gtids); if (variables.eventslog_filename) free(variables.eventslog_filename); if (variables.auditlog_filename) free(variables.auditlog_filename); if (variables.ssl_p2s_ca) free(variables.ssl_p2s_ca); if (variables.ssl_p2s_capath) free(variables.ssl_p2s_capath); if (variables.ssl_p2s_cert) free(variables.ssl_p2s_cert); if (variables.ssl_p2s_key) free(variables.ssl_p2s_key); if (variables.ssl_p2s_cipher) free(variables.ssl_p2s_cipher); if (variables.ssl_p2s_crl) free(variables.ssl_p2s_crl); if (variables.ssl_p2s_crlpath) free(variables.ssl_p2s_crlpath); for (int i=0; ilen) { MySQL_Session *sess=(MySQL_Session *)mysql_sessions->remove_index_fast(0); if (sess->session_type == PROXYSQL_SESSION_ADMIN || sess->session_type == PROXYSQL_SESSION_STATS) { char _buf[1024]; sprintf(_buf,"%s:%d:%s()", __FILE__, __LINE__, __func__); if (GloMyLogger) { GloMyLogger->log_audit_entry(PROXYSQL_MYSQL_AUTH_CLOSE, sess, NULL, _buf); } } delete sess; } delete mysql_sessions; mysql_sessions=NULL; GloMyQPro->end_thread(); // only for real threads } if (mirror_queue_mysql_sessions) { while(mirror_queue_mysql_sessions->len) { MySQL_Session *sess=(MySQL_Session *)mirror_queue_mysql_sessions->remove_index_fast(0); delete sess; } delete mirror_queue_mysql_sessions; mirror_queue_mysql_sessions=NULL; } if (mirror_queue_mysql_sessions_cache) { while(mirror_queue_mysql_sessions_cache->len) { MySQL_Session *sess=(MySQL_Session *)mirror_queue_mysql_sessions_cache->remove_index_fast(0); delete sess; } delete mirror_queue_mysql_sessions_cache; mirror_queue_mysql_sessions_cache=NULL; } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { if (idle_mysql_sessions) { while(idle_mysql_sessions->len) { MySQL_Session *sess=(MySQL_Session *)idle_mysql_sessions->remove_index_fast(0); delete sess; } delete idle_mysql_sessions; } if (resume_mysql_sessions) { while(resume_mysql_sessions->len) { MySQL_Session *sess=(MySQL_Session *)resume_mysql_sessions->remove_index_fast(0); delete sess; } delete resume_mysql_sessions; } if (myexchange.idle_mysql_sessions) { while(myexchange.idle_mysql_sessions->len) { MySQL_Session *sess=(MySQL_Session *)myexchange.idle_mysql_sessions->remove_index_fast(0); delete sess; } delete myexchange.idle_mysql_sessions; } if (myexchange.resume_mysql_sessions) { while(myexchange.resume_mysql_sessions->len) { MySQL_Session *sess=(MySQL_Session *)myexchange.resume_mysql_sessions->remove_index_fast(0); delete sess; } delete myexchange.resume_mysql_sessions; } } #endif // IDLE_THREADS if (cached_connections) { return_local_connections(); delete cached_connections; } unsigned int i; for (i=0;imyds_type==MYDS_LISTENER) { delete mypolls.myds[i]; } } if (my_idle_conns) free(my_idle_conns); if (mysql_thread___monitor_username) { free(mysql_thread___monitor_username); mysql_thread___monitor_username=NULL; } if (mysql_thread___monitor_password) { free(mysql_thread___monitor_password); mysql_thread___monitor_password=NULL; } if (mysql_thread___monitor_replication_lag_use_percona_heartbeat) { free(mysql_thread___monitor_replication_lag_use_percona_heartbeat); mysql_thread___monitor_replication_lag_use_percona_heartbeat=NULL; } if (mysql_thread___default_schema) { free(mysql_thread___default_schema); mysql_thread___default_schema=NULL; } if (mysql_thread___server_version) { free(mysql_thread___server_version); mysql_thread___server_version=NULL; } if (mysql_thread___keep_multiplexing_variables) { free(mysql_thread___keep_multiplexing_variables); mysql_thread___keep_multiplexing_variables=NULL; } if (mysql_thread___default_authentication_plugin) { free(mysql_thread___default_authentication_plugin); mysql_thread___default_authentication_plugin=NULL; } if (mysql_thread___proxy_protocol_networks) { free(mysql_thread___proxy_protocol_networks); mysql_thread___proxy_protocol_networks=NULL; } if (mysql_thread___firewall_whitelist_errormsg) { free(mysql_thread___firewall_whitelist_errormsg); mysql_thread___firewall_whitelist_errormsg=NULL; } if (mysql_thread___init_connect) { free(mysql_thread___init_connect); mysql_thread___init_connect=NULL; } if (mysql_thread___ldap_user_variable) { free(mysql_thread___ldap_user_variable); mysql_thread___ldap_user_variable=NULL; } if (mysql_thread___add_ldap_user_comment) { free(mysql_thread___add_ldap_user_comment); mysql_thread___add_ldap_user_comment=NULL; } if (mysql_thread___default_session_track_gtids) { free(mysql_thread___default_session_track_gtids); mysql_thread___default_session_track_gtids=NULL; } for (int i=0; iinit_thread(); refresh_variables(); i=pipe(pipefd); ioctl_FIONBIO(pipefd[0],1); ioctl_FIONBIO(pipefd[1],1); mypolls.add(POLLIN, pipefd[0], NULL, 0); assert(i==0); thr_SetParser = new MySQL_Set_Stmt_Parser(""); match_regexes=(Session_Regex **)malloc(sizeof(Session_Regex *)*4); // match_regexes[0]=new Session_Regex((char *)"^SET (|SESSION |@@|@@session.)SQL_LOG_BIN( *)(:|)=( *)"); match_regexes[0] = NULL; // NOTE: historically we used match_regexes[0] for SET SQL_LOG_BIN . Not anymore std::stringstream ss; ss << "^SET (|SESSION |@@|@@session.|@@local.)`?(" << mysql_variables.variables_regexp << "SESSION_TRACK_GTIDS|TX_ISOLATION|TX_READ_ONLY|TRANSACTION_ISOLATION|TRANSACTION_READ_ONLY)`?( *)(:|)=( *)"; match_regexes[1]=new Session_Regex((char *)ss.str().c_str()); match_regexes[2]=new Session_Regex((char *)"^SET(?: +)(|SESSION +)TRANSACTION(?: +)(?:(?:(ISOLATION(?: +)LEVEL)(?: +)(REPEATABLE(?: +)READ|READ(?: +)COMMITTED|READ(?: +)UNCOMMITTED|SERIALIZABLE))|(?:(READ)(?: +)(WRITE|ONLY)))"); match_regexes[3]=new Session_Regex((char *)"^(set)(?: +)((charset)|(character +set))(?: )"); return true; } struct pollfd * MySQL_Thread::get_pollfd(unsigned int i) { return &mypolls.fds[i]; } void MySQL_Thread::poll_listener_add(int sock) { MySQL_Data_Stream *listener_DS = new MySQL_Data_Stream(); listener_DS->myds_type=MYDS_LISTENER; listener_DS->fd=sock; proxy_debug(PROXY_DEBUG_NET,1,"Created listener %p for socket %d\n", listener_DS, sock); /** * @brief Register listener socket with ProxySQL_Poll for incoming connections * * This usage pattern registers a listener socket file descriptor with the ProxySQL_Poll instance * to monitor for incoming client connections. The listener data stream handles the accept() * operation when connection events are detected. * * Usage pattern: mypolls.add(POLLIN, sock, listener_DS, monotonic_time()) * - POLLIN: Monitor for read events (new connections ready to accept) * - sock: Listener socket file descriptor * - listener_DS: Data stream associated with the listener (accepts connections) * - monotonic_time(): Current timestamp for tracking socket registration time * * Called during: Listener setup and initialization * Purpose: Enables the thread to accept incoming MySQL client connections */ mypolls.add(POLLIN, sock, listener_DS, monotonic_time()); } void MySQL_Thread::poll_listener_del(int sock) { /** * @brief Remove listener socket from the poll set using efficient index lookup * * This usage pattern demonstrates the complete removal workflow for listener sockets: * 1. Find the index of the socket in the poll set using find_index() * 2. Remove the socket using remove_index_fast() with the found index * * Usage pattern: * int i = mypollolls.find_index(sock); // Find index by file descriptor * if (i>=0) { * mypolls.remove_index_fast(i); // Remove by index (O(1) operation) * } * * find_index(sock): Returns index of socket or -1 if not found * remove_index_fast(i): Removes the entry at index i efficiently * * Called during: Listener shutdown and cleanup * Purpose: Properly removes listener sockets from polling to prevent memory leaks */ int i=mypolls.find_index(sock); if (i>=0) { MySQL_Data_Stream *myds=mypolls.myds[i]; mypolls.remove_index_fast(i); #ifdef SO_REUSEPORT if (GloVars.global.reuseport) #else myds->fd=-1; // this to prevent that delete myds will shutdown the fd; #endif delete myds; } } void MySQL_Thread::unregister_session(int idx) { if (mysql_sessions==NULL) return; proxy_debug(PROXY_DEBUG_NET,1,"Thread=%p, Session=%p -- Unregistered session\n", this, mysql_sessions->index(idx)); mysql_sessions->remove_index_fast(idx); } // this function was inline in MySQL_Thread::run() /** * @brief Retrieves multiple idle connections and processes them. * * This method retrieves multiple idle connections from the MySQL Hostgroup Manager (MyHGM) and processes them. * It updates the number of idle connections (num_idles) based on the retrieved connections. For each idle connection, * it creates a new MySQL session, attaches the connection to the appropriate backend, assigns a file descriptor, * sets necessary parameters, registers the session with the connection handler, and initiates handling. * * @param num_idles Reference to an integer to store the number of idle connections retrieved. * * @note This method assumes that MyHGM, my_idle_conns, SESSIONS_FOR_CONNECTIONS_HANDLER, curtime, and relevant MySQL thread variables have been properly initialized and are accessible. */ void MySQL_Thread::run___get_multiple_idle_connections(int& num_idles) { int i; num_idles=MyHGM->get_multiple_idle_connections(-1, curtime-mysql_thread___ping_interval_server_msec*1000, my_idle_conns, SESSIONS_FOR_CONNECTIONS_HANDLER); for (i=0; imybe=sess->find_or_create_backend(mc->parent->myhgc->hid); myds=sess->mybe->server_myds; myds->attach_connection(mc); myds->assign_fd_from_mysql_conn(); myds->myds_type=MYDS_BACKEND; sess->to_process=1; myds->wait_until=curtime+mysql_thread___ping_timeout_server*1000; // max_timeout mc->last_time_used=curtime; myds->myprot.init(&myds, myds->myconn->userinfo, NULL); sess->status=PINGING_SERVER; myds->DSS=STATE_MARIADB_PING; register_session_connection_handler(sess,true); int rc=sess->handler(); if (rc==-1) { unsigned int sess_idx=mysql_sessions->len-1; unregister_session(sess_idx); delete sess; } } processing_idles=true; last_processing_idles=curtime; } // this function was inline in MySQL_Thread::run() /** * @brief Cleans up the mirror queue by removing excess sessions. * * This function checks if the length of the mirror queue exceeds the maximum concurrency limit. * If it does, it iteratively removes sessions from the mirror queue until its length matches * the maximum concurrency limit or falls below it. For each removed session, it updates * the mirror sessions current count and increments the mirror concurrency gauge accordingly. */ void MySQL_Thread::run___cleanup_mirror_queue() { unsigned int l = (unsigned int)mysql_thread___mirror_max_concurrency; if (mirror_queue_mysql_sessions_cache->len > l) { while (mirror_queue_mysql_sessions_cache->len > mirror_queue_mysql_sessions->len && mirror_queue_mysql_sessions_cache->len > l) { MySQL_Session *newsess=(MySQL_Session *)mirror_queue_mysql_sessions_cache->remove_index_fast(0); __sync_add_and_fetch(&GloMTH->status_variables.mirror_sessions_current,1); GloMTH->status_variables.p_gauge_array[p_th_gauge::mirror_concurrency]->Increment(); delete newsess; } } } #ifdef IDLE_THREADS /** * @brief Handles session assignment and retrieval between worker and idle threads. * * This block of code checks if the global configuration allows idle threads and if the current thread * is not an idle maintenance thread. If both conditions are met, it randomly selects an idle worker thread * and assigns sessions to it. Then, it retrieves sessions from the idle thread. * * @note This functionality is part of the management of worker and idle threads in the MySQL thread pool. * It facilitates the distribution of sessions between active worker threads and idle threads to optimize resource utilization. * * @param idle_maintenance_thread Flag indicating whether the current thread is an idle maintenance thread. * @param GloVars Global configuration variables for the MySQL thread. * @param GloMTH Global MySQL thread handlers object. */ void MySQL_Thread::run_MoveSessionsBetweenThreads() { if (GloVars.global.idle_threads) { int r=rand()%(GloMTH->num_threads); MySQL_Thread *thr=GloMTH->mysql_threads_idles[r].worker; worker_thread_assigns_sessions_to_idle_thread(thr); worker_thread_gets_sessions_from_idle_thread(); } } void MySQL_Thread::run_Handle_epoll_wait(int rc) { if (rc) { int i; for (i=0; ilen && maintenance_loop) { if (curtime == last_maintenance_time) { idle_thread_to_kill_idle_sessions(); } } } #endif // IDLE_THREADS void MySQL_Thread::run_BootstrapListener() { unsigned int n; while ( // spin here if ... (n=__sync_add_and_fetch(&mypolls.pending_listener_add,0)) // there is a new listener to add || (GloMTH->bootstrapping_listeners == true) // MySQL_Thread_Handlers has more listeners to configure ) { if (n) { poll_listener_add(n); assert(__sync_bool_compare_and_swap(&mypolls.pending_listener_add,n,0)); } // The delay for the active-wait is a fraction of 'poll_timeout'. Since other // threads may be waiting on poll for further operations, checks are meaningless // until that timeout expires (other workers make progress). usleep(std::min(std::max(mysql_thread___poll_timeout/20, 10000), 40000) + (rand() % 2000)); } } int MySQL_Thread::run_ComputePollTimeout() { proxy_debug(PROXY_DEBUG_NET, 7, "poll_timeout=%u\n", mypolls.poll_timeout); if (mysql_thread___wait_timeout==0) { // we should be going into PAUSE mode if (mypolls.poll_timeout==0 || mypolls.poll_timeout > 100000) { mypolls.poll_timeout=100000; } } proxy_debug(PROXY_DEBUG_NET, 7, "poll_timeout=%u\n", mypolls.poll_timeout); pre_poll_time=curtime; int ttw = ( mypolls.poll_timeout ? ( mypolls.poll_timeout/1000 < (unsigned int) mysql_thread___poll_timeout ? mypolls.poll_timeout/1000 : mysql_thread___poll_timeout ) : mysql_thread___poll_timeout ); return ttw; } void MySQL_Thread::run_StopListener() { unsigned int n; while ((n=__sync_add_and_fetch(&mypolls.pending_listener_del,0))) { // spin here if (static_cast(n) == -1) { for (unsigned int i = 0; i < mypolls.len; i++) { if (mypolls.myds[i] && mypolls.myds[i]->myds_type == MYDS_LISTENER) { poll_listener_del(mypolls.myds[i]->fd); } } } else { poll_listener_del(n); } assert(__sync_bool_compare_and_swap(&mypolls.pending_listener_del,n,0)); } } // main loop /** * @brief Main loop for the MySQL thread. * * This method represents the main loop executed by the MySQL thread. It performs various tasks including handling idle connections, * processing sessions, performing maintenance, and updating variables. The loop continues executing until shutdown is initiated. * * @note This method assumes that relevant variables, mutexes, and objects have been properly initialized. */ void MySQL_Thread::run() { int rc; #ifdef IDLE_THREADS bool idle_maintenance_thread=epoll_thread; if (idle_maintenance_thread) { // we check if it is the first time we are called if (efd==-1) { efd = EPOLL_CREATE; int fd=pipefd[0]; struct epoll_event event; memset(&event,0,sizeof(event)); // let's make valgrind happy event.events = EPOLLIN; event.data.u32=0; // special value to point to the pipe epoll_ctl(efd, EPOLL_CTL_ADD, fd, &event); } } #endif // IDLE_THREADS curtime=monotonic_time(); atomic_curtime=curtime; pthread_mutex_lock(&thread_mutex); while (shutdown==0) { #ifdef IDLE_THREADS if (idle_maintenance_thread) { goto __run_skip_1; } #endif // IDLE_THREADS int num_idles; if (processing_idles==true && (last_processing_idles < curtime-mysql_thread___ping_timeout_server*1000)) { processing_idles=false; } if (processing_idles==false && (last_processing_idles < curtime-mysql_thread___ping_interval_server_msec*1000) ) { run___get_multiple_idle_connections(num_idles); } #ifdef IDLE_THREADS __run_skip_1: if (idle_maintenance_thread) { idle_thread_gets_sessions_from_worker_thread(); } else { #endif // IDLE_THREADS handle_mirror_queue_mysql_sessions(); ProcessAllMyDS_BeforePoll(); #ifdef IDLE_THREADS run_MoveSessionsBetweenThreads(); } #endif // IDLE_THREADS pthread_mutex_unlock(&thread_mutex); run_BootstrapListener(); // flush mysql log file GloMyLogger->flush(); int ttw = run_ComputePollTimeout(); #ifdef IDLE_THREADS if (GloVars.global.idle_threads && idle_maintenance_thread) { memset(events,0,sizeof(struct epoll_event)*MY_EPOLL_THREAD_MAXEVENTS); // let's make valgrind happy. It also seems that needs to be zeroed anyway // we call epoll() rc = epoll_wait (efd, events, MY_EPOLL_THREAD_MAXEVENTS, mysql_thread___poll_timeout); } else { #endif // IDLE_THREADS //this is the only portion of code not protected by a global mutex proxy_debug(PROXY_DEBUG_NET,5,"Calling poll with timeout %d\n", ttw ); // poll is called with a timeout of mypolls.poll_timeout if set , or mysql_thread___poll_timeout /** * @brief Execute main poll() loop to monitor all registered FDs * * This usage pattern demonstrates the core polling mechanism that drives ProxySQL's event loop. * The poll() system call blocks until one of the registered file descriptors becomes ready * or the timeout expires. * * Usage pattern: rc = poll(mypolls.fds, mypolls.len, ttw) * - mypollolls.fds: Array of pollfd structures containing file descriptors and events * - mypolls.len: Number of file descriptors to monitor * - ttw: Timeout in milliseconds (mydynamic poll timeout) * * Return codes: * - > 0: Number of file descriptors with events * - 0: Timeout occurred * - -1: Error (errno set) * * Called during: Main event loop iteration * Purpose: Enables efficient I/O multiplexing across all connections */ rc=poll(mypolls.fds,mypolls.len, ttw); proxy_debug(PROXY_DEBUG_NET,5,"%s\n", "Returning poll"); #ifdef IDLE_THREADS } #endif // IDLE_THREADS if (unlikely(maintenance_loop == true)) { if (unlikely(__sync_add_and_fetch(&mypolls.pending_listener_del,0))) { run_StopListener(); } } pthread_mutex_lock(&thread_mutex); if (shutdown == 1) { return; } mypolls.poll_timeout=0; // always reset this to 0 . If a session needs a specific timeout, it will set this one curtime=monotonic_time(); atomic_curtime=curtime; poll_timeout_bool=false; if ( #ifdef IDLE_THREADS idle_maintenance_thread==false && #endif // IDLE_THREADS (curtime >= (pre_poll_time + ttw))) { poll_timeout_bool=true; } unsigned long long maintenance_interval = 1000000; // hardcoded value for now #ifdef IDLE_THREADS if (idle_maintenance_thread) { maintenance_interval=maintenance_interval*2; } #endif // IDLE_THREADS if (curtime > last_maintenance_time + maintenance_interval) { last_maintenance_time=curtime; maintenance_loop=true; servers_table_version_previous = servers_table_version_current; servers_table_version_current = MyHGM->get_servers_table_version(); // during a maintenance loop (every 1 second) we read has_gtid_port from MyHGM retrieve_gtids_required = MyHGM->has_gtid_port; } else { maintenance_loop=false; } handle_kill_queues(); // update polls statistics mypolls.loops++; mypolls.loop_counters->incr(curtime/1000000); if (maintenance_loop == true #ifdef IDLE_THREADS // in case of idle thread // do not run any mirror cleanup and do not // update query processor stats && idle_maintenance_thread == false #endif // IDLE_THREADS ) { // house keeping run___cleanup_mirror_queue(); GloMyQPro->update_query_processor_stats(); } if (rc == -1 && errno == EINTR) // poll() timeout, try again continue; if (rc == -1) { // LCOV_EXCL_START // error , exit perror("poll()"); exit(EXIT_FAILURE); // LCOV_EXCL_STOP } if (__sync_add_and_fetch(&__global_MySQL_Thread_Variables_version,0) > __thread_MySQL_Thread_Variables_version) { refresh_variables(); } run_SetAllSession_ToProcess0(); #ifdef IDLE_THREADS // here we handle epoll_wait() if (GloVars.global.idle_threads && idle_maintenance_thread) { run_Handle_epoll_wait(rc); unsigned int w=rand()%(GloMTH->num_threads); MySQL_Thread *thr=GloMTH->mysql_threads[w].worker; if (resume_mysql_sessions->len) { idle_thread_assigns_sessions_to_worker_thread(thr); } else { idle_thread_check_if_worker_thread_has_unprocess_resumed_sessions_and_signal_it(thr); } } else { #endif // IDLE_THREADS ProcessAllMyDS_AfterPoll(); // iterate through all sessions and process the session logic process_all_sessions(); return_local_connections(); #ifdef IDLE_THREADS } #endif // IDLE_THREADS #ifdef DEBUG // This block is only used for Watchdog unit tests: // Specifically for PROXYSQLTEST cases 55 0 and 55 1. while (watchdog_test__simulated_delay_ms > 0) { std::this_thread::sleep_for(std::chrono::milliseconds(100)); } #endif } } // end of ::run() #ifdef IDLE_THREADS /** * @brief Moves idle threads to kill idle sessions. * * This function iterates through a portion of MySQL sessions to scan for idle sessions. * If the current time exceeds the wait timeout threshold, it marks sessions idle for longer * than the timeout as killed. It then removes the killed sessions from the session map, * adjusts the map if necessary, unregisters the sessions, and adds them to the list of * sessions to be resumed. Additionally, it removes associated data streams from the poll list * and epoll control, updating relevant indices and pointers. * * @note This function assumes that MySQL sessions and related data structures have been * initialized and are accessible within the MySQL Thread. */ void MySQL_Thread::idle_thread_to_kill_idle_sessions() { if (mysess_idx + SESS_TO_SCAN_idle_thread > mysql_sessions->len) { mysess_idx=0; } unsigned int i; if (curtime < (unsigned long long)mysql_thread___wait_timeout*1000) { return; // this should never happen //min_idle = curtime - (unsigned long long)mysql_thread___wait_timeout*1000; } for (i=0 ; i < SESS_TO_SCAN_idle_thread && mysess_idx < mysql_sessions->len; i++) { uint32_t sess_pos=mysess_idx; MySQL_Session *mysess=(MySQL_Session *)mysql_sessions->index(sess_pos); unsigned long long effective_wait_timeout = std::min( static_cast(mysql_thread___wait_timeout), static_cast(mysess->wait_timeout) ); unsigned long long min_idle = 0; min_idle = curtime - (unsigned long long)effective_wait_timeout*1000; if (mysess->idle_since < min_idle) { unsigned long long sess_time = curtime - mysess->idle_since; proxy_warning("Killing client connection %s:%d because inactive for %llums\n", mysess->client_myds->addr.addr, mysess->client_myds->addr.port, sess_time/1000); mysess->killed=true; // Increment counter for timeout-terminated connections mysess->thread->status_variables.stvar[st_var_timeout_terminated_connections]++; } if (mysess->killed==true) { // because idle or for any other reason MySQL_Data_Stream *tmp_myds=mysess->client_myds; int dsidx=tmp_myds->poll_fds_idx; //fprintf(stderr,"Removing session %p, DS %p idx %d\n",mysess,tmp_myds,dsidx); mypolls.remove_index_fast(dsidx); tmp_myds->mypolls=NULL; mysess->thread=NULL; // we first delete the association in sessmap sessmap.erase(mysess->thread_session_id); if (mysql_sessions->len > 1) { // take the last element and adjust the map MySQL_Session *mysess_last=(MySQL_Session *)mysql_sessions->index(mysql_sessions->len-1); if (mysess->thread_session_id != mysess_last->thread_session_id) sessmap[mysess_last->thread_session_id]=sess_pos; } unregister_session(sess_pos); resume_mysql_sessions->add(mysess); epoll_ctl(efd, EPOLL_CTL_DEL, tmp_myds->fd, NULL); } mysess_idx++; } } void MySQL_Thread::idle_thread_prepares_session_to_send_to_worker_thread(int i) { // NOTE: not sure why, sometime events returns odd values. If set, we take it out as normal worker threads know how to handle it if (events[i].events) { uint32_t sess_thr_id=events[i].data.u32; uint32_t sess_pos=sessmap[sess_thr_id]; MySQL_Session *mysess=(MySQL_Session *)mysql_sessions->index(sess_pos); MySQL_Data_Stream *tmp_myds=mysess->client_myds; int dsidx=tmp_myds->poll_fds_idx; //fprintf(stderr,"Removing session %p, DS %p idx %d\n",mysess,tmp_myds,dsidx); mypolls.remove_index_fast(dsidx); tmp_myds->mypolls=NULL; mysess->thread=NULL; // we first delete the association in sessmap sessmap.erase(mysess->thread_session_id); if (mysql_sessions->len > 1) { // take the last element and adjust the map MySQL_Session *mysess_last=(MySQL_Session *)mysql_sessions->index(mysql_sessions->len-1); if (mysess->thread_session_id != mysess_last->thread_session_id) sessmap[mysess_last->thread_session_id]=sess_pos; } unregister_session(sess_pos); resume_mysql_sessions->add(mysess); epoll_ctl(efd, EPOLL_CTL_DEL, tmp_myds->fd, NULL); } } void MySQL_Thread::idle_thread_check_if_worker_thread_has_unprocess_resumed_sessions_and_signal_it(MySQL_Thread *thr) { pthread_mutex_lock(&thr->myexchange.mutex_resumes); if (shutdown==0 && thr->shutdown==0 && thr->myexchange.resume_mysql_sessions->len) { unsigned char c=0; int fd=thr->pipefd[1]; if (write(fd,&c,1)==-1) { //proxy_error("Error while signaling maintenance thread\n"); } } pthread_mutex_unlock(&thr->myexchange.mutex_resumes); } /** * @brief Assigns idle sessions to a worker thread for processing. * * This function is executed by an idle thread to assign idle sessions to a specified worker thread * for processing. It locks the mutex associated with the worker thread's session exchange mechanism, * checks if there are sessions to resume, and if so, transfers them to the worker thread's list * of sessions to resume. After transferring sessions, it determines whether to send a signal to the * worker thread to inform it of the presence of new sessions. * * @param thr The worker thread to which idle sessions will be assigned. */ void MySQL_Thread::idle_thread_assigns_sessions_to_worker_thread(MySQL_Thread *thr) { bool send_signal = false; // send_signal variable will control if we need to signal or not // the worker thread pthread_mutex_lock(&thr->myexchange.mutex_resumes); if (shutdown==0 && thr->shutdown==0) if (resume_mysql_sessions->len) { while (resume_mysql_sessions->len) { MySQL_Session *mysess=(MySQL_Session *)resume_mysql_sessions->remove_index_fast(0); thr->myexchange.resume_mysql_sessions->add(mysess); } send_signal=true; // signal only if there are sessions to resume } pthread_mutex_unlock(&thr->myexchange.mutex_resumes); if (send_signal) { // signal only if there are sessions to resume unsigned char c=0; //MySQL_Thread *thr=GloMTH->mysql_threads[w].worker; // we signal the thread to inform there are sessions int fd=thr->pipefd[1]; if (write(fd,&c,1)==-1) { //proxy_error("Error while signaling maintenance thread\n"); } } } /** * @brief Assigns idle sessions to an idle thread for processing. * * This function is executed by a worker thread to assign idle sessions to a specified idle thread * for processing. It checks if both the current thread and the idle thread are not in shutdown mode, * and if there are idle sessions to assign. If conditions are met, it locks the mutex associated with * the idle thread's session exchange mechanism, transfers idle sessions to the idle thread's list of * sessions to process, and sends a signal to the idle thread if its session queue was empty before * transferring sessions. * * @param thr The idle thread to which idle sessions will be assigned. */ void MySQL_Thread::worker_thread_assigns_sessions_to_idle_thread(MySQL_Thread *thr) { if (shutdown==0 && thr->shutdown==0 && idle_mysql_sessions->len) { pthread_mutex_lock(&thr->myexchange.mutex_idles); bool empty_queue=true; if (thr->myexchange.idle_mysql_sessions->len) { // there are already sessions in the queues. We assume someone already notified worker 0 empty_queue=false; } while (idle_mysql_sessions->len) { MySQL_Session *mysess=(MySQL_Session *)idle_mysql_sessions->remove_index_fast(0); thr->myexchange.idle_mysql_sessions->add(mysess); } pthread_mutex_unlock(&thr->myexchange.mutex_idles); if (empty_queue==true) { unsigned char c=1; int fd=thr->pipefd[1]; if (write(fd,&c,1)==-1) { //proxy_error("Error while signaling maintenance thread\n"); } } } } /** * @brief Worker thread retrieves sessions from the idle thread for processing. * * This function is executed by a worker thread to retrieve sessions from the idle thread * for processing. It locks the mutex associated with the session exchange mechanism, * checks if there are sessions to resume, and if so, retrieves them from the idle thread's * list of sessions to be resumed. For each retrieved session, it registers the session, * adds its associated data stream to the poll list for monitoring read events, and updates * the poll timestamp. After processing all available sessions, it unlocks the mutex. * * @note This function assumes that the worker thread's session exchange mechanism has been * initialized and is accessible within the MySQL Thread. */ void MySQL_Thread::worker_thread_gets_sessions_from_idle_thread() { pthread_mutex_lock(&myexchange.mutex_resumes); if (myexchange.resume_mysql_sessions->len) { //unsigned int maxsess=GloMTH->resume_mysql_sessions->len; while (myexchange.resume_mysql_sessions->len) { MySQL_Session *mysess=(MySQL_Session *)myexchange.resume_mysql_sessions->remove_index_fast(0); register_session(this, mysess, false); MySQL_Data_Stream *myds=mysess->client_myds; /** * @brief Add client session to poll set for resumed connections * * This usage pattern registers a client data stream for resumed connections * during session restoration in IDLE_THREADS mode. * * Usage pattern: mypolls.add(POLLIN, myds->fd, myds, monotonic_time()) * - POLLIN: Monitor for read events (client data available) * - myds->fd: Client socket file descriptor * - myds: MySQL_Data_Stream instance for the client session * - monotonic_time(): Current timestamp for tracking session registration time * * Called during: Session restoration in IDLE_THREADS mode * Purpose: Enables the thread to receive and process client MySQL protocol data * for resumed sessions */ mypolls.add(POLLIN, myds->fd, myds, monotonic_time()); } } pthread_mutex_unlock(&myexchange.mutex_resumes); } #endif // IDLE_THREADS bool MySQL_Thread::process_data_on_data_stream(MySQL_Data_Stream *myds, unsigned int n) { if (mypolls.fds[n].revents) { #ifdef IDLE_THREADS if (myds->myds_type==MYDS_FRONTEND) { if (epoll_thread) { mypolls.remove_index_fast(n); myds->mypolls=NULL; unsigned int i; for (i=0;ilen;i++) { MySQL_Session *mysess=(MySQL_Session *)mysql_sessions->index(i); if (mysess==myds->sess) { mysess->thread=NULL; unregister_session(i); //exit_cond=true; resume_mysql_sessions->add(myds->sess); return false; } } } } #endif // IDLE_THREADS mypolls.last_recv[n]=curtime; myds->revents=mypolls.fds[n].revents; myds->sess->to_process=1; assert(myds->sess->status!=session_status___NONE); } else { // no events if (myds->wait_until && curtime > myds->wait_until) { // timeout myds->sess->to_process=1; assert(myds->sess->status!=session_status___NONE); } else { if (myds->sess->pause_until && curtime > myds->sess->pause_until) { // timeout myds->sess->to_process=1; } } } if (myds->myds_type==MYDS_BACKEND && myds->sess->status!=FAST_FORWARD) { if (mypolls.fds[n].revents) { // this part of the code fixes an important bug // if a connection in use but idle (ex: running a transaction) // get data, immediately destroy the session // // this can happen, for example, with a low wait_timeout and running transaction if (myds->sess->status==WAITING_CLIENT_DATA) { if (myds->myconn && myds->myconn->async_state_machine==ASYNC_IDLE) { proxy_warning("Detected broken idle connection on %s:%d\n", myds->myconn->parent->address, myds->myconn->parent->port); myds->destroy_MySQL_Connection_From_Pool(false); myds->sess->set_unhealthy(); return false; } } } return true; } if (myds->myds_type==MYDS_INTERNAL_GENAI) { // INTERNAL_GENAI doesn't need special idle connection handling return true; } if (mypolls.fds[n].revents) { if (mypolls.myds[n]->DSS < STATE_MARIADB_BEGIN || mypolls.myds[n]->DSS > STATE_MARIADB_END) { // only if we aren't using MariaDB Client Library int rb = 0; do { rb = myds->read_from_net(); if (rb > 0 && myds->myds_type == MYDS_FRONTEND) { status_variables.stvar[st_var_queries_frontends_bytes_recv] += rb; } myds->read_pkts(); if (rb > 0 && myds->myds_type == MYDS_BACKEND) { if (myds->sess->session_fast_forward) { if (myds->encrypted == true) { // we are in fast_forward mode and encrypted == true // PMC-10004 // we probably should use SSL_pending() and/or SSL_has_pending() to determine // if there is more data to be read, but it doesn't seem to be working. // Therefore we try to call read_from_net() again as long as there is data. // Previously we hardcoded 16KB but it seems that it can return in smaller // chunks of 4KB. // We finally removed the chunk size as it seems that any size is possible. /* int sslp = SSL_pending(myds->ssl); int sslhp = SSL_has_pending(myds->ssl); proxy_debug(PROXY_DEBUG_NET, 5, "Session=%p: in fast_forward mode and SSL read %d bytes , SSL_pending: %d bytes , SSL_has_pending: %d\n", myds->sess, rb, sslp, sslhp); */ proxy_debug(PROXY_DEBUG_NET, 5, "Session=%p, DataStream=%p , thread_session_id=%u -- in fast_forward mode and SSL read %d bytes\n", myds->sess, myds, myds->sess->thread_session_id, rb); while (rb > 0) { rb = myds->read_from_net(); if (rb > 0 && myds->myds_type == MYDS_FRONTEND) { status_variables.stvar[st_var_queries_frontends_bytes_recv] += rb; } proxy_debug(PROXY_DEBUG_NET, 5, "Session=%p, DataStream=%p -- in fast_forward mode and SSL read %d bytes\n", myds->sess, myds, rb); myds->read_pkts(); } rb = 0; // exit loop } else { // we are in fast_forward mode and encrypted == false struct pollfd _fds; nfds_t _nfds = 1; _fds.fd = mypolls.fds[n].fd; _fds.events = POLLIN; _fds.revents = 0; int _rc = poll(&_fds, _nfds, 0); if ((_rc > 0) && _fds.revents == POLLIN) { // there is more data myds->revents = _fds.revents; } else { rb = 0; // exit loop } } } else { rb = 0; // exit loop } } else { bool set_rb_zero = true; if (rb > 0 && myds->myds_type == MYDS_FRONTEND) { if (myds->encrypted == true) { if (SSL_is_init_finished(myds->ssl)) { if (myds->data_in_rbio()) { set_rb_zero = false; } } } } if (set_rb_zero) rb = 0; // exit loop } } while (rb > 0); } else { if (mypolls.fds[n].revents) { myds->myconn->handler(mypolls.fds[n].revents); } } if ( (mypolls.fds[n].events & POLLOUT) && ( (mypolls.fds[n].revents & POLLERR) || (mypolls.fds[n].revents & POLLHUP) ) ) { myds->set_net_failure(); } myds->check_data_flow(); } if (myds->active==0) { if (myds->sess->client_myds==myds) { proxy_debug(PROXY_DEBUG_NET,1, "Session=%p, DataStream=%p -- Deleting FD %d\n", myds->sess, myds, myds->fd); myds->sess->set_unhealthy(); } else { // if this is a backend with fast_forward, set unhealthy // if this is a backend without fast_forward, do not set unhealthy: it will be handled by client library if (myds->sess->session_fast_forward) { // if fast forward if (myds->myds_type==MYDS_BACKEND) { // and backend // myds->sess->set_unhealthy(); // set unhealthy // Fast Forward Grace Close Logic: // If the backend closed during fast forward mode, we defer session closure to allow // pending client output buffers to drain, preventing data loss. // Detect if backend closed during fast forward if (myds->sess->backend_closed_in_fast_forward == false) { myds->sess->backend_closed_in_fast_forward = true; //cerr << __FILE__ << ":" << __LINE__ << " grace_start_time from " << myds->sess->fast_forward_grace_start_time << " to " << curtime << endl; myds->sess->fast_forward_grace_start_time = curtime; } if (myds->sess->backend_closed_in_fast_forward) { if (myds->PSarrayIN->len == 0 && myds->sess->client_myds->PSarrayOUT->len == 0 && (myds->sess->client_myds->queueOUT.head - myds->sess->client_myds->queueOUT.tail) == 0) { // buffers empty, close myds->sess->set_unhealthy(); // set unhealthy } else if (curtime - myds->sess->fast_forward_grace_start_time > (unsigned long long)mysql_thread___fast_forward_grace_close_ms * 1000) { // timeout, close myds->sess->set_unhealthy(); // set unhealthy } } } } } } return true; } // this function was inline in MySQL_Thread::process_all_sessions() void MySQL_Thread::ProcessAllSessions_CompletedMirrorSession(unsigned int& n, MySQL_Session *sess) { unregister_session(n); n--; unsigned int l = (unsigned int)mysql_thread___mirror_max_concurrency; if (mirror_queue_mysql_sessions->len*0.3 > l) l=mirror_queue_mysql_sessions->len*0.3; if (mirror_queue_mysql_sessions_cache->len <= l) { bool to_cache=true; if (sess->mybe) { if (sess->mybe->server_myds) { to_cache=false; } } if (to_cache) { __sync_sub_and_fetch(&GloMTH->status_variables.mirror_sessions_current,1); GloMTH->status_variables.p_gauge_array[p_th_gauge::mirror_concurrency]->Decrement(); mirror_queue_mysql_sessions_cache->add(sess); } else { delete sess; } } else { delete sess; } } // this function was inline in MySQL_Thread::process_all_sessions() /** * @brief Processes a session in the maintenance loop. * * This function performs maintenance tasks for a session within the maintenance loop. It handles checks related to session * timeouts, active transactions, and server table version changes. Depending on the conditions, it may kill the session, * simulate data in failed backend connections, or update expired connections if multiplexing is enabled. * * @param sess The MySQL session to process. * @param sess_time The time elapsed since the session started, in milliseconds. * @param total_active_transactions_ Reference to the total number of active transactions across all sessions. */ void MySQL_Thread::ProcessAllSessions_MaintenanceLoop(MySQL_Session *sess, unsigned long long sess_time, unsigned int& total_active_transactions_) { unsigned int numTrx=0; total_active_transactions_ += sess->active_transactions; sess->to_process=1; /** * @brief Handles session timeout conditions and associated actions. * * This block of code evaluates whether the session has exceeded either the maximum transaction idle time * or the wait timeout duration. If either condition is met, it takes appropriate action: * * - If the session has active transactions, it checks if the maximum transaction time has been exceeded * and kills the session if necessary. * - If the session does not have active transactions, it kills the session if it has been inactive for longer * than the wait timeout duration. * * If none of the timeout conditions are met, it continues to evaluate the session's active transactions * against the maximum transaction time criteria and kills the session if necessary. * * @param sess_time The time elapsed since the session started, in milliseconds. * @param curtime The current time, in milliseconds. * @param sess The MySQL session to handle. */ unsigned long long effective_wait_timeout = std::min( static_cast(mysql_thread___wait_timeout), static_cast(sess->wait_timeout) ); if ((sess_time/1000 > static_cast(mysql_thread___max_transaction_idle_time)) || (sess_time/1000 > effective_wait_timeout)) { //numTrx = sess->NumActiveTransactions(); numTrx = sess->active_transactions; if (numTrx) { // the session has idle transactions, kill it if (sess_time/1000 > (unsigned long long)mysql_thread___max_transaction_idle_time) { sess->killed=true; if (sess->client_myds) { proxy_warning("Killing client connection %s:%d because of (possible) transaction idle for %llums\n",sess->client_myds->addr.addr,sess->client_myds->addr.port, sess_time/1000); } } } else { // the session is idle, kill it if (sess_time/1000 > effective_wait_timeout) { sess->killed=true; if (sess->client_myds) { proxy_warning("Killing client connection %s:%d because inactive for %llums\n",sess->client_myds->addr.addr,sess->client_myds->addr.port, sess_time/1000); } // Increment counter for timeout-terminated connections sess->thread->status_variables.stvar[st_var_timeout_terminated_connections]++; } } } else { if (sess->active_transactions > 0) { // here is all the logic related to max_transaction_time unsigned long long trx_started = sess->transaction_started_at; if (trx_started > 0 && curtime > trx_started) { unsigned long long trx_time = curtime - trx_started; unsigned long long trx_time_ms = trx_time/1000; if (trx_time_ms > (unsigned long long)mysql_thread___max_transaction_time) { sess->killed=true; if (sess->client_myds) { proxy_warning("Killing client connection %s:%d because of (possible) transaction running for %llums\n",sess->client_myds->addr.addr,sess->client_myds->addr.port, trx_time_ms); } } } } } /** * @brief Handles server table version change and its associated actions. * * This block of code checks if the current server table version differs from the previous version, * indicating a change in server configurations. If there is a change, it performs the following actions: * * - If the session is in fast forward mode and has offline backends, it immediately kills the client connection. * - If the session is not in fast forward mode, it searches for connections that should be terminated, * and simulates data in them by failing the backend connections instead of killing the sessions. * * This block also addresses bug fix #1085 related to handling client connections using an OFFLINE node. */ if (servers_table_version_current != servers_table_version_previous) { // bug fix for #1085 // Immediatelly kill all client connections using an OFFLINE node when session_fast_forward == true if (sess->session_fast_forward) { if (sess->HasOfflineBackends()) { sess->killed=true; proxy_warning("Killing client connection %s:%d due to 'session_fast_forward' and offline backends\n", sess->client_myds->addr.addr, sess->client_myds->addr.port); } } else { // Search for connections that should be terminated, and simulate data in them // the following 2 lines of code replace the previous 2 lines // instead of killing the sessions, fails the backend connections if (sess->SetEventInOfflineBackends()) { sess->to_process=1; } } } // Perform the maintenance for expired connections on the session /** * @brief Checks for expired connections and updates them if multiplexing is enabled. * * If multiplexing is enabled, this block of code defines lambda functions to check for expired connections * based on auto-increment delay and connection delay multiplexing criteria. It then creates a vector of these * functions and passes it to the `update_expired_conns` method of the session object to update expired connections. */ if (mysql_thread___multiplexing) { const auto auto_incr_delay_multiplex_check = [curtime=this->curtime] (MySQL_Connection* myconn) -> bool { const uint64_t multiplex_timeout_ms = mysql_thread___auto_increment_delay_multiplex_timeout_ms; const bool multiplex_delayed_enabled = multiplex_timeout_ms != 0 && myconn->auto_increment_delay_token > 0; const bool timeout_expired = multiplex_delayed_enabled && myconn->myds->wait_until != 0 && myconn->myds->wait_until < curtime; return timeout_expired; }; const auto conn_delay_multiplex = [curtime=this->curtime] (MySQL_Connection* myconn) -> bool { const bool multiplex_delayed = mysql_thread___connection_delay_multiplex_ms != 0 && myconn->multiplex_delayed == true; const bool timeout_expired = multiplex_delayed && myconn->myds->wait_until != 0 && myconn->myds->wait_until < curtime; return timeout_expired; }; const vector> expire_conn_checks { auto_incr_delay_multiplex_check, conn_delay_multiplex }; sess->update_expired_conns(expire_conn_checks); } } void MySQL_Thread::ProcessAllSessions_Healthy0(MySQL_Session *sess, unsigned int& n) { char _buf[1024]; if (sess->client_myds) { if (mysql_thread___log_unhealthy_connections) { if (sess->session_fast_forward == SESSION_FORWARD_TYPE_NONE) { proxy_warning( "Closing unhealthy client connection %s:%d\n", sess->client_myds->addr.addr, sess->client_myds->addr.port ); } else { string extra_info = ""; if (sess->backend_closed_in_fast_forward == true) { unsigned long long lapse = curtime - sess->fast_forward_grace_start_time; extra_info = "Yes , " + to_string(lapse/1000) + " ms ago"; } else { extra_info = "No"; } proxy_warning( "Closing 'fast_forward' client connection %s:%d . Backend already close: %s\n", sess->client_myds->addr.addr, sess->client_myds->addr.port, extra_info.c_str() ); } } } sprintf(_buf,"%s:%d:%s()", __FILE__, __LINE__, __func__); GloMyLogger->log_audit_entry(PROXYSQL_MYSQL_AUTH_CLOSE, sess, NULL, _buf); unregister_session(n); n--; delete sess; } /** * @brief Processes all active sessions within the MySQL thread. * * This function iterates through all active sessions within the MySQL thread and performs various actions based on the session state and conditions. * * If the session sorting flag is enabled and there are more than three sessions, it sorts the sessions. * * For each session, it performs the following tasks: * - Checks if the session is a mirror session and handles completed mirror sessions accordingly. * - Handles client connection establishment timeout if the session is in the CONNECTING_CLIENT state. * - Executes maintenance tasks on sessions if the MySQL thread is in maintenance mode. * - Handles unhealthy sessions by closing them and logging audit entries. * - Executes the session handler if it needs processing and is not paused, handling session termination if needed. * * After processing all sessions, if the MySQL thread is in maintenance mode, it updates the total active transactions. * * @param sess_sort Flag indicating whether session sorting is enabled. * @param mysql_sessions Pointer to the list of MySQL sessions. * @param mysql_thread___connect_timeout_client Timeout value for establishing client connections. * @param mysql_thread___wait_timeout Timeout value for inactive sessions. * @param mysql_thread___log_unhealthy_connections Flag indicating whether to log unhealthy connections. * @param curtime Current timestamp. * @param maintenance_loop Flag indicating whether the MySQL thread is in maintenance mode. * @param status_variables Struct containing status variables for the MySQL thread. * @param total_active_transactions_ Reference variable to store the total active transactions. * @param rc Variable to store the return code of session handlers. */ void MySQL_Thread::process_all_sessions() { unsigned int n; unsigned int total_active_transactions_=0; #ifdef IDLE_THREADS bool idle_maintenance_thread=epoll_thread; #endif // IDLE_THREADS int rc; bool sess_sort=mysql_thread___sessions_sort; #ifdef IDLE_THREADS if (idle_maintenance_thread) { sess_sort=false; } #endif // IDLE_THREADS if (sess_sort && mysql_sessions->len > 3) { ProcessAllSessions_SortingSessions(); } for (n=0; nlen; n++) { MySQL_Session *sess=(MySQL_Session *)mysql_sessions->index(n); #ifdef DEBUG if(sess==sess_stopat) { sess_stopat=sess; } #endif if (sess->mirror==true) { // this is a mirror session if (sess->status==WAITING_CLIENT_DATA) { // the mirror session has completed ProcessAllSessions_CompletedMirrorSession(n, sess); continue; } } if (sess->status == CONNECTING_CLIENT) { unsigned long long sess_time = sess->IdleTime(); if (sess_time/1000 > (unsigned long long)mysql_thread___connect_timeout_client) { proxy_warning("Closing not established client connection %s:%d after %llums\n",sess->client_myds->addr.addr,sess->client_myds->addr.port, sess_time/1000); sess->healthy = 0; if (mysql_thread___client_host_cache_size) { GloMTH->update_client_host_cache(sess->client_myds->client_addr, true); } } } if (maintenance_loop) { unsigned long long sess_time = sess->IdleTime(); #ifdef IDLE_THREADS if (idle_maintenance_thread==false) #endif // IDLE_THREADS { ProcessAllSessions_MaintenanceLoop(sess, sess_time, total_active_transactions_); } #ifdef IDLE_THREADS else { unsigned long long effective_wait_timeout = std::min( static_cast(mysql_thread___wait_timeout), static_cast(sess->wait_timeout) ); if ( (sess_time/1000 > effective_wait_timeout) ) { sess->killed=true; sess->to_process=1; proxy_warning("Killing client connection %s:%d because inactive for %llums\n", sess->client_myds->addr.addr, sess->client_myds->addr.port, sess_time/1000); // Increment counter for timeout-terminated connections sess->thread->status_variables.stvar[st_var_timeout_terminated_connections]++; } } #endif // IDLE_THREADS } else { // NOTE: we used the special value -1 to inform MySQL_Session::handler() to recompute it // removing this logic in 2.0.15 //sess->active_transactions = -1; } if (unlikely(sess->healthy==0)) { ProcessAllSessions_Healthy0(sess, n); } else { if (sess->to_process==1) { if (sess->pause_until <= curtime) { rc=sess->handler(); if (rc==-1 || sess->killed==true) { char _buf[1024]; if (sess->client_myds && sess->killed) proxy_warning("Closing killed client connection %s:%d\n",sess->client_myds->addr.addr,sess->client_myds->addr.port); sprintf(_buf,"%s:%d:%s()", __FILE__, __LINE__, __func__); GloMyLogger->log_audit_entry(PROXYSQL_MYSQL_AUTH_CLOSE, sess, NULL, _buf); unregister_session(n); n--; delete sess; } } } else { if (unlikely(sess->killed==true)) { // this is a special cause, if killed the session needs to be executed no matter if paused sess->handler(); char _buf[1024]; if (sess->client_myds) proxy_warning("Closing killed client connection %s:%d\n",sess->client_myds->addr.addr,sess->client_myds->addr.port); sprintf(_buf,"%s:%d:%s()", __FILE__, __LINE__, __func__); GloMyLogger->log_audit_entry(PROXYSQL_MYSQL_AUTH_CLOSE, sess, NULL, _buf); unregister_session(n); n--; delete sess; } } } } if (maintenance_loop) { unsigned int total_active_transactions_tmp; total_active_transactions_tmp=__sync_add_and_fetch(&status_variables.active_transactions,0); __sync_bool_compare_and_swap(&status_variables.active_transactions,total_active_transactions_tmp,total_active_transactions_); } } /** * @brief Refreshes MySQL thread variables from global MySQL thread handler. * * This method locks the global MySQL thread handler mutex and refreshes various MySQL thread variables * from the global MySQL thread handler. It retrieves values for variables such as maximum allowed packet size, * automatic SQL injection detection, firewall whitelist status, TCP keepalive usage, TCP keepalive time, * connection throttling per second to host groups, maximum transaction idle time, maximum transaction time, * threshold query length, threshold result set size, maximum query digest length, maximum query length for digests, * wait timeout, default variables, replication lag evaluation on server load, and session debugging mode. * * @note This method assumes that the global MySQL thread handler (GloMTH) and relevant variables such as * mysql_thread___default_variables have been properly initialized and are accessible. */ void MySQL_Thread::refresh_variables() { pthread_mutex_lock(&GloVars.global.ext_glomth_mutex); if (GloMTH==NULL) { return; } GloMTH->wrlock(); __thread_MySQL_Thread_Variables_version=__global_MySQL_Thread_Variables_version; REFRESH_VARIABLE_INT (max_allowed_packet); REFRESH_VARIABLE_BOOL(automatic_detect_sqli); REFRESH_VARIABLE_BOOL(firewall_whitelist_enabled); REFRESH_VARIABLE_BOOL(use_tcp_keepalive); REFRESH_VARIABLE_INT(tcp_keepalive_time); REFRESH_VARIABLE_INT(throttle_connections_per_sec_to_hostgroup); REFRESH_VARIABLE_INT(max_transaction_idle_time); REFRESH_VARIABLE_INT(max_transaction_time); REFRESH_VARIABLE_INT(threshold_query_length); REFRESH_VARIABLE_INT(threshold_resultset_size); REFRESH_VARIABLE_INT(query_digests_max_digest_length); REFRESH_VARIABLE_INT(query_digests_max_query_length); REFRESH_VARIABLE_INT(wait_timeout); REFRESH_VARIABLE_INT(throttle_max_bytes_per_second_to_client); REFRESH_VARIABLE_INT(throttle_ratio_server_to_client); REFRESH_VARIABLE_INT(max_connections); REFRESH_VARIABLE_INT(max_stmts_per_connection); REFRESH_VARIABLE_INT(max_stmts_cache); REFRESH_VARIABLE_INT(mirror_max_concurrency); REFRESH_VARIABLE_INT(mirror_max_queue_length); REFRESH_VARIABLE_INT(default_query_delay); REFRESH_VARIABLE_INT(default_query_timeout); REFRESH_VARIABLE_INT(query_processor_iterations); REFRESH_VARIABLE_INT(query_processor_regex); REFRESH_VARIABLE_INT(set_query_lock_on_hostgroup); REFRESH_VARIABLE_INT(set_parser_algorithm); REFRESH_VARIABLE_INT(reset_connection_algorithm); REFRESH_VARIABLE_INT(auto_increment_delay_multiplex); REFRESH_VARIABLE_INT(auto_increment_delay_multiplex_timeout_ms); REFRESH_VARIABLE_INT(default_max_latency_ms); REFRESH_VARIABLE_INT(long_query_time); REFRESH_VARIABLE_INT(query_cache_size_MB); REFRESH_VARIABLE_INT(query_cache_soft_ttl_pct); REFRESH_VARIABLE_INT(query_cache_handle_warnings); REFRESH_VARIABLE_INT(ping_interval_server_msec); REFRESH_VARIABLE_INT(ping_timeout_server); REFRESH_VARIABLE_INT(shun_on_failures); REFRESH_VARIABLE_INT(shun_recovery_time_sec); REFRESH_VARIABLE_INT(unshun_algorithm); REFRESH_VARIABLE_INT(query_retries_on_failure); REFRESH_VARIABLE_INT(connect_retries_on_failure); REFRESH_VARIABLE_INT(connection_delay_multiplex_ms); REFRESH_VARIABLE_INT(connection_max_age_ms); REFRESH_VARIABLE_INT(connect_timeout_client); REFRESH_VARIABLE_INT(connect_timeout_server); REFRESH_VARIABLE_INT(connect_timeout_server_max); REFRESH_VARIABLE_INT(free_connections_pct); REFRESH_VARIABLE_INT(fast_forward_grace_close_ms); #ifdef IDLE_THREADS REFRESH_VARIABLE_INT(session_idle_ms); #endif // IDLE_THREADS REFRESH_VARIABLE_INT(connect_retries_delay); REFRESH_VARIABLE_CHAR(monitor_username); REFRESH_VARIABLE_CHAR(monitor_password); REFRESH_VARIABLE_CHAR(monitor_replication_lag_use_percona_heartbeat); // SSL proxy to server REFRESH_VARIABLE_CHAR(ssl_p2s_ca); REFRESH_VARIABLE_CHAR(ssl_p2s_capath); REFRESH_VARIABLE_CHAR(ssl_p2s_cert); REFRESH_VARIABLE_CHAR(ssl_p2s_key); REFRESH_VARIABLE_CHAR(ssl_p2s_cipher); REFRESH_VARIABLE_CHAR(ssl_p2s_crl); REFRESH_VARIABLE_CHAR(ssl_p2s_crlpath); REFRESH_VARIABLE_BOOL(monitor_wait_timeout); REFRESH_VARIABLE_BOOL(monitor_writer_is_also_reader); REFRESH_VARIABLE_BOOL(monitor_enabled); REFRESH_VARIABLE_INT(monitor_history); REFRESH_VARIABLE_INT(monitor_connect_interval); REFRESH_VARIABLE_INT(monitor_connect_timeout); REFRESH_VARIABLE_INT(monitor_ping_interval); REFRESH_VARIABLE_INT(monitor_ping_max_failures); REFRESH_VARIABLE_INT(monitor_ping_timeout); REFRESH_VARIABLE_INT(monitor_aws_rds_topology_discovery_interval); REFRESH_VARIABLE_INT(monitor_read_only_interval); REFRESH_VARIABLE_INT(monitor_read_only_timeout); REFRESH_VARIABLE_INT(monitor_read_only_max_timeout_count); REFRESH_VARIABLE_BOOL(monitor_replication_lag_group_by_host); REFRESH_VARIABLE_INT(monitor_replication_lag_interval); REFRESH_VARIABLE_INT(monitor_replication_lag_timeout); REFRESH_VARIABLE_INT(monitor_replication_lag_count); REFRESH_VARIABLE_INT(monitor_groupreplication_healthcheck_interval); REFRESH_VARIABLE_INT(monitor_groupreplication_healthcheck_timeout); REFRESH_VARIABLE_INT(monitor_groupreplication_healthcheck_max_timeout_count); REFRESH_VARIABLE_INT(monitor_groupreplication_max_transactions_behind_count); REFRESH_VARIABLE_INT(monitor_groupreplication_max_transactions_behind_for_read_only); REFRESH_VARIABLE_INT(monitor_galera_healthcheck_interval); REFRESH_VARIABLE_INT(monitor_galera_healthcheck_timeout); REFRESH_VARIABLE_INT(monitor_galera_healthcheck_max_timeout_count); REFRESH_VARIABLE_INT(monitor_query_interval); REFRESH_VARIABLE_INT(monitor_query_timeout); REFRESH_VARIABLE_INT(monitor_slave_lag_when_null); REFRESH_VARIABLE_INT(monitor_threads_min); REFRESH_VARIABLE_INT(monitor_threads_max); REFRESH_VARIABLE_INT(monitor_threads_queue_maxsize); REFRESH_VARIABLE_INT(monitor_local_dns_cache_ttl); REFRESH_VARIABLE_INT(monitor_local_dns_cache_refresh_interval); REFRESH_VARIABLE_INT(monitor_local_dns_resolver_queue_maxsize); REFRESH_VARIABLE_CHAR(firewall_whitelist_errormsg); REFRESH_VARIABLE_CHAR(init_connect); REFRESH_VARIABLE_CHAR(ldap_user_variable); REFRESH_VARIABLE_CHAR(add_ldap_user_comment); REFRESH_VARIABLE_CHAR(default_session_track_gtids); for (int i=0; iget_variable_string(buf); } } REFRESH_VARIABLE_CHAR(server_version); REFRESH_VARIABLE_INT(eventslog_filesize); REFRESH_VARIABLE_INT(eventslog_table_memory_size); REFRESH_VARIABLE_INT(eventslog_buffer_history_size); { int elmhs = mysql_thread___eventslog_buffer_history_size; if (GloMyLogger->MyLogCB->getBufferSize() != elmhs) { GloMyLogger->MyLogCB->setBufferSize(elmhs); } } REFRESH_VARIABLE_INT(eventslog_buffer_max_query_length); REFRESH_VARIABLE_INT(eventslog_default_log); REFRESH_VARIABLE_INT(eventslog_format); REFRESH_VARIABLE_INT(eventslog_stmt_parameters); REFRESH_VARIABLE_CHAR(eventslog_filename); REFRESH_VARIABLE_INT(auditlog_filesize); REFRESH_VARIABLE_CHAR(auditlog_filename); GloMyLogger->events_set_base_filename(); // both filename and filesize are set here GloMyLogger->audit_set_base_filename(); // both filename and filesize are set here REFRESH_VARIABLE_CHAR(default_schema); REFRESH_VARIABLE_CHAR(keep_multiplexing_variables); REFRESH_VARIABLE_CHAR(proxy_protocol_networks); REFRESH_VARIABLE_CHAR(default_authentication_plugin); mysql_thread___default_authentication_plugin_int = GloMTH->variables.default_authentication_plugin_int; mysql_thread___server_capabilities=GloMTH->get_variable_uint16((char *)"server_capabilities"); REFRESH_VARIABLE_INT(handle_unknown_charset); REFRESH_VARIABLE_INT(poll_timeout); REFRESH_VARIABLE_INT(poll_timeout_on_failure); REFRESH_VARIABLE_BOOL(have_compress); REFRESH_VARIABLE_INT(protocol_compression_level); REFRESH_VARIABLE_BOOL(have_ssl); REFRESH_VARIABLE_BOOL(multiplexing); REFRESH_VARIABLE_BOOL(log_unhealthy_connections); REFRESH_VARIABLE_BOOL(connection_warming); REFRESH_VARIABLE_BOOL(enforce_autocommit_on_reads); REFRESH_VARIABLE_BOOL(autocommit_false_not_reusable); REFRESH_VARIABLE_BOOL(autocommit_false_is_transaction); REFRESH_VARIABLE_BOOL(verbose_query_error); REFRESH_VARIABLE_BOOL(commands_stats); REFRESH_VARIABLE_BOOL(query_digests); REFRESH_VARIABLE_BOOL(query_digests_lowercase); REFRESH_VARIABLE_BOOL(query_digests_replace_null); REFRESH_VARIABLE_BOOL(query_digests_no_digits); REFRESH_VARIABLE_BOOL(query_digests_normalize_digest_text); REFRESH_VARIABLE_BOOL(query_digests_track_hostname); REFRESH_VARIABLE_INT(query_digests_grouping_limit); REFRESH_VARIABLE_INT(query_digests_groups_grouping_limit); REFRESH_VARIABLE_BOOL(query_digests_keep_comment); REFRESH_VARIABLE_BOOL(parse_failure_logs_digest); variables.min_num_servers_lantency_awareness=GloMTH->get_variable_int((char *)"min_num_servers_lantency_awareness"); variables.aurora_max_lag_ms_only_read_from_replicas=GloMTH->get_variable_int((char *)"aurora_max_lag_ms_only_read_from_replicas"); variables.stats_time_backend_query=(bool)GloMTH->get_variable_int((char *)"stats_time_backend_query"); variables.stats_time_query_processor=(bool)GloMTH->get_variable_int((char *)"stats_time_query_processor"); variables.query_cache_stores_empty_result=(bool)GloMTH->get_variable_int((char *)"query_cache_stores_empty_result"); REFRESH_VARIABLE_INT(hostgroup_manager_verbose); REFRESH_VARIABLE_BOOL(kill_backend_connection_when_disconnect); REFRESH_VARIABLE_BOOL(client_session_track_gtid); REFRESH_VARIABLE_BOOL(sessions_sort); REFRESH_VARIABLE_BOOL(servers_stats); REFRESH_VARIABLE_BOOL(default_reconnect); REFRESH_VARIABLE_BOOL(enable_client_deprecate_eof); REFRESH_VARIABLE_BOOL(enable_server_deprecate_eof); REFRESH_VARIABLE_BOOL(enable_load_data_local_infile); REFRESH_VARIABLE_BOOL(log_mysql_warnings_enabled); REFRESH_VARIABLE_INT(client_host_cache_size); REFRESH_VARIABLE_INT(client_host_error_counts); REFRESH_VARIABLE_INT(handle_warnings); REFRESH_VARIABLE_INT(evaluate_replication_lag_on_servers_load); REFRESH_VARIABLE_BOOL(ignore_min_gtid_annotations); #ifdef DEBUG REFRESH_VARIABLE_BOOL(session_debug); #endif /* DEBUG */ GloMTH->wrunlock(); pthread_mutex_unlock(&GloVars.global.ext_glomth_mutex); } MySQL_Thread::MySQL_Thread() { pthread_mutex_init(&thread_mutex,NULL); my_idle_conns=NULL; cached_connections=NULL; mysql_sessions=NULL; mirror_queue_mysql_sessions=NULL; mirror_queue_mysql_sessions_cache=NULL; #ifdef IDLE_THREADS efd=-1; epoll_thread=false; mysess_idx=0; idle_mysql_sessions=NULL; resume_mysql_sessions=NULL; myexchange.idle_mysql_sessions=NULL; myexchange.resume_mysql_sessions=NULL; #endif // IDLE_THREADS processing_idles=false; last_processing_idles=0; __thread_MySQL_Thread_Variables_version=0; mysql_thread___server_version=NULL; mysql_thread___init_connect=NULL; mysql_thread___ldap_user_variable=NULL; mysql_thread___add_ldap_user_comment=NULL; mysql_thread___eventslog_filename=NULL; mysql_thread___auditlog_filename=NULL; // SSL proxy to server mysql_thread___ssl_p2s_ca=NULL; mysql_thread___ssl_p2s_capath=NULL; mysql_thread___ssl_p2s_cert=NULL; mysql_thread___ssl_p2s_key=NULL; mysql_thread___ssl_p2s_cipher=NULL; mysql_thread___ssl_p2s_crl=NULL; mysql_thread___ssl_p2s_crlpath=NULL; mysql_thread___protocol_compression_level=3; last_maintenance_time=0; last_move_to_idle_thread_time=0; maintenance_loop=true; retrieve_gtids_required = false; servers_table_version_previous=0; servers_table_version_current=0; status_variables.active_transactions=0; for (unsigned int i = 0; i < MY_st_var_END ; i++) { status_variables.stvar[i] = 0; } match_regexes=NULL; variables.min_num_servers_lantency_awareness = 1000; variables.aurora_max_lag_ms_only_read_from_replicas = 2; variables.stats_time_backend_query=false; variables.stats_time_query_processor=false; variables.query_cache_stores_empty_result=true; for (int i=0; ithread=this; _sess->connections_handler=true; assert(_new); mysql_sessions->add(_sess); } /** * @brief Unregisters a session from the connection handler. * * This method unregisters a session from the connection handler of the MySQL thread. It removes the session * from the MySQL sessions list based on the provided index. * * @param idx Index of the session in the MySQL sessions list to unregister. * @param _new Boolean flag indicating whether the session is new. * * @note This method assumes that the MySQL sessions list (mysql_sessions) has been properly initialized and is accessible. */ void MySQL_Thread::unregister_session_connection_handler(int idx, bool _new) { assert(_new); mysql_sessions->remove_index_fast(idx); } void MySQL_Thread::listener_handle_new_connection(MySQL_Data_Stream *myds, unsigned int n) { int c; union { struct sockaddr_in in; struct sockaddr_in6 in6; } custom_sockaddr; struct sockaddr *addr=(struct sockaddr *)malloc(sizeof(custom_sockaddr)); socklen_t addrlen=sizeof(custom_sockaddr); memset(addr, 0, sizeof(custom_sockaddr)); if (GloMTH->num_threads > 1) { // there are more than 1 thread . We pause for a little bit to avoid all connections to be handled by the same thread #ifdef SO_REUSEPORT if (GloVars.global.reuseport==false) { // only if reuseport is not enabled //usleep(10+rand()%50); } #else //usleep(10+rand()%50); #endif /* SO_REUSEPORT */ } c=accept(myds->fd, addr, &addrlen); if (c>-1) { // accept() succeeded if (mysql_thread___client_host_cache_size) { MySQL_Client_Host_Cache_Entry client_host_entry = GloMTH->find_client_host_cache(addr); if ( client_host_entry.updated_at != 0 && client_host_entry.error_count >= static_cast(mysql_thread___client_host_error_counts) ) { std::string client_addr = get_client_addr(addr); proxy_error( "Closing connection because client '%s' reached 'mysql-client_host_error_counts': %d\n", client_addr.c_str(), mysql_thread___client_host_error_counts ); close(c); free(addr); status_variables.stvar[st_var_client_host_error_killed_connections] += 1; return; } } // create a new client connection mypolls.fds[n].revents=0; MySQL_Session *sess=create_new_session_and_client_data_stream(c); __sync_add_and_fetch(&MyHGM->status.client_connections_created,1); if (__sync_add_and_fetch(&MyHGM->status.client_connections,1) > mysql_thread___max_connections) { sess->max_connections_reached=true; } sess->client_myds->client_addrlen=addrlen; sess->client_myds->client_addr=addr; switch (sess->client_myds->client_addr->sa_family) { case AF_INET: { struct sockaddr_in *ipv4 = (struct sockaddr_in *)sess->client_myds->client_addr; char buf[INET_ADDRSTRLEN]; inet_ntop(sess->client_myds->client_addr->sa_family, &ipv4->sin_addr, buf, INET_ADDRSTRLEN); sess->client_myds->addr.addr = strdup(buf); sess->client_myds->addr.port = htons(ipv4->sin_port); break; } case AF_INET6: { struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)sess->client_myds->client_addr; char buf[INET6_ADDRSTRLEN]; inet_ntop(sess->client_myds->client_addr->sa_family, &ipv6->sin6_addr, buf, INET6_ADDRSTRLEN); sess->client_myds->addr.addr = strdup(buf); sess->client_myds->addr.port = htons(ipv6->sin6_port); break; } default: sess->client_myds->addr.addr = strdup("localhost"); break; } iface_info *ifi=NULL; ifi=GloMTH->MLM_find_iface_from_fd(myds->fd); // here we try to get the info about the proxy bind address if (ifi) { sess->client_myds->proxy_addr.addr=strdup(ifi->address); sess->client_myds->proxy_addr.port=ifi->port; } sess->client_myds->myprot.generate_pkt_initial_handshake(true,NULL,NULL, &sess->thread_session_id, true); ioctl_FIONBIO(sess->client_myds->fd, 1); /** * @brief Add client socket to poll set with both read and write monitoring * * This usage pattern registers a client socket with both POLLIN and POLLOUT events, * which is typically done during initial client setup when we need to send the * initial handshake packet and also be ready to receive client responses. * * Usage pattern: mypolls.add(POLLIN|POLLOUT, sess->client_myds->fd, sess->client_myds, curtime) * - POLLIN|POLLOUT: Monitor both read and write events * - sess->client_myds->fd: Client socket file descriptor * - sess->client_myds: MySQL_Data_Stream instance for the client * - curtime: Current timestamp for tracking * * Called during: Initial client connection setup after handshake packet generation * Purpose: Enables bidirectional communication with the client during setup phase * * Note: This ensures we can send the initial handshake immediately and also handle * any client packets that might arrive before the handshake is complete. */ mypolls.add(POLLIN|POLLOUT, sess->client_myds->fd, sess->client_myds, curtime); proxy_debug(PROXY_DEBUG_NET,1,"Session=%p -- Adding client FD %d\n", sess, sess->client_myds->fd); // we now enforce sending the 'initial handshake packet' as soon as it's generated. This // is done to prevent situations in which a client sends a packet *before* receiving // this 'initial handshake', leading to invalid state in dataflow, since it will be // data in both ends of the datastream. For more details see #3342. sess->writeout(); } else { free(addr); // if we arrive here, accept() failed // because multiple threads try to handle the same incoming connection, this is OK } } SQLite3_result * MySQL_Threads_Handler::SQL3_GlobalStatus(bool _memory) { const int colnum=2; char buf[256]; char **pta=(char **)malloc(sizeof(char *)*colnum); if (_memory == true) { Get_Memory_Stats(); } proxy_debug(PROXY_DEBUG_MYSQL_CONNECTION, 4, "Dumping MySQL Global Status\n"); SQLite3_result *result=new SQLite3_result(colnum); result->add_column_definition(SQLITE_TEXT,"Variable_Name"); result->add_column_definition(SQLITE_TEXT,"Variable_Value"); // NOTE: as there is no string copy, we do NOT free pta[0] and pta[1] { // uptime unsigned long long t1=monotonic_time(); pta[0] = (char *)"ProxySQL_Uptime"; sprintf(buf,"%llu",(t1-GloVars.global.start_time)/1000/1000); pta[1]=buf; result->add_row(pta); } { // Active Transactions pta[0]=(char *)"Active_Transactions"; sprintf(buf,"%u",get_active_transations()); pta[1]=buf; result->add_row(pta); } { // Connections created pta[0]=(char *)"Client_Connections_aborted"; sprintf(buf,"%lu",MyHGM->status.client_connections_aborted); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Client_Connections_connected"; sprintf(buf,"%d",MyHGM->status.client_connections); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Client_Connections_connected_prim_pass"; sprintf(buf,"%d",MyHGM->status.client_connections_prim_pass); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Client_Connections_connected_addl_pass"; sprintf(buf,"%d",MyHGM->status.client_connections_addl_pass); pta[1]=buf; result->add_row(pta); } { // Connections created pta[0]=(char *)"Client_Connections_created"; sprintf(buf,"%lu",MyHGM->status.client_connections_created); pta[1]=buf; result->add_row(pta); } { // Connections created using cached 'clear_text_passwords' pta[0]=(char *)"Client_Connections_sha2cached"; sprintf(buf,"%lu",MyHGM->status.client_connections_sha2cached); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Server_Connections_aborted"; sprintf(buf,"%lu",MyHGM->status.server_connections_aborted); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Server_Connections_connected"; sprintf(buf,"%lu",MyHGM->status.server_connections_connected); pta[1]=buf; result->add_row(pta); } { // Connections pta[0]=(char *)"Server_Connections_created"; sprintf(buf,"%lu",MyHGM->status.server_connections_created); pta[1]=buf; result->add_row(pta); } { // Connections delayed pta[0]=(char *)"Server_Connections_delayed"; sprintf(buf,"%lu",MyHGM->status.server_connections_delayed); pta[1]=buf; result->add_row(pta); } #ifdef IDLE_THREADS { // Connections non idle pta[0]=(char *)"Client_Connections_non_idle"; sprintf(buf,"%u",get_non_idle_client_connections()); pta[1]=buf; result->add_row(pta); } #endif // IDLE_THREADS { // MySQL Backend buffers bytes pta[0]=(char *)"mysql_backend_buffers_bytes"; sprintf(buf,"%llu",get_mysql_backend_buffers_bytes()); pta[1]=buf; result->add_row(pta); } { // MySQL Frontend buffers bytes pta[0]=(char *)"mysql_frontend_buffers_bytes"; sprintf(buf,"%llu",get_mysql_frontend_buffers_bytes()); pta[1]=buf; result->add_row(pta); } { // MySQL Frontend buffers bytes pta[0]=(char *)"mysql_session_internal_bytes"; sprintf(buf,"%llu",get_mysql_session_internal_bytes()); pta[1]=buf; result->add_row(pta); } { // Queries autocommit pta[0]=(char *)"Com_autocommit"; sprintf(buf,"%llu",MyHGM->status.autocommit_cnt); pta[1]=buf; result->add_row(pta); } { // Queries filtered autocommit pta[0]=(char *)"Com_autocommit_filtered"; sprintf(buf,"%llu",MyHGM->status.autocommit_cnt_filtered); pta[1]=buf; result->add_row(pta); } { // Queries commit pta[0]=(char *)"Com_commit"; sprintf(buf,"%llu",MyHGM->status.commit_cnt); pta[1]=buf; result->add_row(pta); } { // Queries filtered commit pta[0]=(char *)"Com_commit_filtered"; sprintf(buf,"%llu",MyHGM->status.commit_cnt_filtered); pta[1]=buf; result->add_row(pta); } { // Queries rollback pta[0]=(char *)"Com_rollback"; sprintf(buf,"%llu",MyHGM->status.rollback_cnt); pta[1]=buf; result->add_row(pta); } { // Queries filtered rollback pta[0]=(char *)"Com_rollback_filtered"; sprintf(buf,"%llu",MyHGM->status.rollback_cnt_filtered); pta[1]=buf; result->add_row(pta); } { // Queries backend CHANGE_USER pta[0]=(char *)"Com_backend_change_user"; sprintf(buf,"%llu",MyHGM->status.backend_change_user); pta[1]=buf; result->add_row(pta); } { // Queries backend INIT DB pta[0]=(char *)"Com_backend_init_db"; sprintf(buf,"%llu",MyHGM->status.backend_init_db); pta[1]=buf; result->add_row(pta); } { // Queries backend SET NAMES pta[0]=(char *)"Com_backend_set_names"; sprintf(buf,"%llu",MyHGM->status.backend_set_names); pta[1]=buf; result->add_row(pta); } { // Queries frontend INIT DB pta[0]=(char *)"Com_frontend_init_db"; sprintf(buf,"%llu",MyHGM->status.frontend_init_db); pta[1]=buf; result->add_row(pta); } { // Queries frontend SET NAMES pta[0]=(char *)"Com_frontend_set_names"; sprintf(buf,"%llu",MyHGM->status.frontend_set_names); pta[1]=buf; result->add_row(pta); } { // Queries frontend USE DB pta[0]=(char *)"Com_frontend_use_db"; sprintf(buf,"%llu",MyHGM->status.frontend_use_db); pta[1]=buf; result->add_row(pta); } for (unsigned int i=0; iadd_row(pta); } } } // Gauge variables for (unsigned int i=0; iadd_row(pta); } } } { // Mirror current concurrency pta[0]=(char *)"Mirror_concurrency"; sprintf(buf,"%u",status_variables.mirror_sessions_current); pta[1]=buf; result->add_row(pta); } { // Mirror queue length pta[0]=(char *)"Mirror_queue_length"; sprintf(buf,"%llu",get_total_mirror_queue()); pta[1]=buf; result->add_row(pta); } { // Queries that are SELECT for update or equivalent pta[0]=(char *)"Selects_for_update__autocommit0"; sprintf(buf,"%llu",MyHGM->status.select_for_update_or_equivalent); pta[1]=buf; result->add_row(pta); } { // Servers_table_version pta[0]=(char *)"Servers_table_version"; sprintf(buf,"%u",MyHGM->get_servers_table_version()); pta[1]=buf; result->add_row(pta); } { // MySQL Threads workers pta[0]=(char *)"MySQL_Thread_Workers"; sprintf(buf,"%d",num_threads); pta[1]=buf; result->add_row(pta); } { // Access_Denied_Wrong_Password pta[0]=(char *)"Access_Denied_Wrong_Password"; sprintf(buf,"%llu",MyHGM->status.access_denied_wrong_password); pta[1]=buf; result->add_row(pta); } { // Access_Denied_Max_Connections pta[0]=(char *)"Access_Denied_Max_Connections"; sprintf(buf,"%llu",MyHGM->status.access_denied_max_connections); pta[1]=buf; result->add_row(pta); } { // Access_Denied_Max_User_Connections pta[0]=(char *)"Access_Denied_Max_User_Connections"; sprintf(buf,"%llu",MyHGM->status.access_denied_max_user_connections); pta[1]=buf; result->add_row(pta); } if (GloMyMon) { { // MySQL Monitor workers pta[0]=(char *)"MySQL_Monitor_Workers"; sprintf(buf,"%d",( variables.monitor_enabled ? GloMyMon->num_threads : 0)); pta[1]=buf; result->add_row(pta); } { // MySQL Monitor workers pta[0]=(char *)"MySQL_Monitor_Workers_Aux"; sprintf(buf,"%d",( variables.monitor_enabled ? GloMyMon->aux_threads : 0)); pta[1]=buf; result->add_row(pta); } { // MySQL Monitor workers pta[0]=(char *)"MySQL_Monitor_Workers_Started"; sprintf(buf,"%d",( variables.monitor_enabled ? GloMyMon->started_threads : 0)); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_connect_check_OK"; sprintf(buf,"%llu", GloMyMon->connect_check_OK); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_connect_check_ERR"; sprintf(buf,"%llu", GloMyMon->connect_check_ERR); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_ping_check_OK"; sprintf(buf,"%llu", GloMyMon->ping_check_OK); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_ping_check_ERR"; sprintf(buf,"%llu", GloMyMon->ping_check_ERR); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_read_only_check_OK"; sprintf(buf,"%llu", GloMyMon->read_only_check_OK); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_read_only_check_ERR"; sprintf(buf,"%llu", GloMyMon->read_only_check_ERR); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_replication_lag_check_OK"; sprintf(buf,"%llu", GloMyMon->replication_lag_check_OK); pta[1]=buf; result->add_row(pta); } { pta[0]=(char *)"MySQL_Monitor_replication_lag_check_ERR"; sprintf(buf,"%llu", GloMyMon->replication_lag_check_ERR); pta[1]=buf; result->add_row(pta); } { pta[0] = (char*)"MySQL_Monitor_dns_cache_queried"; sprintf(buf, "%llu", GloMyMon->dns_cache_queried); pta[1] = buf; result->add_row(pta); } { pta[0] = (char*)"MySQL_Monitor_dns_cache_lookup_success"; sprintf(buf, "%llu", GloMyMon->dns_cache_lookup_success); pta[1] = buf; result->add_row(pta); } { pta[0] = (char*)"MySQL_Monitor_dns_cache_record_updated"; sprintf(buf, "%llu", GloMyMon->dns_cache_record_updated); pta[1] = buf; result->add_row(pta); } } free(pta); return result; } /** * @brief Retrieves memory statistics from all MySQL threads. * * This function iterates through all MySQL threads, including both active worker threads and idle maintenance threads if enabled, * to retrieve memory statistics using the `Get_Memory_Stats()` function for each thread. * * It first determines the total number of threads to iterate based on the number of active worker threads (`num_threads`) and * whether idle threads are enabled (`GloVars.global.idle_threads`). If idle threads are enabled, it doubles the count to include * both active worker threads and idle maintenance threads. * * For each thread, it acquires a lock on the thread mutex to safely retrieve memory statistics and then releases the lock. * If any thread is found to be NULL during iteration, indicating that it is not ready, the function exits early. * * @note This function assumes that the `Get_Memory_Stats()` function is implemented for MySQL_Thread objects to retrieve memory statistics. * * @param num_threads Number of active worker threads. * @param mysql_threads Array of active worker threads. * @param mysql_threads_idles Array of idle maintenance threads. * @param GloVars Struct containing global variables. */ void MySQL_Threads_Handler::Get_Memory_Stats() { unsigned int i; unsigned int j; j=num_threads; #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { j+=num_threads; } #endif // IDLE_THREADS for (i=0;ithread_mutex); thr->Get_Memory_Stats(); pthread_mutex_unlock(&thr->thread_mutex); } } SQLite3_result * MySQL_Threads_Handler::SQL3_Processlist(processlist_config_t args) { const int colnum=16; char port[NI_MAXSERV]; proxy_debug(PROXY_DEBUG_MYSQL_CONNECTION, 4, "Dumping MySQL Processlist\n"); SQLite3_result *result=new SQLite3_result(colnum); result->add_column_definition(SQLITE_TEXT,"ThreadID"); result->add_column_definition(SQLITE_TEXT,"SessionID"); result->add_column_definition(SQLITE_TEXT,"user"); result->add_column_definition(SQLITE_TEXT,"db"); result->add_column_definition(SQLITE_TEXT,"cli_host"); result->add_column_definition(SQLITE_TEXT,"cli_port"); result->add_column_definition(SQLITE_TEXT,"hostgroup"); result->add_column_definition(SQLITE_TEXT,"l_srv_host"); result->add_column_definition(SQLITE_TEXT,"l_srv_port"); result->add_column_definition(SQLITE_TEXT,"srv_host"); result->add_column_definition(SQLITE_TEXT,"srv_port"); result->add_column_definition(SQLITE_TEXT,"command"); result->add_column_definition(SQLITE_TEXT,"time_ms"); result->add_column_definition(SQLITE_TEXT,"info"); result->add_column_definition(SQLITE_TEXT,"status_flags"); result->add_column_definition(SQLITE_TEXT,"extended_info"); unsigned int i; unsigned int i2; // signal_all_threads(1); i2=num_threads; #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { i2+=num_threads; } #endif // IDLE_THREADS for (i=0;ithread_mutex); unsigned int j; for (j=0; jmysql_sessions->len; j++) { MySQL_Session *sess=(MySQL_Session *)thr->mysql_sessions->pdata[j]; if (sess->client_myds) { char buf[1024]; char **pta=(char **)malloc(sizeof(char *)*colnum); sprintf(buf,"%d", i); pta[0]=strdup(buf); sprintf(buf,"%u", sess->thread_session_id); pta[1]=strdup(buf); MySQL_Connection_userinfo *ui=sess->client_myds->myconn->userinfo; pta[2]=NULL; pta[3]=NULL; if (ui) { if (ui->username) { pta[2]=strdup(ui->username); } else { pta[2]=strdup("unauthenticated user"); } if (ui->schemaname) { pta[3]=strdup(ui->schemaname); } } if (sess->mirror==false) { switch (sess->client_myds->client_addr->sa_family) { case AF_INET: if (sess->client_myds->addr.addr != NULL) { pta[4] = strdup(sess->client_myds->addr.addr); sprintf(port, "%d", sess->client_myds->addr.port); pta[5] = strdup(port); } else { struct sockaddr_in *ipv4 = (struct sockaddr_in *)sess->client_myds->client_addr; inet_ntop(sess->client_myds->client_addr->sa_family, &ipv4->sin_addr, buf, INET_ADDRSTRLEN); pta[4] = strdup(buf); sprintf(port, "%d", ntohs(ipv4->sin_port)); pta[5] = strdup(port); } break; case AF_INET6: if (sess->client_myds->addr.addr != NULL) { pta[4] = strdup(sess->client_myds->addr.addr); sprintf(port, "%d", sess->client_myds->addr.port); pta[5] = strdup(port); } else { struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)sess->client_myds->client_addr; inet_ntop(sess->client_myds->client_addr->sa_family, &ipv6->sin6_addr, buf, INET6_ADDRSTRLEN); pta[4] = strdup(buf); sprintf(port, "%d", ntohs(ipv6->sin6_port)); pta[5] = strdup(port); } break; default: pta[4] = strdup("localhost"); pta[5] = NULL; break; } } else { pta[4] = strdup("mirror_internal"); pta[5] = NULL; } sprintf(buf,"%d", sess->current_hostgroup); pta[6]=strdup(buf); if (sess->mybe && sess->mybe->server_myds && sess->mybe->server_myds->myconn) { MySQL_Connection *mc=sess->mybe->server_myds->myconn; struct sockaddr addr; socklen_t addr_len=sizeof(struct sockaddr); memset(&addr,0,addr_len); int rc; rc=getsockname(mc->fd, &addr, &addr_len); if (rc==0) { switch (addr.sa_family) { case AF_INET: { struct sockaddr_in *ipv4 = (struct sockaddr_in *)&addr; inet_ntop(addr.sa_family, &ipv4->sin_addr, buf, INET_ADDRSTRLEN); pta[7] = strdup(buf); sprintf(port, "%d", ntohs(ipv4->sin_port)); pta[8] = strdup(port); break; } case AF_INET6: { struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)&addr; inet_ntop(addr.sa_family, &ipv6->sin6_addr, buf, INET6_ADDRSTRLEN); pta[7] = strdup(buf); sprintf(port, "%d", ntohs(ipv6->sin6_port)); pta[8] = strdup(port); break; } default: pta[7] = strdup("localhost"); pta[8] = NULL; break; } } else { pta[7]=NULL; pta[8]=NULL; } sprintf(buf,"%s", mc->parent->address); pta[9]=strdup(buf); sprintf(buf,"%d", mc->parent->port); pta[10]=strdup(buf); pta[13] = sess->get_current_query(args.max_query_length); sprintf(buf,"%d", mc->status_flags); pta[14]=strdup(buf); } else { pta[7]=NULL; pta[8]=NULL; pta[9]=NULL; pta[10]=NULL; pta[13]=NULL; pta[14]=NULL; } switch (sess->status) { case CONNECTING_SERVER: pta[11]=strdup("Connect"); break; case PROCESSING_QUERY: if (sess->pause_until > sess->thread->curtime) { pta[11]=strdup("Delay"); } else { pta[11]=strdup("Query"); } break; case WAITING_CLIENT_DATA: pta[11]=strdup("Sleep"); break; case CHANGING_USER_SERVER: pta[11]=strdup("Changing user server"); break; case CHANGING_USER_CLIENT: pta[11]=strdup("Change user client"); break; case RESETTING_CONNECTION: pta[11]=strdup("Resetting connection"); break; case CHANGING_SCHEMA: pta[11]=strdup("InitDB"); break; case PROCESSING_STMT_EXECUTE: pta[11]=strdup("Execute"); break; case PROCESSING_STMT_PREPARE: pta[11]=strdup("Prepare"); break; case CONNECTING_CLIENT: pta[11]=strdup("Connecting client"); break; case PINGING_SERVER: pta[11]=strdup("Pinging server"); break; case WAITING_SERVER_DATA: pta[11]=strdup("Waiting server data"); break; case CHANGING_CHARSET: pta[11]=strdup("Changing charset"); break; case CHANGING_AUTOCOMMIT: pta[11]=strdup("Changing autocommit"); break; case SETTING_INIT_CONNECT: pta[11]=strdup("Setting init connect"); break; /* case SETTING_SQL_LOG_BIN: pta[11]=strdup("Set log bin"); break; case SETTING_SQL_MODE: pta[11]=strdup("Set SQL mode"); break; case SETTING_TIME_ZONE: pta[11]=strdup("Set TZ"); break; */ case SETTING_VARIABLE: { int idx = sess->changing_variable_idx; if (idx < SQL_NAME_LAST_HIGH_WM) { char buf[128]; sprintf(buf, "Setting variable %s", mysql_tracked_variables[idx].set_variable_name); pta[11]=strdup(buf); } else { pta[11]=strdup("Setting variable"); } } break; case FAST_FORWARD: pta[11]=strdup("Fast forward"); break; case session_status___NONE: pta[11]=strdup("None"); break; default: sprintf(buf,"%d", sess->status); pta[11]=strdup(buf); break; } if (sess->mirror==false) { int idx=sess->client_myds->poll_fds_idx; unsigned long long last_sent=sess->thread->mypolls.last_sent[idx]; unsigned long long last_recv=sess->thread->mypolls.last_recv[idx]; unsigned long long last_time=(last_sent > last_recv ? last_sent : last_recv); if (last_time>sess->thread->curtime) { last_time=sess->thread->curtime; } sprintf(buf,"%llu", (sess->thread->curtime - last_time)/1000 ); } else { // for mirror session we only consider the start time sprintf(buf,"%llu", (sess->thread->curtime - sess->start_time)/1000 ); } pta[12]=strdup(buf); pta[15]=NULL; if (args.show_extended) { json j; sess->generate_proxysql_internal_session_json(j); if (args.show_extended == 2) { std::string s = j.dump(4, ' ', false, json::error_handler_t::replace); pta[15] = strdup(s.c_str()); } else { std::string s = j.dump(-1, ' ', false, json::error_handler_t::replace); pta[15] = strdup(s.c_str()); } } result->add_row(pta); unsigned int k; for (k=0; kthread_mutex); } return result; } void MySQL_Threads_Handler::signal_all_threads(unsigned char _c) { unsigned int i; unsigned char c=_c; if (mysql_threads==0) return; for (i=0;ipipefd[1]; if (write(fd,&c,1)==-1) { proxy_error("Error during write in signal_all_threads()\n"); } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;ipipefd[1]; if (write(fd,&c,1)==-1) { proxy_error("Error during write in signal_all_threads()\n"); } } #endif // IDLE_THREADS } void MySQL_Threads_Handler::kill_connection_or_query(uint32_t _thread_session_id, bool query, char *username) { unsigned int i; for (i=0;iid = _thread_session_id; tu->username = strdup(username); pthread_mutex_lock(&thr->kq.m); if (query) { thr->kq.query_ids.push_back(tu); } else { thr->kq.conn_ids.push_back(tu); } pthread_mutex_unlock(&thr->kq.m); } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { for (i=0;iid = _thread_session_id; tu->username = strdup(username); pthread_mutex_lock(&thr->kq.m); if (query) { thr->kq.query_ids.push_back(tu); } else { thr->kq.conn_ids.push_back(tu); } pthread_mutex_unlock(&thr->kq.m); } } #endif signal_all_threads(0); } bool MySQL_Threads_Handler::kill_session(uint32_t _thread_session_id) { bool ret=false; unsigned int i; signal_all_threads(1); for (i=0;ithread_mutex); } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;ithread_mutex); } #endif // IDLE_THREADS for (i=0;imysql_sessions->len; j++) { MySQL_Session *sess=(MySQL_Session *)thr->mysql_sessions->pdata[j]; if (sess->thread_session_id==_thread_session_id) { sess->killed=true; ret=true; goto __exit_kill_session; } } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;imysql_sessions->len; j++) { MySQL_Session *sess=(MySQL_Session *)thr->mysql_sessions->pdata[j]; if (sess->thread_session_id==_thread_session_id) { sess->killed=true; ret=true; goto __exit_kill_session; } } } #endif // IDLE_THREADS __exit_kill_session: for (i=0;ithread_mutex); } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;ithread_mutex); } #endif // IDLE_THREADS return ret; } unsigned long long MySQL_Threads_Handler::get_total_mirror_queue() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;imirror_queue_mysql_sessions->len; // this is a dirty read } } this->status_variables.p_gauge_array[p_th_gauge::mirror_queue_lengths]->Set(q); return q; } unsigned long long MySQL_Threads_Handler::get_status_variable( enum MySQL_Thread_status_variable v_idx, p_th_counter::metric m_idx, unsigned long long conv ) { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.stvar[v_idx],0); } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;istatus_variables.stvar[v_idx],0); } } #endif // IDLE_THREADS if (m_idx != p_th_counter::__size) { const auto& cur_val = status_variables.p_counter_array[m_idx]->Value(); double final_val = 0; if (conv != 0) { final_val = (q - (cur_val * conv)) / conv; } else { final_val = q - cur_val; } status_variables.p_counter_array[m_idx]->Increment(final_val); } return q; } unsigned long long MySQL_Threads_Handler::get_status_variable( enum MySQL_Thread_status_variable v_idx, p_th_gauge::metric m_idx, unsigned long long conv ) { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.stvar[v_idx],0); } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;istatus_variables.stvar[v_idx],0); } } #endif // IDLE_THREADS if (m_idx != p_th_gauge::__size) { double final_val = 0; if (conv != 0) { final_val = q / static_cast(conv); } else { final_val = q; } status_variables.p_gauge_array[m_idx]->Set(final_val); } return q; } unsigned int MySQL_Threads_Handler::get_active_transations() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.active_transactions,0); } } this->status_variables.p_gauge_array[p_th_gauge::active_transactions]->Set(q); return q; } #ifdef IDLE_THREADS unsigned int MySQL_Threads_Handler::get_non_idle_client_connections() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;imysql_sessions->len,0); } } this->status_variables.p_gauge_array[p_th_gauge::client_connections_non_idle]->Set(q); return q; } #endif // IDLE_THREADS unsigned long long MySQL_Threads_Handler::get_mysql_backend_buffers_bytes() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.stvar[st_var_mysql_backend_buffers_bytes],0); } } const auto& cur_val = this->status_variables.p_counter_array[p_th_gauge::mysql_backend_buffers_bytes]->Value(); this->status_variables.p_counter_array[p_th_gauge::mysql_backend_buffers_bytes]->Increment(q - cur_val); return q; } unsigned long long MySQL_Threads_Handler::get_mysql_frontend_buffers_bytes() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.stvar[st_var_mysql_frontend_buffers_bytes],0); } } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) for (i=0;istatus_variables.stvar[st_var_mysql_frontend_buffers_bytes],0); } } #endif // IDLE_THREADS this->status_variables.p_counter_array[p_th_gauge::mysql_frontend_buffers_bytes]->Increment(q); return q; } unsigned long long MySQL_Threads_Handler::get_mysql_session_internal_bytes() { if ((__sync_fetch_and_add(&status_variables.threads_initialized, 0) == 0) || this->shutdown_) return 0; unsigned long long q=0; unsigned int i; for (i=0;istatus_variables.stvar[st_var_mysql_session_internal_bytes],0); } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) if (mysql_threads_idles) { MySQL_Thread *thr=(MySQL_Thread *)mysql_threads_idles[i].worker; if (thr) q+=__sync_fetch_and_add(&thr->status_variables.stvar[st_var_mysql_session_internal_bytes],0); } #endif // IDLE_THREADS } this->status_variables.p_gauge_array[p_th_gauge::mysql_session_internal_bytes]->Set(q); return q; } void MySQL_Threads_Handler::p_update_metrics() { get_total_mirror_queue(); get_active_transations(); #ifdef IDLE_THREADS get_non_idle_client_connections(); #endif // IDLE_THREADS get_mysql_backend_buffers_bytes(); get_mysql_frontend_buffers_bytes(); get_mysql_session_internal_bytes(); for (unsigned int i=0; istatus_variables.p_gauge_array[p_th_gauge::mysql_wait_timeout]->Set(this->variables.wait_timeout); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_ping_interval]->Set(this->variables.monitor_ping_interval/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_max_connections]->Set(this->variables.max_connections); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_enabled]->Set(this->variables.monitor_enabled); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_ping_timeout]->Set(this->variables.monitor_ping_timeout/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_ping_max_failures]->Set(this->variables.monitor_ping_max_failures); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_aws_rds_topology_discovery_interval]->Set(this->variables.monitor_aws_rds_topology_discovery_interval); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_read_only_interval]->Set(this->variables.monitor_read_only_interval/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_read_only_timeout]->Set(this->variables.monitor_read_only_timeout/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_writer_is_also_reader]->Set(this->variables.monitor_writer_is_also_reader); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_replication_lag_group_by_host]->Set(this->variables.monitor_replication_lag_group_by_host); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_replication_lag_interval]->Set(this->variables.monitor_replication_lag_interval/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_replication_lag_timeout]->Set(this->variables.monitor_replication_lag_timeout/1000.0); this->status_variables.p_gauge_array[p_th_gauge::mysql_monitor_history]->Set(this->variables.monitor_history/1000.0); } /** * @brief Retrieves memory statistics for the MySQL thread. * * This function calculates memory statistics for the MySQL thread, including backend and frontend buffers, * as well as memory usage for internal session structures. * * Memory statistics are stored in the `status_variables` object, particularly in the `stvar` array. * * If there are active MySQL sessions associated with the thread, additional memory statistics are computed * based on session-related data structures and buffers. * * @note This function assumes that the `Memory_Stats()` function is implemented for MySQL_Session objects to retrieve session memory statistics. */ void MySQL_Thread::Get_Memory_Stats() { unsigned int i; status_variables.stvar[st_var_mysql_backend_buffers_bytes]=0; status_variables.stvar[st_var_mysql_frontend_buffers_bytes]=0; status_variables.stvar[st_var_mysql_session_internal_bytes]=sizeof(MySQL_Thread); if (mysql_sessions) { status_variables.stvar[st_var_mysql_session_internal_bytes]+=(mysql_sessions->size)*sizeof(MySQL_Session *); if (epoll_thread==false) { for (i=0; ilen; i++) { MySQL_Session *sess=(MySQL_Session *)mysql_sessions->index(i); sess->Memory_Stats(); } } else { status_variables.stvar[st_var_mysql_frontend_buffers_bytes]+=(mysql_sessions->len * QUEUE_T_DEFAULT_SIZE * 2); status_variables.stvar[st_var_mysql_session_internal_bytes]+=(mysql_sessions->len * sizeof(MySQL_Connection)); #if !defined(__FreeBSD__) && !defined(__APPLE__) status_variables.stvar[st_var_mysql_session_internal_bytes]+=((sizeof(int) + sizeof(int) + sizeof(std::_Rb_tree_node_base)) * mysql_sessions->len ); #else status_variables.stvar[st_var_mysql_session_internal_bytes]+=((sizeof(int) + sizeof(int) + 32) * mysql_sessions->len ); #endif } } } /** * @brief Retrieves a MySQL connection from the local cache based on specified criteria. * * This function retrieves a MySQL connection from the local cache managed by the MySQL_Thread instance. * It searches for a suitable connection based on the provided parameters such as host group ID, session information, * GTID UUID, GTID transaction ID, and maximum lag time. If a matching connection is found, it is removed from the * cache and returned to the caller. * * @param _hid The host group ID to which the connection belongs. * @param sess The MySQL session associated with the connection. * @param gtid_uuid The GTID UUID used for replication, or NULL if not used. * @param gtid_trxid The GTID transaction ID. * @param max_lag_ms The maximum lag time allowed for the connection in milliseconds. * @return A pointer to the retrieved MySQL connection if found; otherwise, NULL. */ MySQL_Connection * MySQL_Thread::get_MyConn_local(unsigned int _hid, MySQL_Session *sess, char *gtid_uuid, uint64_t gtid_trxid, int max_lag_ms) { // some sanity check if (sess == NULL) return NULL; if (sess->client_myds == NULL) return NULL; if (sess->client_myds->myconn == NULL) return NULL; if (sess->client_myds->myconn->userinfo == NULL) return NULL; unsigned int i; std::vector parents; // this is a vector of srvers that needs to be excluded in case gtid_uuid is used MySQL_Connection *c=NULL; for (i=0; ilen; i++) { c=(MySQL_Connection *)cached_connections->index(i); if (c->parent->myhgc->hid==_hid && sess->client_myds->myconn->match_tracked_options(c)) { // options are all identical if ( (gtid_uuid == NULL) || // gtid_uuid is not used (gtid_uuid && find(parents.begin(), parents.end(), c->parent) == parents.end()) // the server is currently not excluded ) { MySQL_Connection *client_conn = sess->client_myds->myconn; if (c->requires_CHANGE_USER(client_conn)==false) { // CHANGE_USER is not required char *schema = client_conn->userinfo->schemaname; if (strcmp(c->userinfo->schemaname,schema)==0) { // same schema unsigned int not_match = 0; // number of not matching session variables c->number_of_matching_session_variables(client_conn, not_match); if (not_match == 0) { // all session variables match if (gtid_uuid) { // gtid_uuid is used // we first check if we already excluded this parent (MySQL Server) MySrvC *mysrvc = c->parent; std::vector::iterator it; it = find(parents.begin(), parents.end(), mysrvc); if (it != parents.end()) { // we didn't exclude this server (yet?) bool gtid_found = false; gtid_found = MyHGM->gtid_exists(mysrvc, gtid_uuid, gtid_trxid); if (gtid_found) { // this server has the correct GTID c=(MySQL_Connection *)cached_connections->remove_index_fast(i); return c; } else { parents.push_back(mysrvc); // stop evaluating this server } } } else { // gtid_is not used if (max_lag_ms >= 0) { if ((unsigned int)max_lag_ms < (c->parent->aws_aurora_current_lag_us / 1000)) { status_variables.stvar[st_var_aws_aurora_replicas_skipped_during_query]++; continue; } } // return the connection c=(MySQL_Connection *)cached_connections->remove_index_fast(i); return c; } } } } } } } return NULL; } /** * @brief Pushes a MySQL connection to the local connection pool. * * This function is responsible for adding a MySQL connection to the local connection pool. * It resets the insert_id and checks if the associated server is online and the connection is idle * before adding it to the pool. * * If the server is online and the connection is idle, the connection is added to the cached connections pool. * Otherwise, if the server is not online or the connection is not idle, the connection is pushed to the * global connection pool managed by MySQL_Host_Group_Manager. * * @param c Pointer to the MySQL_Connection object to be pushed to the local connection pool. */ void MySQL_Thread::push_MyConn_local(MySQL_Connection *c) { MySrvC *mysrvc=NULL; mysrvc=(MySrvC *)c->parent; // reset insert_id #1093 c->mysql->insert_id = 0; if (mysrvc->get_status() == MYSQL_SERVER_STATUS_ONLINE) { if (c->async_state_machine==ASYNC_IDLE) { cached_connections->add(c); return; // all went well } } MyHGM->push_MyConn_to_pool(c); } /** * @brief Returns all locally cached MySQL connections to the global connection pool. * * This function is responsible for returning all locally cached MySQL connections to the global connection pool. * It checks if there are any cached connections available, and if so, it pushes them back to the global connection pool * managed by MySQL_Host_Group_Manager. After returning the connections, it clears the local cached connections pool. */ void MySQL_Thread::return_local_connections() { if (cached_connections->len==0) { return; } MyHGM->push_MyConn_to_pool_array((MySQL_Connection **)cached_connections->pdata, cached_connections->len); while (cached_connections->len) { cached_connections->remove_index_fast(0); } } /** * @brief Scans sessions to kill based on connection and query IDs stored in the kill queue. * * This function scans sessions to kill based on the connection and query IDs stored in the kill queue. * It iterates over the MySQL sessions and checks if any session matches the IDs stored in the kill queue. * If a session matches, it is marked for termination. * * For idle threads, it also scans sessions in the idle and resume sessions queues maintained by the session exchange. * * @note This function assumes that the kill queue (kq) contains connection and query IDs to be scanned. * * @see MySQL_Thread::Scan_Sessions_to_Kill */ void MySQL_Thread::Scan_Sessions_to_Kill_All() { if (kq.conn_ids.size() + kq.query_ids.size()) { Scan_Sessions_to_Kill(mysql_sessions); } #ifdef IDLE_THREADS if (GloVars.global.idle_threads) { if (kq.conn_ids.size() + kq.query_ids.size()) { Scan_Sessions_to_Kill(idle_mysql_sessions); } if (kq.conn_ids.size() + kq.query_ids.size()) { Scan_Sessions_to_Kill(resume_mysql_sessions); } if (kq.conn_ids.size() + kq.query_ids.size()) { pthread_mutex_lock(&myexchange.mutex_idles); Scan_Sessions_to_Kill(myexchange.idle_mysql_sessions); pthread_mutex_unlock(&myexchange.mutex_idles); } if (kq.conn_ids.size() + kq.query_ids.size()) { pthread_mutex_lock(&myexchange.mutex_resumes); Scan_Sessions_to_Kill(myexchange.resume_mysql_sessions); pthread_mutex_unlock(&myexchange.mutex_resumes); } } #endif for (std::vector::iterator it=kq.conn_ids.begin(); it!=kq.conn_ids.end(); ++it) { thr_id_usr *t = *it; free(t->username); free(t); } for (std::vector::iterator it=kq.query_ids.begin(); it!=kq.query_ids.end(); ++it) { thr_id_usr *t = *it; free(t->username); free(t); } kq.conn_ids.clear(); kq.query_ids.clear(); } /** * @brief Scans sessions in the provided session array to mark sessions for termination based on kill queue IDs. * * This function scans sessions in the provided session array (mysess) to identify sessions that match the connection * and query IDs stored in the kill queue (kq). If a session matches a connection or query ID, it is marked for termination. * * @param mysess Pointer to the session array to be scanned. * * @note This function assumes that the kill queue (kq) contains connection and query IDs to be scanned. */ void MySQL_Thread::Scan_Sessions_to_Kill(PtrArray *mysess) { for (unsigned int n=0; nlen && ( kq.conn_ids.size() + kq.query_ids.size() ) ; n++) { MySQL_Session *_sess=(MySQL_Session *)mysess->index(n); bool cont=true; for (std::vector::iterator it=kq.conn_ids.begin(); cont && it!=kq.conn_ids.end(); ++it) { thr_id_usr *t = *it; if (t->id == _sess->thread_session_id) { if (_sess->client_myds) { if (strcmp(t->username,_sess->client_myds->myconn->userinfo->username)==0) { _sess->killed=true; } } cont=false; free(t->username); free(t); kq.conn_ids.erase(it); } } for (std::vector::iterator it=kq.query_ids.begin(); cont && it!=kq.query_ids.end(); ++it) { thr_id_usr *t = *it; if (t->id == _sess->thread_session_id) { proxy_info("Killing query %d\n", t->id); if (_sess->client_myds) { if (strcmp(t->username,_sess->client_myds->myconn->userinfo->username)==0) { if (_sess->mybe) { if (_sess->mybe->server_myds) { _sess->mybe->server_myds->wait_until=curtime; _sess->mybe->server_myds->kill_type=1; } } } } cont=false; free(t->username); free(t); kq.query_ids.erase(it); } } } } #ifdef IDLE_THREADS /** * @brief Moves sessions from the idle thread's session array to the worker thread's session array. * * This function is called by the idle maintenance thread to transfer sessions from the idle session array * managed by the exchange structure to the worker thread's session array. It locks the mutex associated * with the idle session array, iterates through the sessions, registers each session, adds it to the worker * thread's poll array, and adds it to the epoll set for monitoring read events. Finally, it updates the session * map to map thread IDs to the positions in the worker thread's session array. */ void MySQL_Thread::idle_thread_gets_sessions_from_worker_thread() { pthread_mutex_lock(&myexchange.mutex_idles); while (myexchange.idle_mysql_sessions->len) { MySQL_Session *mysess=(MySQL_Session *)myexchange.idle_mysql_sessions->remove_index_fast(0); register_session(this, mysess, false); MySQL_Data_Stream *myds=mysess->client_myds; mypolls.add(POLLIN, myds->fd, myds, monotonic_time()); // add in epoll() struct epoll_event event; memset(&event,0,sizeof(event)); // let's make valgrind happy event.data.u32=mysess->thread_session_id; event.events = EPOLLIN; epoll_ctl (efd, EPOLL_CTL_ADD, myds->fd, &event); // we map thread_id -> position in mysql_session (end of the list) sessmap[mysess->thread_session_id]=mysql_sessions->len-1; //fprintf(stderr,"Adding session %p idx, DS %p idx %d\n",mysess,myds,myds->poll_fds_idx); } pthread_mutex_unlock(&myexchange.mutex_idles); } #endif // IDLE_THREADS void MySQL_Thread::handle_mirror_queue_mysql_sessions() { while (mirror_queue_mysql_sessions->len) { if (__sync_add_and_fetch(&GloMTH->status_variables.mirror_sessions_current,1) > (unsigned int)mysql_thread___mirror_max_concurrency ) { __sync_sub_and_fetch(&GloMTH->status_variables.mirror_sessions_current,1); //goto __mysql_thread_exit_add_mirror; // we can't add more mirror sessions at runtime return; } else { int idx; idx=fastrand()%(mirror_queue_mysql_sessions->len); MySQL_Session *newsess=(MySQL_Session *)mirror_queue_mysql_sessions->remove_index_fast(idx); register_session(this, newsess); newsess->handler(); // execute immediately if (newsess->status==WAITING_CLIENT_DATA) { // the mirror session has completed unregister_session(mysql_sessions->len-1); unsigned int l = (unsigned int)mysql_thread___mirror_max_concurrency; if (mirror_queue_mysql_sessions->len*0.3 > l) l=mirror_queue_mysql_sessions->len*0.3; if (mirror_queue_mysql_sessions_cache->len <= l) { bool to_cache=true; if (newsess->mybe) { if (newsess->mybe->server_myds) { to_cache=false; } } if (to_cache) { __sync_sub_and_fetch(&GloMTH->status_variables.mirror_sessions_current,1); mirror_queue_mysql_sessions_cache->add(newsess); } else { delete newsess; } } else { delete newsess; } } //newsess->to_process=0; } } } /** * @brief Handles the kill queues by scanning sessions to kill and setting the maintenance loop flag. * * This function is responsible for handling the kill queues. It locks the mutex associated with the kill queues, * scans the sessions to kill if there are pending connections or queries in the kill queues, and sets the maintenance * loop flag to true to initiate maintenance tasks. After processing the kill queues, it releases the mutex. */ void MySQL_Thread::handle_kill_queues() { pthread_mutex_lock(&kq.m); if (kq.conn_ids.size() + kq.query_ids.size()) { Scan_Sessions_to_Kill_All(); maintenance_loop=true; } pthread_mutex_unlock(&kq.m); }