diff options
Diffstat (limited to 'fluent-bit/plugins/out_influxdb')
| -rw-r--r-- | fluent-bit/plugins/out_influxdb/CMakeLists.txt | 5 | ||||
| -rw-r--r-- | fluent-bit/plugins/out_influxdb/influxdb.c | 682 | ||||
| -rw-r--r-- | fluent-bit/plugins/out_influxdb/influxdb.h | 78 | ||||
| -rw-r--r-- | fluent-bit/plugins/out_influxdb/influxdb_bulk.c | 233 | ||||
| -rw-r--r-- | fluent-bit/plugins/out_influxdb/influxdb_bulk.h | 54 |
5 files changed, 0 insertions, 1052 deletions
diff --git a/fluent-bit/plugins/out_influxdb/CMakeLists.txt b/fluent-bit/plugins/out_influxdb/CMakeLists.txt deleted file mode 100644 index 75d85a6b5..000000000 --- a/fluent-bit/plugins/out_influxdb/CMakeLists.txt +++ /dev/null @@ -1,5 +0,0 @@ -set(src - influxdb_bulk.c - influxdb.c) - -FLB_PLUGIN(out_influxdb "${src}" "") diff --git a/fluent-bit/plugins/out_influxdb/influxdb.c b/fluent-bit/plugins/out_influxdb/influxdb.c deleted file mode 100644 index 71e489fbc..000000000 --- a/fluent-bit/plugins/out_influxdb/influxdb.c +++ /dev/null @@ -1,682 +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_utils.h> -#include <fluent-bit/flb_http_client.h> -#include <fluent-bit/flb_time.h> -#include <fluent-bit/flb_metrics.h> -#include <fluent-bit/flb_log_event_decoder.h> - -#include <msgpack.h> - -#include "influxdb.h" -#include "influxdb_bulk.h" - -#include <stdio.h> - -/* - * Returns FLB_TRUE when the specified key is in Tag_Keys list, - * otherwise FLB_FALSE - */ -static int is_tagged_key(struct flb_influxdb *ctx, - const char *key, int kl, int type); - -/* - * Increments the timestamp when it is duplicated - */ -static void influxdb_tsmod(struct flb_time *ts, struct flb_time *dupe, - struct flb_time *last) { - if (flb_time_equal(ts, last) || flb_time_equal(ts, dupe)) { - ++dupe->tm.tv_nsec; - flb_time_copy(last, ts); - flb_time_copy(ts, dupe); - } - else { - flb_time_copy(last, ts); - flb_time_copy(dupe, ts); - } -} - -/* - * Convert the internal Fluent Bit data representation to the required one - * by InfluxDB. - */ -static char *influxdb_format(const char *tag, int tag_len, - const void *data, size_t bytes, size_t *out_size, - struct flb_influxdb *ctx) -{ - int i; - int ret; - int n_size; - uint64_t seq = 0; - char *buf; - char *str = NULL; - size_t str_size; - char tmp[128]; - msgpack_object map; - struct flb_time tm; - struct influxdb_bulk *bulk = NULL; - struct influxdb_bulk *bulk_head = NULL; - struct influxdb_bulk *bulk_body = NULL; - struct flb_log_event_decoder log_decoder; - struct flb_log_event log_event; - - ret = flb_log_event_decoder_init(&log_decoder, (char *) data, bytes); - - if (ret != FLB_EVENT_DECODER_SUCCESS) { - flb_plg_error(ctx->ins, - "Log event decoder initialization error : %d", ret); - - return NULL; - } - - /* Create the bulk composer */ - bulk = influxdb_bulk_create(); - if (!bulk) { - goto error; - } - - bulk_head = influxdb_bulk_create(); - if (!bulk_head) { - goto error; - } - - bulk_body = influxdb_bulk_create(); - if (!bulk_body) { - goto error; - } - - while ((ret = flb_log_event_decoder_next( - &log_decoder, - &log_event)) == FLB_EVENT_DECODER_SUCCESS) { - flb_time_copy(&tm, &log_event.timestamp); - - map = *log_event.body; - n_size = map.via.map.size + 1; - - seq = ctx->seq; - if (ctx->seq + 1 >= 100000) { - seq = 1; - } - else { - ctx->seq++; - } - - ret = influxdb_bulk_append_header(bulk_head, - tag, tag_len, - seq, - ctx->seq_name, ctx->seq_len); - if (ret == -1) { - goto error; - } - - for (i = 0; i < n_size - 1; i++) { - msgpack_object *k = &map.via.map.ptr[i].key; - msgpack_object *v = &map.via.map.ptr[i].val; - - if (k->type != MSGPACK_OBJECT_BIN && k->type != MSGPACK_OBJECT_STR) { - continue; - } - - int quote = FLB_FALSE; - - /* key */ - const char *key = NULL; - int key_len; - - /* val */ - const char *val = NULL; - int val_len; - - if (k->type == MSGPACK_OBJECT_STR) { - key = k->via.str.ptr; - key_len = k->via.str.size; - } - else { - key = k->via.bin.ptr; - key_len = k->via.bin.size; - } - - /* Store value */ - if (v->type == MSGPACK_OBJECT_NIL) { - /* Missing values are Null by default in InfluxDB */ - continue; - } - else if (v->type == MSGPACK_OBJECT_BOOLEAN) { - if (v->via.boolean) { - val = "TRUE"; - val_len = 4; - } - else { - val = "FALSE"; - val_len = 5; - } - } - else if (v->type == MSGPACK_OBJECT_POSITIVE_INTEGER) { - val = tmp; - val_len = snprintf(tmp, sizeof(tmp) - 1, "%" PRIu64, v->via.u64); - } - else if (v->type == MSGPACK_OBJECT_NEGATIVE_INTEGER) { - val = tmp; - val_len = snprintf(tmp, sizeof(tmp) - 1, "%" PRId64, v->via.i64); - } - else if (v->type == MSGPACK_OBJECT_FLOAT || v->type == MSGPACK_OBJECT_FLOAT32) { - val = tmp; - val_len = snprintf(tmp, sizeof(tmp) - 1, "%f", v->via.f64); - } - else if (v->type == MSGPACK_OBJECT_STR) { - /* String value */ - quote = FLB_TRUE; - val = v->via.str.ptr; - val_len = v->via.str.size; - } - else if (v->type == MSGPACK_OBJECT_BIN) { - /* Bin value */ - quote = FLB_TRUE; - val = v->via.bin.ptr; - val_len = v->via.bin.size; - } - - if (!val || !key) { - continue; - } - - /* is this a string ? */ - if (quote == FLB_TRUE) { - ret = flb_utils_write_str_buf(val, val_len, - &str, &str_size); - if (ret == -1) { - flb_errno(); - goto error; - } - - val = str; - val_len = str_size; - } - - if (is_tagged_key(ctx, key, key_len, v->type)) { - /* Append key/value data into the bulk_head */ - ret = influxdb_bulk_append_kv(bulk_head, - key, key_len, - val, val_len, - false); - } - else { - /* Append key/value data into the bulk_body */ - ret = influxdb_bulk_append_kv(bulk_body, - key, key_len, - val, val_len, - quote); - } - - if (quote == FLB_TRUE) { - flb_free(str); - str_size = 0; - } - - if (ret == -1) { - flb_plg_error(ctx->ins, "cannot append key/value"); - goto error; - } - } - - /* Check have data fields */ - if (bulk_body->len > 0) { - /* Modify timestamp in avoidance of duplication */ - influxdb_tsmod(&tm, &ctx->ts_dupe, &ctx->ts_last); - /* Append the timestamp */ - ret = influxdb_bulk_append_timestamp(bulk_body, &tm); - if (ret == -1) { - flb_plg_error(ctx->ins, "cannot append timestamp"); - goto error; - } - - /* Append collected data to final bulk */ - if (influxdb_bulk_append_bulk(bulk, bulk_head, '\n') != 0 || - influxdb_bulk_append_bulk(bulk, bulk_body, ' ') != 0) { - goto error; - } - } - else { - flb_plg_warn(ctx->ins, "skip send record, " - "since no record available " - "or all fields are tagged in record"); - /* Following records maybe ok, so continue processing */ - } - - /* Reset bulk_head and bulk_body */ - bulk_head->len = 0; - bulk_body->len = 0; - } - - flb_log_event_decoder_destroy(&log_decoder); - - *out_size = bulk->len; - buf = bulk->ptr; - - /* - * Note: we don't destroy the bulk as we need to keep the allocated - * buffer with the data. Instead we just release the bulk context and - * return the bulk->ptr buffer - */ - flb_free(bulk); - influxdb_bulk_destroy(bulk_head); - influxdb_bulk_destroy(bulk_body); - - return buf; - -error: - if (bulk != NULL) { - influxdb_bulk_destroy(bulk); - } - if (bulk_head != NULL) { - influxdb_bulk_destroy(bulk_head); - } - if (bulk_body != NULL) { - influxdb_bulk_destroy(bulk_body); - } - - flb_log_event_decoder_destroy(&log_decoder); - - return NULL; -} - -static int cb_influxdb_init(struct flb_output_instance *ins, struct flb_config *config, - void *data) -{ - int ret; - int io_flags = 0; - const char *tmp; - struct flb_upstream *upstream; - struct flb_influxdb *ctx; - - /* Set default network configuration */ - flb_output_net_default(FLB_INFLUXDB_HOST, FLB_INFLUXDB_PORT, ins); - - /* Allocate plugin context */ - ctx = flb_calloc(1, sizeof(struct flb_influxdb)); - if (!ctx) { - flb_errno(); - return -1; - } - ctx->ins = ins; - - /* Register context with plugin instance */ - flb_output_set_context(ins, ctx); - - /* - * This plugin instance uses the HTTP client interface, let's register - * it debugging callbacks. - */ - flb_output_set_http_debug_callbacks(ins); - - /* Load config map */ - ret = flb_output_config_map_set(ins, (void *) ctx); - if (ret == -1) { - return -1; - } - - if (ins->use_tls == FLB_TRUE) { - io_flags = FLB_IO_TLS; - } - else { - io_flags = FLB_IO_TCP; - } - - /* sequence tag */ - tmp = flb_output_get_property("sequence_tag", ins); - if (!tmp) { - ctx->seq_name = flb_strdup("_seq"); - } - else if (strcmp(tmp, "off") == 0) { - ctx->seq_name = flb_strdup(""); - } - else { - ctx->seq_name = flb_strdup(tmp); - } - ctx->seq_len = strlen(ctx->seq_name); - - if (ctx->custom_uri) { - /* custom URI endpoint (e.g: Grafana */ - if (ctx->custom_uri[0] != '/') { - flb_plg_error(ctx->ins, - "'custom_uri' value must start wih a forward slash '/'"); - return -1; - } - snprintf(ctx->uri, sizeof(ctx->uri) - 1, "%s", ctx->custom_uri); - } - else if (ctx->bucket) { - /* bucket: api v2 */ - snprintf(ctx->uri, sizeof(ctx->uri) - 1, - "/api/v2/write?org=%s&bucket=%s&precision=ns", - ctx->organization, ctx->bucket); - } - else { - snprintf(ctx->uri, sizeof(ctx->uri) - 1, - "/write?db=%s&precision=n", - ctx->database); - } - - if (ins->host.ipv6 == FLB_TRUE) { - io_flags |= FLB_IO_IPV6; - } - - - /* Tag_Keys */ - tmp = flb_output_get_property("tag_keys", ins); - if (tmp) { - ctx->tag_keys = flb_utils_split(tmp, ' ', 256); - } - else { - ctx->tag_keys = NULL; - } - - /* Prepare an upstream handler */ - upstream = flb_upstream_create(config, - ins->host.name, - ins->host.port, - io_flags, - ins->tls); - if (!upstream) { - flb_free(ctx); - return -1; - } - ctx->u = upstream; - ctx->seq = 0; - flb_output_upstream_set(ctx->u, ins); - - flb_time_zero(&ctx->ts_dupe); - flb_time_zero(&ctx->ts_last); - - flb_plg_debug(ctx->ins, "host=%s port=%i", ins->host.name, ins->host.port); - - return 0; -} - -static int format_metrics(struct flb_output_instance *ins, - const void *data, size_t bytes, - char **out_buf, size_t *out_size) -{ - int ret; - size_t off = 0; - cfl_sds_t text; - struct cmt *cmt = NULL; - - /* get cmetrics context */ - ret = cmt_decode_msgpack_create(&cmt, (char *) data, bytes, &off); - if (ret != 0) { - flb_plg_error(ins, "could not process metrics payload"); - return -1; - } - - /* convert to text representation */ - text = cmt_encode_influx_create(cmt); - if (!text) { - cmt_destroy(cmt); - return -1; - } - - /* destroy cmt context */ - cmt_destroy(cmt); - - *out_buf = text; - *out_size = flb_sds_len(text); - - return 0; -} - -static void cb_influxdb_flush(struct flb_event_chunk *event_chunk, - struct flb_output_flush *out_flush, - struct flb_input_instance *i_ins, - void *out_context, - struct flb_config *config) -{ - int ret; - int out_ret = FLB_OK; - int is_metric = FLB_FALSE; - size_t b_sent; - size_t bytes_out; - char *pack; - char tmp[128]; - struct mk_list *head; - struct flb_connection *u_conn; - struct flb_http_client *c; - struct flb_config_map_val *mv; - struct flb_slist_entry *key = NULL; - struct flb_slist_entry *val = NULL; - struct flb_influxdb *ctx = out_context; - - /* Convert format: metrics / logs */ - if (event_chunk->type == FLB_EVENT_TYPE_METRICS) { - /* format metrics */ - ret = format_metrics(ctx->ins, - (char *) event_chunk->data, - event_chunk->size, - &pack, &bytes_out); - if (ret == -1) { - FLB_OUTPUT_RETURN(FLB_ERROR); - } - is_metric = FLB_TRUE; - } - else { - /* format logs */ - pack = influxdb_format(event_chunk->tag, flb_sds_len(event_chunk->tag), - event_chunk->data, event_chunk->size, - &bytes_out, ctx); - if (!pack) { - FLB_OUTPUT_RETURN(FLB_ERROR); - } - } - - /* Get upstream connection */ - u_conn = flb_upstream_conn_get(ctx->u); - if (!u_conn) { - if (is_metric) { - cmt_encode_influx_destroy(pack); - } - else { - flb_free(pack); - } - FLB_OUTPUT_RETURN(FLB_RETRY); - } - - /* Compose HTTP Client request */ - c = flb_http_client(u_conn, FLB_HTTP_POST, ctx->uri, - pack, bytes_out, NULL, 0, NULL, 0); - flb_http_add_header(c, "User-Agent", 10, "Fluent-Bit", 10); - - if (ctx->http_token) { - ret = snprintf(tmp, sizeof(tmp) - 1, "Token %s", ctx->http_token); - flb_http_add_header(c, FLB_HTTP_HEADER_AUTH, sizeof FLB_HTTP_HEADER_AUTH - 1, tmp, ret); - } - else if (ctx->http_user && ctx->http_passwd) { - flb_http_basic_auth(c, ctx->http_user, ctx->http_passwd); - } - - /* Append custom headers if any */ - flb_config_map_foreach(head, mv, ctx->headers) { - key = mk_list_entry_first(mv->val.list, struct flb_slist_entry, _head); - val = mk_list_entry_last(mv->val.list, struct flb_slist_entry, _head); - - flb_http_add_header(c, - key->str, flb_sds_len(key->str), - val->str, flb_sds_len(val->str)); - } - - /* Map debug callbacks */ - flb_http_client_debug(c, ctx->ins->callback); - - ret = flb_http_do(c, &b_sent); - if (ret == 0) { - if (c->resp.status != 200 && c->resp.status != 204) { - if (c->resp.payload_size > 0) { - flb_plg_error(ctx->ins, "http_status=%i\n%s", - c->resp.status, c->resp.payload); - } - else { - flb_plg_debug(ctx->ins, "http_status=%i", - c->resp.status); - } - } - flb_plg_debug(ctx->ins, "http_do=%i OK", ret); - } - else { - flb_plg_error(ctx->ins, "http_do=%i", ret); - out_ret = FLB_RETRY; - } - - flb_http_client_destroy(c); - - if (is_metric) { - cmt_encode_influx_destroy(pack); - } - else { - flb_free(pack); - } - - /* Release the connection */ - flb_upstream_conn_release(u_conn); - - FLB_OUTPUT_RETURN(out_ret); -} - -static int cb_influxdb_exit(void *data, struct flb_config *config) -{ - struct flb_influxdb *ctx = data; - - if (!ctx) { - return 0; - } - - if (ctx->tag_keys) { - flb_utils_split_free(ctx->tag_keys); - } - - flb_upstream_destroy(ctx->u); - flb_free(ctx); - - return 0; -} - -int is_tagged_key(struct flb_influxdb *ctx, const char *key, int kl, int type) -{ - if (type == MSGPACK_OBJECT_STR) { - if (ctx->auto_tags) { - return FLB_TRUE; - } - } - - struct mk_list *head; - struct flb_split_entry *entry; - - if (ctx->tag_keys) { - mk_list_foreach(head, ctx->tag_keys) { - entry = mk_list_entry(head, struct flb_split_entry, _head); - if (kl == entry->len && strncmp(key, entry->value, kl) == 0) { - return FLB_TRUE; - } - } - } - - return FLB_FALSE; -} - -/* Configuration properties map */ -static struct flb_config_map config_map[] = { - { - FLB_CONFIG_MAP_STR, "database", "fluentbit", - 0, FLB_TRUE, offsetof(struct flb_influxdb, database), - "Set the database name." - }, - { - FLB_CONFIG_MAP_STR, "bucket", NULL, - 0, FLB_TRUE, offsetof(struct flb_influxdb, bucket), - "Specify the bucket name, used on InfluxDB API v2." - }, - - { - FLB_CONFIG_MAP_STR, "org", "fluent", - 0, FLB_TRUE, offsetof(struct flb_influxdb, organization), - "Set the Organization name." - }, - - { - FLB_CONFIG_MAP_STR, "sequence_tag", NULL, - 0, FLB_FALSE, 0, - "Specify the sequence tag." - }, - - { - FLB_CONFIG_MAP_STR, "uri", NULL, - 0, FLB_TRUE, offsetof(struct flb_influxdb, custom_uri), - "Specify a custom URI endpoint (must start with '/')." - }, - - { - FLB_CONFIG_MAP_STR, "http_user", NULL, - 0, FLB_TRUE, offsetof(struct flb_influxdb, http_user), - "HTTP Basic Auth username." - }, - - { - FLB_CONFIG_MAP_STR, "http_passwd", "", - 0, FLB_TRUE, offsetof(struct flb_influxdb, http_passwd), - "HTTP Basic Auth password." - }, - - { - FLB_CONFIG_MAP_STR, "http_token", NULL, - 0, FLB_TRUE, offsetof(struct flb_influxdb, http_token), - "Set InfluxDB HTTP Token API v2." - }, - - { - FLB_CONFIG_MAP_SLIST_1, "http_header", NULL, - FLB_CONFIG_MAP_MULT, FLB_TRUE, offsetof(struct flb_influxdb, headers), - "Add a HTTP header key/value pair. Multiple headers can be set" - }, - - { - FLB_CONFIG_MAP_BOOL, "auto_tags", "false", - 0, FLB_TRUE, offsetof(struct flb_influxdb, auto_tags), - "Automatically tag keys where value is string." - }, - - { - FLB_CONFIG_MAP_BOOL, "tag_keys", NULL, - 0, FLB_FALSE, 0, - "Space separated list of keys that needs to be tagged." - }, - - /* EOF */ - {0} -}; - -struct flb_output_plugin out_influxdb_plugin = { - .name = "influxdb", - .description = "InfluxDB Time Series", - .cb_init = cb_influxdb_init, - .cb_pre_run = NULL, - .cb_flush = cb_influxdb_flush, - .cb_exit = cb_influxdb_exit, - .config_map = config_map, - .flags = FLB_OUTPUT_NET | FLB_IO_OPT_TLS, - .event_type = FLB_OUTPUT_LOGS | FLB_OUTPUT_METRICS -}; diff --git a/fluent-bit/plugins/out_influxdb/influxdb.h b/fluent-bit/plugins/out_influxdb/influxdb.h deleted file mode 100644 index 13a72d4b8..000000000 --- a/fluent-bit/plugins/out_influxdb/influxdb.h +++ /dev/null @@ -1,78 +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. - */ - -#ifndef FLB_OUT_INFLUXDB_H -#define FLB_OUT_INFLUXDB_H - -#include <fluent-bit/flb_output.h> -#include <fluent-bit/flb_time.h> - -#define FLB_INFLUXDB_HOST "127.0.0.1" -#define FLB_INFLUXDB_PORT 8086 - -struct flb_influxdb { - uint64_t seq; - - char uri[2048]; - - /* v1 */ - /* database */ - flb_sds_t database; - - /* HTTP Auth */ - flb_sds_t http_user; - flb_sds_t http_passwd; - - // v2 - /* bucket */ - flb_sds_t bucket; - - /* organization */ - flb_sds_t organization; - - /* custom HTTP URI */ - flb_sds_t custom_uri; - - /* HTTP Token */ - flb_sds_t http_token; - - /* sequence tag */ - char *seq_name; - int seq_len; - - /* auto_tags: on/off */ - int auto_tags; - - /* tag_keys: space separated list of key */ - struct mk_list *tag_keys; - - /* Arbitrary HTTP headers */ - struct mk_list *headers; - - /* Upstream connection to the backend server */ - struct flb_upstream *u; - - /* used for incrementing identical timestamps */ - struct flb_time ts_dupe; - struct flb_time ts_last; - - struct flb_output_instance *ins; -}; - -#endif diff --git a/fluent-bit/plugins/out_influxdb/influxdb_bulk.c b/fluent-bit/plugins/out_influxdb/influxdb_bulk.c deleted file mode 100644 index 2542ee3c4..000000000 --- a/fluent-bit/plugins/out_influxdb/influxdb_bulk.c +++ /dev/null @@ -1,233 +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 <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <ctype.h> - -#include <fluent-bit.h> -#include "influxdb.h" -#include "influxdb_bulk.h" - -static const uint64_t ONE_BILLION = 1000000000; - -static int influxdb_escape(char *out, const char *str, int size, bool quote) { - int out_size = 0; - int i; - for (i = 0; i < size; ++i) { - char ch = str[i]; - if (quote ? (ch == '"' || ch == '\\') : (isspace(ch) || ch == ',' || ch == '=')) { - out[out_size++] = '\\'; - } else if (ch == '\\') { - out[out_size++] = '\\'; - } - out[out_size++] = ch; - } - return out_size; -} - -static int influxdb_bulk_buffer(struct influxdb_bulk *bulk, int required) -{ - int new_size; - int available; - char *ptr; - - available = (bulk->size - bulk->len); - if (available < required) { - new_size = bulk->size + available + required + INFLUXDB_BULK_CHUNK; - ptr = flb_realloc(bulk->ptr, new_size); - if (!ptr) { - flb_errno(); - return -1; - } - bulk->ptr = ptr; - bulk->size = new_size; - } - - return 0; -} - -struct influxdb_bulk *influxdb_bulk_create() -{ - struct influxdb_bulk *b; - - b = flb_malloc(sizeof(struct influxdb_bulk)); - if (!b) { - perror("calloc"); - return NULL; - } - - b->ptr = flb_malloc(INFLUXDB_BULK_CHUNK); - if (!b->ptr) { - perror("malloc"); - flb_free(b); - return NULL; - } - - b->size = INFLUXDB_BULK_CHUNK; - b->len = 0; - - return b; -} - -void influxdb_bulk_destroy(struct influxdb_bulk *bulk) -{ - if (bulk->size > 0) { - flb_free(bulk->ptr); - } - flb_free(bulk); -} - -int influxdb_bulk_append_header(struct influxdb_bulk *bulk, - const char *tag, int tag_len, - uint64_t seq_n, const char *seq, int seq_len) -{ - int ret; - int required; - - required = tag_len + 1 + seq_len + 1 + 32; - - /* Make sure we have enough space */ - ret = influxdb_bulk_buffer(bulk, required); - if (ret != 0) { - return -1; - } - - /* Tag, sequence and final space */ - memcpy(bulk->ptr + bulk->len, tag, tag_len); - bulk->len += tag_len; - - if (seq_len != 0) { - bulk->ptr[bulk->len] = ','; - bulk->len++; - - /* Sequence number */ - memcpy(bulk->ptr + bulk->len, seq, seq_len); - bulk->len += seq_len; - - bulk->ptr[bulk->len] = '='; - bulk->len++; - - ret = snprintf(bulk->ptr + bulk->len, 32, "%" PRIu64, seq_n); - bulk->len += ret; - } - - /* Add a NULL byte for debugging purposes */ - bulk->ptr[bulk->len] = '\0'; - - return 0; -} - -int influxdb_bulk_append_kv(struct influxdb_bulk *bulk, - const char *key, int k_len, - const char *val, int v_len, - int quote) -{ - int ret; - int required; - - /* Reserve double space for keys and values in case of escaping */ - required = k_len * 2 + 1 + v_len * 2 + 1 + 1; - if (quote) { - required += 2; - } - - /* Make sure we have enough space */ - ret = influxdb_bulk_buffer(bulk, required); - if (ret != 0) { - return -1; - } - - if (bulk->len > 0) { - bulk->ptr[bulk->len] = ','; - bulk->len++; - } - - /* key */ - bulk->len += influxdb_escape(bulk->ptr + bulk->len, key, k_len, false); - - /* separator */ - bulk->ptr[bulk->len] = '='; - bulk->len++; - - /* value */ - if (quote) { - bulk->ptr[bulk->len] = '"'; - bulk->len++; - } - bulk->len += influxdb_escape(bulk->ptr + bulk->len, val, v_len, quote); - if (quote) { - bulk->ptr[bulk->len] = '"'; - bulk->len++; - } - - /* Add a NULL byte for debugging purposes */ - bulk->ptr[bulk->len] = '\0'; - - return 0; -}; - -int influxdb_bulk_append_timestamp(struct influxdb_bulk *bulk, - struct flb_time *t) -{ - int ret; - int len; - uint64_t timestamp; - - /* Make sure we have enough space */ - ret = influxdb_bulk_buffer(bulk, 128); - if (ret != 0) { - return -1; - } - - /* Timestamp is in Nanoseconds */ - timestamp = (t->tm.tv_sec * ONE_BILLION) + t->tm.tv_nsec; - len = snprintf(bulk->ptr + bulk->len, 127, " %" PRIu64, timestamp); - if (len == -1) { - return -1; - } - bulk->len += len; - bulk->ptr[bulk->len] = '\0'; - - return 0; -}; - -int influxdb_bulk_append_bulk(struct influxdb_bulk *bulk_to, - struct influxdb_bulk *bulk_from, - char separator) -{ - if (influxdb_bulk_buffer(bulk_to, bulk_from->len + 2) != 0) { - return -1; - } - - if (bulk_to->len > 0) { - bulk_to->ptr[bulk_to->len] = separator; - bulk_to->len += 1; - } - - memcpy(bulk_to->ptr + bulk_to->len, - bulk_from->ptr, bulk_from->len * sizeof(char)); - bulk_to->len += bulk_from->len; - - /* Add a NULL byte for always terminating with NULL */ - bulk_to->ptr[bulk_to->len] = '\0'; - - return 0; -}; diff --git a/fluent-bit/plugins/out_influxdb/influxdb_bulk.h b/fluent-bit/plugins/out_influxdb/influxdb_bulk.h deleted file mode 100644 index 66f914528..000000000 --- a/fluent-bit/plugins/out_influxdb/influxdb_bulk.h +++ /dev/null @@ -1,54 +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. - */ - -#ifndef FLB_INFLUXDB_BULK_H -#define FLB_INFLUXDB_BULK_H - -#include <fluent-bit/flb_info.h> -#include <fluent-bit/flb_time.h> -#include <inttypes.h> - -#define INFLUXDB_BULK_CHUNK 4096 /* 4KB of buffer chunks */ - -struct influxdb_bulk { - char *ptr; - uint32_t len; - uint32_t size; -}; - -struct influxdb_bulk *influxdb_bulk_create(); - -int influxdb_bulk_append_header(struct influxdb_bulk *bulk, - const char *tag, int tag_len, - uint64_t seq_n, const char *seq, int seq_len); - -int influxdb_bulk_append_kv(struct influxdb_bulk *bulk, - const char *key, int k_len, - const char *val, int v_len, - int quote); - -int influxdb_bulk_append_bulk(struct influxdb_bulk *bulk_to, - struct influxdb_bulk *bulk_from, - char separator); - -void influxdb_bulk_destroy(struct influxdb_bulk *bulk); -int influxdb_bulk_append_timestamp(struct influxdb_bulk *bulk, - struct flb_time *t); - -#endif |