summaryrefslogtreecommitdiffstats
path: root/src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c')
-rw-r--r--src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c268
1 files changed, 0 insertions, 268 deletions
diff --git a/src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c b/src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c
deleted file mode 100644
index 7ff3f8200..000000000
--- a/src/fluent-bit/plugins/out_prometheus_exporter/prom_http.c
+++ /dev/null
@@ -1,268 +0,0 @@
-/* -*- 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_output_plugin.h>
-#include <fluent-bit/flb_http_server.h>
-#include "prom.h"
-#include "prom_http.h"
-
-pthread_key_t ph_metrics_key;
-
-/* Return the newest storage metrics buffer */
-static struct prom_http_buf *metrics_get_latest()
-{
- struct prom_http_buf *buf;
- struct mk_list *metrics_list;
-
- metrics_list = pthread_getspecific(ph_metrics_key);
- if (!metrics_list) {
- return NULL;
- }
-
- if (mk_list_size(metrics_list) == 0) {
- return NULL;
- }
-
- buf = mk_list_entry_last(metrics_list, struct prom_http_buf, _head);
- return buf;
-}
-
-/* Delete unused metrics, note that we only care about the latest node */
-static int cleanup_metrics()
-{
- int c = 0;
- struct mk_list *tmp;
- struct mk_list *head;
- struct mk_list *metrics_list;
- struct prom_http_buf *last;
- struct prom_http_buf *entry;
-
- metrics_list = pthread_getspecific(ph_metrics_key);
- if (!metrics_list) {
- return -1;
- }
-
- last = metrics_get_latest();
- if (!last) {
- return -1;
- }
-
- mk_list_foreach_safe(head, tmp, metrics_list) {
- entry = mk_list_entry(head, struct prom_http_buf, _head);
- if (entry != last && entry->users == 0) {
- mk_list_del(&entry->_head);
- flb_free(entry->buf_data);
- flb_free(entry);
- c++;
- }
- }
-
- return c;
-}
-
-/* destructor callback */
-static void destruct_metrics(void *data)
-{
- struct mk_list *tmp;
- struct mk_list *head;
- struct mk_list *metrics_list = (struct mk_list*)data;
- struct prom_http_buf *entry;
-
- if (!metrics_list) {
- return;
- }
-
- mk_list_foreach_safe(head, tmp, metrics_list) {
- entry = mk_list_entry(head, struct prom_http_buf, _head);
- mk_list_del(&entry->_head);
- flb_free(entry->buf_data);
- flb_free(entry);
- }
-
- flb_free(metrics_list);
-}
-
-/*
- * Callback invoked every time a new payload of Metrics is received from
- * Fluent Bit engine through Message Queue channel.
- *
- * This function runs in a Monkey HTTP thread worker and it purpose is
- * to take the metrics data and store it locally for every thread, so then
- * it can be available on 'cb_metrics()' to serve it as a response.
- */
-static void cb_mq_metrics(mk_mq_t *queue, void *data, size_t size)
-{
- struct prom_http_buf *buf;
- struct mk_list *metrics_list = NULL;
-
- metrics_list = pthread_getspecific(ph_metrics_key);
- if (!metrics_list) {
- metrics_list = flb_malloc(sizeof(struct mk_list));
- if (!metrics_list) {
- flb_errno();
- return;
- }
- mk_list_init(metrics_list);
- pthread_setspecific(ph_metrics_key, metrics_list);
- }
-
- /* FIXME: convert data ? */
- buf = flb_malloc(sizeof(struct prom_http_buf));
- if (!buf) {
- flb_errno();
- return;
- }
- buf->users = 0;
- buf->buf_data = flb_malloc(size);
- if (!buf->buf_data) {
- flb_errno();
- flb_free(buf);
- return;
- }
- memcpy(buf->buf_data, data, size);
- buf->buf_size = size;
-
- mk_list_add(&buf->_head, metrics_list);
- cleanup_metrics();
-}
-
-/* Create message queue to receive Metrics payload from the engine */
-static int http_server_mq_create(struct prom_http *ph)
-{
- int ret;
-
- pthread_key_create(&ph_metrics_key, destruct_metrics);
-
- ret = mk_mq_create(ph->ctx, "/metrics", cb_mq_metrics, NULL);
- if (ret == -1) {
- return -1;
- }
- ph->qid_metrics = ret;
- return 0;
-}
-
-/* HTTP endpoint: /metrics */
-static void cb_metrics(mk_request_t *request, void *data)
-{
- struct prom_http_buf *buf;
- (void) data;
-
- buf = metrics_get_latest();
- if (!buf) {
- mk_http_status(request, 404);
- mk_http_done(request);
- return;
- }
-
- buf->users++;
-
- mk_http_status(request, 200);
- flb_hs_add_content_type_to_req(request, FLB_HS_CONTENT_TYPE_PROMETHEUS);
- mk_http_send(request, buf->buf_data, buf->buf_size, NULL);
- mk_http_done(request);
-
- buf->users--;
-}
-
-/* HTTP endpoint: / (root) */
-static void cb_root(mk_request_t *request, void *data)
-{
- (void) data;
-
- mk_http_status(request, 200);
- mk_http_send(request, "Fluent Bit Prometheus Exporter\n", 31, NULL);
- mk_http_done(request);
-}
-
-struct prom_http *prom_http_server_create(struct prom_exporter *ctx,
- const char *listen,
- int tcp_port,
- struct flb_config *config)
-{
- int ret;
- int vid;
- char tmp[32];
- struct prom_http *ph;
-
- ph = flb_malloc(sizeof(struct prom_http));
- if (!ph) {
- flb_errno();
- return NULL;
- }
- ph->config = config;
-
- /* HTTP Server context */
- ph->ctx = mk_create();
- if (!ph->ctx) {
- flb_free(ph);
- return NULL;
- }
-
- /* Compose listen address */
- snprintf(tmp, sizeof(tmp) -1, "%s:%d", listen, tcp_port);
- mk_config_set(ph->ctx,
- "Listen", tmp,
- "Workers", "1",
- NULL);
-
- /* Virtual host */
- vid = mk_vhost_create(ph->ctx, NULL);
- ph->vid = vid;
-
- /* Set HTTP URI callbacks */
- mk_vhost_handler(ph->ctx, vid, "/metrics", cb_metrics, NULL);
- mk_vhost_handler(ph->ctx, vid, "/", cb_root, NULL);
-
- /* Create a Message Queue to push 'metrics' to HTTP workers */
- ret = http_server_mq_create(ph);
- if (ret == -1) {
- mk_destroy(ph->ctx);
- flb_free(ph);
- return NULL;
- }
-
- return ph;
-}
-
-void prom_http_server_destroy(struct prom_http *ph)
-{
- if (ph) {
- /* TODO: release mk_vhost */
- if (ph->ctx) {
- mk_destroy(ph->ctx);
- }
- flb_free(ph);
- }
-}
-
-int prom_http_server_start(struct prom_http *ph)
-{
- return mk_start(ph->ctx);
-}
-
-int prom_http_server_stop(struct prom_http *ph)
-{
- return mk_stop(ph->ctx);
-}
-
-int prom_http_server_mq_push_metrics(struct prom_http *ph,
- void *data, size_t size)
-{
- return mk_mq_send(ph->ctx, ph->qid_metrics, data, size);
-}