src/tokyotyrant.c

   1 /**
   2  * collectd - src/tokyotyrant.c
   3  * Copyright (C) 2009 Paul Sadauskas
   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; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Paul Sadauskas <psadauskas@gmail.com>
  20  **/
  21
  22 #include "collectd.h"
  23
  24 #include "common.h"
  25 #include "plugin.h"
  26 #include "utils_cache.h"
  27
  28 #include <tcrdb.h>
  29
  30 #define DEFAULT_HOST "127.0.0.1"
  31 #define DEFAULT_PORT 1978
  32
  33 static const char *config_keys[] = {"Host", "Port"};
  34 static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
  35
  36 static char *config_host = NULL;
  37 static char *config_port = NULL;
  38
  39 static TCRDB *rdb = NULL;
  40
  41 static int tt_config(const char *key, const char *value) {
  42   if (strcasecmp("Host", key) == 0) {
  43     char *temp;
  44
  45     temp = strdup(value);
  46     if (temp == NULL) {
  47       ERROR("tokyotyrant plugin: Host strdup failed.");
  48       return (1);
  49     }
  50     sfree(config_host);
  51     config_host = temp;
  52   } else if (strcasecmp("Port", key) == 0) {
  53     char *temp;
  54
  55     temp = strdup(value);
  56     if (temp == NULL) {
  57       ERROR("tokyotyrant plugin: Port strdup failed.");
  58       return (1);
  59     }
  60     sfree(config_port);
  61     config_port = temp;
  62   } else {
  63     ERROR("tokyotyrant plugin: error: unrecognized configuration key %s", key);
  64     return (-1);
  65   }
  66
  67   return (0);
  68 }
  69
  70 static void printerr(void) {
  71   int ecode = tcrdbecode(rdb);
  72   ERROR("tokyotyrant plugin: error: %d, %s", ecode, tcrdberrmsg(ecode));
  73 }
  74
  75 static void tt_submit(gauge_t val, const char *type) {
  76   value_t values[1];
  77   value_list_t vl = VALUE_LIST_INIT;
  78
  79   values[0].gauge = val;
  80
  81   vl.values = values;
  82   vl.values_len = STATIC_ARRAY_SIZE(values);
  83
  84   sstrncpy(vl.host, config_host, sizeof(vl.host));
  85   sstrncpy(vl.plugin, "tokyotyrant", sizeof(vl.plugin));
  86   sstrncpy(vl.plugin_instance, config_port, sizeof(vl.plugin_instance));
  87   sstrncpy(vl.type, type, sizeof(vl.type));
  88
  89   plugin_dispatch_values(&vl);
  90 }
  91
  92 static void tt_open_db(void) {
  93   const char *host;
  94   int port = DEFAULT_PORT;
  95
  96   if (rdb != NULL)
  97     return;
  98
  99   host = ((config_host != NULL) ? config_host : DEFAULT_HOST);
 100
 101   if (config_port != NULL) {
 102     port = service_name_to_port_number(config_port);
 103     if (port <= 0)
 104       return;
 105   }
 106
 107   rdb = tcrdbnew();
 108   if (rdb == NULL)
 109     return;
 110   else if (!tcrdbopen(rdb, host, port)) {
 111     printerr();
 112     tcrdbdel(rdb);
 113     rdb = NULL;
 114   }
 115 } /* void tt_open_db */
 116
 117 static int tt_read(void) {
 118   gauge_t rnum, size;
 119
 120   tt_open_db();
 121   if (rdb == NULL)
 122     return (-1);
 123
 124   rnum = tcrdbrnum(rdb);
 125   tt_submit(rnum, "records");
 126
 127   size = tcrdbsize(rdb);
 128   tt_submit(size, "file_size");
 129
 130   return (0);
 131 }
 132
 133 static int tt_shutdown(void) {
 134   sfree(config_host);
 135   sfree(config_port);
 136
 137   if (rdb != NULL) {
 138     if (!tcrdbclose(rdb)) {
 139       printerr();
 140       tcrdbdel(rdb);
 141       return (1);
 142     }
 143     tcrdbdel(rdb);
 144     rdb = NULL;
 145   }
 146
 147   return (0);
 148 }
 149
 150 void module_register(void) {
 151   plugin_register_config("tokyotyrant", tt_config, config_keys,
 152                          config_keys_num);
 153   plugin_register_read("tokyotyrant", tt_read);
 154   plugin_register_shutdown("tokyotyrant", tt_shutdown);
 155 }
 156
 157 /* vim: set sw=8 ts=8 tw=78 : */