diff --git a/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/cassandra/TraceReactiveSession.java b/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/cassandra/TraceReactiveSession.java index dd94e0ec3..2d35f63fe 100644 --- a/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/cassandra/TraceReactiveSession.java +++ b/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/cassandra/TraceReactiveSession.java @@ -19,10 +19,12 @@ package org.springframework.cloud.sleuth.instrument.cassandra; import java.util.Map; import java.util.Optional; +import com.datastax.oss.driver.api.core.CqlIdentifier; import com.datastax.oss.driver.api.core.context.DriverContext; import com.datastax.oss.driver.api.core.cql.PreparedStatement; import com.datastax.oss.driver.api.core.cql.SimpleStatement; import com.datastax.oss.driver.api.core.cql.Statement; +import com.datastax.oss.driver.api.core.metadata.Metadata; import reactor.core.publisher.Mono; import reactor.util.context.ContextView; @@ -77,6 +79,16 @@ public class TraceReactiveSession implements ReactiveSession { return this.delegate.getContext(); } + @Override + public Optional getKeyspace() { + return this.delegate.getKeyspace(); + } + + @Override + public Metadata getMetadata() { + return this.delegate.getMetadata(); + } + @Override public Mono execute(String cql) { return execute(SimpleStatement.newInstance(cql));