fix: import path

go-kod · Oct 28, 2024 · 660ecc7 · 660ecc7
1 parent 07ef6a3
commit 660ecc7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cmd/kod/internal/generate_generator.go b/cmd/kod/internal/generate_generator.go
@@ -701,7 +701,6 @@ func (g *generator) generateRegisteredComponents(p printFn) {
 		name := comp.intfName()
 		myName := comp.fullIntfName()
 
-		g.interceptor()
 		localStubFn := fmt.Sprintf(`func(ctx context.Context, info *kod.LocalStubFnInfo) any {
 			return %s_local_stub{
 				impl: info.Impl.(%s),
@@ -765,15 +764,16 @@ func (g *generator) generateLocalStubs(p printFn) {
 	p(``)
 	p(``)
 	p(`// Local stub implementations.`)
-	p(``)
-	g.tset.importPackage("context", "context")
 
 	var b strings.Builder
 	for _, comp := range g.components {
 		if comp.isMain {
 			continue
 		}
 
+		g.tset.importPackage("context", "context")
+		g.interceptor()
+
 		stub := notExported(comp.intfName()) + "_local_stub"
 		p(`// %s is a local stub implementation of [%s].`, stub, g.tset.genTypeString(comp.intf))
 		p(`type %s struct{`, stub)