/proc/self/cwd/source/extensions/tracers/skywalking/tracer.cc
Line | Count | Source (jump to first uncovered line) |
1 | | #include "source/extensions/tracers/skywalking/tracer.h" |
2 | | |
3 | | #include <string> |
4 | | |
5 | | namespace Envoy { |
6 | | namespace Extensions { |
7 | | namespace Tracers { |
8 | | namespace SkyWalking { |
9 | | |
10 | | namespace { |
11 | | static constexpr absl::string_view StatusCodeTag = "status_code"; |
12 | | static constexpr absl::string_view UrlTag = "url"; |
13 | | } // namespace |
14 | | |
15 | 0 | const Http::LowerCaseString& skywalkingPropagationHeaderKey() { |
16 | 0 | CONSTRUCT_ON_FIRST_USE(Http::LowerCaseString, "sw8"); |
17 | 0 | } |
18 | | |
19 | 0 | void Span::setTag(absl::string_view name, absl::string_view value) { |
20 | 0 | if (name == Tracing::Tags::get().HttpUrl) { |
21 | 0 | span_entity_->addTag(UrlTag.data(), std::string(value)); |
22 | 0 | } else if (name == Tracing::Tags::get().HttpStatusCode) { |
23 | 0 | span_entity_->addTag(StatusCodeTag.data(), std::string(value)); |
24 | 0 | } else if (name == Tracing::Tags::get().Error) { |
25 | 0 | span_entity_->setErrorStatus(); |
26 | 0 | span_entity_->addTag(std::string(name), std::string(value)); |
27 | 0 | } else if (name == Tracing::Tags::get().PeerAddress) { |
28 | 0 | span_entity_->setPeer(std::string(value)); |
29 | 0 | } else { |
30 | 0 | span_entity_->addTag(std::string(name), std::string(value)); |
31 | 0 | } |
32 | 0 | } |
33 | | |
34 | 0 | void Span::setSampled(bool do_sample) { |
35 | | // Sampling status is always true on SkyWalking. But with disabling skip_analysis, |
36 | | // this span can't be analyzed. |
37 | 0 | if (!do_sample) { |
38 | 0 | span_entity_->setSkipAnalysis(); |
39 | 0 | } |
40 | 0 | } |
41 | | |
42 | 0 | void Span::log(SystemTime, const std::string& event) { span_entity_->addLog(EMPTY_STRING, event); } |
43 | | |
44 | 0 | void Span::finishSpan() { |
45 | 0 | span_entity_->endSpan(); |
46 | 0 | parent_tracer_.sendSegment(tracing_context_); |
47 | 0 | } |
48 | | |
49 | | void Span::injectContext(Tracing::TraceContext& trace_context, |
50 | 0 | const Upstream::HostDescriptionConstSharedPtr& upstream) { |
51 | 0 | absl::string_view remote_address = |
52 | 0 | upstream != nullptr ? upstream->address()->asStringView() : trace_context.host(); |
53 | |
|
54 | 0 | auto sw8_header = |
55 | 0 | tracing_context_->createSW8HeaderValue({remote_address.data(), remote_address.size()}); |
56 | 0 | if (sw8_header.has_value()) { |
57 | 0 | trace_context.setByReferenceKey(skywalkingPropagationHeaderKey(), sw8_header.value()); |
58 | | |
59 | | // Rewrite operation name with latest upstream request path for the EXIT span. |
60 | 0 | absl::string_view upstream_request_path = trace_context.path(); |
61 | 0 | span_entity_->setOperationName({upstream_request_path.data(), upstream_request_path.size()}); |
62 | 0 | } |
63 | 0 | } |
64 | | |
65 | 0 | Tracing::SpanPtr Span::spawnChild(const Tracing::Config&, const std::string&, SystemTime) { |
66 | | // Reuse operation name of parent span by default. |
67 | 0 | return std::make_unique<Span>(span_entity_->operationName(), span_entity_->spanLayer(), *this, |
68 | 0 | tracing_context_, parent_tracer_); |
69 | 0 | } |
70 | | |
71 | 0 | Tracer::Tracer(TraceSegmentReporterPtr reporter) : reporter_(std::move(reporter)) {} |
72 | | |
73 | 0 | void Tracer::sendSegment(TracingContextPtr segment_context) { |
74 | 0 | ASSERT(reporter_); |
75 | 0 | if (segment_context->readyToSend()) { |
76 | 0 | reporter_->report(std::move(segment_context)); |
77 | 0 | } |
78 | 0 | } |
79 | | |
80 | | Tracing::SpanPtr Tracer::startSpan(absl::string_view name, absl::string_view protocol, |
81 | 0 | TracingContextPtr tracing_context) { |
82 | 0 | return std::make_unique<Span>(name, protocol, tracing_context, *this); |
83 | 0 | } |
84 | | } // namespace SkyWalking |
85 | | } // namespace Tracers |
86 | | } // namespace Extensions |
87 | | } // namespace Envoy |