span_guard.hpp

#pragma once
 
#include <exception>
 
#include <grpcpp/support/status.h>
 
#include <viam/sdk/common/grpc_fwd.hpp>
 
#ifdef VIAMCPPSDK_OPENTELEMETRY_TRACING
#include <opentelemetry/trace/scope.h>
#include <opentelemetry/trace/span.h>
#endif
 
namespace viam {
namespace sdk {
namespace impl {
 
class ServerSpanGuard {
   public:
    explicit ServerSpanGuard(const GrpcServerContext* ctx, const char* method) noexcept;
    ~ServerSpanGuard() noexcept;  // NOLINT(performance-trivially-destructible)
 
    ::grpc::Status commit(::grpc::Status status) noexcept;
 
    void record_exception(const std::exception& xcp) noexcept;
 
    void record_unknown_exception() noexcept;
 
    ServerSpanGuard(const ServerSpanGuard&) = delete;
    ServerSpanGuard& operator=(const ServerSpanGuard&) = delete;
 
#ifdef VIAMCPPSDK_OPENTELEMETRY_TRACING
   private:
    // Builds the span and makes it active; runs before `scope_` in the constructor init list.
    static opentelemetry::nostd::shared_ptr<opentelemetry::trace::Span> start_span_(
        const GrpcServerContext* ctx, const char* method) noexcept;
 
    opentelemetry::nostd::shared_ptr<opentelemetry::trace::Span> span_;
    opentelemetry::trace::Scope scope_;
    bool committed_ = false;
#endif
};
 
void inject_trace_context(GrpcClientContext* ctx) noexcept;
 
#ifdef VIAMCPPSDK_OPENTELEMETRY_TRACING
void record_exception(opentelemetry::trace::Span* span, const std::exception& xcp) noexcept;
 
void record_unknown_exception(opentelemetry::trace::Span* span) noexcept;
#endif
 
}  // namespace impl
}  // namespace sdk
}  // namespace viam