diff --git a/src/go/mongolib/profiler/profiler.go b/src/go/mongolib/profiler/profiler.go index 320aeb99..be03d4cb 100644 --- a/src/go/mongolib/profiler/profiler.go +++ b/src/go/mongolib/profiler/profiler.go @@ -18,6 +18,7 @@ var ( type Profiler interface { GetLastError() error QueriesChan() chan stats.Queries + TimeoutsChan() <-chan time.Time FlushQueries() Start() Stop() @@ -93,6 +94,15 @@ func (p *Profile) Stop() { } } +func (p *Profile) TimeoutsChan() <-chan time.Time { + p.lock.Lock() + defer p.lock.Unlock() + if p.timeoutsChan == nil { + p.timeoutsChan = make(chan time.Time) + } + return p.timeoutsChan +} + func (p *Profile) getData() { go p.getDocs() p.stopWaitGroup.Add(1)