Merge 5acbe4b into 1a75356

openzipkin · May 14, 2019 · 267294f · 267294f
2 parents 1a75356 + 5acbe4b
commit 267294f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/span_implementation.go b/span_implementation.go
@@ -85,6 +85,15 @@ func (s *spanImpl) Finish() {
 	}
 }
 
+func (s *spanImpl) FinishedWithDuration(d time.Duration) {
+	if atomic.CompareAndSwapInt32(&s.mustCollect, 1, 0) {
+		s.Duration = d
+		if s.flushOnFinish {
+			s.tracer.reporter.Send(s.SpanModel)
+		}
+	}
+}
+
 func (s *spanImpl) Flush() {
 	if s.SpanModel.Debug || (s.SpanModel.Sampled != nil && *s.SpanModel.Sampled) {
 		s.tracer.reporter.Send(s.SpanModel)