src/mysql.c

   1 /**
   2  * collectd - src/mysql.c
   3  * Copyright (C) 2005  Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; either version 2 of the License, or (at your
   8  * option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  18  *
  19  * Authors:
  20  *   Florian octo Forster <octo at verplant.org>
  21  **/
  22
  23 #include "collectd.h"
  24 #include "common.h"
  25 #include "plugin.h"
  26 #include "configfile.h"
  27
  28 #ifdef HAVE_MYSQL_MYSQL_H
  29 #include <mysql/mysql.h>
  30 #endif
  31
  32 #define MODULE_NAME "mysql"
  33
  34 #if COLLECT_LIBMYSQL
  35 # define MYSQL_HAVE_READ 1
  36 #else
  37 # define MYSQL_HAVE_READ 0
  38 #endif
  39
  40 #define BUFSIZE 512
  41
  42 static char *host;
  43 static char *user;
  44 static char *pass;
  45 static char *db = NULL;
  46
  47 static char  init_suceeded = 0;
  48
  49 static char *commands_file = "mysql_commands.rrd";
  50 static char *traffic_file  = "mysql_traffic.rrd";
  51
  52 static char *commands_ds_def[] =
  53 {
  54         "DS:insert:COUNTER:25:0:U",
  55         "DS:select:COUNTER:25:0:U",
  56         "DS:show:COUNTER:25:0:U",
  57         "DS:update:COUNTER:25:0:U",
  58         "DS:other:COUNTER:25:0:U",
  59         NULL
  60 };
  61 static int commands_ds_num = 5;
  62
  63 static char *traffic_ds_def[] =
  64 {
  65         "DS:incoming:COUNTER:25:0:U",
  66         "DS:outgoing:COUNTER:25:0:U",
  67         NULL
  68 };
  69 static int traffic_ds_num = 2;
  70
  71 static char *config_keys[] =
  72 {
  73         "Host",
  74         "User",
  75         "Password",
  76         "Database",
  77         NULL
  78 };
  79 static int config_keys_num = 4;
  80
  81 #if MYSQL_HAVE_READ
  82 static MYSQL *getconnection (void)
  83 {
  84         static MYSQL *con;
  85         static int    state;
  86
  87         if (state != 0)
  88         {
  89                 int err;
  90                 if ((err = mysql_ping (con)) != 0)
  91                 {
  92                         syslog (LOG_WARNING, "mysql_ping failed: %s", mysql_error (con));
  93                         state = 0;
  94                 }
  95                 else
  96                 {
  97                         state = 1;
  98                         return (con);
  99                 }
 100         }
 101
 102         if ((con = mysql_init (con)) == NULL)
 103         {
 104                 syslog (LOG_ERR, "mysql_init failed: %s", mysql_error (con));
 105                 state = 0;
 106                 return (NULL);
 107         }
 108
 109         if (mysql_real_connect (con, host, user, pass, db, 0, NULL, 0) == NULL)
 110         {
 111                 syslog (LOG_ERR, "mysql_real_connect failed: %s", mysql_error (con));
 112                 state = 0;
 113                 return (NULL);
 114         }
 115         else
 116         {
 117                 state = 1;
 118                 return (con);
 119         }
 120 } /* static MYSQL *getconnection (void) */
 121 #endif /* MYSQL_HAVE_READ */
 122
 123 static void init (void)
 124 {
 125 #if MYSQL_HAVE_READ
 126         if (getconnection () != NULL)
 127                 init_suceeded = 1;
 128         else
 129                 init_suceeded = 0;
 130 #endif /* MYSQL_HAVE_READ */
 131
 132         return;
 133 }
 134
 135 static int config (char *key, char *value)
 136 {
 137         if (strcasecmp (key, "host") == 0)
 138                 return ((host = strdup (value)) == NULL ? 1 : 0);
 139         else if (strcasecmp (key, "user") == 0)
 140                 return ((user = strdup (value)) == NULL ? 1 : 0);
 141         else if (strcasecmp (key, "password") == 0)
 142                 return ((pass = strdup (value)) == NULL ? 1 : 0);
 143         else if (strcasecmp (key, "database") == 0)
 144                 return ((db = strdup (value)) == NULL ? 1 : 0);
 145         else
 146                 return (-1);
 147 }
 148
 149 static void commands_write (char *host, char *inst, char *val)
 150 {
 151         rrd_update_file (host, commands_file, val, commands_ds_def, commands_ds_num);
 152 }
 153
 154 static void traffic_write (char *host, char *inst, char *val)
 155 {
 156         rrd_update_file (host, traffic_file, val, traffic_ds_def, traffic_ds_num);
 157 }
 158
 159 #if MYSQL_HAVE_READ
 160 static void commands_submit (unsigned long long insert,
 161                 unsigned long long select,
 162                 unsigned long long show,
 163                 unsigned long long update,
 164                 unsigned long long other)
 165 {
 166         char buf[BUFSIZE];
 167         int  status;
 168
 169         status = snprintf (buf, BUFSIZE, "%u:%llu:%llu:%llu:%llu:%llu", (unsigned int) curtime,
 170                         insert, select, show, update, other);
 171
 172         if (status < 0)
 173         {
 174                 syslog (LOG_ERR, "snprintf failed");
 175                 return;
 176         }
 177         else if (status >= BUFSIZE)
 178         {
 179                 syslog (LOG_WARNING, "snprintf was truncated");
 180                 return;
 181         }
 182
 183         plugin_submit ("mysql_commands", "-", buf);
 184 }
 185
 186 static void traffic_submit (unsigned long long incoming,
 187                 unsigned long long outgoing)
 188 {
 189         char buf[BUFSIZE];
 190         int  status;
 191
 192         status = snprintf (buf, BUFSIZE, "%u:%llu:%llu", (unsigned int) curtime,
 193                         incoming, outgoing);
 194
 195         if (status < 0)
 196         {
 197                 syslog (LOG_ERR, "snprintf failed");
 198                 return;
 199         }
 200         else if (status >= BUFSIZE)
 201         {
 202                 syslog (LOG_WARNING, "snprintf was truncated");
 203                 return;
 204         }
 205
 206         plugin_submit ("mysql_traffic", "-", buf);
 207 }
 208
 209 static void mysql_read (void)
 210 {
 211         MYSQL     *con;
 212         MYSQL_RES *res;
 213         MYSQL_ROW  row;
 214         char      *query;
 215         int        query_len;
 216         int        field_num;
 217
 218         unsigned long long cmd_insert = 0LL;
 219         unsigned long long cmd_select = 0LL;
 220         unsigned long long cmd_show   = 0LL;
 221         unsigned long long cmd_update = 0LL;
 222         unsigned long long cmd_other  = 0LL;
 223
 224         unsigned long long traffic_incoming = 0LL;
 225         unsigned long long traffic_outgoing = 0LL;
 226
 227         if (init_suceeded == 0)
 228                 return;
 229
 230         /* An error message will have been printed in this case */
 231         if ((con = getconnection ()) == NULL)
 232                 return;
 233
 234         query = "SHOW STATUS";
 235         query_len = strlen (query);
 236
 237         if (mysql_real_query (con, query, query_len))
 238         {
 239                 syslog (LOG_ERR, "mysql_real_query failed: %s\n", mysql_error (con));
 240                 return;
 241         }
 242
 243         if ((res = mysql_store_result (con)) == NULL)
 244         {
 245                 syslog (LOG_ERR, "mysql_store_result failed: %s\n", mysql_error (con));
 246                 return;
 247         }
 248
 249         field_num = mysql_num_fields (res);
 250         while ((row = mysql_fetch_row (res)))
 251         {
 252                 char *key;
 253                 unsigned long long val;
 254
 255                 key = row[0];
 256                 val = atoll (row[1]);
 257
 258                 if (strncmp (key, "Com_", 4) == 0)
 259                 {
 260                         if (strncmp (key, "Com_insert", 10) == 0)
 261                                 cmd_insert += val;
 262                         else if (strncmp (key, "Com_select", 10) == 0)
 263                                 cmd_select += val;
 264                         else if (strncmp (key, "Com_show", 8) == 0)
 265                                 cmd_show += val;
 266                         else if (strncmp (key, "Com_update", 10) == 0)
 267                                 cmd_update += val;
 268                         else if (strncmp (key, "Com_stmt_", 9) == 0)
 269                         { /* do nothing */ }
 270                         else
 271                                 cmd_other += val;
 272                 }
 273                 else if (strncmp (key, "Bytes_", 6) == 0)
 274                 {
 275                         if (strncmp (key, "Bytes_received", 14) == 0)
 276                                 traffic_incoming += val;
 277                         else if (strncmp (key, "Bytes_sent", 10) == 0)
 278                                 traffic_outgoing += val;
 279                 }
 280         }
 281         mysql_free_result (res); res = NULL;
 282
 283         commands_submit (cmd_insert, cmd_select, cmd_show, cmd_update, cmd_other);
 284         traffic_submit  (traffic_incoming, traffic_outgoing);
 285
 286         /* mysql_close (con); */
 287
 288         return;
 289 }
 290 #else
 291 # define mysql_read NULL
 292 #endif /* MYSQL_HAVE_READ */
 293
 294 void module_register (void)
 295 {
 296         plugin_register (MODULE_NAME, init, mysql_read, NULL);
 297         plugin_register ("mysql_commands", NULL, NULL, commands_write);
 298         plugin_register ("mysql_traffic", NULL, NULL, traffic_write);
 299         cf_register (MODULE_NAME, config, config_keys, config_keys_num);
 300 }
 301
 302 #undef BUFSIZE
 303 #undef MODULE_NAME