protogen: rewrite _ imports as well
Call the import-rewriting function on _ imports as well as named ones.
Change-Id: I2fe2c90fe4b201e96a243f231b54387c6b65f7a5
Reviewed-on: https://go-review.googlesource.com/c/139878
Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
diff --git a/protogen/protogen.go b/protogen/protogen.go
index fdacc43..4e76c30 100644
--- a/protogen/protogen.go
+++ b/protogen/protogen.go
@@ -896,7 +896,7 @@
if _, ok := g.packageNames[importPath]; ok {
continue
}
- astutil.AddNamedImport(fset, file, "_", string(importPath))
+ astutil.AddNamedImport(fset, file, "_", rewriteImport(string(importPath)))
}
ast.SortImports(fset, file)