From 602749a8d148c281e9f519a1ef362744eeff1f44 Mon Sep 17 00:00:00 2001
From: spencergibb <sgibb@vmware.com>
Date: Fri, 9 Jul 2021 14:55:27 -0400
Subject: [PATCH] Implements new methods and delegates them.

---
 .../instrument/cassandra/TraceReactiveSession.java   | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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<CqlIdentifier> getKeyspace() {
+		return this.delegate.getKeyspace();
+	}
+
+	@Override
+	public Metadata getMetadata() {
+		return this.delegate.getMetadata();
+	}
+
 	@Override
 	public Mono<ReactiveResultSet> execute(String cql) {
 		return execute(SimpleStatement.newInstance(cql));