Fix the queue attributes used in heartbeat ping

This commit is contained in:
Les Orchard 2018-06-28 16:53:10 -04:00
Родитель 4ed5f5304f
Коммит 5728560144
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 6DA657B1385DFCAE
2 изменённых файлов: 10 добавлений и 6 удалений

Просмотреть файл

@ -122,11 +122,7 @@ const sendHeartbeatMetrics = async (
const { QueueUrl } = await SQS.getQueueUrl({
QueueName: QUEUE_NAME
}).promise();
const {
ApproximateNumberOfMessages: items_in_queue,
ApproximateNumberOfMessagesDelayed: items_in_waiting,
ApproximateNumberOfMessagesNotVisible: items_in_progress
} = await SQS.getQueueAttributes({
const attribsResult = await SQS.getQueueAttributes({
QueueUrl,
AttributeNames: [
"ApproximateNumberOfMessages",
@ -134,6 +130,12 @@ const sendHeartbeatMetrics = async (
"ApproximateNumberOfMessagesNotVisible"
]
}).promise();
const {
ApproximateNumberOfMessages: items_in_queue,
ApproximateNumberOfMessagesDelayed: items_in_waiting,
ApproximateNumberOfMessagesNotVisible: items_in_progress
} =
attribsResult.Attributes || {};
await Metrics.pollerHeartbeat({
poller_id,
items_in_queue,

Просмотреть файл

@ -79,7 +79,9 @@ global.resetMocks = () => {
mocks.deleteItem.returns(makePromiseFn({}));
mocks.putItem.returns(makePromiseFn({}));
mocks.getItem.returns(makePromiseFn({}));
mocks.getQueueAttributes.returns(makePromiseFn({ QueueAttributes }));
mocks.getQueueAttributes.returns(
makePromiseFn({ Attributes: QueueAttributes })
);
mocks.getQueueUrl.returns(makePromiseFn({ QueueUrl }));
mocks.getSignedUrl.returns("");
mocks.putObject.returns(makePromiseFn({ ETag }));