Update main.go

This commit is contained in:
RamiBerm 2021-07-19 09:08:21 +03:00
parent dc62195a8f
commit eb02ecda20

View File

@ -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)