1
0
mirror of https://github.com/astaxie/beego.git synced 2024-12-26 09:00:48 +00:00

Merge pull request #3996 from bharrat/3995-fix-middleware-not-working-with-graceful

Fixes #3995 Use handlers with middleware when starting Graceful server
This commit is contained in:
Ming Deng 2020-05-31 22:23:05 +08:00 committed by GitHub
commit 11740cede6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

4
app.go
View File

@ -123,7 +123,7 @@ func (app *App) Run(mws ...MiddleWare) {
httpsAddr = fmt.Sprintf("%s:%d", BConfig.Listen.HTTPSAddr, BConfig.Listen.HTTPSPort)
app.Server.Addr = httpsAddr
}
server := grace.NewServer(httpsAddr, app.Handlers)
server := grace.NewServer(httpsAddr, app.Server.Handler)
server.Server.ReadTimeout = app.Server.ReadTimeout
server.Server.WriteTimeout = app.Server.WriteTimeout
if BConfig.Listen.EnableMutualHTTPS {
@ -152,7 +152,7 @@ func (app *App) Run(mws ...MiddleWare) {
}
if BConfig.Listen.EnableHTTP {
go func() {
server := grace.NewServer(addr, app.Handlers)
server := grace.NewServer(addr, app.Server.Handler)
server.Server.ReadTimeout = app.Server.ReadTimeout
server.Server.WriteTimeout = app.Server.WriteTimeout
if BConfig.Listen.ListenTCP4 {