improve vulnerability processing

cmd/vulnerability-db-consumer/config.go

@@ -5,7 +5,7 @@ Copyright 2020 Adevinta
 package main
 
 import (
-	"io/ioutil"
+	"io"
 	"os"
 
 	log "github.com/sirupsen/logrus"
@@ -39,8 +39,8 @@ type dbConfig struct {
 
 type sqsConfig struct {
 	NProcessors uint8 `toml:"number_of_processors"`
-	WaitTime    uint8 `toml:"wait_time"`
-	Timeout     uint8
+	WaitTime    int   `toml:"wait_time"`
+	Timeout     int
 	QueueARN    string `toml:"queue_arn"`
 	Endpoint    string `toml:"endpoint"`
 }
@@ -73,7 +73,7 @@ func parseConfig(cfgFilePath string) (*config, error) {
 	}
 	defer cfgFile.Close()
 
-	cfgData, err := ioutil.ReadAll(cfgFile)
+	cfgData, err := io.ReadAll(cfgFile)
 
 	var conf config
 	if _, err := toml.Decode(string(cfgData[:]), &conf); err != nil {

config.toml

@@ -17,7 +17,7 @@ name = "$PG_NAME"
 [sqs]
 number_of_processors = $SQS_NUMBER_OF_PROCESSORS
 wait_time = 20
-timeout = 30
+timeout = 600
 queue_arn = "$SQS_QUEUE_ARN"
 endpoint = "$AWS_SQS_ENDPOINT"
 

pkg/processor/processor.go

@@ -148,10 +148,11 @@ func (p *CheckProcessor) ProcessMessage(m string) error {
 	l.Debugf("Processing source with %d source findings: %#v", len(sourceFindings), source)
 	source, err = p.store.ProcessSourceExecution(source, sourceFindings)
 	if err != nil {
-		l.Errorf("Error while processing source: %#v", err)
-		if !store.IsDuplicateErr(err) {
-			return err
+		if store.IsDuplicateErr(err) {
+			l.Warnf("duplicated source: %#v", err)
+			return nil
 		}
+		return err
 	}
 
 	l.Debug("Sending open findings")

pkg/store/issues.go

@@ -37,20 +37,18 @@ func (db *psqlxStore) UpdateIssueLabels(issueID string, labels []string) error {
 		return err
 	}
 
-	_, err = tx.Exec("DELETE FROM issue_labels WHERE issue_id = $1", issueID)
+	l := pq.Array(labels)
+	_, err = tx.Exec("DELETE FROM issue_labels WHERE issue_id = $1 and label <> ANY($2::TEXT[])", issueID, l)
 	if err != nil {
 		tx.Rollback()
 		return err
 	}
 
-	for _, label := range labels {
-		_, err := tx.Exec("INSERT INTO issue_labels (issue_id, label) VALUES ($1, $2)", issueID, label)
-		if err != nil {
-			tx.Rollback()
-			return err
-		}
+	_, err = tx.Exec("INSERT INTO issue_labels SELECT $1, unnest($2::TEXT[]) ON CONFLICT DO NOTHING", issueID, l)
+	if err != nil {
+		tx.Rollback()
+		return err
 	}
-
 	return tx.Commit()
 }