options: Fix #8 Add WaitTimeSeconds

nabeken · May 4, 2020 · b159bc6 · b159bc6
1 parent 56bf52b
commit b159bc6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/queue/option/option.go b/queue/option/option.go
@@ -34,6 +34,14 @@ func MaxNumberOfMessages(n int64) ReceiveMessageInput {
 	}
 }
 
+// WaitTimeSeconds returns a ReceiveMessageInput that
+// changes WaitTimeSeconds parameter.
+func WaitTimeSeconds(n int64) ReceiveMessageInput {
+	return func(req *sqs.ReceiveMessageInput) {
+		req.WaitTimeSeconds = aws.Int64(n)
+	}
+}
+
 // UseAllAttribute returns a ReceiveMessageInput that
 // changes a parameter to receive all messages regardless of attributes.
 func UseAllAttribute() ReceiveMessageInput {