From 9b4211de448a860770309f007d2bbeb78dc445a3 Mon Sep 17 00:00:00 2001 From: stephen mcquay Date: Wed, 24 Feb 2016 22:09:54 -0800 Subject: [PATCH] Added -http flag for specifying listening address --- main.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 4eb54f6..347f009 100644 --- a/main.go +++ b/main.go @@ -1,6 +1,7 @@ package main import ( + "flag" "fmt" "log" "net/http" @@ -13,6 +14,8 @@ import ( "github.com/boltdb/bolt" ) +var addr = flag.String("http", ":8000", "HTTP listen address") + func makeHandler(name string, fn func(http.ResponseWriter, *http.Request, string)) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { validPath := regexp.MustCompile(fmt.Sprintf(`^/%s/([a-zA-Z0-9\-_\/\.]+)$`, name)) @@ -71,6 +74,7 @@ func initDB() error { } func main() { + flag.Parse() if err := os.MkdirAll("repos/src/github.com", 0755); err != nil && !os.IsExist(err) { log.Fatal("ERROR: could not create repos dir: ", err) } @@ -89,6 +93,6 @@ func main() { http.HandleFunc("/about/", handlers.AboutHandler) http.HandleFunc("/", handlers.HomeHandler) - fmt.Println("Running on :8080...") - log.Fatal(http.ListenAndServe(":8080", nil)) + log.Printf("Running on %s ...", *addr) + log.Fatal(http.ListenAndServe(*addr, nil)) }