sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server: Return ErrServerClosed

Browse Source
This commit is contained in:
Adnan Maolood 2021-02-21 00:51:02 -05:00
parent 3fa55b52dd
commit e3d48b2cad
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.go
View File

@ -219,6 +219,9 @@ func (srv *Server) Serve(ctx context.Context, l net.Listener) error {
select { select {
case <-lnctx.Done(): case <-lnctx.Done():
if atomic.LoadInt32(&srv.closed) == 1 {
return ErrServerClosed
}
return lnctx.Err() return lnctx.Err()
case err := <-errch: case err := <-errch:
return err return err
@ -234,6 +237,9 @@ func (srv *Server) serve(ctx context.Context, l net.Listener) error {
if err != nil { if err != nil {
select { select {
case <-ctx.Done(): case <-ctx.Done():
if atomic.LoadInt32(&srv.closed) == 1 {
return ErrServerClosed
}
return ctx.Err() return ctx.Err()
default: default:
} }