Add reset visibility param to aws_sqs input to potentially avoid rese…

…tting visibility timeout
chunchun010 · Oct 20, 2021 · 9aedddd · 9aedddd
1 parent 0150fae
commit 9aedddd
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/lib/input/aws_sqs.go b/lib/input/aws_sqs.go
@@ -71,17 +71,19 @@ You can access these metadata fields using
 
 // AWSSQSConfig contains configuration values for the input type.
 type AWSSQSConfig struct {
- sess.Config `json:",inline" yaml:",inline"`
- URL string `json:"url" yaml:"url"`
- DeleteMessage bool `json:"delete_message" yaml:"delete_message"`
+ sess.Config `json:",inline" yaml:",inline"`
+ URL string `json:"url" yaml:"url"`
+ DeleteMessage bool `json:"delete_message" yaml:"delete_message"`
+ ResetVisibility bool `json:"reset_visibility" yaml:"reset_visibility"`
 }
 
 // NewAWSSQSConfig creates a new Config with default values.
 func NewAWSSQSConfig() AWSSQSConfig {
  return AWSSQSConfig{
- Config: sess.NewConfig(),
- URL: "",
- DeleteMessage: true,
+ Config: sess.NewConfig(),
+ URL: "",
+ DeleteMessage: true,
+ ResetVisibility: true,
  }
 }
 
@@ -405,6 +407,9 @@ func (a *awsSQS) ReadWithContext(ctx context.Context) (types.Message, reader.Asy
  return nil
  }
 
+ if !a.conf.ResetVisibility {
+ return nil
+ }
  select {
  case <-rctx.Done():
  return rctx.Err()