From eb02ecda206a007ba5aca672204c1ca3ddb4b31a Mon Sep 17 00:00:00 2001 From: RamiBerm Date: Mon, 19 Jul 2021 09:08:21 +0300 Subject: [PATCH] Update main.go --- api/main.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/api/main.go b/api/main.go index 40c930b73..c09af2ecc 100644 --- a/api/main.go +++ b/api/main.go @@ -10,6 +10,7 @@ import ( "github.com/romana/rlog" "github.com/up9inc/mizu/shared" "github.com/up9inc/mizu/tap" + "log" "mizuserver/pkg/api" "mizuserver/pkg/middleware" "mizuserver/pkg/models" @@ -18,6 +19,7 @@ import ( "mizuserver/pkg/utils" "os" "os/signal" + "runtime/pprof" "strings" ) @@ -65,6 +67,12 @@ func main() { go pipeChannelToSocket(socketConnection, harOutputChannel) go api.StartReadingOutbound(outboundLinkOutputChannel) } else if *aggregator { + f, err := os.Create("/app/profile.pprof") + if err != nil { + log.Fatal(err) + } + fmt.Println("Starting profiling4") + pprof.StartCPUProfile(f) socketHarOutChannel := make(chan *tap.OutputChannelItem, 1000) filteredHarChannel := make(chan *tap.OutputChannelItem)