zishuo/go-pkg
1
0
Fork 0
mirror of https://github.com/pyihe/go-pkg.git synced 2025-11-01 19:52:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feature(https): add timeout

Browse Source
This commit is contained in:
pyihe
2022-08-10 17:17:04 +08:00
parent 1c9ec64011
commit eb6e4faeb5
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
https/http_api/http.go
View File

@@ -3,6 +3,7 @@ package http_api
import (
"context"
"net/http"
"time"
"github.com/gin-gonic/gin"
"github.com/pyihe/go-pkg/errors"
@@ -85,7 +86,10 @@ func (s *HttpServer) AddHandler(handler APIHandler) {
}
func (s *HttpServer) Stop() error {
if err := s.server.Shutdown(context.Background()); err != nil {
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := s.server.Shutdown(ctx); err != nil {
return err
}
s.wg.Wait()