Skip to content

Commit 73f4e47

Browse files
author
Karolina
committed
add number of cache reads received in custom payload to the trace
1 parent 821a267 commit 73f4e47

File tree

5 files changed

+27
-0
lines changed

5 files changed

+27
-0
lines changed

driver-core/src/main/java/com/datastax/driver/core/RequestHandler.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1199,6 +1199,7 @@ private void setFinalResult(Connection connection, Message.Response response) {
11991199
}
12001200
}
12011201
parentTracingInfo.setReplicas(replicasBuilder.toString());
1202+
parentTracingInfo.setCacheCounter(buf.getInt());
12021203
}
12031204

12041205
parentTracingInfo.tracingFinished();

driver-core/src/main/java/com/datastax/driver/core/tracing/NoopTracingInfoFactory.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,9 @@ public void setTable(String table) {}
7878
@Override
7979
public void setReplicas(String replicas) {}
8080

81+
@Override
82+
public void setCacheCounter(int cacheCounter) {}
83+
8184
@Override
8285
public void recordException(Exception exception) {}
8386

driver-core/src/main/java/com/datastax/driver/core/tracing/TracingInfo.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,13 @@ enum StatusCode {
162162
*/
163163
void setReplicas(String replicas);
164164

165+
/**
166+
* Adds provided cache reads counter to the trace.
167+
*
168+
* @param cacheCounter the counter to be set.
169+
*/
170+
void setCacheCounter(int cacheCounter);
171+
165172
/**
166173
* Records in the trace that the provided exception occured.
167174
*

driver-core/src/test/java/com/datastax/driver/core/tracing/TestTracingInfo.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ public class TestTracingInfo implements TracingInfo {
5151
private String partitionKey;
5252
private String table;
5353
private String replicas;
54+
private Integer cacheCounter;
5455

5556
public TestTracingInfo(PrecisionLevel precision) {
5657
this.precision = precision;
@@ -159,6 +160,11 @@ public void setReplicas(String replicas) {
159160
this.replicas = replicas;
160161
}
161162

163+
@Override
164+
public void setCacheCounter(int cacheCounter) {
165+
this.cacheCounter = cacheCounter;
166+
}
167+
162168
@Override
163169
public void recordException(Exception exception) {
164170
if (this.exceptions == null) {
@@ -267,6 +273,10 @@ public String getReplicas() {
267273
return replicas;
268274
}
269275

276+
public Integer getCacheCounter() {
277+
return cacheCounter;
278+
}
279+
270280
public StatusCode getStatusCode() {
271281
return statusCode;
272282
}

driver-opentelemetry/src/main/java/com/datastax/driver/opentelemetry/OpenTelemetryTracingInfo.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,12 @@ public void setReplicas(String replicas) {
184184
span.setAttribute("db.scylla.replicas", replicas);
185185
}
186186

187+
@Override
188+
public void setCacheCounter(int cacheCounter) {
189+
assertStarted();
190+
span.setAttribute("db.scylla.cache_counter", cacheCounter);
191+
}
192+
187193
private io.opentelemetry.api.trace.StatusCode mapStatusCode(StatusCode code) {
188194
switch (code) {
189195
case OK:

0 commit comments

Comments
 (0)