refactor: move server-running code to separate function

1 files changed, 9 insertions, 4 deletions

diff --git a/main.go b/main.go
index ec799df..bda2d87 100644
--- a/main.go
+++ b/main.go
@@ -22,11 +22,16 @@ func main() {
 
 	mux.Handle("/{$}", &handler{})
 
+	log.Fatal(runServer(mux, *port))
+}
+
+func runServer(h http.Handler, port string) error {
 	srv := http.Server{
-		Addr:    ":" + *port,
-		Handler: mux,
+		Addr:    ":" + port,
+		Handler: h,
 	}
 
-	log.Printf("Serving on port %s...\n", *port)
-	log.Fatal(srv.ListenAndServe())
+	log.Printf("Serving on port %s...\n", port)
+
+	return srv.ListenAndServe()
 }