fix http restart memory leak #1056
This commit is contained in:
+11
-4
@@ -11,6 +11,7 @@ import (
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/alireza0/s-ui/api"
|
||||
"github.com/alireza0/s-ui/config"
|
||||
@@ -200,20 +201,26 @@ func (s *Server) Start() (err error) {
|
||||
}
|
||||
|
||||
func (s *Server) Stop() error {
|
||||
s.cancel()
|
||||
var err error
|
||||
if s.httpServer != nil {
|
||||
err = s.httpServer.Shutdown(s.ctx)
|
||||
shutdownCtx, cancelShutdown := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
err = s.httpServer.Shutdown(shutdownCtx)
|
||||
cancelShutdown()
|
||||
if err != nil {
|
||||
s.cancel()
|
||||
if s.listener != nil {
|
||||
_ = s.listener.Close()
|
||||
}
|
||||
return err
|
||||
}
|
||||
}
|
||||
if s.listener != nil {
|
||||
} else if s.listener != nil {
|
||||
err = s.listener.Close()
|
||||
if err != nil {
|
||||
s.cancel()
|
||||
return err
|
||||
}
|
||||
}
|
||||
s.cancel()
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user