93 lines
2.8 KiB
Ruby
93 lines
2.8 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Jaeger
|
||
|
|
module Encoders
|
||
|
|
class ThriftEncoder
|
||
|
|
def initialize(service_name:)
|
||
|
|
@service_name = service_name
|
||
|
|
@tags = [
|
||
|
|
Jaeger::Thrift::Tag.new(
|
||
|
|
'key' => 'jaeger.version',
|
||
|
|
'vType' => Jaeger::Thrift::TagType::STRING,
|
||
|
|
'vStr' => 'Ruby-' + Jaeger::Client::VERSION
|
||
|
|
),
|
||
|
|
Jaeger::Thrift::Tag.new(
|
||
|
|
'key' => 'hostname',
|
||
|
|
'vType' => Jaeger::Thrift::TagType::STRING,
|
||
|
|
'vStr' => Socket.gethostname
|
||
|
|
)
|
||
|
|
]
|
||
|
|
ipv4 = Socket.ip_address_list.find { |ai| ai.ipv4? && !ai.ipv4_loopback? }
|
||
|
|
unless ipv4.nil? # rubocop:disable Style/GuardClause
|
||
|
|
@tags << Jaeger::Thrift::Tag.new(
|
||
|
|
'key' => 'ip',
|
||
|
|
'vType' => Jaeger::Thrift::TagType::STRING,
|
||
|
|
'vStr' => ipv4.ip_address
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def encode(spans)
|
||
|
|
Jaeger::Thrift::Batch.new(
|
||
|
|
'process' => Jaeger::Thrift::Process.new(
|
||
|
|
'serviceName' => @service_name,
|
||
|
|
'tags' => @tags
|
||
|
|
),
|
||
|
|
'spans' => spans.map(&method(:encode_span))
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def encode_span(span)
|
||
|
|
context = span.context
|
||
|
|
start_ts, duration = build_timestamps(span)
|
||
|
|
|
||
|
|
Jaeger::Thrift::Span.new(
|
||
|
|
'traceIdLow' => TraceId.uint64_id_to_int64(context.trace_id),
|
||
|
|
'traceIdHigh' => 0,
|
||
|
|
'spanId' => TraceId.uint64_id_to_int64(context.span_id),
|
||
|
|
'parentSpanId' => TraceId.uint64_id_to_int64(context.parent_id),
|
||
|
|
'operationName' => span.operation_name,
|
||
|
|
'references' => build_references(span.references || []),
|
||
|
|
'flags' => context.flags,
|
||
|
|
'startTime' => start_ts,
|
||
|
|
'duration' => duration,
|
||
|
|
'tags' => span.tags,
|
||
|
|
'logs' => span.logs
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def build_references(references)
|
||
|
|
references.map do |ref|
|
||
|
|
Jaeger::Thrift::SpanRef.new(
|
||
|
|
'refType' => span_ref_type(ref.type),
|
||
|
|
'traceIdLow' => TraceId.uint64_id_to_int64(ref.context.trace_id),
|
||
|
|
'traceIdHigh' => 0,
|
||
|
|
'spanId' => TraceId.uint64_id_to_int64(ref.context.span_id)
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def build_timestamps(span)
|
||
|
|
start_ts = (span.start_time.to_f * 1_000_000).to_i
|
||
|
|
end_ts = (span.end_time.to_f * 1_000_000).to_i
|
||
|
|
duration = end_ts - start_ts
|
||
|
|
[start_ts, duration]
|
||
|
|
end
|
||
|
|
|
||
|
|
def span_ref_type(type)
|
||
|
|
case type
|
||
|
|
when OpenTracing::Reference::CHILD_OF
|
||
|
|
Jaeger::Thrift::SpanRefType::CHILD_OF
|
||
|
|
when OpenTracing::Reference::FOLLOWS_FROM
|
||
|
|
Jaeger::Thrift::SpanRefType::FOLLOWS_FROM
|
||
|
|
else
|
||
|
|
warn "Jaeger::Client with format #{type} is not supported yet"
|
||
|
|
nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|