summaryrefslogtreecommitdiffstats
path: root/mocktracer/include/opentracing/mocktracer/recorder.h
blob: d6b8554ce5669281506dbc2fdf5a12bd1e053079 (plain)
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
#ifndef OPENTRACING_MOCKTRACER_RECORDER_H
#define OPENTRACING_MOCKTRACER_RECORDER_H

#include <opentracing/mocktracer/symbols.h>
#include <opentracing/tracer.h>

#include <cstdint>
#include <iosfwd>
#include <map>

namespace opentracing {
BEGIN_OPENTRACING_ABI_NAMESPACE
namespace mocktracer {
struct SpanContextData {
  uint64_t trace_id;
  uint64_t span_id;
  std::map<std::string, std::string> baggage;
};

inline bool operator==(const SpanContextData& lhs, const SpanContextData& rhs) {
  return lhs.trace_id == rhs.trace_id && lhs.span_id == rhs.span_id &&
         lhs.baggage == rhs.baggage;
}

inline bool operator!=(const SpanContextData& lhs, const SpanContextData& rhs) {
  return !(lhs == rhs);
}

std::ostream& operator<<(std::ostream& out,
                         const SpanContextData& span_context_data);

struct SpanReferenceData {
  SpanReferenceType reference_type;
  uint64_t trace_id;
  uint64_t span_id;
};

inline bool operator==(const SpanReferenceData& lhs,
                       const SpanReferenceData& rhs) {
  return lhs.reference_type == rhs.reference_type &&
         lhs.trace_id == rhs.trace_id && lhs.span_id == rhs.span_id;
}

inline bool operator!=(const SpanReferenceData& lhs,
                       const SpanReferenceData& rhs) {
  return !(lhs == rhs);
}

struct SpanData {
  SpanContextData span_context;
  std::vector<SpanReferenceData> references;
  std::string operation_name;
  SystemTime start_timestamp;
  SteadyClock::duration duration;
  std::map<std::string, Value> tags;
  std::vector<LogRecord> logs;
};

inline bool operator==(const SpanData& lhs, const SpanData& rhs) {
  return lhs.span_context == rhs.span_context &&
         lhs.references == rhs.references &&
         lhs.operation_name == rhs.operation_name &&
         lhs.start_timestamp == rhs.start_timestamp &&
         lhs.duration == rhs.duration && lhs.tags == rhs.tags &&
         lhs.logs == rhs.logs;
}

inline bool operator!=(const SpanData& lhs, const SpanData& rhs) {
  return !(lhs == rhs);
}

std::ostream& operator<<(std::ostream& out, const SpanData& span_data);

class OPENTRACING_MOCK_TRACER_API Recorder {
 public:
  virtual ~Recorder() = default;

  virtual void RecordSpan(SpanData&& span_data) noexcept = 0;

  virtual void Close() noexcept {}
};

}  // namespace mocktracer
END_OPENTRACING_ABI_NAMESPACE
}  // namespace opentracing

#endif  // OPENTRACING_MOCKTRACER_RECORDER_H