cshared: Make goroutine for Collect to be able to exit

Signed-off-by: Hiroshi Hatake <[email protected]>
chronosphereio · Oct 6, 2023 · 5bd8020 · 5bd8020
1 parent 589518f
commit 5bd8020
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cshared.go b/cshared.go
@@ -201,8 +201,16 @@ func prepareInputCollector() (err error) {
 
 	go func(theChannel chan<- Message) {
 		defer theInputLock.Unlock()
+		for {
+			select {
+			case <-runCtx.Done():
+				log.Printf("goroutine will be stopping: name=%q\n", theName)
+				return
+			default:
+				err = theInput.Collect(runCtx, theChannel)
+			}
+		}
 
-		err := theInput.Collect(runCtx, theChannel)
 		if err != nil {
 			fmt.Fprintf(os.Stderr,
 				"collect error: %s\n", err.Error())