Add gokit/kit/endpoint import for the generated endpoint file

kujtimiihoxha · Sep 8, 2017 · 8dd56d0 · 8dd56d0
1 parent 0327e75
commit 8dd56d0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generator/generator.go b/generator/generator.go
@@ -628,6 +628,7 @@ func (sg *ServiceInitGenerator) generateEndpoints(name string, iface *parser.Int
 	servicePath = strings.Replace(servicePath, "\\", "/", -1)
 	serviceImport := projectPath + "/" + servicePath
 	file.Imports = []parser.NamedTypeValue{
+		parser.NewNameType("", "\"github.com/go-kit/kit/endpoint\""),
 		parser.NewNameType("", "\""+serviceImport+"\""),
 	}
 	file.Methods = []parser.Method{