Add retry

Author: brandboat

README.md

@@ -193,8 +193,8 @@ Now you're ready to go!
 | `ec2-instance-id`                                                                                                                                                            | Required if you use the `stop` mode.       | EC2 Instance Id of the created runner. <br><br> The id is provided by the output of the action in the `start` mode. <br><br> The id is used to terminate the EC2 instance when the runner is not needed anymore.                                                                                                                      |
 | `iam-role-name`                                                                                                                                                              | Optional. Used only with the `start` mode. | IAM role name to attach to the created EC2 runner. <br><br> This allows the runner to have permissions to run additional actions within the AWS account, without having to manage additional GitHub secrets and AWS users. <br><br> Setting this requires additional AWS permissions for the role launching the instance (see above). |
 | `aws-resource-tags`                                                                                                                                                          | Optional. Used only with the `start` mode. | Specifies tags to add to the EC2 instance and any attached storage. <br><br> This field is a stringified JSON array of tag objects, each containing a `Key` and `Value` field (see example below). <br><br> Setting this requires additional AWS permissions for the role launching the instance (see above).                         |
-| `runner-home-dir`                                                                                                                                                              | Optional. Used only with the `start` mode. | Specifies a directory where pre-installed actions-runner software and scripts are located.<br><br> |
-
+| `runner-home-dir`                                                                                                                                                            | Optional. Used only with the `start` mode. | Specifies a directory where pre-installed actions-runner software and scripts are located.<br><br>                                                                                                                                                                                                                                    |
+| `max_attempts`                                                                                                                                                               | Optional. Default is 1.                    | Specifies the amount attempts to retry `start` or `stop` mode.                                                                                                                                                                                                                                                                        |
 ### Environment variables
 
 In addition to the inputs described above, the action also requires the following environment variables to access your AWS account:

action.yml

@@ -22,7 +22,7 @@ inputs:
     required: false
   ec2-instance-type:
     description: >-
-      EC2 Instance Type. 
+      EC2 Instance Type.
       This input is required if you use the 'start' mode.
     required: false
   subnet-id:
@@ -32,7 +32,7 @@ inputs:
     required: false
   security-group-id:
     description: >-
-      EC2 Security Group Id. 
+      EC2 Security Group Id.
       The security group should belong to the same VPC as the specified subnet.
       The runner doesn't require any inbound traffic. However, outbound traffic should be allowed.
       This input is required if you use the 'start' mode.
@@ -65,6 +65,10 @@ inputs:
     description: >-
       Directory that contains actions-runner software and scripts. E.g. /home/runner/actions-runner.
     required: false
+  max_attempts:
+    description: Number of attempts. Default is 1
+    required: false
+    default: 1
 outputs:
   label:
     description: >-

src/index.js

@@ -23,10 +23,24 @@ async function stop() {
 }
 
 (async function () {
-  try {
-    config.input.mode === 'start' ? await start() : await stop();
-  } catch (error) {
-    core.error(error);
-    core.setFailed(error.message);
-  }
+  const MAX_ATTEMPTS = Number.parseInt(core.getInput('max_attempts'));
+  let attempt = 0;
+  let hasSucceeded = false;
+  do {
+    try {
+      config.input.mode === 'start' ? await start() : await stop();
+      hasSucceeded = true;
+    } catch (error) {
+      await sleep(5000);
+      attempt += 1;
+      core.warning(`Attempt ${attempt} of ${MAX_ATTEMPTS}`);
+      if (attempt >= MAX_ATTEMPTS) {
+        core.error('Max attempts exceeded');
+        core.error(error);
+        core.setFailed(error.message);
+      } else {
+        core.warning(`${error} - ${error.message}`);
+      }
+    }
+  } while (attempt < MAX_ATTEMPTS && !hasSucceeded);
 })();