From 473255dbde814c3f7d2959723d1f7affc7ae10de Mon Sep 17 00:00:00 2001
From: Ben Visness <bvisness@gmail.com>
Date: Thu, 24 Jun 2021 08:10:44 -0500
Subject: [PATCH] Add pprof

---
 src/config/types.go    | 1 +
 src/website/website.go | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/src/config/types.go b/src/config/types.go
index 461e4f4..4b8dee0 100644
--- a/src/config/types.go
+++ b/src/config/types.go
@@ -18,6 +18,7 @@ const (
 type HMNConfig struct {
 	Env          Environment
 	Addr         string
+	PrivateAddr  string
 	BaseUrl      string
 	LogLevel     zerolog.Level
 	Postgres     PostgresConfig
diff --git a/src/website/website.go b/src/website/website.go
index 350f035..81073ce 100644
--- a/src/website/website.go
+++ b/src/website/website.go
@@ -3,7 +3,9 @@ package website
 import (
 	"context"
 	"errors"
+	"log"
 	"net/http"
+	_ "net/http/pprof"
 	"os"
 	"os/signal"
 	"time"
@@ -58,6 +60,10 @@ var WebsiteCommand = &cobra.Command{
 			os.Exit(1)
 		}()
 
+		go func() {
+			log.Println(http.ListenAndServe(config.Config.PrivateAddr, nil))
+		}()
+
 		logging.Info().Str("addr", config.Config.Addr).Msg("Serving the website")
 		serverErr := server.ListenAndServe()
 		if !errors.Is(serverErr, http.ErrServerClosed) {