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

OpenAPITools · Oct 1, 2019 · c8ac41c · c8ac41c
1 parent 775d96c
commit c8ac41c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...openapi-generator/src/main/java/org/openapitools/codegen/languages/AbstractGoCodegen.java b/...openapi-generator/src/main/java/org/openapitools/codegen/languages/AbstractGoCodegen.java
@@ -377,6 +377,11 @@ public Map<String, Object> postProcessOperationsWithModels(Map<String, Object> o
         boolean addedOSImport = false;
         boolean addedReflectImport = false;
         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) {
                 // import "os" if the operation uses files
                 if (!addedOSImport && "*os.File".equals(param.dataType)) {