1 files changed, 94 insertions, 0 deletions
diff --git a/src/fluent-bit/plugins/out_td/td_http.c b/src/fluent-bit/plugins/out_td/td_http.c
new file mode 100644
index 000000000..539408044
--- /dev/null
+++ b/src/fluent-bit/plugins/out_td/td_http.c
@@ -0,0 +1,94 @@
+/* -*- 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_gzip.h>
+#include <fluent-bit/flb_config.h>
+#include <fluent-bit/flb_http_client.h>
+
+#include "td_config.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+
+#define TD_HTTP_HEADER_SIZE  512
+
+struct flb_http_client *td_http_client(struct flb_connection *u_conn,
+                                       void *data, size_t len,
+                                       char **body,
+                                       struct flb_td *ctx,
+                                       struct flb_config *config)
+{
+    int ret;
+    int pos = 0;
+    int api_len;
+    size_t gz_size;
+    void *gz_data;
+    char *tmp;
+    struct flb_http_client *c;
+
+    /* Compress data */
+    ret = flb_gzip_compress(data, len, &gz_data, &gz_size);
+    if (ret == -1) {
+        flb_plg_error(ctx->ins, "error compressing data");
+        return NULL;
+    }
+
+    /* Compose URI */
+    tmp = flb_malloc(512);
+    if (!tmp) {
+        flb_free(gz_data);
+        return NULL;
+    }
+    snprintf(tmp, 256,
+             "/v3/table/import/%s/%s/msgpack.gz",
+             ctx->db_name, ctx->db_table);
+
+    /* Create client */
+    c = flb_http_client(u_conn, FLB_HTTP_PUT, tmp,
+                        gz_data, gz_size, NULL, 0, NULL, 0);
+    if (!c) {
+        flb_free(tmp);
+        flb_free(gz_data);
+        return NULL;
+    }
+
+    /* Add custom headers */
+    tmp[pos++] = 'T';
+    tmp[pos++] = 'D';
+    tmp[pos++] = '1';
+    tmp[pos++] = ' ';
+
+    api_len = strlen(ctx->api);
+    memcpy(tmp + pos, ctx->api, api_len);
+    pos += api_len;
+
+    flb_http_add_header(c,
+                        "Authorization", 13,
+                        tmp, pos);
+    flb_http_add_header(c,
+                        "Content-Type", 12,
+                        "application/gzip", 16);
+    flb_free(tmp);
+    *body = gz_data;
+
+    return c;
+}