From 2ebae2aaefe86e3b620d3a1f4a80826923906d04 Mon Sep 17 00:00:00 2001
From: Greg Burd <gburd@onshape.com>
Date: Wed, 9 Aug 2017 09:34:20 -0400
Subject: [PATCH] Changed Zipkin API a bit, now requires a TraceContext rather
 than a Span before sync/async calls.

---
 .../core/operation/AbstractOptionalOperation.java      |  4 ++--
 .../core/operation/AbstractStatementOperation.java     | 10 +++++-----
 .../core/operation/AbstractStreamOperation.java        |  4 ++--
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/main/java/net/helenus/core/operation/AbstractOptionalOperation.java b/src/main/java/net/helenus/core/operation/AbstractOptionalOperation.java
index 5c56442..507fd69 100644
--- a/src/main/java/net/helenus/core/operation/AbstractOptionalOperation.java
+++ b/src/main/java/net/helenus/core/operation/AbstractOptionalOperation.java
@@ -53,7 +53,7 @@ public abstract class AbstractOptionalOperation<E, O extends AbstractOptionalOpe
 
 	public Optional<E> sync() {
         Tracer tracer = this.sessionOps.getZipkinTracer();
-        final Span cassandraSpan = (tracer != null && span != null) ? tracer.newChild(span.context()) : null;
+        final Span cassandraSpan = (tracer != null && traceContext != null) ? tracer.newChild(traceContext) : null;
         if (cassandraSpan != null) {
             cassandraSpan.name("cassandra");
             cassandraSpan.start();
@@ -71,7 +71,7 @@ public abstract class AbstractOptionalOperation<E, O extends AbstractOptionalOpe
 
 	public ListenableFuture<Optional<E>> async() {
         final Tracer tracer = this.sessionOps.getZipkinTracer();
-        final Span cassandraSpan = (tracer != null && span != null) ? tracer.newChild(span.context()) : null;
+        final Span cassandraSpan = (tracer != null && traceContext != null) ? tracer.newChild(traceContext) : null;
         if (cassandraSpan != null) {
             cassandraSpan.name("cassandra");
             cassandraSpan.start();
diff --git a/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java b/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java
index 1e4a70a..9cdf890 100644
--- a/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java
+++ b/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java
@@ -15,8 +15,8 @@
  */
 package net.helenus.core.operation;
 
-import brave.Span;
 import brave.Tracer;
+import brave.propagation.TraceContext;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -44,7 +44,7 @@ public abstract class AbstractStatementOperation<E, O extends AbstractStatementO
 	public abstract Statement buildStatement();
 
 	protected boolean showValues = true;
-    protected Span span;
+    protected TraceContext traceContext;
 	private ConsistencyLevel consistencyLevel;
 	private ConsistencyLevel serialConsistencyLevel;
 	private RetryPolicy retryPolicy;
@@ -215,11 +215,11 @@ public abstract class AbstractStatementOperation<E, O extends AbstractStatementO
 		return statement;
 	}
 
-    public O withinSpan(Span span) {
-        if (span != null) {
+    public O zipkinContext(TraceContext traceContext) {
+        if (traceContext != null) {
             Tracer tracer = this.sessionOps.getZipkinTracer();
             if (tracer != null) {
-                this.span = span;
+                this.traceContext = traceContext;
             }
         }
 
diff --git a/src/main/java/net/helenus/core/operation/AbstractStreamOperation.java b/src/main/java/net/helenus/core/operation/AbstractStreamOperation.java
index da9fd16..c82af73 100644
--- a/src/main/java/net/helenus/core/operation/AbstractStreamOperation.java
+++ b/src/main/java/net/helenus/core/operation/AbstractStreamOperation.java
@@ -54,7 +54,7 @@ public abstract class AbstractStreamOperation<E, O extends AbstractStreamOperati
 
     public Stream<E> sync() {
         Tracer tracer = this.sessionOps.getZipkinTracer();
-        final Span cassandraSpan = (tracer != null && span != null) ? tracer.newChild(span.context()) : null;
+        final Span cassandraSpan = (tracer != null && traceContext != null) ? tracer.newChild(traceContext) : null;
         if (cassandraSpan != null) {
             cassandraSpan.name("cassandra");
             cassandraSpan.start();
@@ -72,7 +72,7 @@ public abstract class AbstractStreamOperation<E, O extends AbstractStreamOperati
 
 	public ListenableFuture<Stream<E>> async() {
         Tracer tracer = this.sessionOps.getZipkinTracer();
-        final Span cassandraSpan = (tracer != null && span != null) ? tracer.newChild(span.context()) : null;
+        final Span cassandraSpan = (tracer != null && traceContext != null) ? tracer.newChild(traceContext) : null;
         if (cassandraSpan != null) {
             cassandraSpan.name("cassandra");
             cassandraSpan.start();