Add check for nil otherTypeDef (#1372)

fix bug: more than two models have the same short package name
swaggo · Oct 30, 2022 · 07690e9 · 07690e9
1 parent 38b5f15
commit 07690e9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages.go b/packages.go
@@ -140,7 +140,10 @@ func (pkgDefs *PackagesDefinitions) parseTypesFromFile(astFile *ast.File, packag
 
 					anotherTypeDef, ok := pkgDefs.uniqueDefinitions[fullName]
 					if ok {
-						if typeSpecDef.PkgPath != anotherTypeDef.PkgPath {
+						if anotherTypeDef == nil {
+							typeSpecDef.NotUnique = true
+							pkgDefs.uniqueDefinitions[typeSpecDef.TypeName()] = typeSpecDef
+						} else if typeSpecDef.PkgPath != anotherTypeDef.PkgPath {
 							anotherTypeDef.NotUnique = true
 							typeSpecDef.NotUnique = true
 							pkgDefs.uniqueDefinitions[fullName] = nil
@@ -197,7 +200,10 @@ func (pkgDefs *PackagesDefinitions) parseFunctionScopedTypesFromFile(astFile *as
 
 								anotherTypeDef, ok := pkgDefs.uniqueDefinitions[fullName]
 								if ok {
-									if typeSpecDef.PkgPath != anotherTypeDef.PkgPath {
+									if anotherTypeDef == nil {
+										typeSpecDef.NotUnique = true
+										pkgDefs.uniqueDefinitions[typeSpecDef.TypeName()] = typeSpecDef
+									} else if typeSpecDef.PkgPath != anotherTypeDef.PkgPath {
 										anotherTypeDef.NotUnique = true
 										typeSpecDef.NotUnique = true
 										pkgDefs.uniqueDefinitions[fullName] = nil