X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fcore%2Fplugin.c;h=4de1c3a4d93f3a8f502844bc67d4fe5801889b6b;hb=f05f6dfc4e48309c916c0dcb158df66c80de6c71;hp=36f80f6e6c29296322fac9979632a0e32ea05f29;hpb=c460163d13ccecdaf71ff5fc0fea802bf4360dc5;p=sysdb.git diff --git a/src/core/plugin.c b/src/core/plugin.c index 36f80f6..4de1c3a 100644 --- a/src/core/plugin.c +++ b/src/core/plugin.c @@ -50,29 +50,14 @@ #include +/* helper to access info attributes */ +#define INFO_GET(i, attr) \ + ((i)->attr ? (i)->attr : #attr" not set") + /* * private data types */ -struct sdb_plugin_info { - char *plugin_name; - char *filename; - - /* public attributes */ - char *description; - char *copyright; - char *license; - - int version; - int plugin_version; -}; -#define SDB_PLUGIN_INFO_INIT { \ - /* plugin_name */ NULL, /* filename */ NULL, /* desc */ NULL, \ - /* copyright */ NULL, /* license */ NULL, \ - /* version */ -1, /* plugin_version */ -1 } -#define INFO_GET(i, attr) \ - ((i)->attr ? (i)->attr : #attr" not set") - typedef struct { sdb_object_t super; sdb_plugin_ctx_t public; @@ -99,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; @@ -108,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_object_t super; + sdb_store_writer_t impl; + sdb_object_t *user_data; + ctx_t *ctx; +} sdb_plugin_writer_t; +#define SDB_PLUGIN_WRITER(obj) ((sdb_plugin_writer_t *)(obj)) + /* * private variables */ @@ -122,7 +114,7 @@ static sdb_plugin_ctx_t plugin_default_ctx = SDB_PLUGIN_CTX_INIT; static sdb_plugin_info_t plugin_default_info = SDB_PLUGIN_INFO_INIT; static pthread_key_t plugin_ctx_key; -static _Bool plugin_ctx_key_initialized = 0; +static bool plugin_ctx_key_initialized = 0; /* a list of the plugin contexts of all registered plugins */ static sdb_llist_t *all_plugins = NULL; @@ -133,17 +125,21 @@ static sdb_llist_t *collector_list = NULL; 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 }, }; /* @@ -413,6 +409,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)->ctx = ctx_get(); + sdb_object_ref(SDB_OBJ(SDB_PLUGIN_WRITER(obj)->ctx)); + + sdb_object_ref(ud); + SDB_PLUGIN_WRITER(obj)->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)->user_data); + sdb_object_deref(SDB_OBJ(SDB_PLUGIN_WRITER(obj)->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) { @@ -766,12 +811,16 @@ sdb_plugin_register_collector(const char *name, sdb_plugin_collector_cb callback if (interval) SDB_PLUGIN_CCB(obj)->ccb_interval = *interval; else { - sdb_time_t tmp = sdb_plugin_get_ctx().interval; + ctx_t *ctx = ctx_get(); - if (tmp > 0) - SDB_PLUGIN_CCB(obj)->ccb_interval = tmp; - else - SDB_PLUGIN_CCB(obj)->ccb_interval = 0; + if (! ctx) { + sdb_log(SDB_LOG_ERR, "core: Cannot determine interval " + "for collector %s; none specified and no plugin " + "context found", cb_name); + return -1; + } + + SDB_PLUGIN_CCB(obj)->ccb_interval = ctx->public.interval; } if (! (SDB_PLUGIN_CCB(obj)->ccb_next_update = sdb_gettime())) { @@ -797,6 +846,49 @@ sdb_plugin_register_collector(const char *name, sdb_plugin_collector_cb callback return 0; } /* sdb_plugin_register_collector */ +int +sdb_plugin_register_ts_fetcher(const char *name, + sdb_plugin_fetch_ts_cb callback, sdb_object_t *user_data) +{ + return plugin_add_callback(&ts_fetcher_list, "time-series fetcher", + 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 */ + sdb_plugin_ctx_t sdb_plugin_get_ctx(void) { @@ -829,6 +921,16 @@ sdb_plugin_set_ctx(sdb_plugin_ctx_t ctx, sdb_plugin_ctx_t *old) return 0; } /* sdb_plugin_set_ctx */ +const sdb_plugin_info_t * +sdb_plugin_current(void) +{ + ctx_t *ctx = ctx_get(); + + if (! ctx) + return NULL; + return &ctx->info; +} /* sdb_plugin_current */ + int sdb_plugin_configure(const char *name, oconfig_item_t *ci) { @@ -1132,7 +1234,7 @@ sdb_plugin_log(int prio, const char *msg) sdb_llist_iter_t *iter; int ret = -1; - _Bool logged = 0; + bool logged = 0; if (! msg) return 0; @@ -1202,5 +1304,33 @@ sdb_plugin_logf(int prio, const char *fmt, ...) return ret; } /* sdb_plugin_logf */ +sdb_timeseries_t * +sdb_plugin_fetch_timeseries(const char *type, const char *id, + sdb_timeseries_opts_t *opts) +{ + sdb_plugin_cb_t *plugin; + sdb_plugin_fetch_ts_cb callback; + sdb_timeseries_t *ts; + + ctx_t *old_ctx; + + if ((! type) || (! id) || (! opts)) + return NULL; + + plugin = SDB_PLUGIN_CB(sdb_llist_search_by_name(ts_fetcher_list, type)); + if (! plugin) { + sdb_log(SDB_LOG_ERR, "core: Cannot fetch time-series of type %s: " + "no such plugin loaded", type); + errno = ENOENT; + return NULL; + } + + old_ctx = ctx_set(plugin->cb_ctx); + callback = (sdb_plugin_fetch_ts_cb)plugin->cb_callback; + ts = callback(id, opts, plugin->cb_user_data); + ctx_set(old_ctx); + return ts; +} /* sdb_plugin_fetch_timeseries */ + /* vim: set tw=78 sw=4 ts=4 noexpandtab : */