Small concurrency tweaks

Signed-off-by: egibs <20933572+egibs@users.noreply.github.com>

Author: egibs

cmd/mal/mal.go

@@ -244,16 +244,21 @@ func main() {
 				returnCode = ExitInvalidRules
 			}
 
+			concurrency := concurrencyFlag
+			if concurrency < 1 {
+				concurrency = 1
+			}
+
 			var pool *malcontent.ScannerPool
 			if mc.ScannerPool == nil {
-				pool, err = malcontent.NewScannerPool(yrs, runtime.NumCPU())
+				pool, err = malcontent.NewScannerPool(yrs, concurrency)
 				if err != nil {
 					returnCode = ExitInvalidRules
 				}
 			}
 
 			mc = malcontent.Config{
-				Concurrency:           concurrencyFlag,
+				Concurrency:           concurrency,
 				ExitFirstHit:          exitFirstHitFlag,
 				ExitFirstMiss:         exitFirstMissFlag,
 				IgnoreSelf:            ignoreSelfFlag,

pkg/action/scan.go

@@ -57,7 +57,7 @@ func scanSinglePath(ctx context.Context, c malcontent.Config, path string, ruleF
 
 	var pool *malcontent.ScannerPool
 	if c.ScannerPool == nil {
-		pool, err = malcontent.NewScannerPool(yrs, runtime.NumCPU())
+		pool, err = malcontent.NewScannerPool(yrs, c.Concurrency)
 		if err != nil {
 			return nil, fmt.Errorf("failed to create scanner pool: %w", err)
 		}

pkg/refresh/action.go

@@ -67,6 +67,7 @@ func actionRefresh(ctx context.Context) ([]TestData, error) {
 		}
 
 		c := &malcontent.Config{
+			Concurrency:           runtime.NumCPU(),
 			IgnoreSelf:            false,
 			MinFileRisk:           0,
 			MinRisk:               0,

pkg/refresh/diff.go

@@ -193,6 +193,7 @@ func diffRefresh(ctx context.Context, rc Config) ([]TestData, error) {
 		}
 
 		c := &malcontent.Config{
+			Concurrency:           runtime.NumCPU(),
 			FileRiskChange:        td.riskChange,
 			FileRiskIncrease:      td.riskIncrease,
 			MinFileRisk:           minFileRisk,

pkg/refresh/refresh.go

@@ -72,6 +72,7 @@ func discoverTestData(rc Config) (map[string]string, error) {
 // newConfig returns a new malcontent Config with parent configurations.
 func newConfig(rc Config) *malcontent.Config {
 	return &malcontent.Config{
+		Concurrency:           runtime.NumCPU(),
 		IgnoreTags:            []string{"harmless"},
 		MinFileRisk:           1,
 		MinRisk:               1,