20
20
#include "config.h"
22
#include <boost/scoped_array.hpp>
24
21
#include <drizzled/plugin/logging.h>
25
22
#include <drizzled/gettext.h>
26
23
#include <drizzled/session.h>
27
#include <boost/date_time.hpp>
28
#include <boost/program_options.hpp>
29
#include <drizzled/module/option_map.h>
30
25
#include <libgearman/gearman.h>
31
26
#include <limits.h>
32
28
#include <sys/types.h>
33
29
#include <sys/stat.h>
40
namespace drizzle_plugin
43
namespace po= boost::program_options;
33
using namespace drizzled;
45
36
/* TODO make this dynamic as needed */
46
37
static const int MAX_MSG_LEN= 32*1024;
39
static bool sysvar_logging_gearman_enable= false;
40
static char* sysvar_logging_gearman_host= NULL;
41
static char* sysvar_logging_gearman_function= NULL;
44
/* stolen from mysys/my_getsystime
45
until the Session has a good utime "now" we can use
46
will have to use this instead */
48
static uint64_t get_microtime()
50
#if defined(HAVE_GETHRTIME)
51
return gethrtime()/1000;
56
The following loop is here because gettimeofday may fail on some systems
58
while (gettimeofday(&t, NULL) != 0) {}
59
newtime= (uint64_t)t.tv_sec * 1000000 + t.tv_usec;
61
#endif /* defined(HAVE_GETHRTIME) */
48
64
/* quote a string to be safe to include in a CSV line
49
65
that means backslash quoting all commas, doublequotes, backslashes,
50
66
and all the ASCII unprintable characters
156
class LoggingGearman :
157
public drizzled::plugin::Logging
172
class LoggingGearman : public plugin::Logging
160
const std::string _host;
161
const std::string _function;
163
int _gearman_client_ok;
164
gearman_client_st _gearman_client;
167
LoggingGearman(const LoggingGearman&);
175
int gearman_client_ok;
176
gearman_client_st gearman_client;
171
LoggingGearman(const std::string &host,
172
const std::string &function) :
173
drizzled::plugin::Logging("LoggingGearman"),
176
_gearman_client_ok(0),
181
: plugin::Logging("LoggingGearman"),
179
184
gearman_return_t ret;
182
if (gearman_client_create(&_gearman_client) == NULL)
186
if (sysvar_logging_gearman_enable == false)
189
if (sysvar_logging_gearman_host == NULL)
193
if (gearman_client_create(&gearman_client) == NULL)
184
char errmsg[STRERROR_MAX];
185
strerror_r(errno, errmsg, sizeof(errmsg));
186
drizzled::errmsg_printf(ERRMSG_LVL_ERROR, _("fail gearman_client_create(): %s"),
195
errmsg_printf(ERRMSG_LVL_ERROR, _("fail gearman_client_create(): %s"),
191
200
/* TODO, be able to override the port */
192
201
/* TODO, be able send to multiple servers */
193
ret= gearman_client_add_server(&_gearman_client,
202
ret= gearman_client_add_server(&gearman_client,
203
sysvar_logging_gearman_host, 0);
195
204
if (ret != GEARMAN_SUCCESS)
197
drizzled::errmsg_printf(ERRMSG_LVL_ERROR, _("fail gearman_client_add_server(): %s"),
198
gearman_client_error(&_gearman_client));
206
errmsg_printf(ERRMSG_LVL_ERROR, _("fail gearman_client_add_server(): %s"),
207
gearman_client_error(&gearman_client));
202
_gearman_client_ok= 1;
211
gearman_client_ok= 1;
206
215
~LoggingGearman()
208
if (_gearman_client_ok)
217
if (gearman_client_ok)
210
gearman_client_free(&_gearman_client);
219
gearman_client_free(&gearman_client);
214
virtual bool post(drizzled::Session *session)
223
virtual bool post(Session *session)
216
boost::scoped_array<char> msgbuf(new char[MAX_MSG_LEN]);
225
char msgbuf[MAX_MSG_LEN];
217
226
int msgbuf_len= 0;
219
228
assert(session != NULL);
222
231
but that crashes the server, so for now, we just lie a little bit
225
if (not _gearman_client_ok)
234
if (!gearman_client_ok)
228
237
/* TODO, the session object should have a "utime command completed"
229
238
inside itself, so be more accurate, and so this doesnt have to
230
239
keep calling current_utime, which can be slow */
232
boost::posix_time::ptime mytime(boost::posix_time::microsec_clock::local_time());
233
boost::posix_time::ptime epoch(boost::gregorian::date(1970,1,1));
234
uint64_t t_mark= (mytime-epoch).total_microseconds();
241
uint64_t t_mark= get_microtime();
237
243
// buffer to quotify the query
238
244
unsigned char qs[255];
240
246
// to avoid trying to printf %s something that is potentially NULL
241
drizzled::util::string::const_shared_ptr dbs(session->schema());
247
const char *dbs= session->db.empty() ? "" : session->db.c_str();
244
snprintf(msgbuf.get(), MAX_MSG_LEN,
250
snprintf(msgbuf, MAX_MSG_LEN,
245
251
"%"PRIu64",%"PRIu64",%"PRIu64",\"%.*s\",\"%s\",\"%.*s\","
246
252
"%"PRIu64",%"PRIu64",%"PRIu64",%"PRIu64",%"PRIu64","
247
253
"%"PRIu32",%"PRIu32",%"PRIu32",\"%s\"",
249
255
session->thread_id,
250
256
session->getQueryId(),
251
257
// dont need to quote the db name, always CSV safe
252
(int)dbs->size(), dbs->c_str(),
258
(int)session->db.length(), dbs,
253
259
// do need to quote the query
254
quotify((const unsigned char *)session->getQueryString()->c_str(), session->getQueryString()->length(), qs, sizeof(qs)),
260
quotify((const unsigned char *)session->getQueryString().c_str(),
261
session->getQueryLength(), qs, sizeof(qs)),
255
262
// command_name is defined in drizzled/sql_parse.cc
256
263
// dont need to quote the command name, always CSV safe
257
(int)drizzled::command_name[session->command].length,
258
drizzled::command_name[session->command].str,
264
(int)command_name[session->command].length,
265
command_name[session->command].str,
259
266
// counters are at end, to make it easier to add more
260
267
(t_mark - session->getConnectMicroseconds()),
261
268
(t_mark - session->start_utime),
265
272
session->tmp_table,
266
273
session->total_warn_count,
267
274
session->getServerId(),
268
drizzled::glob_hostname
271
278
char job_handle[GEARMAN_JOB_HANDLE_SIZE];
273
(void) gearman_client_do_background(&_gearman_client,
280
(void) gearman_client_do_background(&gearman_client,
281
sysvar_logging_gearman_function,
276
(void *) msgbuf.get(),
277
284
(size_t) msgbuf_len,
284
291
static LoggingGearman *handler= NULL;
286
static int logging_gearman_plugin_init(drizzled::module::Context &context)
293
static int logging_gearman_plugin_init(module::Context &context)
288
const drizzled::module::option_map &vm= context.getOptions();
290
handler= new LoggingGearman(vm["host"].as<std::string>(),
291
vm["function"].as<std::string>());
295
handler= new LoggingGearman();
292
296
context.add(handler);
293
context.registerVariable(new drizzled::sys_var_const_string_val("host", vm["host"].as<std::string>()));
294
context.registerVariable(new drizzled::sys_var_const_string_val("function", vm["function"].as<std::string>()));
299
static void init_options(drizzled::module::option_context &context)
302
po::value<std::string>()->default_value("localhost"),
303
N_("Hostname for logging to a Gearman server"));
305
po::value<std::string>()->default_value("drizzlelog"),
306
N_("Gearman Function to send logging to"));
309
} /* namespace drizzle_plugin */
301
static DRIZZLE_SYSVAR_BOOL(
303
sysvar_logging_gearman_enable,
305
N_("Enable logging to a gearman server"),
306
NULL, /* check func */
307
NULL, /* update func */
308
false /* default */);
310
static DRIZZLE_SYSVAR_STR(
312
sysvar_logging_gearman_host,
314
N_("Hostname for logging to a Gearman server"),
315
NULL, /* check func */
316
NULL, /* update func*/
317
"localhost" /* default */);
319
static DRIZZLE_SYSVAR_STR(
321
sysvar_logging_gearman_function,
323
N_("Gearman Function to send logging to"),
324
NULL, /* check func */
325
NULL, /* update func*/
326
"drizzlelog" /* default */);
328
static drizzle_sys_var* logging_gearman_system_variables[]= {
329
DRIZZLE_SYSVAR(enable),
330
DRIZZLE_SYSVAR(host),
331
DRIZZLE_SYSVAR(function),
311
335
DRIZZLE_DECLARE_PLUGIN
313
337
DRIZZLE_VERSION_ID,
316
340
"Mark Atwood <mark@fallenpegasus.com>",
317
341
N_("Log queries to a Gearman server"),
318
drizzled::PLUGIN_LICENSE_GPL,
319
drizzle_plugin::logging_gearman_plugin_init,
321
drizzle_plugin::init_options
343
logging_gearman_plugin_init,
344
logging_gearman_system_variables,
323
347
DRIZZLE_DECLARE_PLUGIN_END;