Fix import for self

Signed-off-by: Edward McFarlane <[email protected]>
connectrpc · Dec 19, 2024 · a2b22c5 · a2b22c5
1 parent 7b04e44
commit a2b22c5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cmd/protoc-gen-connect-go/main.go b/cmd/protoc-gen-connect-go/main.go
@@ -132,7 +132,9 @@ func generate(plugin *protogen.Plugin, file *protogen.File, samePackage bool) {
 		file.GeneratedFilenamePrefix+generatedFilenameExtension,
 		goImportPath,
 	)
-	generatedFile.Import(file.GoImportPath)
+	if !samePackage {
+		generatedFile.Import(file.GoImportPath)
+	}
 	generatePreamble(generatedFile, file)
 	generateServiceNameConstants(generatedFile, file.Services)
 	generateServiceNameVariables(generatedFile, file)