loafle/openapi-generator
3
0
Fork 0
forked from loafle/openapi-generator-original
Code Pull Requests Activity

go: add os import in case of File return type (#3996)

Browse Source
This commit is contained in:
Nándor István Krácser 2019-10-01 05:31:55 +02:00 committed by William Cheng
parent 775d96c8e6
commit c8ac41c051
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/AbstractGoCodegen.java
View File

@ -377,6 +377,11 @@ public abstract class AbstractGoCodegen extends DefaultCodegen implements Codege
boolean addedOSImport = false; boolean addedOSImport = false;
boolean addedReflectImport = false; boolean addedReflectImport = false;
for (CodegenOperation operation : operations) { for (CodegenOperation operation : operations) {
// import "os" if the operation uses files
if (!addedOSImport && "*os.File".equals(operation.returnType)) {
imports.add(createMapping("import", "os"));
addedOSImport = true;
}
for (CodegenParameter param : operation.allParams) { for (CodegenParameter param : operation.allParams) {
// import "os" if the operation uses files // import "os" if the operation uses files
if (!addedOSImport && "*os.File".equals(param.dataType)) { if (!addedOSImport && "*os.File".equals(param.dataType)) {