Optimize startup time (machulav#223)

* added startup optimization config

* added startup optimization config

* added startup optimization config

Author: Preen

action.yml

@@ -74,7 +74,19 @@ inputs:
       Specifies the market (purchasing) option for the instance:
           - 'spot' - Use a spot instance
     required: false
-
+  startup-quiet-period-seconds:
+    description: >-
+      Specifies the quiet period in seconds after the instance starts.
+      The runner will not be registered during this period.
+    required: false
+  startup-retry-interval-seconds:
+    description: >-
+      Specifies the retry interval in seconds to register the runner after the quiet period.
+    required: false
+  startup-timeout-seconds:
+    description: >-
+      Specifies the timeout in seconds to register the runner after the quiet period.
+    required: false
 outputs:
   label:
     description: >-

dist/index.js

@@ -145145,18 +145145,21 @@ const github = __nccwpck_require__(5438);
 class Config {
   constructor() {
     this.input = {
-      mode: core.getInput('mode'),
-      githubToken: core.getInput('github-token'),
       ec2ImageId: core.getInput('ec2-image-id'),
-      ec2InstanceType: core.getInput('ec2-instance-type'),
-      subnetId: core.getInput('subnet-id'),
-      securityGroupId: core.getInput('security-group-id'),
-      label: core.getInput('label'),
       ec2InstanceId: core.getInput('ec2-instance-id'),
+      ec2InstanceType: core.getInput('ec2-instance-type'),
+      githubToken: core.getInput('github-token'),
       iamRoleName: core.getInput('iam-role-name'),
-      runnerHomeDir: core.getInput('runner-home-dir'),
-      preRunnerScript: core.getInput('pre-runner-script'),
+      label: core.getInput('label'),
       marketType: core.getInput('market-type'),
+      mode: core.getInput('mode'),
+      preRunnerScript: core.getInput('pre-runner-script'),
+      runnerHomeDir: core.getInput('runner-home-dir'),
+      securityGroupId: core.getInput('security-group-id'),
+      startupQuietPeriodSeconds: core.getInput('startup-quiet-period-seconds'),
+      startupRetryIntervalSeconds: core.getInput('startup-retry-interval-seconds'),
+      startupTimeoutMinutes: core.getInput('startup-timeout-minutes'),
+      subnetId: core.getInput('subnet-id'),
     };
 
     const tags = JSON.parse(core.getInput('aws-resource-tags'));
@@ -145277,9 +145280,10 @@ async function removeRunner() {
 }
 
 async function waitForRunnerRegistered(label) {
-  const timeoutMinutes = 5;
-  const retryIntervalSeconds = 10;
-  const quietPeriodSeconds = 30;
+  const timeoutMinutes = config.input.startupTimeoutMinutes ? config.input.startupTimeoutMinutes : 5;
+  const retryIntervalSeconds = config.input.startupRetryIntervalSeconds ? config.input.startupRetryIntervalSeconds : 10;
+  const quietPeriodSeconds = config.input.startupQuietPeriodSeconds ? config.input.startupQuietPeriodSeconds : 30;
+
   let waitSeconds = 0;
 
   core.info(`Waiting ${quietPeriodSeconds}s for the AWS EC2 instance to be registered in GitHub as a new self-hosted runner`);

src/config.js

@@ -4,18 +4,21 @@ const github = require('@actions/github');
 class Config {
   constructor() {
     this.input = {
-      mode: core.getInput('mode'),
-      githubToken: core.getInput('github-token'),
       ec2ImageId: core.getInput('ec2-image-id'),
-      ec2InstanceType: core.getInput('ec2-instance-type'),
-      subnetId: core.getInput('subnet-id'),
-      securityGroupId: core.getInput('security-group-id'),
-      label: core.getInput('label'),
       ec2InstanceId: core.getInput('ec2-instance-id'),
+      ec2InstanceType: core.getInput('ec2-instance-type'),
+      githubToken: core.getInput('github-token'),
       iamRoleName: core.getInput('iam-role-name'),
-      runnerHomeDir: core.getInput('runner-home-dir'),
-      preRunnerScript: core.getInput('pre-runner-script'),
+      label: core.getInput('label'),
       marketType: core.getInput('market-type'),
+      mode: core.getInput('mode'),
+      preRunnerScript: core.getInput('pre-runner-script'),
+      runnerHomeDir: core.getInput('runner-home-dir'),
+      securityGroupId: core.getInput('security-group-id'),
+      startupQuietPeriodSeconds: core.getInput('startup-quiet-period-seconds'),
+      startupRetryIntervalSeconds: core.getInput('startup-retry-interval-seconds'),
+      startupTimeoutMinutes: core.getInput('startup-timeout-minutes'),
+      subnetId: core.getInput('subnet-id'),
     };
 
     const tags = JSON.parse(core.getInput('aws-resource-tags'));

src/gh.js

@@ -52,9 +52,10 @@ async function removeRunner() {
 }
 
 async function waitForRunnerRegistered(label) {
-  const timeoutMinutes = 5;
-  const retryIntervalSeconds = 10;
-  const quietPeriodSeconds = 30;
+  const timeoutMinutes = config.input.startupTimeoutMinutes ? config.input.startupTimeoutMinutes : 5;
+  const retryIntervalSeconds = config.input.startupRetryIntervalSeconds ? config.input.startupRetryIntervalSeconds : 10;
+  const quietPeriodSeconds = config.input.startupQuietPeriodSeconds ? config.input.startupQuietPeriodSeconds : 30;
+
   let waitSeconds = 0;
 
   core.info(`Waiting ${quietPeriodSeconds}s for the AWS EC2 instance to be registered in GitHub as a new self-hosted runner`);