diff --git a/snippetbox/cmd/web/main.go b/snippetbox/cmd/web/main.go index d86d205..a7d9d57 100644 --- a/snippetbox/cmd/web/main.go +++ b/snippetbox/cmd/web/main.go @@ -1,18 +1,23 @@ package main import ( + "flag" "log" "net/http" + "os" ) func main() { + addr := flag.String("addr", ":4000", "HTTP network address") + flag.Parse() + infoLog := log.New(os.Stdout, "INFO\t", log.Ldate|log.Ltime) mux := http.NewServeMux() fileServer := http.FileServer(http.Dir("./ui/static/")) mux.Handle("/static/", http.StripPrefix("/static", fileServer)) mux.HandleFunc("/", home) mux.HandleFunc("/snippet/view", snippetView) mux.HandleFunc("/snippet/create", snippetCreate) - log.Print("Starting server on :4000") - err := http.ListenAndServe(":4000", mux) + log.Printf("Starting server on %s", *addr) + err := http.ListenAndServe(*addr, mux) log.Fatal(err) }