Initialize sessions maps if not set. (#79)

bazelbuild · Jan 5, 2017 · 0ca115e · 0ca115e
1 parent 03f111d
commit 0ca115e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/go/launcher/proxy/driver_hub.go b/go/launcher/proxy/driver_hub.go
@@ -80,13 +80,19 @@ func (h *WebDriverHub) Name() string {
 func (h *WebDriverHub) AddSession(id string, session http.Handler) {
 	h.mu.Lock()
 	defer h.mu.Unlock()
+	if h.sessions == nil {
+		h.sessions = map[string]http.Handler{}
+	}
 	h.sessions[id] = session
 }
 
 // RemoveSession removes a session from WebDriverHub.
 func (h *WebDriverHub) RemoveSession(id string) {
 	h.mu.Lock()
 	defer h.mu.Unlock()
+	if h.sessions == nil {
+		h.sessions = map[string]http.Handler{}
+	}
 	delete(h.sessions, id)
 }