add Profiler

Author: topi314

cmd/dbot/main.go

@@ -3,18 +3,22 @@ package main
 import (
 	"context"
 	"flag"
+	"net/http"
 	"os"
 	"os/signal"
 	"syscall"
 
+	"github.com/disgoorg/log"
+	"github.com/disgoorg/snowflake/v2"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+	_ "github.com/lib/pq"
+
 	"github.com/KittyBot-Org/KittyBotGo/config"
 	"github.com/KittyBot-Org/KittyBotGo/db"
 	"github.com/KittyBot-Org/KittyBotGo/dbot"
 	"github.com/KittyBot-Org/KittyBotGo/dbot/commands"
 	"github.com/KittyBot-Org/KittyBotGo/dbot/listeners"
-	"github.com/disgoorg/log"
-	"github.com/disgoorg/snowflake/v2"
-	_ "github.com/lib/pq"
 )
 
 var (
@@ -35,6 +39,15 @@ func main() {
 	logger := log.New(log.Ldate | log.Ltime | log.Lshortfile)
 	logger.Info("Starting discord dbot version: ", version)
 
+	go func() {
+		r := chi.NewRouter()
+		r.Use(middleware.Logger)
+		r.Use(middleware.Recoverer)
+		r.Use(middleware.Heartbeat("/ping"))
+		r.Mount("/debug", middleware.Profiler())
+		log.Error(http.ListenAndServe("0.0.0.0:6060", r))
+	}()
+
 	var cfg dbot.Config
 	if err := config.LoadConfig(&cfg); err != nil {
 		logger.Fatal("Failed to load config: ", err)

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/disgoorg/snowflake/v2 v2.0.1
 	github.com/disgoorg/source-plugins v1.3.1-0.20221009162536-28cea64ea737
 	github.com/disgoorg/utils/paginator v0.0.0-20220811191046-2474e390fa3b
+	github.com/go-chi/chi/v5 v5.0.8
 	github.com/go-jet/jet/v2 v2.9.0
 	github.com/gorilla/mux v1.8.0
 	github.com/lib/pq v1.10.7

go.sum

@@ -82,6 +82,8 @@ github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.m
 github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=
 github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
 github.com/friendsofgo/errors v0.9.2/go.mod h1:yCvFW5AkDIL9qn7suHVLiI/gH228n7PC4Pn44IGoTOI=
+github.com/go-chi/chi/v5 v5.0.8 h1:lD+NLqFcAi1ovnVZpsnObHGW4xb4J8lNmoYVfECH1Y0=
+github.com/go-chi/chi/v5 v5.0.8/go.mod h1:DslCQbL2OYiznFReuXYUmQ2hGd1aDpCnlMNITLSKoi8=
 github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20191125211704-12ad95a8df72/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=