Added cmd line port option.

oliverjanik · Apr 2, 2014 · e1e0d57 · e1e0d57
1 parent 50dda8e
commit e1e0d57
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,3 @@
-file-server
-===========
+# Simple file server
 
-Simple file server
+Wanted to write a file server in Go. Turns out there's one already built in.
diff --git a/file-server.go b/file-server.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
 	"net/http"
@@ -24,17 +25,19 @@ func wrapHandler(handler http.Handler) http.Handler {
 	})
 }
 
+var port = flag.Int("p", 8001, "Port to run on")
+
 func main() {
+	flag.Parse()
+
 	wd, err := os.Getwd()
 	if err != nil {
 		log.Fatal(err)
 	}
 
-	port := 8001
-
-	log.Printf("Starting server on port %d", port)
+	log.Printf("Starting server on port %d", *port)
 
-	addr := fmt.Sprintf(":%d", port)
+	addr := fmt.Sprintf(":%d", *port)
 
 	log.Fatal(http.ListenAndServe(addr, wrapHandler(http.FileServer(http.Dir(wd)))))
 }