diff options
Diffstat (limited to 'src/fluent-bit/plugins/in_fluentbit_metrics/metrics.c')
| -rw-r--r-- | src/fluent-bit/plugins/in_fluentbit_metrics/metrics.c | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/src/fluent-bit/plugins/in_fluentbit_metrics/metrics.c b/src/fluent-bit/plugins/in_fluentbit_metrics/metrics.c new file mode 100644 index 000000000..779a03651 --- /dev/null +++ b/src/fluent-bit/plugins/in_fluentbit_metrics/metrics.c @@ -0,0 +1,201 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +/* Fluent Bit + * ========== + * Copyright (C) 2015-2022 The Fluent Bit Authors + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <fluent-bit/flb_input_plugin.h> +#include <fluent-bit/flb_config.h> +#include <fluent-bit/flb_metrics.h> +#include <fluent-bit/flb_metrics_exporter.h> + +struct flb_in_metrics { + /* config map options */ + int scrape_on_start; + int scrape_interval; + + /* internal */ + int coll_fd_start; + int coll_fd_runtime; + struct cmt_counter *c; + struct flb_input_instance *ins; +}; + +static int scrape_metrics(struct flb_config *config, struct flb_in_metrics *ctx) +{ + int ret; + size_t ts; + char *name; + struct cmt *cmt; + + /* Update internal metric */ + ts = cfl_time_now(); + name = (char *) flb_input_name(ctx->ins); + cmt_counter_inc(ctx->c, ts, 1, (char *[]) {name}); + + + cmt = flb_me_get_cmetrics(config); + if (!cmt) { + flb_plg_error(ctx->ins, "could not scrape metrics"); + return 0; + } + + /* Append the updated metrics */ + ret = flb_input_metrics_append(ctx->ins, NULL, 0, cmt); + if (ret != 0) { + flb_plg_error(ctx->ins, "could not append metrics"); + } + cmt_destroy(cmt); + + return 0; +} + +/* + * Update the metrics, this function is invoked every time 'scrape_interval' + * expires. + */ +static int cb_metrics_collect_runtime(struct flb_input_instance *ins, + struct flb_config *config, void *in_context) +{ + return scrape_metrics(config, in_context); +} + +static int cb_metrics_collect_start(struct flb_input_instance *ins, + struct flb_config *config, void *in_context) +{ + struct flb_in_metrics *ctx = in_context; + + /* pause collector */ + flb_input_collector_pause(ctx->coll_fd_start, ctx->ins); + + return scrape_metrics(config, ctx); +} + +static int in_metrics_init(struct flb_input_instance *in, + struct flb_config *config, void *data) +{ + int ret; + struct flb_in_metrics *ctx; + + /* Create plugin context */ + ctx = flb_calloc(1, sizeof(struct flb_in_metrics)); + if (!ctx) { + flb_errno(); + return -1; + } + ctx->ins = in; + + /* Load the config map */ + ret = flb_input_config_map_set(in, (void *) ctx); + if (ret == -1) { + flb_free(ctx); + return -1; + } + + /* Associate context with the instance */ + flb_input_set_context(in, ctx); + + /* Scrape metrics on start / collector */ + if (ctx->scrape_interval > 2 && ctx->scrape_on_start) { + ret = flb_input_set_collector_time(in, + cb_metrics_collect_start, + 5, 0, + config); + if (ret == -1) { + flb_plg_error(ctx->ins, + "could not set collector on start for Fluent Bit " + "metrics plugin"); + return -1; + } + ctx->coll_fd_start = ret; + } + + /* Create the runtime collector */ + ret = flb_input_set_collector_time(in, + cb_metrics_collect_runtime, + ctx->scrape_interval, 0, + config); + if (ret == -1) { + flb_plg_error(ctx->ins, + "could not set collector for Fluent Bit metrics plugin"); + return -1; + } + ctx->coll_fd_runtime = ret; + + /* Internal metrics */ + ctx->c = cmt_counter_create(ctx->ins->cmt, + "fluentbit", "input_metrics", "scrapes_total", + "Number of total metrics scrapes", + 1, (char *[]) {"name"}); + return 0; +} + +static int in_metrics_exit(void *data, struct flb_config *config) +{ + struct flb_in_metrics *ctx = data; + + if (!ctx) { + return 0; + } + + flb_free(ctx); + return 0; +} + +static void in_metrics_pause(void *data, struct flb_config *config) +{ + struct flb_in_metrics *ctx = data; + + flb_input_collector_pause(ctx->coll_fd_runtime, ctx->ins); +} + +static void in_metrics_resume(void *data, struct flb_config *config) +{ + struct flb_in_metrics *ctx = data; + + flb_input_collector_resume(ctx->coll_fd_runtime, ctx->ins); +} + +/* Configuration properties map */ +static struct flb_config_map config_map[] = { + { + FLB_CONFIG_MAP_TIME, "scrape_interval", "2", + 0, FLB_TRUE, offsetof(struct flb_in_metrics, scrape_interval), + "scrape interval to collect the internal metrics of Fluent Bit." + }, + + { + FLB_CONFIG_MAP_BOOL, "scrape_on_start", "false", + 0, FLB_TRUE, offsetof(struct flb_in_metrics, scrape_on_start), + "scrape metrics upon start, useful to avoid waiting for 'scrape_interval' " + "for the first round of metrics." + }, + + /* EOF */ + {0} +}; + +struct flb_input_plugin in_fluentbit_metrics_plugin = { + .name = "fluentbit_metrics", + .description = "Fluent Bit internal metrics", + .cb_init = in_metrics_init, + .cb_pre_run = NULL, + .cb_flush_buf = NULL, + .config_map = config_map, + .cb_pause = in_metrics_pause, + .cb_resume = in_metrics_resume, + .cb_exit = in_metrics_exit, +}; |