From 59b496b66ec6c0cd1ec44cfbbe98e2c6b16c2e8c Mon Sep 17 00:00:00 2001 From: Stephen McQuay Date: Mon, 21 Oct 2013 19:21:59 -0700 Subject: [PATCH] Let user specify port --- main.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/main.go b/main.go index e39e65f..66dbe31 100644 --- a/main.go +++ b/main.go @@ -1,10 +1,14 @@ package main import ( + "flag" + "fmt" "log" "net/http" ) +var port = flag.Int("port", 8000, "port from which to serve") + func logger(h http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { log.Printf("%s: %s\n", r.RemoteAddr, r.URL) @@ -13,6 +17,11 @@ func logger(h http.Handler) http.Handler { } func main() { + flag.Parse() + addr := fmt.Sprintf(":%d", *port) fh := http.FileServer(http.Dir("./")) http.ListenAndServe(":8080", logger(fh)) + if err := http.ListenAndServe(addr, logger(fh)); err != nil { + log.Fatal(err) + } }