← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/logging_query/logging_query.cc

  • Committer: Brian Aker
  • Date: 2009-02-10 00:14:40 UTC
  • Revision ID: brian@tangent.org-20090210001440-qjg8eofh3h93064b
Adding Multi-threaded Scheduler into the system.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/logging_query/logging_query.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2008, 2009 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2008,2009 Mark Atwood
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#include <config.h>
21
 
#include <drizzled/plugin/logging.h>
 
20
#include <drizzled/server_includes.h>
 
21
#include <drizzled/plugin_logging.h>
22
22
#include <drizzled/gettext.h>
23
23
#include <drizzled/session.h>
24
 
#include PCRE_HEADER
25
 
#include <limits.h>
 
24
 
 
25
/* TODO make this dynamic as needed */
 
26
static const int MAX_MSG_LEN= 32*1024;
 
27
 
 
28
static bool sysvar_logging_query_enable= false;
 
29
static char* sysvar_logging_query_filename= NULL;
 
30
/* TODO fix these to not be unsigned long one we have sensible sys_var system */
 
31
static unsigned long sysvar_logging_query_threshold_slow= 0;
 
32
static unsigned long sysvar_logging_query_threshold_big_resultset= 0;
 
33
static unsigned long sysvar_logging_query_threshold_big_examined= 0;
 
34
 
 
35
static int fd= -1;
 
36
 
 
37
/* stolen from mysys/my_getsystime
 
38
   until the Session has a good utime "now" we can use
 
39
   will have to use this instead */
 
40
 
26
41
#include <sys/time.h>
27
 
#include <sys/types.h>
28
 
#include <sys/stat.h>
29
 
#include <fcntl.h>
30
 
#include <string>
31
 
#include <boost/format.hpp>
32
 
#include <boost/program_options.hpp>
33
 
#include <drizzled/module/option_map.h>
34
 
#include <cstdio>
35
 
#include <cerrno>
36
 
 
37
 
namespace po= boost::program_options;
38
 
using namespace drizzled;
39
 
using namespace std;
40
 
 
41
 
#define ESCAPE_CHAR      '\\'
42
 
#define SEPARATOR_CHAR   ','
43
 
 
44
 
namespace drizzle_plugin
 
42
static uint64_t get_microtime()
45
43
{
46
 
 
47
 
static bool sysvar_logging_query_enable= false;
48
 
/* TODO fix these to not be unsigned long once we have sensible sys_var system */
49
 
static uint32_constraint sysvar_logging_query_threshold_slow;
50
 
static uint32_constraint sysvar_logging_query_threshold_big_resultset;
51
 
static uint32_constraint sysvar_logging_query_threshold_big_examined;
 
44
#if defined(HAVE_GETHRTIME)
 
45
  return gethrtime()/1000;
 
46
#else
 
47
  uint64_t newtime;
 
48
  struct timeval t;
 
49
  /*
 
50
    The following loop is here because gettimeofday may fail on some systems
 
51
  */
 
52
  while (gettimeofday(&t, NULL) != 0) {}
 
53
  newtime= (uint64_t)t.tv_sec * 1000000 + t.tv_usec;
 
54
  return newtime;
 
55
#endif  /* defined(HAVE_GETHRTIME) */
 
56
}
52
57
 
53
58
/* quote a string to be safe to include in a CSV line
54
59
   that means backslash quoting all commas, doublequotes, backslashes,
66
71
 
67
72
*/
68
73
 
69
 
static void quotify(const string &src, string &dst)
 
74
static unsigned char *quotify (const unsigned char *src, size_t srclen,
 
75
                               unsigned char *dst, size_t dstlen)
70
76
{
71
77
  static const char hexit[]= { '0', '1', '2', '3', '4', '5', '6', '7',
72
78
                          '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };
73
 
  string::const_iterator src_iter;
74
 
  
75
 
  for (src_iter= src.begin(); src_iter < src.end(); ++src_iter)
 
79
  size_t dst_ndx;  /* ndx down the dst */
 
80
  size_t src_ndx;  /* ndx down the src */
 
81
 
 
82
  assert(dst);
 
83
  assert(dstlen > 0);
 
84
 
 
85
  for (dst_ndx= 0,src_ndx= 0; src_ndx < srclen; src_ndx++)
76
86
  {
77
 
    if (static_cast<unsigned char>(*src_iter) > 0x7f)
78
 
    {
79
 
      dst.push_back(*src_iter);
80
 
    }
81
 
    else if (*src_iter == 0x00)  // null
82
 
    {
83
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('0');
84
 
    }
85
 
    else if (*src_iter == 0x07)  // bell
86
 
    {
87
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('a');
88
 
    }
89
 
    else if (*src_iter == 0x08)  // backspace
90
 
    {
91
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('b');
92
 
    }
93
 
    else if (*src_iter == 0x09)  // horiz tab
94
 
    {
95
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('t');
96
 
    }
97
 
    else if (*src_iter == 0x0a)  // line feed
98
 
    {
99
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('n');
100
 
    }
101
 
    else if (*src_iter == 0x0b)  // vert tab
102
 
    {
103
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('v');
104
 
    }
105
 
    else if (*src_iter == 0x0c)  // formfeed
106
 
    {
107
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('f');
108
 
    }
109
 
    else if (*src_iter == 0x0d)  // carrage return
110
 
    {
111
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('r');
112
 
    }
113
 
    else if (*src_iter == 0x1b)  // escape
114
 
    {
115
 
      dst.push_back(ESCAPE_CHAR); dst.push_back('e');
116
 
    }
117
 
    else if (*src_iter == 0x22)  // quotation mark
118
 
    {
119
 
      dst.push_back(ESCAPE_CHAR); dst.push_back(0x22);
120
 
    }
121
 
    else if (*src_iter == SEPARATOR_CHAR)
122
 
    {
123
 
      dst.push_back(ESCAPE_CHAR); dst.push_back(SEPARATOR_CHAR);
124
 
    }
125
 
    else if (*src_iter == ESCAPE_CHAR)
126
 
    {
127
 
      dst.push_back(ESCAPE_CHAR); dst.push_back(ESCAPE_CHAR);
128
 
    }
129
 
    else if ((*src_iter < 0x20) || (*src_iter == 0x7F))  // other unprintable ASCII
130
 
    {
131
 
      dst.push_back(ESCAPE_CHAR);
132
 
      dst.push_back('x');
133
 
      dst.push_back(hexit[(*src_iter >> 4) & 0x0f]);
134
 
      dst.push_back(hexit[*src_iter & 0x0f]);
 
87
 
 
88
    /* Worst case, need 5 dst bytes for the next src byte.
 
89
       backslash x hexit hexit null
 
90
       so if not enough room, just terminate the string and return
 
91
    */
 
92
    if ((dstlen - dst_ndx) < 5)
 
93
    {
 
94
      dst[dst_ndx]= (unsigned char)0x00;
 
95
      return dst;
 
96
    }
 
97
 
 
98
    if (src[src_ndx] > 0x7f)
 
99
    {
 
100
      // pass thru high bit characters, they are non-ASCII UTF8 Unicode
 
101
      dst[dst_ndx++]= src[src_ndx];
 
102
    }
 
103
    else if (src[src_ndx] == 0x00)  // null
 
104
    {
 
105
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) '0';
 
106
    }
 
107
    else if (src[src_ndx] == 0x07)  // bell
 
108
    {
 
109
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'a';
 
110
    }
 
111
    else if (src[src_ndx] == 0x08)  // backspace
 
112
    {
 
113
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'b';
 
114
    }
 
115
    else if (src[src_ndx] == 0x09)  // horiz tab
 
116
    {
 
117
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 't';
 
118
    }
 
119
    else if (src[src_ndx] == 0x0a)  // line feed
 
120
    {
 
121
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'n';
 
122
    }
 
123
    else if (src[src_ndx] == 0x0b)  // vert tab
 
124
    {
 
125
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'v';
 
126
    }
 
127
    else if (src[src_ndx] == 0x0c)  // formfeed
 
128
    {
 
129
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'f';
 
130
    }
 
131
    else if (src[src_ndx] == 0x0d)  // carrage return
 
132
    {
 
133
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'r';
 
134
    }
 
135
    else if (src[src_ndx] == 0x1b)  // escape
 
136
    {
 
137
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= (unsigned char) 'e';
 
138
    }
 
139
    else if (src[src_ndx] == 0x22)  // quotation mark
 
140
    {
 
141
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= 0x22;
 
142
    }
 
143
    else if (src[src_ndx] == 0x2C)  // comma
 
144
    {
 
145
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= 0x2C;
 
146
    }
 
147
    else if (src[src_ndx] == 0x5C)  // backslash
 
148
    {
 
149
      dst[dst_ndx++]= 0x5C; dst[dst_ndx++]= 0x5C;
 
150
    }
 
151
    else if ((src[src_ndx] < 0x20) || (src[src_ndx] == 0x7F))  // other unprintable ASCII
 
152
    {
 
153
      dst[dst_ndx++]= 0x5C;
 
154
      dst[dst_ndx++]= (unsigned char) 'x';
 
155
      dst[dst_ndx++]= hexit[(src[src_ndx] >> 4) & 0x0f];
 
156
      dst[dst_ndx++]= hexit[src[src_ndx] & 0x0f];
135
157
    }
136
158
    else  // everything else
137
159
    {
138
 
      dst.push_back(*src_iter);
139
 
    }
140
 
  }
141
 
}
142
 
 
143
 
 
144
 
class Logging_query: public drizzled::plugin::Logging
145
 
{
146
 
  const std::string _filename;
147
 
  const std::string _query_pcre;
148
 
  int fd;
149
 
  pcre *re;
150
 
  pcre_extra *pe;
151
 
 
152
 
  /** Format of the output string */
153
 
  boost::format formatter;
154
 
 
155
 
public:
156
 
 
157
 
  Logging_query(const std::string &filename,
158
 
                const std::string &query_pcre) :
159
 
    drizzled::plugin::Logging("Logging_query"),
160
 
    _filename(filename),
161
 
    _query_pcre(query_pcre),
162
 
    fd(-1), re(NULL), pe(NULL),
163
 
    formatter("%1%,%2%,%3%,\"%4%\",\"%5%\",\"%6%\",%7%,%8%,"
164
 
              "%9%,%10%,%11%,%12%,%13%,%14%,\"%15%\"\n")
165
 
  {
166
 
 
167
 
    /* if there is no destination filename, dont bother doing anything */
168
 
    if (_filename.empty())
169
 
      return;
170
 
 
171
 
    fd= open(_filename.c_str(),
172
 
             O_WRONLY | O_APPEND | O_CREAT,
173
 
             S_IRUSR|S_IWUSR);
174
 
 
175
 
    if (fd < 0)
176
 
    {
177
 
      sql_perror( _("fail open()"), _filename);
178
 
      return;
179
 
    }
180
 
 
181
 
    if (not _query_pcre.empty())
182
 
    {
183
 
      const char *this_pcre_error;
184
 
      int this_pcre_erroffset;
185
 
      re= pcre_compile(_query_pcre.c_str(), 0, &this_pcre_error,
186
 
                       &this_pcre_erroffset, NULL);
187
 
      pe= pcre_study(re, 0, &this_pcre_error);
188
 
      /* TODO emit error messages if there is a problem */
189
 
    }
190
 
  }
191
 
 
192
 
  ~Logging_query()
193
 
  {
194
 
    if (fd >= 0)
195
 
    {
196
 
      close(fd);
197
 
    }
198
 
 
199
 
    if (pe != NULL)
200
 
    {
201
 
      pcre_free(pe);
202
 
    }
203
 
 
204
 
    if (re != NULL)
205
 
    {
206
 
      pcre_free(re);
207
 
    }
208
 
  }
209
 
 
210
 
  virtual bool post (Session *session)
211
 
  {
212
 
    size_t wrv;
213
 
 
214
 
    assert(session != NULL);
215
 
 
216
 
    if (fd < 0)
217
 
      return false;
218
 
 
219
 
    /* Yes, we know that checking sysvar_logging_query_enable,
220
 
       sysvar_logging_query_threshold_big_resultset, and
221
 
       sysvar_logging_query_threshold_big_examined is not threadsafe,
222
 
       because some other thread might change these sysvars.  But we
223
 
       don't care.  We might start logging a little late as it spreads
224
 
       to other threads.  Big deal. */
225
 
 
226
 
    // return if not enabled or query was too fast or resultset was too small
227
 
    if (sysvar_logging_query_enable == false)
228
 
      return false;
229
 
    if (session->sent_row_count < sysvar_logging_query_threshold_big_resultset.get())
230
 
      return false;
231
 
    if (session->examined_row_count < sysvar_logging_query_threshold_big_examined.get())
232
 
      return false;
233
 
 
234
 
    /*
235
 
      TODO, the session object should have a "utime command completed"
236
 
      inside itself, so be more accurate, and so this doesnt have to
237
 
      keep calling current_utime, which can be slow.
238
 
    */
239
 
    uint64_t t_mark= session->getCurrentTimestamp(false);
240
 
 
241
 
    if (session->getElapsedTime() < (sysvar_logging_query_threshold_slow.get()))
242
 
      return false;
243
 
 
244
 
    Session::QueryString query_string(session->getQueryString());
245
 
    if (re)
246
 
    {
247
 
      int this_pcre_rc;
248
 
      this_pcre_rc= pcre_exec(re, pe, query_string->c_str(), query_string->length(), 0, 0, NULL, 0);
249
 
      if (this_pcre_rc < 0)
250
 
        return false;
251
 
    }
252
 
 
253
 
    // buffer to quotify the query
254
 
    string qs;
255
 
    
256
 
    // Since quotify() builds the quoted string incrementally, we can
257
 
    // avoid some reallocating if we reserve some space up front.
258
 
    qs.reserve(query_string->length());
259
 
    
260
 
    quotify(*query_string, qs);
261
 
    
262
 
    // to avoid trying to printf %s something that is potentially NULL
263
 
    util::string::const_shared_ptr schema(session->schema());
264
 
    const char *dbs= (schema and not schema->empty()) ? schema->c_str() : "";
265
 
 
266
 
    formatter % t_mark
267
 
              % session->thread_id
268
 
              % session->getQueryId()
269
 
              % dbs
270
 
              % qs
271
 
              % getCommandName(session->command)
272
 
              % (t_mark - session->getConnectMicroseconds())
273
 
              % session->getElapsedTime()
274
 
              % (t_mark - session->utime_after_lock)
275
 
              % session->sent_row_count
276
 
              % session->examined_row_count
277
 
              % session->tmp_table
278
 
              % session->total_warn_count
279
 
              % session->getServerId()
280
 
              % getServerHostname();
281
 
 
282
 
    string msgbuf= formatter.str();
283
 
 
284
 
    // a single write has a kernel thread lock, thus no need mutex guard this
285
 
    wrv= write(fd, msgbuf.c_str(), msgbuf.length());
286
 
    assert(wrv == msgbuf.length());
287
 
 
288
 
    return false;
289
 
  }
 
160
      dst[dst_ndx++]= src[src_ndx];
 
161
    }
 
162
    dst[dst_ndx]= '\0';
 
163
  }
 
164
  return dst;
 
165
}
 
166
 
 
167
 
 
168
/* we could just not have a pre entrypoint at all,
 
169
   and have logging_pre == NULL
 
170
   but we have this here for the sake of being an example */
 
171
bool logging_query_func_pre (Session *)
 
172
{
 
173
  return false;
 
174
}
 
175
 
 
176
bool logging_query_func_post (Session *session)
 
177
{
 
178
  char msgbuf[MAX_MSG_LEN];
 
179
  int msgbuf_len= 0;
 
180
  int wrv;
 
181
 
 
182
  assert(session != NULL);
 
183
 
 
184
  if (fd < 0)
 
185
    return false;
 
186
 
 
187
  /* Yes, we know that checking sysvar_logging_query_enable,
 
188
     sysvar_logging_query_threshold_big_resultset, and
 
189
     sysvar_logging_query_threshold_big_examined is not threadsafe,
 
190
     because some other thread might change these sysvars.  But we
 
191
     don't care.  We might start logging a little late as it spreads
 
192
     to other threads.  Big deal. */
 
193
 
 
194
  // return if not enabled or query was too fast or resultset was too small
 
195
  if (sysvar_logging_query_enable == false)
 
196
    return false;
 
197
  if (session->sent_row_count < sysvar_logging_query_threshold_big_resultset)
 
198
    return false;
 
199
  if (session->examined_row_count < sysvar_logging_query_threshold_big_examined)
 
200
    return false;
 
201
 
 
202
  // logging this is far too verbose
 
203
  if (session->command == COM_FIELD_LIST)
 
204
    return false;
 
205
 
 
206
  /* TODO, looks like connect_utime isnt being set in the session
 
207
     object.  We could store the time this plugin was loaded, but that
 
208
     would just be a dumb workaround. */
 
209
  /* TODO, the session object should have a "utime command completed"
 
210
     inside itself, so be more accurate, and so this doesnt have to
 
211
     keep calling current_utime, which can be slow */
 
212
 
 
213
  uint64_t t_mark= get_microtime();
 
214
 
 
215
  if ((t_mark - session->start_utime) < (sysvar_logging_query_threshold_slow))
 
216
    return false;
 
217
 
 
218
  // buffer to quotify the query
 
219
  unsigned char qs[255];
 
220
 
 
221
  // to avoid trying to printf %s something that is potentially NULL
 
222
  const char *dbs= (session->db) ? session->db : "";
 
223
  int dbl= 0;
 
224
  if (dbs != NULL)
 
225
    dbl= session->db_length;
 
226
 
 
227
  msgbuf_len=
 
228
    snprintf(msgbuf, MAX_MSG_LEN,
 
229
             "%"PRIu64",%"PRIu64",%"PRIu64",\"%.*s\",\"%s\",\"%.*s\","
 
230
             "%"PRIu64",%"PRIu64",%"PRIu64",%"PRIu64",%"PRIu64"\n",
 
231
             t_mark,
 
232
             session->thread_id,
 
233
             session->query_id,
 
234
             // dont need to quote the db name, always CSV safe
 
235
             dbl, dbs,
 
236
             // do need to quote the query
 
237
             quotify((unsigned char *)session->query,
 
238
                     session->query_length, qs, sizeof(qs)),
 
239
             // command_name is defined in drizzled/sql_parse.cc
 
240
             // dont need to quote the command name, always CSV safe
 
241
             (int)command_name[session->command].length,
 
242
             command_name[session->command].str,
 
243
             // counters are at end, to make it easier to add more
 
244
             (t_mark - session->connect_utime),
 
245
             (t_mark - session->start_utime),
 
246
             (t_mark - session->utime_after_lock),
 
247
             session->sent_row_count,
 
248
             session->examined_row_count);
 
249
 
 
250
 
 
251
  // a single write has a kernel thread lock, thus no need mutex guard this
 
252
  wrv= write(fd, msgbuf, msgbuf_len);
 
253
  assert(wrv == msgbuf_len);
 
254
 
 
255
  return false;
 
256
}
 
257
 
 
258
static int logging_query_plugin_init(void *p)
 
259
{
 
260
  logging_t *l= static_cast<logging_t *>(p);
 
261
 
 
262
  if (sysvar_logging_query_filename == NULL)
 
263
  {
 
264
    /* no destination filename was specified via system variables
 
265
       return now, dont set the callback pointers
 
266
    */
 
267
    return 0;
 
268
  }
 
269
 
 
270
  fd= open(sysvar_logging_query_filename,
 
271
           O_WRONLY | O_APPEND | O_CREAT,
 
272
           S_IRUSR|S_IWUSR);
 
273
  if (fd < 0)
 
274
  {
 
275
    errmsg_printf(ERRMSG_LVL_ERROR, _("fail open() fn=%s er=%s\n"),
 
276
                  sysvar_logging_query_filename,
 
277
                  strerror(errno));
 
278
 
 
279
    /* TODO
 
280
       we should return an error here, so the plugin doesnt load
 
281
       but this causes Drizzle to crash
 
282
       so until that is fixed,
 
283
       just return a success,
 
284
       but leave the function pointers as NULL and the fd as -1
 
285
    */
 
286
    return 0;
 
287
  }
 
288
 
 
289
  l->logging_pre= logging_query_func_pre;
 
290
  l->logging_post= logging_query_func_post;
 
291
 
 
292
  return 0;
 
293
}
 
294
 
 
295
static int logging_query_plugin_deinit(void *p)
 
296
{
 
297
  logging_st *l= (logging_st *) p;
 
298
 
 
299
  if (fd >= 0)
 
300
  {
 
301
    close(fd);
 
302
    fd= -1;
 
303
  }
 
304
 
 
305
  l->logging_pre= NULL;
 
306
  l->logging_post= NULL;
 
307
 
 
308
  return 0;
 
309
}
 
310
 
 
311
static DRIZZLE_SYSVAR_BOOL(
 
312
  enable,
 
313
  sysvar_logging_query_enable,
 
314
  PLUGIN_VAR_NOCMDARG,
 
315
  N_("Enable logging to CSV file"),
 
316
  NULL, /* check func */
 
317
  NULL, /* update func */
 
318
  false /* default */);
 
319
 
 
320
static DRIZZLE_SYSVAR_STR(
 
321
  filename,
 
322
  sysvar_logging_query_filename,
 
323
  PLUGIN_VAR_READONLY,
 
324
  N_("File to log to"),
 
325
  NULL, /* check func */
 
326
  NULL, /* update func*/
 
327
  NULL /* default */);
 
328
 
 
329
static DRIZZLE_SYSVAR_ULONG(
 
330
  threshold_slow,
 
331
  sysvar_logging_query_threshold_slow,
 
332
  PLUGIN_VAR_OPCMDARG,
 
333
  N_("Threshold for logging slow queries, in microseconds"),
 
334
  NULL, /* check func */
 
335
  NULL, /* update func */
 
336
  0, /* default */
 
337
  0, /* min */
 
338
  UINT32_MAX, /* max */
 
339
  0 /* blksiz */);
 
340
 
 
341
static DRIZZLE_SYSVAR_ULONG(
 
342
  threshold_big_resultset,
 
343
  sysvar_logging_query_threshold_big_resultset,
 
344
  PLUGIN_VAR_OPCMDARG,
 
345
  N_("Threshold for logging big queries, for rows returned"),
 
346
  NULL, /* check func */
 
347
  NULL, /* update func */
 
348
  0, /* default */
 
349
  0, /* min */
 
350
  UINT32_MAX, /* max */
 
351
  0 /* blksiz */);
 
352
 
 
353
static DRIZZLE_SYSVAR_ULONG(
 
354
  threshold_big_examined,
 
355
  sysvar_logging_query_threshold_big_examined,
 
356
  PLUGIN_VAR_OPCMDARG,
 
357
  N_("Threshold for logging big queries, for rows examined"),
 
358
  NULL, /* check func */
 
359
  NULL, /* update func */
 
360
  0, /* default */
 
361
  0, /* min */
 
362
  UINT32_MAX, /* max */
 
363
  0 /* blksiz */);
 
364
 
 
365
static struct st_mysql_sys_var* logging_query_system_variables[]= {
 
366
  DRIZZLE_SYSVAR(enable),
 
367
  DRIZZLE_SYSVAR(filename),
 
368
  DRIZZLE_SYSVAR(threshold_slow),
 
369
  DRIZZLE_SYSVAR(threshold_big_resultset),
 
370
  DRIZZLE_SYSVAR(threshold_big_examined),
 
371
  NULL
290
372
};
291
373
 
292
 
static int logging_query_plugin_init(drizzled::module::Context &context)
293
 
{
294
 
 
295
 
  const module::option_map &vm= context.getOptions();
296
 
 
297
 
  if (vm.count("filename") > 0)
298
 
  {
299
 
    context.add(new Logging_query(vm["filename"].as<string>(),
300
 
                                  vm["pcre"].as<string>()));
301
 
    context.registerVariable(new sys_var_bool_ptr("enable", &sysvar_logging_query_enable));
302
 
    context.registerVariable(new sys_var_const_string_val("filename", vm["filename"].as<string>()));
303
 
    context.registerVariable(new sys_var_const_string_val("pcre", vm["pcre"].as<string>()));
304
 
    context.registerVariable(new sys_var_constrained_value<uint32_t>("threshold_slow", sysvar_logging_query_threshold_slow));
305
 
    context.registerVariable(new sys_var_constrained_value<uint32_t>("threshold_big_resultset", sysvar_logging_query_threshold_big_resultset));
306
 
    context.registerVariable(new sys_var_constrained_value<uint32_t>("threshold_big_examined", sysvar_logging_query_threshold_big_examined));
307
 
  }
308
 
 
309
 
  return 0;
310
 
}
311
 
 
312
 
static void init_options(drizzled::module::option_context &context)
313
 
{
314
 
  context("enable",
315
 
          po::value<bool>(&sysvar_logging_query_enable)->default_value(false)->zero_tokens(),
316
 
          _("Enable logging to CSV file"));
317
 
  context("filename",
318
 
          po::value<string>(),
319
 
          _("File to log to"));
320
 
  context("pcre",
321
 
          po::value<string>()->default_value(""),
322
 
          _("PCRE to match the query against"));
323
 
  context("threshold-slow",
324
 
          po::value<uint32_constraint>(&sysvar_logging_query_threshold_slow)->default_value(0),
325
 
          _("Threshold for logging slow queries, in microseconds"));
326
 
  context("threshold-big-resultset",
327
 
          po::value<uint32_constraint>(&sysvar_logging_query_threshold_big_resultset)->default_value(0),
328
 
          _("Threshold for logging big queries, for rows returned"));
329
 
  context("threshold-big-examined",
330
 
          po::value<uint32_constraint>(&sysvar_logging_query_threshold_big_examined)->default_value(0),
331
 
          _("Threshold for logging big queries, for rows examined"));
332
 
}
333
 
 
334
 
} /* namespace drizzle_plugin */
335
 
 
336
 
DRIZZLE_DECLARE_PLUGIN
337
 
{
338
 
  DRIZZLE_VERSION_ID,
339
 
  "logging-query",
 
374
drizzle_declare_plugin(logging_query)
 
375
{
 
376
  DRIZZLE_LOGGER_PLUGIN,
 
377
  "logging_query",
340
378
  "0.2",
341
379
  "Mark Atwood <mark@fallenpegasus.com>",
342
380
  N_("Log queries to a CSV file"),
343
381
  PLUGIN_LICENSE_GPL,
344
 
  drizzle_plugin::logging_query_plugin_init,
345
 
  NULL,
346
 
  drizzle_plugin::init_options
 
382
  logging_query_plugin_init,
 
383
  logging_query_plugin_deinit,
 
384
  NULL,   /* status variables */
 
385
  logging_query_system_variables,
 
386
  NULL
347
387
}
348
 
DRIZZLE_DECLARE_PLUGIN_END;
 
388
drizzle_declare_plugin_end;