aboutsummaryrefslogtreecommitdiff
path: root/src/zenserver/diag/otlphttp.cpp
blob: d62ccccb620bde60decd58f1efb9e02008219a66 (plain) (blame)
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
// 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>

#if ZEN_WITH_OTEL

namespace zen::logging {

//////////////////////////////////////////////////////////////////////////

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::RecordSpans(zen::otel::TraceId Trace, std::span<const zen::otel::Span*> Spans)
{
	std::string Data = m_Encoder.FormatOtelTrace(Trace, Spans);

	IoBuffer Payload{IoBuffer::Wrap, Data.data(), Data.size()};
	Payload.SetContentType(ZenContentType::kProtobuf);

	auto Result = m_OtelHttp.Post("/v1/traces", Payload);
}

void
OtelHttpProtobufSink::TraceRecorder::RecordSpans(zen::otel::TraceId Trace, std::span<const zen::otel::Span*> Spans)
{
	m_Sink->RecordSpans(Trace, Spans);
}

void
OtelHttpProtobufSink::log(const spdlog::details::log_msg& Msg)
{
	{
		std::string Data = m_Encoder.FormatOtelProtobuf(Msg);

		IoBuffer Payload{IoBuffer::Wrap, Data.data(), Data.size()};
		Payload.SetContentType(ZenContentType::kProtobuf);

		auto Result = m_OtelHttp.Post("/v1/logs", Payload);
	}

	{
		std::string Data = m_Encoder.FormatOtelMetrics();

		IoBuffer Payload{IoBuffer::Wrap, Data.data(), Data.size()};
		Payload.SetContentType(ZenContentType::kProtobuf);

		auto Result = m_OtelHttp.Post("/v1/metrics", Payload);
	}
}
void
OtelHttpProtobufSink::flush()
{
}

}  // namespace zen::logging

#endif