From cfdafecc4a3292ca408feb88468da161a737310a Mon Sep 17 00:00:00 2001
From: Nimrod Gilboa Markevich <nimrod@up9.com>
Date: Mon, 16 May 2022 15:51:30 +0300
Subject: [PATCH] Add flag to dump mem profile every minute

---
 tap/passive_tapper.go | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/tap/passive_tapper.go b/tap/passive_tapper.go
index 613cde644..faf1119d0 100644
--- a/tap/passive_tapper.go
+++ b/tap/passive_tapper.go
@@ -80,6 +80,17 @@ func StartPassiveTapper(opts *TapOpts, outputItems chan *api.OutputChannelItem,
 		}()
 	}
 
+	if *memprofile != "" {
+		go func() {
+			for {
+				profilePath := profile.ProfilePath(*memprofile + "/" + time.Now().Format("2006-01-02T15:04:05-07:00"))
+				profile.Start(profile.MemProfileHeap, profilePath, profile.NoShutdownHook).Stop()
+				<-time.After(time.Second * 60)
+				logger.Log.Info("Stopped Mem profiling")
+			}
+		}()
+	}
+
 	extensions = extensionsRef
 	filteringOptions = options
 
@@ -251,9 +262,9 @@ func startPassiveTapper(streamsMap api.TcpStreamMap, assembler *tcpAssembler) {
 		assembler.dumpStreamPool()
 	}
 
-	if err := diagnose.DumpMemoryProfile(*memprofile); err != nil {
-		logger.Log.Errorf("Error dumping memory profile %v", err)
-	}
+	// if err := diagnose.DumpMemoryProfile(*memprofile); err != nil {
+	// 	logger.Log.Errorf("Error dumping memory profile %v", err)
+	// }
 
 	assembler.waitAndDump()