diff --git a/api/main.tpl b/api/main.tpl index b64b7d5..6c4de4a 100644 --- a/api/main.tpl +++ b/api/main.tpl @@ -3,6 +3,7 @@ package main import ( "flag" "fmt" + "net/http" {{.importPackages}} ) @@ -14,9 +15,16 @@ func main() { var c config.Config conf.MustLoad(*configFile, &c) - - server := rest.MustNewServer(c.RestConf) + server := *rest.Server + // dev 模式下开启swagger + if c.Mode == "dev" { + server = rest.MustNewServer(c.RestConf, rest.WithCors(), rest.WithCorsHeaders("*"), rest.WithFileServer("/swagger", http.Dir("./swagger"))) + } else { + server = rest.MustNewServer(c.RestConf, rest.WithCors()) + } defer server.Stop() + // rest.MustNewServer(c.RestConf) + // defer server.Stop() ctx := svc.NewServiceContext(c) handler.RegisterHandlers(server, ctx)