fix: dynamic schemes bugfix (#450)

swaggo · Jul 11, 2019 · 6de832a · 6de832a
1 parent 4f5a434
commit 6de832a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gen/gen.go b/gen/gen.go
@@ -69,7 +69,6 @@ func (g *Gen) Build(config *Config) error {
 	for _, scheme := range swagger.Schemes {
 		schemes = append(schemes, fmt.Sprintf("%q", scheme))
 	}
-	swagger.Schemes = []string{}
 
 	b, err := json.MarshalIndent(swagger, "", "    ")
 	if err != nil {
@@ -111,6 +110,12 @@ func (g *Gen) Build(config *Config) error {
 		return err
 	}
 
+	swagger.Schemes = []string{}
+	b, err = json.MarshalIndent(swagger, "", "    ")
+	if err != nil {
+		return err
+	}
+
 	if err := packageTemplate.Execute(docs, struct {
 		Timestamp time.Time
 		Doc       string