From 771d8c826aadadf277ac08a5b39cc00090f1190a Mon Sep 17 00:00:00 2001
From: Karmanyaah Malhotra <karmanyaahm@gmail.com>
Date: Sun, 28 Feb 2021 15:02:34 -0500
Subject: [PATCH] Profile for debugging cpu usage

---
 main.go | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index fe7c449..3ca459c 100644
--- a/main.go
+++ b/main.go
@@ -21,6 +21,7 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
+	"runtime/pprof"
 	"strings"
 	"sync"
 	"syscall"
@@ -397,6 +398,8 @@ func (bridge *Bridge) Stop() {
 	}
 }
 
+var cpuprofile = flag.MakeFull("", "cpuprofile", "write cpu profile to `file`", "").String()
+
 func (bridge *Bridge) Main() {
 
 	if *generateRegistration {
@@ -411,6 +414,19 @@ func (bridge *Bridge) Main() {
 	bridge.Start()
 	bridge.Log.Infoln("Bridge started!")
 
+	if *cpuprofile != "" {
+		println("profiling")
+		f, err := os.Create(*cpuprofile)
+		if err != nil {
+			log.Fatal("could not create CPU profile: ", err)
+		}
+		defer f.Close() // error handling omitted for example
+		if err := pprof.StartCPUProfile(f); err != nil {
+			log.Fatal("could not start CPU profile: ", err)
+		}
+		defer pprof.StopCPUProfile()
+	}
+
 	c := make(chan os.Signal)
 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
 	<-c
@@ -418,7 +434,7 @@ func (bridge *Bridge) Main() {
 	bridge.Log.Infoln("Interrupt received, stopping...")
 	bridge.Stop()
 	bridge.Log.Infoln("Bridge stopped.")
-	os.Exit(0)
+	//os.Exit(0)
 }
 
 func main() {