improvements

Author: lunny

main.go

@@ -44,7 +44,6 @@ func main() {
 	}
 	app := cmd.NewMainApp(cmd.AppVersion{Version: Version, Extra: formatBuiltWith()})
 	_ = cmd.RunMainApp(app, os.Args...) // all errors should have been handled by the RunMainApp
-	log.GetManager().Close()
 }
 
 func formatBuiltWith() string {

modules/graceful/server.go

@@ -214,6 +214,10 @@ func newWrappedListener(l net.Listener, srv *Server) *wrappedListener {
 }
 
 func (wl *wrappedListener) Accept() (net.Conn, error) {
+	if wl.server.getState() != stateRunning || wl.stopped {
+		return nil, syscall.EINVAL
+	}
+
 	var c net.Conn
 	// Set keepalive on TCPListeners connections.
 	if tcl, ok := wl.Listener.(*net.TCPListener); ok {