19
19
20
20
import com .google .common .collect .ImmutableMap ;
21
21
import com .google .common .primitives .Primitives ;
22
- import io .grpc .Context ;
23
- import io .opentelemetry .common .Attributes ;
22
+ import io .opentelemetry .api .common .Attributes ;
24
23
import io .opentelemetry .context .Scope ;
25
- import io .opentelemetry .trace .SpanContext ;
26
- import io .opentelemetry .trace .StatusCanonicalCode ;
27
- import io .opentelemetry .trace .Tracer ;
24
+ import io .opentelemetry .context .Context ;
25
+ import io .opentelemetry .api .trace .SpanContext ;
26
+ import io .opentelemetry .api .trace .StatusCode ;
27
+ import io .opentelemetry .api .trace .Tracer ;
28
28
import org .openqa .selenium .internal .Require ;
29
29
import org .openqa .selenium .remote .tracing .EventAttributeValue ;
30
30
import org .openqa .selenium .remote .tracing .Span ;
35
35
36
36
class OpenTelemetrySpan extends OpenTelemetryContext implements AutoCloseable , Span {
37
37
38
- private final io .opentelemetry .trace .Span span ;
38
+ private final io .opentelemetry .api . trace .Span span ;
39
39
private final Scope scope ;
40
40
41
- public OpenTelemetrySpan (Tracer tracer , Context context , io .opentelemetry .trace .Span span , Scope scope ) {
41
+ public OpenTelemetrySpan (Tracer tracer , Context context , io .opentelemetry .api . trace .Span span , Scope scope ) {
42
42
super (tracer , context );
43
43
this .span = Require .nonNull ("Span" , span );
44
44
this .scope = Require .nonNull ("Scope" , scope );
@@ -90,42 +90,42 @@ public Span addEvent(String name) {
90
90
public Span addEvent (String name , Map <String , EventAttributeValue > attributeMap ) {
91
91
Require .nonNull ("Name" , name );
92
92
Require .nonNull ("Event Attribute Map" , attributeMap );
93
- Attributes .Builder otAttributes = Attributes .newBuilder ();
93
+ Attributes .Builder otAttributes = Attributes .builder ();
94
94
95
95
attributeMap .forEach (
96
96
(key , value ) -> {
97
97
Require .nonNull ("Event Attribute Value" , value );
98
98
switch (value .getAttributeType ()) {
99
99
case BOOLEAN :
100
- otAttributes .setAttribute (key , value .getBooleanValue ());
100
+ otAttributes .put (key , value .getBooleanValue ());
101
101
break ;
102
102
103
103
case BOOLEAN_ARRAY :
104
- otAttributes .setAttribute (key , value .getBooleanArrayValue ());
104
+ otAttributes .put (key , value .getBooleanArrayValue ());
105
105
break ;
106
106
107
107
case DOUBLE :
108
- otAttributes .setAttribute (key , value .getNumberValue ().doubleValue ());
108
+ otAttributes .put (key , value .getNumberValue ().doubleValue ());
109
109
break ;
110
110
111
111
case DOUBLE_ARRAY :
112
- otAttributes .setAttribute (key , value .getDoubleArrayValue ());
112
+ otAttributes .put (key , value .getDoubleArrayValue ());
113
113
break ;
114
114
115
115
case LONG :
116
- otAttributes .setAttribute (key , value .getNumberValue ().longValue ());
116
+ otAttributes .put (key , value .getNumberValue ().longValue ());
117
117
break ;
118
118
119
119
case LONG_ARRAY :
120
- otAttributes .setAttribute (key , value .getLongArrayValue ());
120
+ otAttributes .put (key , value .getLongArrayValue ());
121
121
break ;
122
122
123
123
case STRING :
124
- otAttributes .setAttribute (key , value .getStringValue ());
124
+ otAttributes .put (key , value .getStringValue ());
125
125
break ;
126
126
127
127
case STRING_ARRAY :
128
- otAttributes .setAttribute (key , value .getStringArrayValue ());
128
+ otAttributes .put (key , value .getStringArrayValue ());
129
129
break ;
130
130
131
131
default :
@@ -139,35 +139,35 @@ public Span addEvent(String name, Map<String, EventAttributeValue> attributeMap)
139
139
return this ;
140
140
}
141
141
142
- private static final Map <Status .Kind , StatusCanonicalCode > statuses
143
- = new ImmutableMap .Builder <Status .Kind , StatusCanonicalCode >()
144
- .put (Status .Kind .ABORTED , StatusCanonicalCode .ERROR )
145
- .put (Status .Kind .CANCELLED , StatusCanonicalCode .ERROR )
146
- .put (Status .Kind .NOT_FOUND , StatusCanonicalCode .ERROR )
147
- .put (Status .Kind .OK , StatusCanonicalCode .OK )
148
- .put (Status .Kind .RESOURCE_EXHAUSTED , StatusCanonicalCode .ERROR )
149
- .put (Status .Kind .UNKNOWN , StatusCanonicalCode .ERROR )
150
- .put (Status .Kind .INVALID_ARGUMENT , StatusCanonicalCode .ERROR )
151
- .put (Status .Kind .DEADLINE_EXCEEDED , StatusCanonicalCode .ERROR )
152
- .put (Status .Kind .ALREADY_EXISTS , StatusCanonicalCode .ERROR )
153
- .put (Status .Kind .PERMISSION_DENIED , StatusCanonicalCode .ERROR )
154
- .put (Status .Kind .OUT_OF_RANGE , StatusCanonicalCode .ERROR )
155
- .put (Status .Kind .UNIMPLEMENTED , StatusCanonicalCode .ERROR )
156
- .put (Status .Kind .INTERNAL , StatusCanonicalCode .ERROR )
157
- .put (Status .Kind .UNAVAILABLE , StatusCanonicalCode .ERROR )
158
- .put (Status .Kind .UNAUTHENTICATED , StatusCanonicalCode .ERROR )
142
+ private static final Map <Status .Kind , StatusCode > statuses
143
+ = new ImmutableMap .Builder <Status .Kind , StatusCode >()
144
+ .put (Status .Kind .ABORTED , StatusCode .ERROR )
145
+ .put (Status .Kind .CANCELLED , StatusCode .ERROR )
146
+ .put (Status .Kind .NOT_FOUND , StatusCode .ERROR )
147
+ .put (Status .Kind .OK , StatusCode .OK )
148
+ .put (Status .Kind .RESOURCE_EXHAUSTED , StatusCode .ERROR )
149
+ .put (Status .Kind .UNKNOWN , StatusCode .ERROR )
150
+ .put (Status .Kind .INVALID_ARGUMENT , StatusCode .ERROR )
151
+ .put (Status .Kind .DEADLINE_EXCEEDED , StatusCode .ERROR )
152
+ .put (Status .Kind .ALREADY_EXISTS , StatusCode .ERROR )
153
+ .put (Status .Kind .PERMISSION_DENIED , StatusCode .ERROR )
154
+ .put (Status .Kind .OUT_OF_RANGE , StatusCode .ERROR )
155
+ .put (Status .Kind .UNIMPLEMENTED , StatusCode .ERROR )
156
+ .put (Status .Kind .INTERNAL , StatusCode .ERROR )
157
+ .put (Status .Kind .UNAVAILABLE , StatusCode .ERROR )
158
+ .put (Status .Kind .UNAUTHENTICATED , StatusCode .ERROR )
159
159
.build ();
160
160
161
161
@ Override
162
162
public Span setStatus (Status status ) {
163
163
Require .nonNull ("Status" , status );
164
164
165
- StatusCanonicalCode statusCanonicalCode = statuses .get (status .getKind ());
166
- if (statusCanonicalCode == null ) {
165
+ StatusCode statusCode = statuses .get (status .getKind ());
166
+ if (statusCode == null ) {
167
167
throw new IllegalArgumentException ("Unrecognized status kind: " + status .getKind ());
168
168
}
169
169
170
- span .setStatus (statusCanonicalCode ,
170
+ span .setStatus (statusCode ,
171
171
"Kind: " + status .getKind ().toString ()
172
172
+ " Description:"
173
173
+ status .getDescription ());
@@ -183,7 +183,7 @@ public void close() {
183
183
184
184
@ Override
185
185
public String toString () {
186
- SpanContext context = span .getContext ();
186
+ SpanContext context = span .getSpanContext ();
187
187
188
188
return "OpenTelemetrySpan{traceId=" +
189
189
context .getTraceIdAsHexString () +
@@ -203,8 +203,8 @@ public boolean equals(Object o) {
203
203
}
204
204
205
205
OpenTelemetrySpan that = (OpenTelemetrySpan ) o ;
206
- SpanContext thisContext = this .span .getContext ();
207
- SpanContext thatContext = that .span .getContext ();
206
+ SpanContext thisContext = this .span .getSpanContext ();
207
+ SpanContext thatContext = that .span .getSpanContext ();
208
208
209
209
return Objects .equals (thisContext .getSpanIdAsHexString (), thatContext .getSpanIdAsHexString ()) &&
210
210
Objects .equals (thisContext .getTraceIdAsHexString (), thatContext .getTraceIdAsHexString ());
0 commit comments