From a6cc8dec3fb39dd43fd509d62dde755293a2dc27 Mon Sep 17 00:00:00 2001
From: Tom Tan <totan@microsoft.com>
Date: Sat, 17 Feb 2024 11:21:10 -0800
Subject: [PATCH] [geneva-exporter] add traceflag override method to fluentd
 exporter

---
 .../opentelemetry/exporters/fluentd/trace/recordable.h        | 2 ++
 exporters/fluentd/src/trace/fluentd_exporter.cc               | 2 +-
 exporters/fluentd/src/trace/recordable.cc                     | 4 ++++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/exporters/fluentd/include/opentelemetry/exporters/fluentd/trace/recordable.h b/exporters/fluentd/include/opentelemetry/exporters/fluentd/trace/recordable.h
index 4b3fb8294..378f7bcba 100644
--- a/exporters/fluentd/include/opentelemetry/exporters/fluentd/trace/recordable.h
+++ b/exporters/fluentd/include/opentelemetry/exporters/fluentd/trace/recordable.h
@@ -71,6 +71,8 @@ class Recordable final : public sdk::trace::Recordable {
       const opentelemetry::sdk::instrumentationscope::InstrumentationScope
           &instrumentation_scope) noexcept override;
 
+  void SetTraceFlags(opentelemetry::trace::TraceFlags flags) noexcept override;
+
 private:
   std::string tag_;
   nlohmann::json events_;
diff --git a/exporters/fluentd/src/trace/fluentd_exporter.cc b/exporters/fluentd/src/trace/fluentd_exporter.cc
index 485f490da..784294b91 100644
--- a/exporters/fluentd/src/trace/fluentd_exporter.cc
+++ b/exporters/fluentd/src/trace/fluentd_exporter.cc
@@ -60,7 +60,7 @@ FluentdExporter::FluentdExporter()
  */
 std::unique_ptr<sdk::trace::Recordable>
 FluentdExporter::MakeRecordable() noexcept {
-  return std::unique_ptr<sdk::trace::Recordable>(new Recordable);
+  return std::unique_ptr<sdk::trace::Recordable>(new opentelemetry::exporter::fluentd::trace::Recordable());
 }
 
 /**
diff --git a/exporters/fluentd/src/trace/recordable.cc b/exporters/fluentd/src/trace/recordable.cc
index 010c02d4e..77527a8b3 100644
--- a/exporters/fluentd/src/trace/recordable.cc
+++ b/exporters/fluentd/src/trace/recordable.cc
@@ -148,6 +148,10 @@ void Recordable::SetInstrumentationScope(
       instrumentation_scope.GetVersion();
 }
 
+void Recordable::SetTraceFlags(opentelemetry::trace::TraceFlags flags) noexcept {
+    // TODO: process trace flags
+}
+
 } // namespace trace
 } // namespace fluentd
 } // namespace exporter