Merge pull request #220 from bustle/respect_limits

Add automatic retry to deploys
bustle · May 22, 2017 · 95bcc28 · 95bcc28
2 parents 49e1dbe + 6ff3634
commit 95bcc28
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/util/aws/api-gateway.js b/src/util/aws/api-gateway.js
@@ -1,4 +1,7 @@
 import AWS from './'
+import Promise from 'bluebird'
+
+export const DEPLOY_ATTEMPT_MAX = 2
 
 export function exportStage (restApiId, stageName) {
   const apiGateway = new AWS.APIGateway()
@@ -16,9 +19,14 @@ export function exportStage (restApiId, stageName) {
   .get('body')
 }
 
-export function deploy (id, env) {
+export function deploy (id, env, attempts = 1) {
   const apiGateway = new AWS.APIGateway()
   return apiGateway.createDeployment({restApiId: id, stageName: env, variables: { functionAlias: env }}).promise()
+  .catch({ code: 'TooManyRequestsException' }, async ({ retryable, retryDelay }) => {
+    if (!retryable && attempts > DEPLOY_ATTEMPT_MAX) { throw new Error('Amazon limit hit') }
+    await Promise.delay(retryDelay * 1000)
+    return deploy(id, env, attempts++)
+  })
 }
 
 export function pushApi (api, id) {