134 lines
3.9 KiB
C++
134 lines
3.9 KiB
C++
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
|
|
#include "opentelemetry/common/key_value_iterable.h"
|
|
#include "opentelemetry/plugin/detail/tracer_handle.h" // IWYU pragma: export
|
|
#include "opentelemetry/trace/span_context_kv_iterable.h"
|
|
#include "opentelemetry/trace/tracer.h"
|
|
#include "opentelemetry/version.h"
|
|
|
|
OPENTELEMETRY_BEGIN_NAMESPACE
|
|
namespace plugin
|
|
{
|
|
|
|
class DynamicLibraryHandle;
|
|
|
|
class Span final : public trace::Span
|
|
{
|
|
public:
|
|
Span(std::shared_ptr<trace::Tracer> &&tracer, const nostd::shared_ptr<trace::Span> &span) noexcept
|
|
: tracer_{std::move(tracer)}, span_{span}
|
|
{}
|
|
|
|
// trace::Span
|
|
void SetAttribute(nostd::string_view name, const common::AttributeValue &value) noexcept override
|
|
{
|
|
span_->SetAttribute(name, value);
|
|
}
|
|
|
|
void AddEvent(nostd::string_view name) noexcept override { span_->AddEvent(name); }
|
|
|
|
void AddEvent(nostd::string_view name, common::SystemTimestamp timestamp) noexcept override
|
|
{
|
|
span_->AddEvent(name, timestamp);
|
|
}
|
|
|
|
void AddEvent(nostd::string_view name,
|
|
const common::KeyValueIterable &attributes) noexcept override
|
|
{
|
|
span_->AddEvent(name, attributes);
|
|
}
|
|
|
|
void AddEvent(nostd::string_view name,
|
|
common::SystemTimestamp timestamp,
|
|
const common::KeyValueIterable &attributes) noexcept override
|
|
{
|
|
span_->AddEvent(name, timestamp, attributes);
|
|
}
|
|
|
|
#if OPENTELEMETRY_ABI_VERSION_NO >= 2
|
|
void AddLink(const trace::SpanContext &target,
|
|
const common::KeyValueIterable &attrs) noexcept override
|
|
{
|
|
span_->AddLink(target, attrs);
|
|
}
|
|
|
|
void AddLinks(const trace::SpanContextKeyValueIterable &links) noexcept override
|
|
{
|
|
span_->AddLinks(links);
|
|
}
|
|
#endif
|
|
|
|
void SetStatus(trace::StatusCode code, nostd::string_view description) noexcept override
|
|
{
|
|
span_->SetStatus(code, description);
|
|
}
|
|
|
|
void UpdateName(nostd::string_view name) noexcept override { span_->UpdateName(name); }
|
|
|
|
void End(const trace::EndSpanOptions &options = {}) noexcept override { span_->End(options); }
|
|
|
|
bool IsRecording() const noexcept override { return span_->IsRecording(); }
|
|
|
|
trace::SpanContext GetContext() const noexcept override { return span_->GetContext(); }
|
|
|
|
private:
|
|
std::shared_ptr<trace::Tracer> tracer_;
|
|
nostd::shared_ptr<trace::Span> span_;
|
|
};
|
|
|
|
class Tracer final : public trace::Tracer, public std::enable_shared_from_this<Tracer>
|
|
{
|
|
public:
|
|
Tracer(std::shared_ptr<DynamicLibraryHandle> library_handle,
|
|
std::unique_ptr<TracerHandle> &&tracer_handle) noexcept
|
|
: library_handle_{std::move(library_handle)}, tracer_handle_{std::move(tracer_handle)}
|
|
{
|
|
#if OPENTELEMETRY_ABI_VERSION_NO >= 2
|
|
UpdateEnabled(true);
|
|
#endif
|
|
}
|
|
|
|
// trace::Tracer
|
|
nostd::shared_ptr<trace::Span> StartSpan(
|
|
nostd::string_view name,
|
|
const common::KeyValueIterable &attributes,
|
|
const trace::SpanContextKeyValueIterable &links,
|
|
const trace::StartSpanOptions &options = {}) noexcept override
|
|
{
|
|
auto span = tracer_handle_->tracer().StartSpan(name, attributes, links, options);
|
|
if (span == nullptr)
|
|
{
|
|
return nostd::shared_ptr<trace::Span>(nullptr);
|
|
}
|
|
return nostd::shared_ptr<trace::Span>{new (std::nothrow) Span{this->shared_from_this(), span}};
|
|
}
|
|
|
|
#if OPENTELEMETRY_ABI_VERSION_NO == 1
|
|
|
|
void ForceFlushWithMicroseconds(uint64_t timeout) noexcept override
|
|
{
|
|
tracer_handle_->tracer().ForceFlushWithMicroseconds(timeout);
|
|
}
|
|
|
|
void CloseWithMicroseconds(uint64_t timeout) noexcept override
|
|
{
|
|
tracer_handle_->tracer().CloseWithMicroseconds(timeout);
|
|
}
|
|
|
|
#endif /* OPENTELEMETRY_ABI_VERSION_NO */
|
|
|
|
private:
|
|
// Note: The order is important here.
|
|
//
|
|
// It's undefined behavior to close the library while a loaded tracer is still active.
|
|
std::shared_ptr<DynamicLibraryHandle> library_handle_;
|
|
std::unique_ptr<TracerHandle> tracer_handle_;
|
|
};
|
|
} // namespace plugin
|
|
OPENTELEMETRY_END_NAMESPACE
|