From 2d715b5efa839980ba0a96adc29436962df159c7 Mon Sep 17 00:00:00 2001 From: Sreekanth Sivasankaran Date: Fri, 23 Mar 2018 17:10:47 +0530 Subject: [PATCH] adding memprofile to bleve-query --- cmd/bleve-query/main.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/cmd/bleve-query/main.go b/cmd/bleve-query/main.go index b29f616be..26155e6b8 100644 --- a/cmd/bleve-query/main.go +++ b/cmd/bleve-query/main.go @@ -25,6 +25,7 @@ var target = flag.String("index", "bench.bleve", "index filename") var bindHTTP = flag.String("bindHttp", ":1234", "http bind port") var statsFile = flag.String("statsFile", "", "") var cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file") +var memprofile = flag.String("memprofile", "", "write mem profile to file") var qtype = flag.String("queryType", "term", "type of query to execute: term, prefix, query_string") var qfield = flag.String("field", "text", "the field to query, not applicable to query_string queries") var qclients = flag.Int("clients", 1, "the number of query clients") @@ -67,6 +68,15 @@ func main() { defer pprof.StopCPUProfile() } + if *memprofile != "" { + f, err := os.Create(*memprofile) + if err != nil { + log.Fatal(err) + } + defer f.Close() + pprof.WriteHeapProfile(f) + } + if *traceprofile != "" { f, err := os.Create(*traceprofile) if err != nil {