Set SQS visibility TO to 5min

Author: Daniel Jimenez

cmd/vulnerability-db-consumer/config.go

@@ -39,9 +39,9 @@ type dbConfig struct {
 }
 
 type sqsConfig struct {
-	NProcessors uint8 `toml:"number_of_processors"`
-	WaitTime    uint8 `toml:"wait_time"`
-	Timeout     uint8
+	NProcessors uint `toml:"number_of_processors"`
+	WaitTime    uint `toml:"wait_time"`
+	Timeout     uint
 	QueueARN    string `toml:"queue_arn"`
 	Endpoint    string `toml:"endpoint"`
 }

config.toml

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

pkg/queue/sqs.go

@@ -53,7 +53,7 @@ type awsData struct {
 }
 
 // NewSQSConsumerGroup creates a new SQSConsumerGroup.
-func NewSQSConsumerGroup(nConsumers uint8, config SQSConfig, processor Processor, logger *log.Logger) (*SQSConsumerGroup, error) {
+func NewSQSConsumerGroup(nConsumers uint, config SQSConfig, processor Processor, logger *log.Logger) (*SQSConsumerGroup, error) {
 	var consumerGroup SQSConsumerGroup
 
 	awsSess, err := session.NewSession()
@@ -64,7 +64,7 @@ func NewSQSConsumerGroup(nConsumers uint8, config SQSConfig, processor Processor
 	awsData := parseQueueARN(config.QueueArn)
 
 	var consumers []*SQSConsumer
-	for i := uint8(0); i < nConsumers; i++ {
+	for i := uint(0); i < nConsumers; i++ {
 		c, err := newSQSConsumer(awsSess, awsData, config, processor, logger)
 		if err != nil {
 			return nil, err