add port to command line options

1 files changed, 4 insertions, 1 deletions

diff --git a/main.go b/main.go
index afaef44..5f6a5cb 100644
--- a/main.go
+++ b/main.go
@@ -3,6 +3,7 @@ package main
 import (
         "bytes"
         "encoding/xml"
+        "flag"
         "fmt"
         "github.com/sideshowdave7/torpedo"
         "net/url"
@@ -38,11 +39,13 @@ func (ping) Get(values url.Values) (int, string) {
 }
 
 func main() {
+        var port = flag.Int("port", 8000, "Port number to listen on")
+        flag.Parse()
 
         ping := new(ping)
 
         var api = new(torpedo.API)
         api.AddResource(ping, "/rest/ping.view")
-        api.Start(3000)
+        api.Start(*port)
 
 }