#include "common.h"
#include "utils_cache.h"
#include "utils_parse_option.h"
+
#include <tcrdb.h>
#define DEFAULT_HOST "127.0.0.1"
plugin_dispatch_values (&vl);
}
+static void tt_open_db (void)
+{
+ char* host = NULL;
+ int port = DEFAULT_PORT;
+
+ if (rdb != NULL)
+ return;
+
+ host = ((config_host != NULL) ? config_host : DEFAULT_HOST);
+
+ if (config_port != NULL)
+ {
+ port = service_name_to_port_number (config_port);
+ if (port <= 0)
+ return;
+ }
+
+ rdb = tcrdbnew ();
+ if (rdb == NULL)
+ return;
+ else if (!tcrdbopen(rdb, host, port))
+ {
+ printerr ();
+ tcrdbdel (rdb);
+ rdb = NULL;
+ }
+} /* void tt_open_db */
+
static int tt_read (void) {
gauge_t rnum, size;
+ tt_open_db ();
+ if (rdb == NULL)
+ return (-1);
+
rnum = tcrdbrnum(rdb);
tt_submit (rnum, "records");
return (0);
}
-static int tt_init(void)
-{
- char* host = NULL;
- int port;
-
- host = ((config_host != NULL) ? config_host : DEFAULT_HOST);
- port = ((config_port != NULL) ? atoi(config_port) : DEFAULT_PORT);
-
- rdb = tcrdbnew();
-
- if (!tcrdbopen(rdb, host, port))
- {
- printerr ();
- tcrdbdel (rdb);
- return (1);
- }
-
- return(0);
-}
-
static int tt_shutdown(void)
{
- sfree(config_host);
- sfree(config_port);
+ sfree(config_host);
+ sfree(config_port);
- if (!tcrdbclose(rdb))
+ if (rdb != NULL)
{
- printerr ();
+ if (!tcrdbclose(rdb))
+ {
+ printerr ();
+ tcrdbdel (rdb);
+ return (1);
+ }
tcrdbdel (rdb);
- return (1);
+ rdb = NULL;
}
- tcrdbdel (rdb);
-
- return(0);
+ return(0);
}
void module_register (void)
plugin_register_config("tokyotyrant", tt_config,
config_keys, config_keys_num);
plugin_register_read("tokyotyrant", tt_read);
- plugin_register_init("tokyotyrant", tt_init);
- plugin_register_shutdown("tokyotyrant", tt_shutdown);
+ plugin_register_shutdown("tokyotyrant", tt_shutdown);
}
/* vim: set sw=8 ts=8 tw=78 : */