Add the ability to have a delay before retrying

README.md

@@ -104,6 +104,23 @@ By default, requests made with the `github` instance will not be retried. You ca
 
 In this example, request failures from `github.rest.issues.get()` will be retried up to 3 times.
 
+You can also configure the delay (delayInSeconds) before retrying via the `retry-after` option:
+
+```yaml
+- uses: actions/github-script@v6
+  id: my-script
+  with:
+    result-encoding: string
+    retries: 3
+    retry-after: 500
+    script: |
+      github.rest.issues.get({
+        issue_number: context.issue.number,
+        owner: context.repo.owner,
+        repo: context.repo.repo,
+      })
+```
+
 You can also configure which status codes should be exempt from retries via the `retry-exempt-status-codes` option:
 
 ```yaml

__test__/get-retry-options.test.ts

@@ -7,11 +7,13 @@ describe('getRequestOptions', () => {
     const [retryOptions, requestOptions] = getRetryOptions(
       0,
       [400, 500, 502],
+      321,
       []
     )
 
     expect(retryOptions.enabled).toBe(false)
     expect(retryOptions.doNotRetry).toBeFalsy()
+    expect(retryOptions.delay).toBeFalsy()
 
     expect(requestOptions?.retries).toBeFalsy()
   })
@@ -20,39 +22,59 @@ describe('getRequestOptions', () => {
     const [retryOptions, requestOptions] = getRetryOptions(
       1,
       [400, 500, 502],
+      321,
       []
     )
 
     expect(retryOptions.enabled).toBe(true)
     expect(retryOptions.doNotRetry).toEqual([400, 500, 502])
+    expect(retryOptions.delay).toEqual(321)
 
     expect(requestOptions?.retries).toEqual(1)
   })
 
-  test('properties set if retries > 0', async () => {
+  test('delay should be undefined if retryAfter < 0', async () => {
     const [retryOptions, requestOptions] = getRetryOptions(
       1,
       [400, 500, 502],
+      -1,
       []
     )
 
     expect(retryOptions.enabled).toBe(true)
     expect(retryOptions.doNotRetry).toEqual([400, 500, 502])
+    expect(retryOptions.delay).toBeUndefined()
+
+    expect(requestOptions?.retries).toEqual(1)
+  })
+
+  test('delay should be undefined if retryAfter = 0', async () => {
+    const [retryOptions, requestOptions] = getRetryOptions(
+      1,
+      [400, 500, 502],
+      0,
+      []
+    )
+
+    expect(retryOptions.enabled).toBe(true)
+    expect(retryOptions.doNotRetry).toEqual([400, 500, 502])
+    expect(retryOptions.delay).toBeUndefined()
 
     expect(requestOptions?.retries).toEqual(1)
   })
 
   test('retryOptions.doNotRetry not set if exemptStatusCodes isEmpty', async () => {
-    const [retryOptions, requestOptions] = getRetryOptions(1, [], [])
+    const [retryOptions, requestOptions] = getRetryOptions(1, [], 321, [])
 
     expect(retryOptions.enabled).toBe(true)
     expect(retryOptions.doNotRetry).toBeUndefined()
+    expect(retryOptions.delay).toBe(321)
 
     expect(requestOptions?.retries).toEqual(1)
   })
 
   test('requestOptions does not override defaults from @actions/github', async () => {
-    const [retryOptions, requestOptions] = getRetryOptions(1, [], {
+    const [retryOptions, requestOptions] = getRetryOptions(1, [], 321, {
       request: {
         agent: 'default-user-agent'
       },
@@ -60,6 +82,7 @@ describe('getRequestOptions', () => {
     })
 
     expect(retryOptions.enabled).toBe(true)
+    expect(retryOptions.delay).toBe(321)
     expect(retryOptions.doNotRetry).toBeUndefined()
 
     expect(requestOptions?.retries).toEqual(1)

action.yml

@@ -26,6 +26,9 @@ inputs:
   retries:
     description: The number of times to retry a request
     default: "0"
+  retry-after:
+    description: The number of seconds after which the retry attempt should be made. No effect unless `retries` is set
+    default: "1000"
   retry-exempt-status-codes:
     description: A comma separated list of status codes that will NOT be retried e.g. "400,500". No effect unless `retries` is set
     default: 400,401,403,404,422 # from https://github.com/octokit/plugin-retry.js/blob/9a2443746c350b3beedec35cf26e197ea318a261/src/index.ts#L14

src/main.ts

@@ -6,10 +6,10 @@ import * as glob from '@actions/glob'
 import * as io from '@actions/io'
 import { retry } from '@octokit/plugin-retry'
 import { RequestRequestOptions } from '@octokit/types'
+import fetch from 'node-fetch'
 import { callAsyncFunction } from './async-function'
 import { getRetryOptions, parseNumberArray, RetryOptions } from './retry-options'
 import { wrapRequire } from './wrap-require'
-import fetch from 'node-fetch'
 
 process.on('unhandledRejection', handleError)
 main().catch(handleError)
@@ -28,12 +28,14 @@ async function main(): Promise<void> {
   const userAgent = core.getInput('user-agent')
   const previews = core.getInput('previews')
   const retries = parseInt(core.getInput('retries'))
+  const retryAfter = parseInt(core.getInput('retry-after'))
   const exemptStatusCodes = parseNumberArray(
     core.getInput('retry-exempt-status-codes')
   )
   const [retryOpts, requestOpts] = getRetryOptions(
     retries,
     exemptStatusCodes,
+    retryAfter,
     defaultGitHubOptions
   )
 

src/retry-options.ts

@@ -5,11 +5,13 @@ import {RequestRequestOptions} from '@octokit/types'
 export type RetryOptions = {
   doNotRetry?: number[]
   enabled?: boolean
+  delay?: number
 }
 
 export function getRetryOptions(
   retries: number,
   exemptStatusCodes: number[],
+  retryAfter: number,
   defaultOptions: OctokitOptions
 ): [RetryOptions, RequestRequestOptions | undefined] {
   if (retries <= 0) {
@@ -20,6 +22,10 @@ export function getRetryOptions(
     enabled: true
   }
 
+  if (retryAfter > 0) {
+    retryOptions.delay = retryAfter
+  }
+
   if (exemptStatusCodes.length > 0) {
     retryOptions.doNotRetry = exemptStatusCodes
   }
@@ -33,10 +39,9 @@ export function getRetryOptions(
   }
 
   core.debug(
-    `GitHub client configured with: (retries: ${
+    `GitHub client configured with: (retries: ${requestOptions.retries
-      requestOptions.retries
+    }, retryAfter: ${retryOptions.delay ?? 'octokit default: 1000'
-    }, retry-exempt-status-code: ${
+    } retry-exempt-status-code: ${retryOptions?.doNotRetry ?? 'octokit default: [400, 401, 403, 404, 422]'
-      retryOptions?.doNotRetry ?? 'octokit default: [400, 401, 403, 404, 422]'
     })`
   )