1 files changed, 12 insertions, 4 deletions

diff --git a/internal/metrics/metrics.go b/internal/metrics/metrics.go
index 9030775..74f9ffb 100644
--- a/internal/metrics/metrics.go
+++ b/internal/metrics/metrics.go
@@ -288,12 +288,20 @@ const (
 //go:embed dashboard.html
 var dashboardHTML []byte
 
-func (m *Metrics) Serve(addr, path string) error {
-        mux := http.NewServeMux()
-        mux.Handle(path, promhttp.Handler())
-        mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+func (m *Metrics) PrometheusHandler() http.Handler {
+        return promhttp.Handler()
+}
+
+func (m *Metrics) DashboardHandler() http.Handler {
+        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
                 w.Header().Set("Content-Type", "text/html; charset=utf-8")
                 w.Write(dashboardHTML)
         })
+}
+
+func (m *Metrics) Serve(addr, path string) error {
+        mux := http.NewServeMux()
+        mux.Handle(path, m.PrometheusHandler())
+        mux.Handle("/", m.DashboardHandler())
         return http.ListenAndServe(addr, mux)
 }