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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
#include "mock_span.h"
#include <cstdio>
#include <random>
namespace opentracing {
BEGIN_OPENTRACING_ABI_NAMESPACE
namespace mocktracer {
static uint64_t GenerateId() {
static thread_local std::mt19937_64 rand_source{std::random_device()()};
return static_cast<uint64_t>(rand_source());
}
static std::tuple<SystemTime, SteadyTime> ComputeStartTimestamps(
const SystemTime& start_system_timestamp,
const SteadyTime& start_steady_timestamp) {
// If neither the system nor steady timestamps are set, get the tme from the
// respective clocks; otherwise, use the set timestamp to initialize the
// other.
if (start_system_timestamp == SystemTime() &&
start_steady_timestamp == SteadyTime()) {
return std::tuple<SystemTime, SteadyTime>{SystemClock::now(),
SteadyClock::now()};
}
if (start_system_timestamp == SystemTime()) {
return std::tuple<SystemTime, SteadyTime>{
opentracing::convert_time_point<SystemClock>(start_steady_timestamp),
start_steady_timestamp};
}
if (start_steady_timestamp == SteadyTime()) {
return std::tuple<SystemTime, SteadyTime>{
start_system_timestamp,
opentracing::convert_time_point<SteadyClock>(start_system_timestamp)};
}
return std::tuple<SystemTime, SteadyTime>{start_system_timestamp,
start_steady_timestamp};
}
static bool SetSpanReference(
const std::pair<SpanReferenceType, const SpanContext*>& reference,
std::map<std::string, std::string>& baggage,
SpanReferenceData& reference_data) {
reference_data.reference_type = reference.first;
if (reference.second == nullptr) {
return false;
}
auto referenced_context =
dynamic_cast<const MockSpanContext*>(reference.second);
if (referenced_context == nullptr) {
return false;
}
reference_data.trace_id = referenced_context->trace_id();
reference_data.span_id = referenced_context->span_id();
referenced_context->ForeachBaggageItem(
[&baggage](const std::string& key, const std::string& value) {
baggage[key] = value;
return true;
});
return true;
}
MockSpan::MockSpan(std::shared_ptr<const Tracer>&& tracer, Recorder* recorder,
string_view operation_name, const StartSpanOptions& options)
: tracer_{std::move(tracer)}, recorder_{recorder} {
data_.operation_name = operation_name;
// Set start timestamps
std::tie(data_.start_timestamp, start_steady_) = ComputeStartTimestamps(
options.start_system_timestamp, options.start_steady_timestamp);
// Set references
SpanContextData span_context_data;
for (auto& reference : options.references) {
SpanReferenceData reference_data;
if (!SetSpanReference(reference, span_context_data.baggage,
reference_data)) {
continue;
}
data_.references.push_back(reference_data);
}
// Set tags
for (auto& tag : options.tags) {
data_.tags[tag.first] = tag.second;
}
// Set span context
span_context_data.trace_id =
data_.references.empty() ? GenerateId() : data_.references[0].trace_id;
span_context_data.span_id = GenerateId();
span_context_ = MockSpanContext{std::move(span_context_data)};
}
MockSpan::~MockSpan() {
if (!is_finished_) {
Finish();
}
}
void MockSpan::FinishWithOptions(const FinishSpanOptions& options) noexcept {
// Ensure the span is only finished once.
if (is_finished_.exchange(true)) {
return;
}
data_.logs.reserve(data_.logs.size() + options.log_records.size());
for (auto& log_record : options.log_records) {
data_.logs.push_back(log_record);
}
auto finish_timestamp = options.finish_steady_timestamp;
if (finish_timestamp == SteadyTime{}) {
finish_timestamp = SteadyClock::now();
}
data_.duration = finish_timestamp - start_steady_;
span_context_.CopyData(data_.span_context);
if (recorder_ != nullptr) {
recorder_->RecordSpan(std::move(data_));
}
}
void MockSpan::SetOperationName(string_view name) noexcept try {
std::lock_guard<std::mutex> lock_guard{mutex_};
data_.operation_name = name;
} catch (const std::exception& e) {
// Ignore operation
fprintf(stderr, "Failed to set operation name: %s\n", e.what());
}
void MockSpan::SetTag(string_view key,
const opentracing::Value& value) noexcept try {
std::lock_guard<std::mutex> lock_guard{mutex_};
data_.tags[key] = value;
} catch (const std::exception& e) {
// Ignore upon error.
fprintf(stderr, "Failed to set tag: %s\n", e.what());
}
void MockSpan::Log(
std::initializer_list<std::pair<string_view, Value>> fields) noexcept {
Log(SystemClock::now(), fields);
}
void MockSpan::Log(
SystemTime timestamp,
std::initializer_list<std::pair<string_view, Value>> fields) noexcept try {
LogRecord log_record;
log_record.timestamp = timestamp;
log_record.fields.reserve(fields.size());
for (auto& field : fields) {
log_record.fields.emplace_back(field.first, field.second);
}
std::lock_guard<std::mutex> lock_guard{mutex_};
data_.logs.emplace_back(std::move(log_record));
} catch (const std::exception& e) {
// Drop log record upon error.
fprintf(stderr, "Failed to log: %s\n", e.what());
}
void MockSpan::Log(
SystemTime timestamp,
const std::vector<std::pair<string_view, Value>>& fields) noexcept try {
LogRecord log_record;
log_record.timestamp = timestamp;
log_record.fields.reserve(fields.size());
for (auto& field : fields) {
log_record.fields.emplace_back(field.first, field.second);
}
std::lock_guard<std::mutex> lock_guard{mutex_};
data_.logs.emplace_back(std::move(log_record));
} catch (const std::exception& e) {
// Drop log record upon error.
fprintf(stderr, "Failed to log: %s\n", e.what());
}
void MockSpan::SetBaggageItem(string_view restricted_key,
string_view value) noexcept try {
std::lock_guard<std::mutex> lock_guard{span_context_.baggage_mutex_};
span_context_.data_.baggage.emplace(restricted_key, value);
} catch (const std::exception& e) {
// Drop baggage item upon error.
fprintf(stderr, "Failed to set baggage item: %s\n", e.what());
}
std::string MockSpan::BaggageItem(string_view restricted_key) const
noexcept try {
std::lock_guard<std::mutex> lock_guard{span_context_.baggage_mutex_};
auto lookup = span_context_.data_.baggage.find(restricted_key);
if (lookup != span_context_.data_.baggage.end()) {
return lookup->second;
}
return {};
} catch (const std::exception& e) {
// Return empty string upon error.
fprintf(stderr, "Failed to retrieve baggage item: %s\n", e.what());
return {};
}
} // namespace mocktracer
END_OPENTRACING_ABI_NAMESPACE
} // namespace opentracing
|
