Allows plugin invoked via GeneratePlugin to call RecordTypeUse wi…

…thout panicking (gogo#571)
YannickB · May 19, 2019 · 285f4c7 · 285f4c7
1 parent aaa2bc4
commit 285f4c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/protoc-gen-gogo/generator/helper.go b/protoc-gen-gogo/generator/helper.go
@@ -355,6 +355,7 @@ func (g *Generator) generatePlugin(file *FileDescriptor, p Plugin) {
 	g.usedPackages = make(map[GoImportPath]bool)
 	g.packageNames = make(map[GoImportPath]GoPackageName)
 	g.usedPackageNames = make(map[GoPackageName]bool)
+	g.addedImports = make(map[GoImportPath]bool)
 	g.file = file
 
 	// Run the plugins before the imports so we know which imports are necessary.