216cceeb8e5e3e4015abe19dacd345398fb27303
- Author
- Ayman Bagabas <ayman.bagabas@gmail.com>
- Committer
- Ayman Bagabas <ayman.bagabas@gmail.com>
- Date
Message
fix: graceful shutdown
Diff
1diff --git a/server/server.go b/server/server.go
2index cbf40fd66a1f4aacd8e74c9677e05bbb52496d23..fed56f3588a4f037a50ec01d2741a73f07bf5d1f 100644
3--- a/server/server.go
4+++ b/server/server.go
5@@ -67,12 +67,18 @@ func (srv *Server) Reload() error {
6
7 // Start starts the SSH server.
8 func (srv *Server) Start() error {
9- return srv.SSHServer.ListenAndServe()
10+ if err := srv.SSHServer.ListenAndServe(); err != ssh.ErrServerClosed {
11+ return err
12+ }
13+ return nil
14 }
15
16 // Serve serves the SSH server using the provided listener.
17 func (srv *Server) Serve(l net.Listener) error {
18- return srv.SSHServer.Serve(l)
19+ if err := srv.SSHServer.Serve(l); err != ssh.ErrServerClosed {
20+ return err
21+ }
22+ return nil
23 }
24
25 // Shutdown lets the server gracefully shutdown.