Add option to ignore current goroutines

leaks_test.go

@@ -22,6 +22,7 @@ package goleak
 
 import (
 	"fmt"
+	"sync"
 	"testing"
 	"time"
 
@@ -83,6 +84,57 @@ func TestVerifyNone(t *testing.T) {
 	bg.unblock()
 }
 
+func TestIgnoreCurrent(t *testing.T) {
+	t.Run("Should ignore current", func(t *testing.T) {
+		defer VerifyNone(t)
+		done := make(chan struct{})
+		go func() {
+			<-done
+		}()
+		VerifyNone(t, IgnoreCurrent())
+		close(done)
+	})
+
+	t.Run("Should detect new leaks", func(t *testing.T) {
+		done := make(chan struct{})
+		defer VerifyNone(t)
+		defer close(done)
+		defer func(o Option) {
+			err := Find(o)
+			require.Error(t, err)
+		}(IgnoreCurrent())
+		go func() {
+			<-done
+		}()
+	})
+
+	t.Run("Should not ignore false positive", func(t *testing.T) {
+		defer VerifyNone(t)
+		const goroutinesCount = 5
+		wg := sync.WaitGroup{}
+		done := make(chan struct{})
+		for i := 0; i < goroutinesCount; i++ {
+			wg.Add(1)
+			go func() {
+				<-done
+				wg.Done()
+			}()
+		}
+		option := IgnoreCurrent()
+		close(done)
+		wg.Wait()
+		for i := 0; i < goroutinesCount; i++ {
+			ch := make(chan struct{})
+			go func() {
+				<-ch
+			}()
+			require.Error(t, Find(option))
+			close(ch)
+			VerifyNone(t)
+		}
+	})
+}
+
 func TestVerifyParallel(t *testing.T) {
 	t.Run("parallel", func(t *testing.T) {
 		t.Parallel()

options.go

@@ -57,6 +57,18 @@ func IgnoreTopFunction(f string) Option {
 	})
 }
 
+// IgnoreCurrent records all current goroutines when the option is created, and ignores
+// them in any future Find/Verify calls.
+func IgnoreCurrent() Option {
+	excludeIDSet := map[int]bool{}
+	for _, s := range stack.All() {
+		excludeIDSet[s.ID()] = true
+	}
+	return addFilter(func(s stack.Stack) bool {
+		return excludeIDSet[s.ID()]
+	})
+}
+
 func maxSleep(d time.Duration) Option {
 	return optionFunc(func(opts *opts) {
 		opts.maxSleep = d