diff --git a/app.go b/app.go index 78485ed..19d959d 100644 --- a/app.go +++ b/app.go @@ -7,7 +7,6 @@ import ( "sync" "github.com/gorilla/websocket" - // "github.com/pkg/profile" "log" "net" "net/http" @@ -16,9 +15,13 @@ import ( "time" ) +var cert string +var key string var port string func init() { + flag.StringVar(&cert, "cert", "", "give me a certificate") + flag.StringVar(&key, "key", "", "give me a key") flag.StringVar(&port, "port", "80", "give me a port number") } @@ -28,7 +31,6 @@ var upgrader = websocket.Upgrader{ } func main() { - // defer profile.Start().Stop() flag.Parse() http.HandleFunc("/echo", echoHandler) http.HandleFunc("/bench", benchHandler) @@ -36,6 +38,9 @@ func main() { http.HandleFunc("/api", api) http.HandleFunc("/health", healthHandler) fmt.Println("Starting up on port " + port) + if len(cert) > 0 && len(key) > 0 { + log.Fatal(http.ListenAndServeTLS(":"+port, cert, key, nil)) + } log.Fatal(http.ListenAndServe(":"+port, nil)) }