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 **/
22 #include "collectd.h"
24 #include "common.h"
25 #include "plugin.h"
26 #include "utils_cache.h"
28 #include <tcrdb.h>
30 #define DEFAULT_HOST "127.0.0.1"
31 #define DEFAULT_PORT 1978
33 static const char *config_keys[] = {"Host", "Port"};
34 static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
36 static char *config_host = NULL;
37 static char *config_port = NULL;
39 static TCRDB *rdb = NULL;
41 static int tt_config(const char *key, const char *value) {
42 if (strcasecmp("Host", key) == 0) {
43 char *temp;
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;
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 }
67 return (0);
68 }
70 static void printerr(void) {
71 int ecode = tcrdbecode(rdb);
72 ERROR("tokyotyrant plugin: error: %d, %s", ecode, tcrdberrmsg(ecode));
73 }
75 static void tt_submit(gauge_t val, const char *type) {
76 value_t values[1];
77 value_list_t vl = VALUE_LIST_INIT;
79 values[0].gauge = val;
81 vl.values = values;
82 vl.values_len = STATIC_ARRAY_SIZE(values);
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));
89 plugin_dispatch_values(&vl);
90 }
92 static void tt_open_db(void) {
93 const char *host;
94 int port = DEFAULT_PORT;
96 if (rdb != NULL)
97 return;
99 host = ((config_host != NULL) ? config_host : DEFAULT_HOST);
101 if (config_port != NULL) {
102 port = service_name_to_port_number(config_port);
103 if (port <= 0)
104 return;
105 }
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 */
117 static int tt_read(void) {
118 gauge_t rnum, size;
120 tt_open_db();
121 if (rdb == NULL)
122 return (-1);
124 rnum = tcrdbrnum(rdb);
125 tt_submit(rnum, "records");
127 size = tcrdbsize(rdb);
128 tt_submit(size, "file_size");
130 return (0);
131 }
133 static int tt_shutdown(void) {
134 sfree(config_host);
135 sfree(config_port);
137 if (rdb != NULL) {
138 if (!tcrdbclose(rdb)) {
139 printerr();
140 tcrdbdel(rdb);
141 return (1);
142 }
143 tcrdbdel(rdb);
144 rdb = NULL;
145 }
147 return (0);
148 }
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 }
157 /* vim: set sw=8 ts=8 tw=78 : */