fix waiting logic

pkg/scan/scanner.go

@@ -95,24 +95,15 @@ func (s *Scanner) Scan(ctx *tasks.ScanContext, stype ScanType, namespace, id str
 	results := make(chan []output.IOCFound)
 	defer close(results)
 
-	done := make(chan bool)
-	defer close(done)
-
 	m := [2][]output.IOCFound{}
 	i := 0
 
 	wg.Add(1)
 	go func() {
 		defer wg.Done()
-		for {
-			select {
-			case malwares := <-results:
-				for _, malware := range malwares {
-					outputFn(malware, scanID)
-				}
-			case <-done:
-				logrus.Info("scan completed")
-				return
+		for malwares := range results {
+			for _, malware := range malwares {
+				outputFn(malware, scanID)
 			}
 		}
 	}()
@@ -140,8 +131,6 @@ func (s *Scanner) Scan(ctx *tasks.ScanContext, stype ScanType, namespace, id str
 		i %= len(m)
 	})
 
-	done <- true
-
 	wg.Wait()
 	return nil
 }