mirror of
https://github.com/MariaDB/server.git
synced 2025-11-04 12:56:14 +01:00
219 lines
7.4 KiB
C++
219 lines
7.4 KiB
C++
|
|
// vim:sw=2:ai
|
|
|
|
/*
|
|
* Copyright (C) 2010 DeNA Co.,Ltd.. All rights reserved.
|
|
* See COPYRIGHT.txt for details.
|
|
*/
|
|
|
|
#include <my_global.h>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <stdio.h>
|
|
|
|
#include "config.hpp"
|
|
#include "hstcpsvr.hpp"
|
|
#include "string_util.hpp"
|
|
#include "mysql_incl.hpp"
|
|
|
|
#define DBG_LOG \
|
|
if (dena::verbose_level >= 100) { \
|
|
fprintf(stderr, "%s %p\n", __PRETTY_FUNCTION__, this); \
|
|
}
|
|
#define DBG_DO(x) if (dena::verbose_level >= 100) { x; }
|
|
|
|
#define DBG_DIR(x)
|
|
|
|
using namespace dena;
|
|
|
|
static char *handlersocket_address = 0;
|
|
static char *handlersocket_port = 0;
|
|
static char *handlersocket_port_wr = 0;
|
|
static unsigned int handlersocket_epoll = 1;
|
|
static unsigned int handlersocket_threads = 32;
|
|
static unsigned int handlersocket_threads_wr = 1;
|
|
static unsigned int handlersocket_timeout = 30;
|
|
static unsigned int handlersocket_backlog = 32768;
|
|
static unsigned int handlersocket_sndbuf = 0;
|
|
static unsigned int handlersocket_rcvbuf = 0;
|
|
static unsigned int handlersocket_readsize = 0;
|
|
static unsigned int handlersocket_accept_balance = 0;
|
|
static unsigned int handlersocket_wrlock_timeout = 0;
|
|
static char *handlersocket_plain_secret = 0;
|
|
static char *handlersocket_plain_secret_wr = 0;
|
|
|
|
struct daemon_handlersocket_data {
|
|
hstcpsvr_ptr hssvr_rd;
|
|
hstcpsvr_ptr hssvr_wr;
|
|
};
|
|
|
|
static int
|
|
daemon_handlersocket_init(void *p)
|
|
{
|
|
DENA_VERBOSE(10, fprintf(stderr, "handlersocket: initialized\n"));
|
|
config conf;
|
|
conf["use_epoll"] = handlersocket_epoll ? "1" : "0";
|
|
if (handlersocket_address) {
|
|
conf["host"] = handlersocket_address;
|
|
}
|
|
if (handlersocket_port) {
|
|
conf["port"] = handlersocket_port;
|
|
}
|
|
/*
|
|
* unix domain socket
|
|
* conf["host"] = "/";
|
|
* conf["port"] = "/tmp/handlersocket";
|
|
*/
|
|
if (handlersocket_threads > 0) {
|
|
conf["num_threads"] = to_stdstring(handlersocket_threads);
|
|
} else {
|
|
conf["num_threads"] = "1";
|
|
}
|
|
conf["timeout"] = to_stdstring(handlersocket_timeout);
|
|
conf["listen_backlog"] = to_stdstring(handlersocket_backlog);
|
|
conf["sndbuf"] = to_stdstring(handlersocket_sndbuf);
|
|
conf["rcvbuf"] = to_stdstring(handlersocket_rcvbuf);
|
|
conf["readsize"] = to_stdstring(handlersocket_readsize);
|
|
conf["accept_balance"] = to_stdstring(handlersocket_accept_balance);
|
|
conf["wrlock_timeout"] = to_stdstring(handlersocket_wrlock_timeout);
|
|
std::unique_ptr<daemon_handlersocket_data> ap(new daemon_handlersocket_data);
|
|
if (handlersocket_port != 0 && handlersocket_port_wr != handlersocket_port) {
|
|
conf["port"] = handlersocket_port;
|
|
if (handlersocket_plain_secret) {
|
|
conf["plain_secret"] = handlersocket_plain_secret;
|
|
}
|
|
ap->hssvr_rd = hstcpsvr_i::create(conf);
|
|
ap->hssvr_rd->start_listen();
|
|
}
|
|
if (handlersocket_port_wr != 0) {
|
|
if (handlersocket_threads_wr > 0) {
|
|
conf["num_threads"] = to_stdstring(handlersocket_threads_wr);
|
|
}
|
|
conf["port"] = handlersocket_port_wr;
|
|
conf["for_write"] = "1";
|
|
conf["plain_secret"] = "";
|
|
if (handlersocket_plain_secret_wr) {
|
|
conf["plain_secret"] = handlersocket_plain_secret_wr;
|
|
}
|
|
ap->hssvr_wr = hstcpsvr_i::create(conf);
|
|
ap->hssvr_wr->start_listen();
|
|
}
|
|
st_plugin_int *const plugin = static_cast<st_plugin_int *>(p);
|
|
plugin->data = ap.release();
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
daemon_handlersocket_deinit(void *p)
|
|
{
|
|
DENA_VERBOSE(10, fprintf(stderr, "handlersocket: terminated\n"));
|
|
st_plugin_int *const plugin = static_cast<st_plugin_int *>(p);
|
|
daemon_handlersocket_data *ptr =
|
|
static_cast<daemon_handlersocket_data *>(plugin->data);
|
|
delete ptr;
|
|
return 0;
|
|
}
|
|
|
|
static struct st_mysql_daemon daemon_handlersocket_plugin = {
|
|
MYSQL_DAEMON_INTERFACE_VERSION
|
|
};
|
|
|
|
static MYSQL_SYSVAR_UINT(verbose, dena::verbose_level, 0,
|
|
"0..10000", 0, 0, 10 /* default */, 0, 10000, 0);
|
|
static MYSQL_SYSVAR_UINT(epoll, handlersocket_epoll, PLUGIN_VAR_READONLY,
|
|
"0..1", 0, 0, 1 /* default */, 0, 1, 0);
|
|
static MYSQL_SYSVAR_STR(address, handlersocket_address,
|
|
PLUGIN_VAR_READONLY | PLUGIN_VAR_MEMALLOC, "", NULL, NULL, NULL);
|
|
static MYSQL_SYSVAR_STR(port, handlersocket_port,
|
|
PLUGIN_VAR_READONLY | PLUGIN_VAR_MEMALLOC, "", NULL, NULL, NULL);
|
|
static MYSQL_SYSVAR_STR(port_wr, handlersocket_port_wr,
|
|
PLUGIN_VAR_READONLY | PLUGIN_VAR_MEMALLOC, "", NULL, NULL, NULL);
|
|
static MYSQL_SYSVAR_UINT(threads, handlersocket_threads, PLUGIN_VAR_READONLY,
|
|
"1..3000", 0, 0, 16 /* default */, 1, 3000, 0);
|
|
static MYSQL_SYSVAR_UINT(threads_wr, handlersocket_threads_wr,
|
|
PLUGIN_VAR_READONLY, "1..3000", 0, 0, 1 /* default */, 1, 3000, 0);
|
|
static MYSQL_SYSVAR_UINT(timeout, handlersocket_timeout, PLUGIN_VAR_READONLY,
|
|
"30..3600", 0, 0, 300 /* default */, 30, 3600, 0);
|
|
static MYSQL_SYSVAR_UINT(backlog, handlersocket_backlog, PLUGIN_VAR_READONLY,
|
|
"5..1000000", 0, 0, 32768 /* default */, 5, 1000000, 0);
|
|
static MYSQL_SYSVAR_UINT(sndbuf, handlersocket_sndbuf, PLUGIN_VAR_READONLY,
|
|
"0..16777216", 0, 0, 0 /* default */, 0, 16777216, 0);
|
|
static MYSQL_SYSVAR_UINT(rcvbuf, handlersocket_rcvbuf, PLUGIN_VAR_READONLY,
|
|
"0..16777216", 0, 0, 0 /* default */, 0, 16777216, 0);
|
|
static MYSQL_SYSVAR_UINT(readsize, handlersocket_readsize, PLUGIN_VAR_READONLY,
|
|
"0..16777216", 0, 0, 0 /* default */, 0, 16777216, 0);
|
|
static MYSQL_SYSVAR_UINT(accept_balance, handlersocket_accept_balance,
|
|
PLUGIN_VAR_READONLY, "0..10000", 0, 0, 0 /* default */, 0, 10000, 0);
|
|
static MYSQL_SYSVAR_UINT(wrlock_timeout, handlersocket_wrlock_timeout,
|
|
PLUGIN_VAR_READONLY, "0..3600", 0, 0, 12 /* default */, 0, 3600, 0);
|
|
static MYSQL_SYSVAR_STR(plain_secret, handlersocket_plain_secret,
|
|
PLUGIN_VAR_READONLY | PLUGIN_VAR_MEMALLOC, "", NULL, NULL, NULL);
|
|
static MYSQL_SYSVAR_STR(plain_secret_wr, handlersocket_plain_secret_wr,
|
|
PLUGIN_VAR_READONLY | PLUGIN_VAR_MEMALLOC, "", NULL, NULL, NULL);
|
|
|
|
|
|
/* warning: type-punning to incomplete type might break strict-aliasing
|
|
* rules */
|
|
static struct st_mysql_sys_var *daemon_handlersocket_system_variables[] = {
|
|
MYSQL_SYSVAR(verbose),
|
|
MYSQL_SYSVAR(address),
|
|
MYSQL_SYSVAR(port),
|
|
MYSQL_SYSVAR(port_wr),
|
|
MYSQL_SYSVAR(epoll),
|
|
MYSQL_SYSVAR(threads),
|
|
MYSQL_SYSVAR(threads_wr),
|
|
MYSQL_SYSVAR(timeout),
|
|
MYSQL_SYSVAR(backlog),
|
|
MYSQL_SYSVAR(sndbuf),
|
|
MYSQL_SYSVAR(rcvbuf),
|
|
MYSQL_SYSVAR(readsize),
|
|
MYSQL_SYSVAR(accept_balance),
|
|
MYSQL_SYSVAR(wrlock_timeout),
|
|
MYSQL_SYSVAR(plain_secret),
|
|
MYSQL_SYSVAR(plain_secret_wr),
|
|
0
|
|
};
|
|
|
|
static SHOW_VAR hs_status_variables[] = {
|
|
{"table_open", (char*) &open_tables_count, SHOW_LONGLONG},
|
|
{"table_close", (char*) &close_tables_count, SHOW_LONGLONG},
|
|
{"table_lock", (char*) &lock_tables_count, SHOW_LONGLONG},
|
|
{"table_unlock", (char*) &unlock_tables_count, SHOW_LONGLONG},
|
|
#if 0
|
|
{"index_exec", (char*) &index_exec_count, SHOW_LONGLONG},
|
|
#endif
|
|
{NullS, NullS, SHOW_LONG}
|
|
};
|
|
|
|
static int show_hs_vars(THD *thd, SHOW_VAR *var, void *buff,
|
|
struct system_status_var *status_var,
|
|
enum enum_var_type var_type)
|
|
{
|
|
var->type= SHOW_ARRAY;
|
|
var->value= (char *) &hs_status_variables;
|
|
return 0;
|
|
}
|
|
|
|
static SHOW_VAR daemon_handlersocket_status_variables[] = {
|
|
SHOW_FUNC_ENTRY("Hs", &show_hs_vars),
|
|
{NullS, NullS, SHOW_LONG}
|
|
};
|
|
|
|
|
|
maria_declare_plugin(handlersocket)
|
|
{
|
|
MYSQL_DAEMON_PLUGIN,
|
|
&daemon_handlersocket_plugin,
|
|
"handlersocket",
|
|
"higuchi dot akira at dena dot jp",
|
|
"Direct access into InnoDB",
|
|
PLUGIN_LICENSE_BSD,
|
|
daemon_handlersocket_init,
|
|
daemon_handlersocket_deinit,
|
|
0x0100 /* 1.0 */,
|
|
daemon_handlersocket_status_variables,
|
|
daemon_handlersocket_system_variables,
|
|
"1.0",
|
|
MariaDB_PLUGIN_MATURITY_BETA
|
|
}
|
|
maria_declare_plugin_end;
|