Fix package name in codegen

moihn · Mar 21, 2022 · 25e5ec5 · 25e5ec5
1 parent 00d4c61
commit 25e5ec5
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/internal/pkg/codegen/model.go b/internal/pkg/codegen/model.go
@@ -137,7 +137,11 @@ func FromDbTableModel(tableModel dbmodel.DbTableModel, tableDef config.TableDef)
 		}
 		typeTokens := strings.Split(codegenCol.ColumnGoType, ".")
 		if len(typeTokens) > 1 {
-			importPackages[typeTokens[0]] = true
+			packageName := typeTokens[0]
+			if packageName[0:1] == "*" {
+				packageName = packageName[1:]
+			}
+			importPackages[packageName] = true
 		}
 		codegenCol.ColumnFieldName = strcase.ToCamel(codegenCol.ColumnLowerName)
 		codegenCol.ColumnVariableName = strcase.ToLowerCamel(codegenCol.ColumnLowerName)
@@ -262,7 +266,11 @@ func FromDbQueryModel(queryModel dbmodel.DbQueryModel, queryDef config.QueryDef)
 		}
 		typeTokens := strings.Split(codegenCol.ColumnGoType, ".")
 		if len(typeTokens) > 1 {
-			importPackages[typeTokens[0]] = true
+			packageName := typeTokens[0]
+			if packageName[0:1] == "*" {
+				packageName = packageName[1:]
+			}
+			importPackages[packageName] = true
 		}
 		codegenModel.QueryColumnList = append(codegenModel.QueryColumnList, &codegenCol)
 	}