darkkirb/harness-drone
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Limit graceful shutdown duration

Browse source
This commit is contained in:
Marko Gaćeša 2021-06-16 18:40:28 +02:00
parent 1093154331
commit 0fa03b03e0
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
server/server.go
View file

@ -37,6 +37,8 @@ type Server struct {
Handler http.Handler Handler http.Handler
} }
const timeoutGracefulShutdown = 5 * time.Second
// ListenAndServe initializes a server to respond to HTTP network requests. // ListenAndServe initializes a server to respond to HTTP network requests.
func (s Server) ListenAndServe(ctx context.Context) error { func (s Server) ListenAndServe(ctx context.Context) error {
if s.Acme { if s.Acme {
@ -60,7 +62,7 @@ func (s Server) listenAndServe(ctx context.Context) error {
g.Go(func() error { g.Go(func() error {
<-ctx.Done() <-ctx.Done()
ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), time.Minute) ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), timeoutGracefulShutdown)
defer cancelFunc() defer cancelFunc()
return s1.Shutdown(ctxShutdown) return s1.Shutdown(ctxShutdown)
@ -94,7 +96,7 @@ func (s Server) listenAndServeTLS(ctx context.Context) error {
<-ctx.Done() <-ctx.Done()
var gShutdown errgroup.Group var gShutdown errgroup.Group
ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), time.Minute) ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), timeoutGracefulShutdown)
defer cancelFunc() defer cancelFunc()
gShutdown.Go(func() error { gShutdown.Go(func() error {
@ -142,7 +144,7 @@ func (s Server) listenAndServeAcme(ctx context.Context) error {
<-ctx.Done() <-ctx.Done()
var gShutdown errgroup.Group var gShutdown errgroup.Group
ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), time.Minute) ctxShutdown, cancelFunc := context.WithTimeout(context.Background(), timeoutGracefulShutdown)
defer cancelFunc() defer cancelFunc()
gShutdown.Go(func() error { gShutdown.Go(func() error {