From a2e6ca1226526c6633d69fa1ce168c156ceb9876 Mon Sep 17 00:00:00 2001
From: tarat44 <32471142+tarat44@users.noreply.github.com>
Date: Sun, 11 Apr 2021 15:11:00 -0400
Subject: [PATCH] add WaitGroup to h2ping

---
 agent/checks/check.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/agent/checks/check.go b/agent/checks/check.go
index cc44e55f00..47752d9311 100644
--- a/agent/checks/check.go
+++ b/agent/checks/check.go
@@ -519,6 +519,7 @@ type CheckH2PING struct {
 	stop     bool
 	stopCh   chan struct{}
 	stopLock sync.Mutex
+        stopWg   sync.WaitGroup
 }
 
 func (c *CheckH2PING) check() {
@@ -559,9 +560,11 @@ func (c *CheckH2PING) Stop() {
 		c.stop = true
 		close(c.stopCh)
 	}
+        c.stopWg.Wait()
 }
 
 func (c *CheckH2PING) run() {
+        defer c.stopWg.Done()
 	// Get the randomized initial pause time
 	initialPauseTime := lib.RandomStagger(c.Interval)
 	next := time.After(initialPauseTime)
@@ -584,6 +587,7 @@ func (c *CheckH2PING) Start() {
 	}
 	c.stop = false
 	c.stopCh = make(chan struct{})
+        c.stopWg.Add(1)
 	go c.run()
 }