X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Fcore%2Fplugin.c;h=bf0bdb34a20af428c73ab63250c7b0f4d3f4c00b;hb=79ba88585625806862fed2399256692c9469d630;hp=20fe61cd523dfe6db6229e3996e332a13cc7ca5e;hpb=49b5a4d2e8e4fb1e4f67c2a368d8d2e3e76b765f;p=sysdb.git diff --git a/src/core/plugin.c b/src/core/plugin.c index 20fe61c..bf0bdb3 100644 --- a/src/core/plugin.c +++ b/src/core/plugin.c @@ -84,6 +84,8 @@ typedef struct { #define SDB_PLUGIN_CB_INIT { SDB_OBJECT_INIT, \ /* callback = */ NULL, /* user_data = */ NULL, \ SDB_PLUGIN_CTX_INIT } +#define SDB_PLUGIN_CB(obj) ((sdb_plugin_cb_t *)(obj)) +#define SDB_CONST_PLUGIN_CB(obj) ((const sdb_plugin_cb_t *)(obj)) typedef struct { sdb_plugin_cb_t super; @@ -93,12 +95,17 @@ typedef struct { sdb_time_t ccb_interval; sdb_time_t ccb_next_update; } sdb_plugin_collector_cb_t; - -#define SDB_PLUGIN_CB(obj) ((sdb_plugin_cb_t *)(obj)) -#define SDB_CONST_PLUGIN_CB(obj) ((const sdb_plugin_cb_t *)(obj)) #define SDB_PLUGIN_CCB(obj) ((sdb_plugin_collector_cb_t *)(obj)) #define SDB_CONST_PLUGIN_CCB(obj) ((const sdb_plugin_collector_cb_t *)(obj)) +typedef struct { + sdb_plugin_cb_t super; /* cb_callback will always be NULL */ +#define w_user_data super.cb_user_data +#define w_ctx super.cb_ctx + sdb_store_writer_t impl; +} sdb_plugin_writer_t; +#define SDB_PLUGIN_WRITER(obj) ((sdb_plugin_writer_t *)(obj)) + /* * private variables */ @@ -119,17 +126,20 @@ static sdb_llist_t *cname_list = NULL; static sdb_llist_t *shutdown_list = NULL; static sdb_llist_t *log_list = NULL; static sdb_llist_t *ts_fetcher_list = NULL; +static sdb_llist_t *writer_list = NULL; static struct { const char *type; sdb_llist_t **list; } all_lists[] = { - { "config", &config_list }, - { "init", &init_list }, - { "collector", &collector_list }, - { "cname", &cname_list }, - { "shutdown", &shutdown_list }, - { "log", &log_list }, + { "config", &config_list }, + { "init", &init_list }, + { "collector", &collector_list }, + { "cname", &cname_list }, + { "shutdown", &shutdown_list }, + { "log", &log_list }, + { "timeseries fetcher", &ts_fetcher_list }, + { "store writer", &writer_list }, }; /* @@ -241,26 +251,6 @@ plugin_unregister_by_name(const char *plugin_name) /* else: other callbacks still reference it */ } /* plugin_unregister_by_name */ -static void -plugin_unregister_all(void) -{ - size_t i; - - for (i = 0; i < SDB_STATIC_ARRAY_LEN(all_lists); ++i) { - const char *type = all_lists[i].type; - sdb_llist_t *list = *all_lists[i].list; - - size_t len = sdb_llist_len(list); - - if (! len) - continue; - - sdb_llist_clear(list); - sdb_log(SDB_LOG_INFO, "core: Unregistered %zu %s callback%s", - len, type, len == 1 ? "" : "s"); - } -} /* plugin_unregister_all */ - /* * private types */ @@ -399,6 +389,55 @@ static sdb_type_t sdb_plugin_collector_cb_type = { plugin_cb_destroy }; +static int +plugin_writer_init(sdb_object_t *obj, va_list ap) +{ + sdb_store_writer_t *impl = va_arg(ap, sdb_store_writer_t *); + sdb_object_t *ud = va_arg(ap, sdb_object_t *); + + assert(impl); + + if ((! impl->store_host) || (! impl->store_service) + || (! impl->store_metric) || (! impl->store_attribute) + || (! impl->store_service_attr) || (! impl->store_metric_attr)) { + sdb_log(SDB_LOG_ERR, "core: store writer callback '%s' " + "does not fully implement the writer interface.", + obj->name); + return -1; + } + if (sdb_llist_search_by_name(writer_list, obj->name)) { + sdb_log(SDB_LOG_WARNING, "core: store writer callback '%s' " + "has already been registered. Ignoring newly " + "registered version.", obj->name); + return -1; + } + + /* ctx may be NULL if the plugin was not registered by a plugin */ + + SDB_PLUGIN_WRITER(obj)->impl = *impl; + SDB_PLUGIN_WRITER(obj)->w_ctx = ctx_get(); + sdb_object_ref(SDB_OBJ(SDB_PLUGIN_WRITER(obj)->w_ctx)); + + sdb_object_ref(ud); + SDB_PLUGIN_WRITER(obj)->w_user_data = ud; + return 0; +} /* plugin_writer_init */ + +static void +plugin_writer_destroy(sdb_object_t *obj) +{ + assert(obj); + sdb_object_deref(SDB_PLUGIN_WRITER(obj)->w_user_data); + sdb_object_deref(SDB_OBJ(SDB_PLUGIN_WRITER(obj)->w_ctx)); +} /* plugin_writer_destroy */ + +static sdb_type_t sdb_plugin_writer_type = { + sizeof(sdb_plugin_writer_t), + + plugin_writer_init, + plugin_writer_destroy +}; + static int module_init(const char *name, lt_dlhandle lh, sdb_plugin_info_t *info) { @@ -683,7 +722,7 @@ sdb_plugin_register_config(sdb_plugin_config_cb callback) "config callback from outside a plugin"); return -1; } - return plugin_add_callback(&config_list, "init", ctx->info.plugin_name, + return plugin_add_callback(&config_list, "config", ctx->info.plugin_name, (void *)callback, NULL); } /* sdb_plugin_register_config */ @@ -795,6 +834,61 @@ sdb_plugin_register_ts_fetcher(const char *name, name, callback, user_data); } /* sdb_plugin_register_ts_fetcher */ +int +sdb_plugin_register_writer(const char *name, + sdb_store_writer_t *writer, sdb_object_t *user_data) +{ + char cb_name[1024]; + sdb_object_t *obj; + + if ((! name) || (! writer)) + return -1; + + if (! writer_list) + writer_list = sdb_llist_create(); + if (! writer_list) + return -1; + + plugin_get_name(name, cb_name, sizeof(cb_name)); + + obj = sdb_object_create(cb_name, sdb_plugin_writer_type, + writer, user_data); + if (! obj) + return -1; + + if (sdb_llist_append(writer_list, obj)) { + sdb_object_deref(obj); + return -1; + } + + /* pass control to the list */ + sdb_object_deref(obj); + + sdb_log(SDB_LOG_INFO, "core: Registered store writer callback '%s'.", + cb_name); + return 0; +} /* sdb_store_register_writer */ + +void +sdb_plugin_unregister_all(void) +{ + size_t i; + + for (i = 0; i < SDB_STATIC_ARRAY_LEN(all_lists); ++i) { + const char *type = all_lists[i].type; + sdb_llist_t *list = *all_lists[i].list; + + size_t len = sdb_llist_len(list); + + if (! len) + continue; + + sdb_llist_clear(list); + sdb_log(SDB_LOG_INFO, "core: Unregistered %zu %s callback%s", + len, type, len == 1 ? "" : "s"); + } +} /* sdb_plugin_unregister_all */ + sdb_plugin_ctx_t sdb_plugin_get_ctx(void) { @@ -903,7 +997,7 @@ sdb_plugin_reconfigure_init(void) CTX(sdb_llist_iter_get_next(iter))->use_cnt = 0; sdb_llist_iter_destroy(iter); - plugin_unregister_all(); + sdb_plugin_unregister_all(); return 0; } /* sdb_plugin_reconfigure_init */ @@ -1238,5 +1332,144 @@ sdb_plugin_fetch_timeseries(const char *type, const char *id, return ts; } /* sdb_plugin_fetch_timeseries */ +int +sdb_plugin_store_host(const char *name, sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if (! name) + return -1; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_host(name, last_update, writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_host */ + +int +sdb_plugin_store_service(const char *hostname, const char *name, + sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if ((! hostname) || (! name)) + return -1; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_service(hostname, name, last_update, + writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_service */ + +int +sdb_plugin_store_metric(const char *hostname, const char *name, + sdb_metric_store_t *store, sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if ((! hostname) || (! name)) + return -1; + + if ((! store->type) || (! store->id)) + store = NULL; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_metric(hostname, name, store, last_update, + writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_metric */ + +int +sdb_plugin_store_attribute(const char *hostname, const char *key, + const sdb_data_t *value, sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if ((! hostname) || (! key) || (! value)) + return -1; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_attribute(hostname, key, value, last_update, + writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_attribute */ + +int +sdb_plugin_store_service_attribute(const char *hostname, const char *service, + const char *key, const sdb_data_t *value, sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if ((! hostname) || (! service) || (! key) || (! value)) + return -1; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_service_attr(hostname, service, + key, value, last_update, writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_service_attribute */ + +int +sdb_plugin_store_metric_attribute(const char *hostname, const char *metric, + const char *key, const sdb_data_t *value, sdb_time_t last_update) +{ + sdb_llist_iter_t *iter; + int status = 0; + + if ((! hostname) || (! metric) || (! key) || (! value)) + return -1; + + iter = sdb_llist_get_iter(writer_list); + while (sdb_llist_iter_has_next(iter)) { + sdb_plugin_writer_t *writer; + writer = SDB_PLUGIN_WRITER(sdb_llist_iter_get_next(iter)); + assert(writer); + if (writer->impl.store_metric_attr(hostname, metric, + key, value, last_update, writer->w_user_data)) + status = -1; + } + sdb_llist_iter_destroy(iter); + return status; +} /* sdb_plugin_store_metric_attribute */ + /* vim: set tw=78 sw=4 ts=4 noexpandtab : */