1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright Epic Games, Inc. All Rights Reserved.
#include "otlphttp.h"
#include <zencore/config.h>
#include <zencore/process.h>
#include <zencore/session.h>
#include <zencore/system.h>
#include <zentelemetry/otlpencoder.h>
#include <protozero/buffer_string.hpp>
#include <protozero/pbf_builder.hpp>
#include <cstdio>
#if ZEN_WITH_OTEL
namespace zen::logging {
//////////////////////////////////////////////////////////////////////////
//
// Important note: in general we cannot use ZEN_WARN/ZEN_ERROR etc in this
// file as it could cause recursive logging calls when we attempt to log
// errors from the OTLP HTTP client itself.
//
OtelHttpProtobufSink::OtelHttpProtobufSink(const std::string_view& Uri) : m_OtelHttp(Uri)
{
m_Encoder.AddResourceAttribute("service.name", "zenserver");
m_Encoder.AddResourceAttribute("service.instance.id", GetSessionIdString());
m_Encoder.AddResourceAttribute("service.namespace", "zen");
m_Encoder.AddResourceAttribute("service.version", ZEN_CFG_VERSION);
m_Encoder.AddResourceAttribute("host.name", GetMachineName());
m_Encoder.AddResourceAttribute("session.id", GetSessionIdString());
m_Encoder.AddResourceAttribute("process.id", zen::GetCurrentProcessId());
m_TraceRecorder = new TraceRecorder(this);
otel::SetTraceRecorder(m_TraceRecorder);
}
OtelHttpProtobufSink::~OtelHttpProtobufSink()
{
otel::SetTraceRecorder({});
}
void
OtelHttpProtobufSink::CheckPostResult(const HttpClient::Response& Result, const char* Endpoint) noexcept
{
if (!Result.IsSuccess())
{
uint32_t PrevFailures = m_ConsecutivePostFailures.fetch_add(1);
if (PrevFailures < kMaxReportedFailures)
{
fprintf(stderr, "OtelHttpProtobufSink: %s\n", Result.ErrorMessage(Endpoint).c_str());
if (PrevFailures + 1 == kMaxReportedFailures)
{
fprintf(stderr, "OtelHttpProtobufSink: suppressing further export errors\n");
}
}
}
else
{
m_ConsecutivePostFailures.store(0);
}
}
void
OtelHttpProtobufSink::RecordSpans(zen::otel::TraceId Trace, std::span<const zen::otel::Span*> Spans)
{
try
{
std::string Data = m_Encoder.FormatOtelTrace(Trace, Spans);
IoBuffer Payload{IoBuffer::Wrap, Data.data(), Data.size()};
Payload.SetContentType(ZenContentType::kProtobuf);
HttpClient::Response Result = m_OtelHttp.Post("/v1/traces", Payload);
CheckPostResult(Result, "POST /v1/traces");
}
catch (const std::exception& Ex)
{
fprintf(stderr, "OtelHttpProtobufSink: exception exporting traces: %s\n", Ex.what());
}
}
void
OtelHttpProtobufSink::TraceRecorder::RecordSpans(zen::otel::TraceId Trace, std::span<const zen::otel::Span*> Spans)
{
m_Sink->RecordSpans(Trace, Spans);
}
void
OtelHttpProtobufSink::Log(const LogMessage& Msg)
{
try
{
std::string Data = m_Encoder.FormatOtelProtobuf(Msg);
IoBuffer Payload{IoBuffer::Wrap, Data.data(), Data.size()};
Payload.SetContentType(ZenContentType::kProtobuf);
HttpClient::Response Result = m_OtelHttp.Post("/v1/logs", Payload);
CheckPostResult(Result, "POST /v1/logs");
}
catch (const std::exception& Ex)
{
fprintf(stderr, "OtelHttpProtobufSink: exception exporting logs: %s\n", Ex.what());
}
}
void
OtelHttpProtobufSink::Flush()
{
}
} // namespace zen::logging
#endif
|