index.js

const H = require ( 'highland' );
const AWS = require ( 'aws-sdk' );

module.exports = ( { awsConfig, count, queueUrl, visibilityTimeout, waitTimeSeconds, maxRetry } = { } ) => {
  const SQS = new AWS.SQS ( {
    apiVersion: '2012-11-05',
    ...awsConfig
  } );

  const generator = ( attempt ) => {
    return H ( ( push, next ) => {
      return H.wrapCallback ( SQS.receiveMessage.bind ( SQS ) ) ( {
         MaxNumberOfMessages: count || 5,
         QueueUrl: queueUrl,
         VisibilityTimeout: visibilityTimeout || 0,
         WaitTimeSeconds: waitTimeSeconds || 0
      } )
       .each ( response => {
         const items = response.Messages || [];

         if ( items.length ) {
           items.forEach ( item => {
               push ( null, item );
           } );

           return setTimeout ( () => {
               next ( generator ( 0 ) );
           }, 0 );
         }

         if ( attempt < ( maxRetry || 3 ) ) {
           return setTimeout ( () => {
               next ( generator ( attempt + 1 ) );
           }, 0 );
         }

         return push ( null, H.nil );
       } )
    } );
  }

  return generator ( 0 );
};