From ecf1f59648d5d081c812d9c02c63d30082395c92 Mon Sep 17 00:00:00 2001
From: Max Chechel <hexdigest@gmail.com>
Date: Tue, 9 Apr 2019 11:48:42 +0300
Subject: [PATCH] fixes #1

---
 main.go | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 68f3058..8ab6424 100644
--- a/main.go
+++ b/main.go
@@ -26,6 +26,7 @@ const template = `
 	type {{$structName}} struct {
 		next {{$interfaceName}}
 		prefix string
+		tracer opentracing.Tracer
 	}
 
 	// New{{$structName}} creates a {{$structName}}
@@ -33,6 +34,16 @@ const template = `
 		return &{{$structName}} {
 			next: next,
 			prefix: prefix,
+			tracer: opentracing.GlobalTracer(),
+		}
+	}
+
+	// New{{$structName}} creates a {{$structName}}
+	func New{{$structName}}WithTracer(next {{$interfaceName}}, prefix string, tracer opentracing.Tracer) *{{$structName}} {
+		return &{{$structName}} {
+			next: next,
+			prefix: prefix,
+			tracer: tracer,
 		}
 	}
 
@@ -83,7 +94,7 @@ func startSpan(gen *generator.Generator) interface{} {
 			return "", errors.Wrap(err, "failed to parse func params")
 		}
 		ctxName := params[0].Name
-		return "span, " + ctxName + " := opentracing.StartSpanFromContext(" + ctxName + ", t.prefix + \"" + anchor + "\")", nil
+		return "span, " + ctxName + " := opentracing.StartSpanFromContextWithTracer(" + ctxName + ", t.tracer, t.prefix + \"" + anchor + "\")", nil
 	}
 }