Browse Source

refactor httpd

boyska 5 years ago
parent
commit
515e910683
2 changed files with 6 additions and 6 deletions
  1. 5 4
      cmd/circologd/http_log.go
  2. 1 2
      cmd/circologd/main.go

+ 5 - 4
cmd/circologd/http.go → cmd/circologd/http_log.go

@@ -13,10 +13,11 @@ import (
 	"gopkg.in/mcuadros/go-syslog.v2/format"
 )
 
-// TODO: return a server
-func setupHTTP(hub circolog.Hub) {
-	http.HandleFunc("/", getHTTPHandler(hub))
-	http.HandleFunc("/ws", getWSHandler(hub))
+func setupHTTP(hub circolog.Hub) *http.ServeMux {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/", getHTTPHandler(hub))
+	mux.HandleFunc("/ws", getWSHandler(hub))
+	return mux
 }
 
 func parseParameterL(r *http.Request) (int, error) {

+ 1 - 2
cmd/circologd/main.go

@@ -59,8 +59,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	setupHTTP(hub)
-	httpServer := http.Server{Handler: nil}
+	httpServer := http.Server{Handler: setupHTTP(hub)}
 	if *querySocket != "" {
 		fmt.Printf("Binding address `%s` [http]\n", *querySocket)
 		unixListener, err := net.Listen("unix", *querySocket)