File tree Expand file tree Collapse file tree 1 file changed +12
-20
lines changed Expand file tree Collapse file tree 1 file changed +12
-20
lines changed Original file line number Diff line number Diff line change @@ -38,40 +38,32 @@ def valid?
38
38
end
39
39
40
40
def trace_id
41
- value = if @metadata [ GRPC_METADATA_TRACE_ID ] . is_a? ( Array )
42
- @metadata [ GRPC_METADATA_TRACE_ID ] . first . to_i
43
- else
44
- @metadata [ GRPC_METADATA_TRACE_ID ] . to_i
45
- end
41
+ value = metadata_for_key ( GRPC_METADATA_TRACE_ID ) . to_i
46
42
value if ( 1 ..Span ::EXTERNAL_MAX_ID ) . cover? value
47
43
end
48
44
49
45
def parent_id
50
- value = if @metadata [ GRPC_METADATA_PARENT_ID ] . is_a? ( Array )
51
- @metadata [ GRPC_METADATA_PARENT_ID ] . first . to_i
52
- else
53
- @metadata [ GRPC_METADATA_PARENT_ID ] . to_i
54
- end
46
+ value = metadata_for_key ( GRPC_METADATA_PARENT_ID ) . to_i
55
47
value if ( 1 ..Span ::EXTERNAL_MAX_ID ) . cover? value
56
48
end
57
49
58
50
def sampling_priority
59
- value = if @metadata [ GRPC_METADATA_SAMPLING_PRIORITY ] . is_a? ( Array )
60
- @metadata [ GRPC_METADATA_SAMPLING_PRIORITY ] . first
61
- else
62
- @metadata [ GRPC_METADATA_SAMPLING_PRIORITY ]
63
- end
51
+ value = metadata_for_key ( GRPC_METADATA_SAMPLING_PRIORITY )
64
52
value && value . to_i
65
53
end
66
54
67
55
def origin
68
- value = if @metadata [ GRPC_METADATA_ORIGIN ] . is_a? ( Array )
69
- @metadata [ GRPC_METADATA_ORIGIN ] . first
70
- else
71
- @metadata [ GRPC_METADATA_ORIGIN ]
72
- end
56
+ value = metadata_for_key ( GRPC_METADATA_ORIGIN )
73
57
value if value != ''
74
58
end
59
+
60
+ private
61
+
62
+ def metadata_for_key ( key )
63
+ # metadata can be arrays if there's multiple headers with the same values
64
+ return @metadata [ key ] . first if @metadata [ key ] . is_a? ( Array )
65
+ @metadata [ key ]
66
+ end
75
67
end
76
68
end
77
69
end
You can’t perform that action at this time.
0 commit comments