Merge branch 'master' into collectd-4
[collectd.git] / src / mysql.c
index 8400a56..a2604e4 100644 (file)
@@ -1,11 +1,10 @@
 /**
  * collectd - src/mysql.c
- * Copyright (C) 2006  Florian octo Forster
+ * Copyright (C) 2006,2007  Florian octo Forster
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License as published by the
- * Free Software Foundation; either version 2 of the License, or (at your
- * option) any later version.
+ * Free Software Foundation; only version 2 of the License is applicable.
  *
  * This program is distributed in the hope that it will be useful, but
  * WITHOUT ANY WARRANTY; without even the implied warranty of
 #include <mysql/mysql.h>
 #endif
 
-#define MODULE_NAME "mysql"
-
 #if COLLECT_LIBMYSQL
 # define MYSQL_HAVE_READ 1
 #else
 # define MYSQL_HAVE_READ 0
 #endif
 
-#define BUFSIZE 512
+/* TODO: Understand `Select_*' and possibly do that stuff as well.. */
 
-static char *host = "localhost";
-static char *user;
-static char *pass;
-static char *db = NULL;
+static data_source_t data_source_counter[1] =
+{
+       {"value", DS_TYPE_COUNTER, 0, NAN}
+};
 
-#if MYSQL_HAVE_READ
-static char  init_suceeded = 0;
-#endif
+static data_set_t ds_commands =
+{
+       "mysql_commands", 1, data_source_counter
+};
 
-/* TODO
- * understand `Select_*' and possibly do that stuff as well..
- */
+static data_set_t ds_handler =
+{
+       "mysql_handler", 1, data_source_counter
+};
 
-static char *commands_file = "mysql/mysql_commands-%s.rrd";
-static char *handler_file  = "mysql/mysql_handler-%s.rrd";
-static char *qcache_file   = "mysql/mysql_qcache.rrd";
-static char *threads_file  = "mysql/mysql_threads.rrd";
-static char *traffic_file  = "traffic-mysql.rrd";
+static data_source_t data_source_qcache[5] =
+{
+       {"hits",             DS_TYPE_COUNTER, 0, NAN},
+       {"inserts",          DS_TYPE_COUNTER, 0, NAN},
+       {"not_cached",       DS_TYPE_COUNTER, 0, NAN},
+       {"lowmem_prunes",    DS_TYPE_COUNTER, 0, NAN},
+       {"queries_in_cache", DS_TYPE_GAUGE,   0, NAN}
+};
 
-static char *commands_ds_def[] =
+static data_set_t ds_qcache =
 {
-       "DS:value:COUNTER:25:0:U",
-       NULL
+       "mysql_qcache", 5, data_source_qcache
 };
-static int commands_ds_num = 1;
 
-static char *handler_ds_def[] =
+static data_source_t data_source_threads[4] =
 {
-       "DS:value:COUNTER:25:0:U",
-       NULL
+       {"running",   DS_TYPE_GAUGE,   0, NAN},
+       {"connected", DS_TYPE_GAUGE,   0, NAN},
+       {"cached",    DS_TYPE_GAUGE,   0, NAN},
+       {"created",   DS_TYPE_COUNTER, 0, NAN}
 };
-static int handler_ds_num = 1;
 
-static char *qcache_ds_def[] =
+static data_set_t ds_threads =
 {
-       "DS:hits:COUNTER:25:0:U",
-       "DS:inserts:COUNTER:25:0:U",
-       "DS:not_cached:COUNTER:25:0:U",
-       "DS:queries_in_cache:GAUGE:25:0:U",
-       NULL
+       "mysql_threads", 4, data_source_threads
 };
-static int qcache_ds_num = 4;
 
-static char *threads_ds_def[] =
+static data_source_t data_source_octets[2] =
 {
-       "DS:running:GAUGE:25:0:U",
-       "DS:connected:GAUGE:25:0:U",
-       "DS:cached:GAUGE:25:0:U",
-       "DS:created:COUNTER:25:0:u",
-       NULL
+       {"rx", DS_TYPE_COUNTER, 0, 4294967295.0},
+       {"tx", DS_TYPE_COUNTER, 0, 4294967295.0}
 };
-static int threads_ds_num = 4;
 
-static char *traffic_ds_def[] =
+static data_set_t ds_octets =
 {
-       "DS:incoming:COUNTER:25:0:U",
-       "DS:outgoing:COUNTER:25:0:U",
-       NULL
+       "mysql_octets", 2, data_source_octets
 };
-static int traffic_ds_num = 2;
 
-static char *config_keys[] =
+#if MYSQL_HAVE_READ
+static const char *config_keys[] =
 {
        "Host",
        "User",
@@ -110,12 +100,21 @@ static char *config_keys[] =
 };
 static int config_keys_num = 4;
 
-#if MYSQL_HAVE_READ
+static char *host = "localhost";
+static char *user;
+static char *pass;
+static char *db = NULL;
+
 static MYSQL *getconnection (void)
 {
        static MYSQL *con;
        static int    state;
 
+       static int wait_for = 0;
+       static int wait_increase = 60;
+
+       int step;
+
        if (state != 0)
        {
                int err;
@@ -131,6 +130,19 @@ static MYSQL *getconnection (void)
                }
        }
 
+       step = atoi (COLLECTD_STEP);
+
+       if (wait_for > 0)
+       {
+               wait_for -= step;
+               return (NULL);
+       }
+
+       wait_for = wait_increase;
+       wait_increase *= 2;
+       if (wait_increase > 86400)
+               wait_increase = 86400;
+
        if ((con = mysql_init (con)) == NULL)
        {
                syslog (LOG_ERR, "mysql_init failed: %s", mysql_error (con));
@@ -147,27 +159,13 @@ static MYSQL *getconnection (void)
        else
        {
                state = 1;
+               wait_for = 0;
+               wait_increase = 60;
                return (con);
        }
 } /* static MYSQL *getconnection (void) */
-#endif /* MYSQL_HAVE_READ */
 
-static void init (void)
-{
-#if MYSQL_HAVE_READ
-       if (getconnection () != NULL)
-               init_suceeded = 1;
-       else
-       {
-               syslog (LOG_ERR, "The `mysql' plugin will be disabled because `init' failed to connect to `%s'", host);
-               init_suceeded = 0;
-       }
-#endif /* MYSQL_HAVE_READ */
-
-       return;
-}
-
-static int config (char *key, char *value)
+static int config (const char *key, const char *value)
 {
        if (strcasecmp (key, "host") == 0)
                return ((host = strdup (value)) == NULL ? 1 : 0);
@@ -181,159 +179,84 @@ static int config (char *key, char *value)
                return (-1);
 }
 
-static void commands_write (char *host, char *inst, char *val)
+static void counter_submit (const char *type, const char *type_instance,
+               counter_t value)
 {
-       char buf[BUFSIZE];
+       value_t values[1];
+       value_list_t vl = VALUE_LIST_INIT;
 
-       if (snprintf (buf, BUFSIZE, commands_file, inst) >= BUFSIZE)
-               return;
+       values[0].counter = value;
 
-       rrd_update_file (host, buf, val, commands_ds_def, commands_ds_num);
-}
+       vl.values = values;
+       vl.values_len = 1;
+       vl.time = time (NULL);
+       strcpy (vl.host, hostname_g);
+       strcpy (vl.plugin, "mysql");
+       strncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
 
-static void handler_write (char *host, char *inst, char *val)
-{
-       char buf[BUFSIZE];
-
-       if (snprintf (buf, BUFSIZE, handler_file, inst) >= BUFSIZE)
-               return;
+       plugin_dispatch_values (type, &vl);
+} /* void counter_submit */
 
-       rrd_update_file (host, buf, val, handler_ds_def, handler_ds_num);
-}
-
-static void threads_write (char *host, char *inst, char *val)
+static void qcache_submit (counter_t hits, counter_t inserts,
+               counter_t not_cached, counter_t lowmem_prunes,
+               gauge_t queries_in_cache)
 {
-       rrd_update_file (host, threads_file, val,
-                       threads_ds_def, threads_ds_num);
-}
-
-static void traffic_write (char *host, char *inst, char *val)
+       value_t values[5];
+       value_list_t vl = VALUE_LIST_INIT;
+
+       values[0].counter = hits;
+       values[1].counter = inserts;
+       values[2].counter = not_cached;
+       values[3].counter = lowmem_prunes;
+       values[4].gauge   = queries_in_cache;
+
+       vl.values = values;
+       vl.values_len = 5;
+       vl.time = time (NULL);
+       strcpy (vl.host, hostname_g);
+       strcpy (vl.plugin, "mysql");
+
+       plugin_dispatch_values ("mysql_qcache", &vl);
+} /* void qcache_submit */
+
+static void threads_submit (gauge_t running, gauge_t connected, gauge_t cached,
+               counter_t created)
 {
-       rrd_update_file (host, traffic_file, val,
-                       traffic_ds_def, traffic_ds_num);
-}
+       value_t values[4];
+       value_list_t vl = VALUE_LIST_INIT;
 
-static void qcache_write (char *host, char *inst, char *val)
-{
-       rrd_update_file (host, qcache_file, val,
-                       qcache_ds_def, qcache_ds_num);
-}
+       values[0].gauge   = running;
+       values[1].gauge   = connected;
+       values[2].gauge   = cached;
+       values[3].counter = created;
 
-#if MYSQL_HAVE_READ
-static void commands_submit (char *inst, unsigned long long value)
-{
-       char buf[BUFSIZE];
-       int  status;
-
-       status = snprintf (buf, BUFSIZE, "%u:%llu", (unsigned int) curtime, value);
-
-       if (status < 0)
-       {
-               syslog (LOG_ERR, "snprintf failed");
-               return;
-       }
-       else if (status >= BUFSIZE)
-       {
-               syslog (LOG_WARNING, "snprintf was truncated");
-               return;
-       }
-
-       plugin_submit ("mysql_commands", inst, buf);
-}
-
-static void handler_submit (char *inst, unsigned long long value)
-{
-       char buf[BUFSIZE];
-       int  status;
-
-       status = snprintf (buf, BUFSIZE, "%u:%llu", (unsigned int) curtime, value);
-
-       if (status < 0)
-       {
-               syslog (LOG_ERR, "snprintf failed");
-               return;
-       }
-       else if (status >= BUFSIZE)
-       {
-               syslog (LOG_WARNING, "snprintf was truncated");
-               return;
-       }
-
-       plugin_submit ("mysql_handler", inst, buf);
-}
-
-static void qcache_submit (unsigned long long hits, unsigned long long inserts,
-               unsigned long long not_cached, int queries_in_cache)
-{
-       char buf[BUFSIZE];
-       int  status;
-
-       status = snprintf (buf, BUFSIZE, "%u:%llu:%llu:%llu:%i",
-                       (unsigned int) curtime, hits, inserts, not_cached,
-                       queries_in_cache);
-
-       if (status < 0)
-       {
-               syslog (LOG_ERR, "snprintf failed");
-               return;
-       }
-       else if (status >= BUFSIZE)
-       {
-               syslog (LOG_WARNING, "snprintf was truncated");
-               return;
-       }
+       vl.values = values;
+       vl.values_len = 4;
+       vl.time = time (NULL);
+       strcpy (vl.host, hostname_g);
+       strcpy (vl.plugin, "mysql");
 
-       plugin_submit ("mysql_qcache", "-", buf);
-}
+       plugin_dispatch_values ("mysql_threads", &vl);
+} /* void threads_submit */
 
-static void threads_submit (int running, int connected, int cached,
-               unsigned long long created)
+static void traffic_submit (counter_t rx, counter_t tx)
 {
-       char buf[BUFSIZE];
-       int  status;
+       value_t values[2];
+       value_list_t vl = VALUE_LIST_INIT;
 
-       status = snprintf (buf, BUFSIZE, "%u:%i:%i:%i:%llu",
-                       (unsigned int) curtime,
-                       running, connected, cached, created);
+       values[0].counter = rx;
+       values[1].counter = tx;
 
-       if (status < 0)
-       {
-               syslog (LOG_ERR, "snprintf failed");
-               return;
-       }
-       else if (status >= BUFSIZE)
-       {
-               syslog (LOG_WARNING, "snprintf was truncated");
-               return;
-       }
+       vl.values = values;
+       vl.values_len = 2;
+       vl.time = time (NULL);
+       strcpy (vl.host, hostname_g);
+       strcpy (vl.plugin, "mysql");
 
-       plugin_submit ("mysql_threads", "-", buf);
-}
+       plugin_dispatch_values ("mysql_octets", &vl);
+} /* void traffic_submit */
 
-static void traffic_submit (unsigned long long incoming,
-               unsigned long long outgoing)
-{
-       char buf[BUFSIZE];
-       int  status;
-
-       status = snprintf (buf, BUFSIZE, "%u:%llu:%llu", (unsigned int) curtime,
-                       incoming, outgoing);
-
-       if (status < 0)
-       {
-               syslog (LOG_ERR, "snprintf failed");
-               return;
-       }
-       else if (status >= BUFSIZE)
-       {
-               syslog (LOG_WARNING, "snprintf was truncated");
-               return;
-       }
-
-       plugin_submit ("mysql_traffic", "-", buf);
-}
-
-static void mysql_read (void)
+static int mysql_read (void)
 {
        MYSQL     *con;
        MYSQL_RES *res;
@@ -342,9 +265,10 @@ static void mysql_read (void)
        int        query_len;
        int        field_num;
 
-       unsigned long long qcache_hits       = 0ULL;
-       unsigned long long qcache_inserts    = 0ULL;
-       unsigned long long qcache_not_cached = 0ULL;
+       unsigned long long qcache_hits          = 0ULL;
+       unsigned long long qcache_inserts       = 0ULL;
+       unsigned long long qcache_not_cached    = 0ULL;
+       unsigned long long qcache_lowmem_prunes = 0ULL;
        int qcache_queries_in_cache = -1;
 
        int threads_running   = -1;
@@ -355,28 +279,28 @@ static void mysql_read (void)
        unsigned long long traffic_incoming = 0ULL;
        unsigned long long traffic_outgoing = 0ULL;
 
-       if (init_suceeded == 0)
-               return;
-
        /* An error message will have been printed in this case */
        if ((con = getconnection ()) == NULL)
-               return;
+               return (-1);
 
        query = "SHOW STATUS";
+       if (mysql_get_server_version (con) >= 50002)
+               query = "SHOW GLOBAL STATUS";
+
        query_len = strlen (query);
 
        if (mysql_real_query (con, query, query_len))
        {
                syslog (LOG_ERR, "mysql_real_query failed: %s\n",
                                mysql_error (con));
-               return;
+               return (-1);
        }
 
        if ((res = mysql_store_result (con)) == NULL)
        {
                syslog (LOG_ERR, "mysql_store_result failed: %s\n",
                                mysql_error (con));
-               return;
+               return (-1);
        }
 
        field_num = mysql_num_fields (res);
@@ -395,14 +319,14 @@ static void mysql_read (void)
 
                        /* Ignore `prepared statements' */
                        if (strncmp (key, "Com_stmt_", 9) != 0)
-                               commands_submit (key + 4, val);
+                               counter_submit ("mysql_commands", key + 4, val);
                }
                else if (strncmp (key, "Handler_", 8) == 0)
                {
                        if (val == 0ULL)
                                continue;
 
-                       handler_submit (key + 8, val);
+                       counter_submit ("mysql_handler", key + 8, val);
                }
                else if (strncmp (key, "Qcache_", 7) == 0)
                {
@@ -412,6 +336,8 @@ static void mysql_read (void)
                                qcache_inserts = val;
                        else if (strcmp (key, "Qcache_not_cached") == 0)
                                qcache_not_cached = val;
+                       else if (strcmp (key, "Qcache_lowmem_prunes") == 0)
+                               qcache_lowmem_prunes = val;
                        else if (strcmp (key, "Qcache_queries_in_cache") == 0)
                                qcache_queries_in_cache = (int) val;
                }
@@ -438,9 +364,10 @@ static void mysql_read (void)
 
        if ((qcache_hits != 0ULL)
                        || (qcache_inserts != 0ULL)
-                       || (qcache_not_cached != 0ULL))
+                       || (qcache_not_cached != 0ULL)
+                       || (qcache_lowmem_prunes != 0ULL))
                qcache_submit (qcache_hits, qcache_inserts, qcache_not_cached,
-                               qcache_queries_in_cache);
+                               qcache_lowmem_prunes, qcache_queries_in_cache);
 
        if (threads_created != 0ULL)
                threads_submit (threads_running, threads_connected,
@@ -450,22 +377,20 @@ static void mysql_read (void)
 
        /* mysql_close (con); */
 
-       return;
-}
-#else
-# define mysql_read NULL
+       return (0);
+} /* int mysql_read */
 #endif /* MYSQL_HAVE_READ */
 
 void module_register (void)
 {
-       plugin_register (MODULE_NAME, init, mysql_read, NULL);
-       plugin_register ("mysql_commands", NULL, NULL, commands_write);
-       plugin_register ("mysql_handler",  NULL, NULL, handler_write);
-       plugin_register ("mysql_qcache",   NULL, NULL, qcache_write);
-       plugin_register ("mysql_threads",  NULL, NULL, threads_write);
-       plugin_register ("mysql_traffic",  NULL, NULL, traffic_write);
-       cf_register (MODULE_NAME, config, config_keys, config_keys_num);
-}
+       plugin_register_data_set (&ds_commands);
+       plugin_register_data_set (&ds_handler);
+       plugin_register_data_set (&ds_qcache);
+       plugin_register_data_set (&ds_threads);
+       plugin_register_data_set (&ds_octets);
 
-#undef BUFSIZE
-#undef MODULE_NAME
+#if MYSQL_HAVE_READ
+       plugin_register_config ("mysql", config, config_keys, config_keys_num);
+       plugin_register_read ("mysql", mysql_read);
+#endif /* MYSQL_HAVE_READ */
+} /* void module_register */