only add to the wg if the validator is NOT closed

robbyt · Sep 19, 2024 · 03d5ff5 · 03d5ff5
1 parent e54e322
commit 03d5ff5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/proxy/addons/requestAndResponseValidator.go b/proxy/addons/requestAndResponseValidator.go
@@ -67,8 +67,11 @@ func (c *RequestAndResponseValidator) Request(f *px.Flow) {
 // Response validates the response, and does not use the normal logger, because
 // the response may not have all the necessary fields needed to log.
 func (c *RequestAndResponseValidator) Response(f *px.Flow) {
-	c.wg.Add(1)
-	defer c.wg.Done()
+	if !c.closed.Load() {
+		// if the addon is NOT closed, then add to the wait group
+		c.wg.Add(1)
+		defer c.wg.Done()
+	}
 
 	if f.Response != nil {
 		return