|
|
|
@ -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) |
|
|
|
|