Complete server lifecycle

web/web.go

@@ -5,7 +5,11 @@ import (
 	"net/http"
 )
 
-func NewWebServer(port int) (*http.Server, error) {
+type WebServer struct {
+	*http.Server
+}
+
+func NewWebServer(port int) (*WebServer, error) {
 	mux := http.NewServeMux()
 
 	// ToDo: configure routes
@@ -13,10 +17,23 @@ func NewWebServer(port int) (*http.Server, error) {
 		fmt.Fprintf(w, "Hello rmsn!!")
 	})
 
-	s := &http.Server{
-		Addr:    fmt.Sprintf(":%d", port),
-		Handler: mux,
+	s := &WebServer{
+		&http.Server{
+			Addr:    fmt.Sprintf(":%d", port),
+			Handler: mux,
+		},
 	}
 
 	return s, nil
 }
+
+func (w *WebServer) Start() error {
+	if err := w.Server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+		return err
+	}
+	return nil
+}
+
+func (w *WebServer) Close() error {
+	return w.Server.Close()
+}