概要

掲題の通り
公式のコードに変更を加えているので、自己責任で利用してください
Lambda から NewRelic にログを転送するときに任意のタグをつけて送りたかった
- NewRelic の Logs の画面で任意のタグでフィルタリングしたかったため
fluentd ではできそうだった
- http://katzchang.hatenablog.com/entry/2019/12/10/132210

方法

そもそも Lambda のスクリプトは以下にある
- https://github.com/newrelic/aws-log-ingestion.git
今回は、 function.py (下記リンクの箇所)に任意のタグを追加することで実現した
https://github.com/newrelic/aws-log-ingestion/blob/59665cd2951fdc7ea3e50d92d2911700f19dc97e/src/function.py#L448
diff はこんな感じ
- Lambda に CUSTOM_ATTRIBUTES という環境変数を設定することで、任意のタグを注入できる

diff --git a/src/function.py b/src/function.py
index 937c1b6..b242aba 100644
--- a/src/function.py
+++ b/src/function.py
@@ -371,6 +371,12 @@ def _reconstruct_infra_data(context, entry, logEvents):
 #  NR Logging  #
 ################
 
+def _get_custom_atrributes():
+    """
+    This functions get attribute key from env vars.
+    """
+    return os.getenv("CUSTOM_ATTRIBUTES", "{}")
+
 
 def _logging_enabled():
     """
@@ -447,6 +453,7 @@ def _package_log_payload(data):
                         "logStream": entry["logStream"],
                         "logGroup": entry["logGroup"],
                     },
+                    "custom": _get_custom_atrributes(),
                 }
             },
             "logs": log_messages,

結果

CUSTOM_ATTRIBUTES = "{\"group\":\"group-x\"}" とした場合
Logs 画面の UI にも反映されている
ログデータの詳細は以下