netdata/exporting/read_config.c

367 lines
14 KiB
C

// SPDX-License-Identifier: GPL-3.0-or-later
#include "exporting_engine.h"
struct config exporting_config = {.sections = NULL,
.mutex = NETDATA_MUTEX_INITIALIZER,
.index = {.avl_tree = {.root = NULL, .compar = appconfig_section_compare},
.rwlock = AVL_LOCK_INITIALIZER}};
static _CONNECTOR_INSTANCE *find_instance(const char *section)
{
_CONNECTOR_INSTANCE *local_ci;
local_ci = add_connector_instance(NULL, NULL); // Get root section
if (unlikely(!local_ci))
return local_ci;
if (!section)
return local_ci;
while (local_ci) {
if (!strcmp(local_ci->instance_name, section))
break;
local_ci = local_ci->next;
}
return local_ci;
}
char *expconfig_get(struct config *root, const char *section, const char *name, const char *default_value)
{
_CONNECTOR_INSTANCE *local_ci;
if (!strcmp(section, CONFIG_SECTION_EXPORTING))
return appconfig_get(root, CONFIG_SECTION_EXPORTING, name, default_value);
local_ci = find_instance(section);
if (!local_ci)
return NULL; // TODO: Check if it is meaningful to return default_value
return appconfig_get(
root,
local_ci->instance_name,
name,
appconfig_get(
root, local_ci->connector_name, name, appconfig_get(root, CONFIG_SECTION_EXPORTING, name, default_value)));
}
int expconfig_get_boolean(struct config *root, const char *section, const char *name, int default_value)
{
_CONNECTOR_INSTANCE *local_ci;
if (!strcmp(section, CONFIG_SECTION_EXPORTING))
return appconfig_get_boolean(root, CONFIG_SECTION_EXPORTING, name, default_value);
local_ci = find_instance(section);
if (!local_ci)
return 0; // TODO: Check if it is meaningful to return default_value
return appconfig_get_boolean(
root,
local_ci->instance_name,
name,
appconfig_get_boolean(
root,
local_ci->connector_name,
name,
appconfig_get_boolean(root, CONFIG_SECTION_EXPORTING, name, default_value)));
}
long long expconfig_get_number(struct config *root, const char *section, const char *name, long long default_value)
{
_CONNECTOR_INSTANCE *local_ci;
if (!strcmp(section, CONFIG_SECTION_EXPORTING))
return appconfig_get_number(root, CONFIG_SECTION_EXPORTING, name, default_value);
local_ci = find_instance(section);
if (!local_ci)
return 0; // TODO: Check if it is meaningful to return default_value
return appconfig_get_number(
root,
local_ci->instance_name,
name,
appconfig_get_number(
root,
local_ci->connector_name,
name,
appconfig_get_number(root, CONFIG_SECTION_EXPORTING, name, default_value)));
}
/*
* Get the next connector instance that we need to activate
*
* @param @target_ci will be filled with instance name and connector name
*
* @return - 1 if more connectors to be fetched, 0 done
*
*/
int get_connector_instance(struct connector_instance *target_ci)
{
static _CONNECTOR_INSTANCE *local_ci = NULL;
_CONNECTOR_INSTANCE *global_connector_instance;
global_connector_instance = find_instance(NULL); // Fetch head of instances
if (unlikely(!global_connector_instance))
return 0;
if (target_ci == NULL) {
local_ci = NULL;
return 1;
}
if (local_ci == NULL)
local_ci = global_connector_instance;
else {
local_ci = local_ci->next;
if (local_ci == NULL)
return 0;
}
strcpy(target_ci->instance_name, local_ci->instance_name);
strcpy(target_ci->connector_name, local_ci->connector_name);
return 1;
}
/**
* Select Type
*
* Select the connector type based on the user input
*
* @param type is the string that defines the connector type
*
* @return It returns the connector id.
*/
BACKEND_TYPE exporting_select_type(const char *type)
{
if (!strcmp(type, "graphite") || !strcmp(type, "graphite:plaintext")) {
return BACKEND_TYPE_GRAPHITE;
} else if (!strcmp(type, "opentsdb") || !strcmp(type, "opentsdb:telnet")) {
return BACKEND_TYPE_OPENTSDB_USING_TELNET;
} else if (!strcmp(type, "opentsdb:http") || !strcmp(type, "opentsdb:https")) {
return BACKEND_TYPE_OPENTSDB_USING_HTTP;
} else if (!strcmp(type, "json") || !strcmp(type, "json:plaintext")) {
return BACKEND_TYPE_JSON;
} else if (!strcmp(type, "prometheus_remote_write")) {
return BACKEND_TYPE_PROMETHEUS;
} else if (!strcmp(type, "kinesis") || !strcmp(type, "kinesis:plaintext")) {
return BACKEND_TYPE_KINESIS;
} else if (!strcmp(type, "mongodb") || !strcmp(type, "mongodb:plaintext"))
return BACKEND_TYPE_MONGODB;
return BACKEND_TYPE_UNKNOWN;
}
EXPORTING_OPTIONS exporting_parse_data_source(const char *data_source, EXPORTING_OPTIONS exporting_options) {
if(!strcmp(data_source, "raw") || !strcmp(data_source, "as collected") || !strcmp(data_source, "as-collected") || !strcmp(data_source, "as_collected") || !strcmp(data_source, "ascollected")) {
exporting_options |= EXPORTING_SOURCE_DATA_AS_COLLECTED;
exporting_options &= ~(EXPORTING_OPTIONS_SOURCE_BITS ^ EXPORTING_SOURCE_DATA_AS_COLLECTED);
}
else if(!strcmp(data_source, "average")) {
exporting_options |= EXPORTING_SOURCE_DATA_AVERAGE;
exporting_options &= ~(EXPORTING_OPTIONS_SOURCE_BITS ^ EXPORTING_SOURCE_DATA_AVERAGE);
}
else if(!strcmp(data_source, "sum") || !strcmp(data_source, "volume")) {
exporting_options |= EXPORTING_SOURCE_DATA_SUM;
exporting_options &= ~(EXPORTING_OPTIONS_SOURCE_BITS ^ EXPORTING_SOURCE_DATA_SUM);
}
else {
error("EXPORTING: invalid data data_source method '%s'.", data_source);
}
return exporting_options;
}
/**
* Read configuration
*
* Based on read configuration an engine data structure is filled with exporting connector instances.
*
* @return Returns a filled engine data structure or NULL if there are no connector instances configured.
*/
struct engine *read_exporting_config()
{
int instances_to_activate = 0;
int exporting_config_exists = 0;
static struct engine *engine = NULL;
struct connector_instance_list {
struct connector_instance local_ci;
struct connector_instance_list *next;
};
struct connector_instance local_ci;
struct connector_instance_list *tmp_ci_list, *tmp_ci_list1;
struct connector_instance_list **ci_list;
if (unlikely(engine))
return engine;
char *filename = strdupz_path_subpath(netdata_configured_user_config_dir, EXPORTING_CONF);
exporting_config_exists = appconfig_load(&exporting_config, filename, 0, NULL);
if (!exporting_config_exists) {
info("CONFIG: cannot load user exporting config '%s'. Will try the stock version.", filename);
freez(filename);
filename = strdupz_path_subpath(netdata_configured_stock_config_dir, EXPORTING_CONF);
exporting_config_exists = appconfig_load(&exporting_config, filename, 0, NULL);
if (!exporting_config_exists)
info("CONFIG: cannot load stock exporting config '%s'. Running with internal defaults.", filename);
}
freez(filename);
// Will build a list of instances per connector
// TODO: change BACKEND to EXPORTING
ci_list = callocz(BACKEND_TYPE_NUM, sizeof(struct connector_instance_list *));
while (get_connector_instance(&local_ci)) {
BACKEND_TYPE backend_type;
info("Processing connector instance (%s)", local_ci.instance_name);
if (exporter_get_boolean(local_ci.instance_name, "enabled", 0)) {
backend_type = exporting_select_type(local_ci.connector_name);
info(
"Instance (%s) on connector (%s) is enabled and scheduled for activation",
local_ci.instance_name,
local_ci.connector_name);
tmp_ci_list = (struct connector_instance_list *)callocz(1, sizeof(struct connector_instance_list));
memcpy(&tmp_ci_list->local_ci, &local_ci, sizeof(local_ci));
tmp_ci_list->next = ci_list[backend_type];
ci_list[backend_type] = tmp_ci_list;
instances_to_activate++;
} else
info("Instance (%s) on connector (%s) is not enabled", local_ci.instance_name, local_ci.connector_name);
}
if (unlikely(!instances_to_activate)) {
info("No connector instances to activate");
freez(ci_list);
return NULL;
}
engine = (struct engine *)calloc(1, sizeof(struct engine));
// TODO: Check and fill engine fields if actually needed
if (exporting_config_exists) {
engine->config.hostname =
strdupz(exporter_get(CONFIG_SECTION_EXPORTING, "hostname", netdata_configured_hostname));
engine->config.prefix = strdupz(exporter_get(CONFIG_SECTION_EXPORTING, "prefix", "netdata"));
engine->config.update_every =
exporter_get_number(CONFIG_SECTION_EXPORTING, EXPORTER_UPDATE_EVERY, EXPORTER_UPDATE_EVERY_DEFAULT);
}
for (size_t i = 0; i < BACKEND_TYPE_NUM; i++) {
// For each connector build list
tmp_ci_list = ci_list[i];
// If we have a list of instances for this connector then build it
if (tmp_ci_list) {
struct connector *tmp_connector;
tmp_connector = (struct connector *)calloc(1, sizeof(struct connector));
tmp_connector->next = engine->connector_root;
engine->connector_root = tmp_connector;
tmp_connector->config.type = i;
tmp_connector->engine = engine;
while (tmp_ci_list) {
struct instance *tmp_instance;
char *instance_name;
info("Instance %s on %s", tmp_ci_list->local_ci.instance_name, tmp_ci_list->local_ci.connector_name);
tmp_instance = (struct instance *)calloc(1, sizeof(struct instance));
tmp_instance->connector = engine->connector_root;
tmp_instance->next = engine->connector_root->instance_root;
engine->connector_root->instance_root = tmp_instance;
tmp_instance->connector = engine->connector_root;
instance_name = tmp_ci_list->local_ci.instance_name;
tmp_instance->config.name = strdupz(tmp_ci_list->local_ci.instance_name);
tmp_instance->config.destination =
strdupz(exporter_get(instance_name, EXPORTER_DESTINATION, EXPORTER_DESTINATION_DEFAULT));
tmp_instance->config.update_every =
exporter_get_number(instance_name, EXPORTER_UPDATE_EVERY, EXPORTER_UPDATE_EVERY_DEFAULT);
tmp_instance->config.buffer_on_failures =
exporter_get_number(instance_name, EXPORTER_BUF_ONFAIL, EXPORTER_BUF_ONFAIL_DEFAULT);
tmp_instance->config.timeoutms =
exporter_get_number(instance_name, EXPORTER_TIMEOUT_MS, EXPORTER_TIMEOUT_MS_DEFAULT);
tmp_instance->config.charts_pattern = simple_pattern_create(
exporter_get(instance_name, EXPORTER_SEND_CHART_MATCH, EXPORTER_SEND_CHART_MATCH_DEFAULT),
NULL,
SIMPLE_PATTERN_EXACT);
tmp_instance->config.hosts_pattern = simple_pattern_create(
exporter_get(instance_name, EXPORTER_SEND_HOST_MATCH, EXPORTER_SEND_HOST_MATCH_DEFAULT),
NULL,
SIMPLE_PATTERN_EXACT);
char *data_source =
exporter_get(instance_name, EXPORTER_DATA_SOURCE, EXPORTER_DATA_SOURCE_DEFAULT);
tmp_instance->config.options = exporting_parse_data_source(data_source, tmp_instance->config.options);
if (exporter_get_boolean(
instance_name, EXPORTER_SEND_CONFIGURED_LABELS, EXPORTER_SEND_CONFIGURED_LABELS_DEFAULT))
tmp_instance->config.options |= EXPORTING_OPTION_SEND_CONFIGURED_LABELS;
else
tmp_instance->config.options &= ~EXPORTING_OPTION_SEND_CONFIGURED_LABELS;
if (exporter_get_boolean(
instance_name, EXPORTER_SEND_AUTOMATIC_LABELS, EXPORTER_SEND_AUTOMATIC_LABELS_DEFAULT))
tmp_instance->config.options |= EXPORTING_OPTION_SEND_AUTOMATIC_LABELS;
else
tmp_instance->config.options &= ~EXPORTING_OPTION_SEND_AUTOMATIC_LABELS;
if (exporter_get_boolean(instance_name, EXPORTER_SEND_NAMES, EXPORTER_SEND_NAMES_DEFAULT))
tmp_instance->config.options |= EXPORTING_OPTION_SEND_NAMES;
else
tmp_instance->config.options &= ~EXPORTING_OPTION_SEND_NAMES;
#ifdef NETDATA_INTERNAL_CHECKS
info(
" Dest=[%s], upd=[%d], buffer=[%d] timeout=[%ld] options=[%u]",
tmp_instance->config.destination,
tmp_instance->config.update_every,
tmp_instance->config.buffer_on_failures,
tmp_instance->config.timeoutms,
tmp_instance->config.options);
#endif
if (unlikely(!exporting_config_exists) && !engine->config.hostname) {
engine->config.hostname =
strdupz(config_get(instance_name, "hostname", netdata_configured_hostname));
engine->config.prefix = strdupz(config_get(instance_name, "prefix", "netdata"));
engine->config.update_every =
config_get_number(instance_name, EXPORTER_UPDATE_EVERY, EXPORTER_UPDATE_EVERY_DEFAULT);
}
tmp_ci_list1 = tmp_ci_list->next;
freez(tmp_ci_list);
tmp_ci_list = tmp_ci_list1;
}
}
}
freez(ci_list);
return engine;
}