From 5c3b71f1671098387ce88936be5ae027c145b840 Mon Sep 17 00:00:00 2001
From: Athish Pranav D <105591739+Athishpranav2003@users.noreply.github.com>
Date: Fri, 4 Oct 2024 06:56:33 +0530
Subject: [PATCH] Added the prefix tag func (#4655)

Signed-off-by: Athish Pranav D <athishanna@gmail.com>
---
 lib/fluent/plugin/in_http.rb |  5 +++++
 test/plugin/test_in_http.rb  | 23 +++++++++++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/lib/fluent/plugin/in_http.rb b/lib/fluent/plugin/in_http.rb
index 1fe8f9985e..ea549cfbb9 100644
--- a/lib/fluent/plugin/in_http.rb
+++ b/lib/fluent/plugin/in_http.rb
@@ -84,6 +84,8 @@ class HttpInput < Input
     config_param :dump_error_log, :bool, default: true
     desc 'Add QUERY_ prefix query params to record'
     config_param :add_query_params, :bool, default: false
+    desc "Add prefix to incoming tag"
+    config_param :add_tag_prefix, :string, default: nil
 
     config_section :parse do
       config_set_default :@type, 'in_http'
@@ -120,6 +122,8 @@ def configure(conf)
         end
       end
 
+      raise Fluent::ConfigError, "'add_tag_prefix' parameter must not be empty" if @add_tag_prefix && @add_tag_prefix.empty?
+
       m = if @parser_configs.first['@type'] == 'in_http'
             @parser_msgpack = parser_create(usage: 'parser_in_http_msgpack', type: 'msgpack')
             @parser_msgpack.time_key = nil
@@ -203,6 +207,7 @@ def on_request(path_info, params)
       begin
         path = path_info[1..-1]  # remove /
         tag = path.split('/').join('.')
+        tag = "#{@add_tag_prefix}.#{tag}" if @add_tag_prefix
 
         mes = Fluent::MultiEventStream.new
         parse_params(params) do |record_time, record|
diff --git a/test/plugin/test_in_http.rb b/test/plugin/test_in_http.rb
index 88d8eacdad..3b9107f224 100644
--- a/test/plugin/test_in_http.rb
+++ b/test/plugin/test_in_http.rb
@@ -1028,6 +1028,29 @@ def test_add_query_params
     assert_equal events, d.events
   end
 
+  def test_add_tag_prefix
+    d = create_driver(config + "add_tag_prefix test")
+    assert_equal "test", d.instance.add_tag_prefix
+
+    time = event_time("2011-01-02 13:14:15.123 UTC")
+    float_time = time.to_f
+
+    events = [
+      ["tag1", time, {"a"=>1}],
+    ]
+    res_codes = []
+
+    d.run(expect_records: 1) do
+      events.each do |tag, t, record|
+        res = post("/#{tag}", {"json"=>record.to_json, "time"=>float_time.to_s})
+        res_codes << res.code
+      end
+    end
+
+    assert_equal ["200"], res_codes
+    assert_equal [["test.tag1", time, {"a"=>1}]], d.events
+  end
+
   $test_in_http_connection_object_ids = []
   $test_in_http_content_types = []
   $test_in_http_content_types_flag = false