moved ringbuffer eventHandler stop to the probe

removed redundant nil checks
DataDog · Nov 6, 2024 · 750171b · 750171b
1 parent 7aec75b
commit 750171b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/pkg/gpu/consumer.go b/pkg/gpu/consumer.go
@@ -51,7 +51,6 @@ func (c *cudaEventConsumer) Stop() {
 	if c == nil {
 		return
 	}
-	c.eventHandler.Stop()
 	c.once.Do(func() {
 		close(c.closed)
 	})

diff --git a/pkg/gpu/probe.go b/pkg/gpu/probe.go
@@ -163,19 +163,11 @@ func (p *Probe) start() error {
 
 // Close stops the probe
 func (p *Probe) Close() {
-	if p.procMon != nil {
-		p.procMon.Stop()
-	}
-
-	if p.attacher != nil {
-		p.attacher.Stop()
-	}
-
+	p.procMon.Stop()
+	p.attacher.Stop()
 	_ = p.m.Stop(manager.CleanAll)
-
-	if p.consumer != nil {
-		p.consumer.Stop()
-	}
+	p.consumer.Stop()
+	p.eventHandler.Stop()
 }
 
 // GetAndFlush returns the GPU stats