artillery/lib/platform/aws/aws-create-sqs-queue.js

Cloud-scale load testing. https://www.artillery.io

const {
  SQSClient,
  CreateQueueCommand,
  ListQueuesCommand
} = require('@aws-sdk/client-sqs');
const debug = require('debug')('artillery:aws-create-sqs-queue');
const sleep = require('../../util/sleep');

// TODO: Add timestamp to SQS queue name for automatic GC
async function createSQSQueue(region, queueName) {
  const sqs = new SQSClient({
    region
  });

  const params = {
    QueueName: queueName,
    Attributes: {
      FifoQueue: 'true',
      ContentBasedDeduplication: 'false',
      MessageRetentionPeriod: '1800',
      VisibilityTimeout: '600'
    }
  };

  const result = await sqs.send(new CreateQueueCommand(params));
  const sqsQueueUrl = result.QueueUrl;

  // Wait for the queue to be available:
  let waited = 0;
  let ok = false;
  while (waited < 120 * 1000) {
    try {
      const results = await sqs.send(
        new ListQueuesCommand({ QueueNamePrefix: queueName })
      );
      if (results.QueueUrls && results.QueueUrls.length === 1) {
        debug('SQS queue created:', queueName);
        ok = true;
        break;
      } else {
        await sleep(10 * 1000);
        waited += 10 * 1000;
      }
    } catch (_err) {
      await sleep(10 * 1000);
      waited += 10 * 1000;
    }
  }

  if (!ok) {
    debug('Time out waiting for SQS queue:', queueName);
    throw new Error('SQS queue could not be created');
  }

  return sqsQueueUrl;
}

module.exports = createSQSQueue;