fix: panic in logged handler

Author: chetan

daemon_test.go

@@ -12,10 +12,10 @@ func TestListClients(t *testing.T) {
 	response := httptest.NewRecorder()
 
 	vhostMux := CreateVhostMux([]string{}, true)
-	rootMux := NewLoggedHandler(vhostMux)
+	lh := NewLoggedHandler(vhostMux)
 
 	// start daemon
-	d := NewDaemon(vhostMux, rootMux, "127.0.0.1", 80, 443)
+	d := NewDaemon(lh, "127.0.0.1", 80, 443)
 
 	d.listClients(response, request)
 	if response.Code != http.StatusOK {

logged_handler.go

@@ -21,12 +21,13 @@ type LoggedHandler struct {
 }
 
 // NewLoggedHandler wraps the given handler with a request/response logger
-func NewLoggedHandler(handler http.Handler) *LoggedHandler {
+func NewLoggedHandler(vm *VhostMux) *LoggedHandler {
 	lh := &LoggedHandler{
 		ServeMux:          http.NewServeMux(),
 		VhostLogListeners: make(map[string]chan string),
+		vhostMux:          vm,
 	}
-	lh.Handle("/", handler)
+	lh.Handle("/", vm)
 	return lh
 }