← 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: Monty Taylor
  • Date: 2008-10-16 06:32:30 UTC
  • mto: (511.1.5 codestyle)
  • mto: This revision was merged to the branch mainline in revision 521.
  • Revision ID: monty@inaugust.com-20081016063230-4brxsra0qsmsg84q
Added -Wunused-macros.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/logging_query/logging_query.cc
 
1
/* drizzle/plugin/logging_query/logging_query.cc */
 
2
 
 
3
/* need to define DRIZZLE_SERVER to get inside the THD */
 
4
#define DRIZZLE_SERVER 1
 
5
#include <drizzled/server_includes.h>
 
6
#include <drizzled/plugin_logging.h>
 
7
 
 
8
#define MAX_MSG_LEN (32*1024)
 
9
 
 
10
static char* logging_query_filename= NULL;
 
11
 
 
12
static int fd= -1;
 
13
 
 
14
// copied from drizzled/sql_parse.cc
 
15
 
 
16
const LEX_STRING command_name[]={
 
17
  { C_STRING_WITH_LEN("Sleep") },
 
18
  { C_STRING_WITH_LEN("Quit") },
 
19
  { C_STRING_WITH_LEN("InitDB") },
 
20
  { C_STRING_WITH_LEN("Query") },
 
21
  { C_STRING_WITH_LEN("FieldList") },
 
22
  { C_STRING_WITH_LEN("CreateDB") },
 
23
  { C_STRING_WITH_LEN("DropDB") },
 
24
  { C_STRING_WITH_LEN("Refresh") },
 
25
  { C_STRING_WITH_LEN("Shutdown") },
 
26
  { C_STRING_WITH_LEN("Processlist") },
 
27
  { C_STRING_WITH_LEN("Connect") },
 
28
  { C_STRING_WITH_LEN("Kill") },
 
29
  { C_STRING_WITH_LEN("Ping") },
 
30
  { C_STRING_WITH_LEN("Time") },
 
31
  { C_STRING_WITH_LEN("ChangeUser") },
 
32
  { C_STRING_WITH_LEN("BinlogDump") },
 
33
  { C_STRING_WITH_LEN("ConnectOut") },
 
34
  { C_STRING_WITH_LEN("RegisterSlave") },
 
35
  { C_STRING_WITH_LEN("SetOption") },
 
36
  { C_STRING_WITH_LEN("Daemon") },
 
37
  { C_STRING_WITH_LEN("Error") }
 
38
};
 
39
 
 
40
 
 
41
bool logging_query_func_pre (THD *thd)
 
42
{
 
43
  char msgbuf[MAX_MSG_LEN];
 
44
  int msgbuf_len= 0;
 
45
  int wrv;
 
46
 
 
47
  if (fd < 0) 
 
48
    return false;
 
49
 
 
50
  assert(thd != NULL);
 
51
 
 
52
  msgbuf_len=
 
53
    snprintf(msgbuf, MAX_MSG_LEN,
 
54
             "log bgn thread_id=%ld query_id=%ld command=%.*s"
 
55
             " db=\"%.*s\" query=\"%.*s\"\n",
 
56
             (unsigned long) thd->thread_id,
 
57
             (unsigned long) thd->query_id,
 
58
             (uint32_t)command_name[thd->command].length, command_name[thd->command].str,
 
59
             thd->db_length, thd->db,
 
60
             thd->query_length, thd->query);
 
61
  /* a single write has a OS level thread lock
 
62
     so there is no need to have mutexes guarding this write,
 
63
  */
 
64
  wrv= write(fd, msgbuf, msgbuf_len);
 
65
  assert(wrv == msgbuf_len);
 
66
 
 
67
  return false;
 
68
}
 
69
 
 
70
bool logging_query_func_post (THD *thd)
 
71
{
 
72
  char msgbuf[MAX_MSG_LEN];
 
73
  int msgbuf_len= 0;
 
74
  int wrv;
 
75
 
 
76
  if (fd < 0) return false;
 
77
 
 
78
  assert(thd != NULL);
 
79
 
 
80
  msgbuf_len=
 
81
    snprintf(msgbuf, MAX_MSG_LEN,
 
82
             "log end thread_id=%ld query_id=%ld command=%.*s"
 
83
             " rows.sent=%ld rows.exam=%u\n",
 
84
             (unsigned long) thd->thread_id, 
 
85
             (unsigned long) thd->query_id,
 
86
             (uint32_t)command_name[thd->command].length, command_name[thd->command].str,
 
87
             (unsigned long) thd->sent_row_count,
 
88
             (uint32_t) thd->examined_row_count);
 
89
  /* a single write has a OS level thread lock
 
90
     so there is no need to have mutexes guarding this write,
 
91
  */
 
92
  wrv= write(fd, msgbuf, msgbuf_len);
 
93
  assert(wrv == msgbuf_len);
 
94
 
 
95
 
 
96
  return false;
 
97
}
 
98
 
 
99
static int logging_query_plugin_init(void *p)
 
100
{
 
101
  logging_t *l= (logging_t *) p;
 
102
 
 
103
  if (logging_query_filename == NULL)
 
104
  {
 
105
    /* no destination filename was specified via system variables
 
106
       return now, dont set the callback pointers 
 
107
    */
 
108
    return 0;
 
109
  }
 
110
 
 
111
  fd= open(logging_query_filename, O_WRONLY | O_APPEND | O_CREAT);
 
112
  if (fd < 0) 
 
113
  {
 
114
    fprintf(stderr, "fail open fn=%s er=%s\n",
 
115
            logging_query_filename,
 
116
            strerror(errno));
 
117
 
 
118
    /* we should return an error here, so the plugin doesnt load
 
119
       but this causes Drizzle to crash
 
120
       so until that is fixed,
 
121
       just return a success,
 
122
       but leave the function pointers as NULL and the fd as -1
 
123
    */
 
124
    return 0;
 
125
  }
 
126
 
 
127
  l->logging_pre= logging_query_func_pre;
 
128
  l->logging_post= logging_query_func_post;
 
129
 
 
130
  return 0;
 
131
}
 
132
 
 
133
static int logging_query_plugin_deinit(void *p)
 
134
{
 
135
  logging_st *l= (logging_st *) p;
 
136
 
 
137
  if (fd >= 0) 
 
138
  {
 
139
    close(fd);
 
140
    fd= -1;
 
141
  }
 
142
 
 
143
  l->logging_pre= NULL;
 
144
  l->logging_post= NULL;
 
145
 
 
146
  return 0;
 
147
}
 
148
 
 
149
static DRIZZLE_SYSVAR_STR(filename, logging_query_filename,
 
150
  PLUGIN_VAR_READONLY,
 
151
  "File to log queries to.",
 
152
  NULL, NULL, NULL);
 
153
 
 
154
static struct st_mysql_sys_var* logging_query_system_variables[]= {
 
155
  DRIZZLE_SYSVAR(filename),
 
156
  NULL
 
157
};
 
158
 
 
159
mysql_declare_plugin(logging_query)
 
160
{
 
161
  DRIZZLE_LOGGER_PLUGIN,
 
162
  "logging_query",
 
163
  "0.1",
 
164
  "Mark Atwood <mark@fallenpegasus.com>",
 
165
  "Log queries",
 
166
  PLUGIN_LICENSE_GPL,
 
167
  logging_query_plugin_init,
 
168
  logging_query_plugin_deinit,
 
169
  NULL,   /* status variables */
 
170
  logging_query_system_variables,
 
171
  NULL
 
172
}
 
173
mysql_declare_plugin_end;