feat: runner timeout

Author: xDarksome

.github/workflows/run-task.yml .github/workflows/setup-runners.yml

@@ -1,12 +1,9 @@
-name: Run ECS Fargate Github Action Runner task(s)
-
 on:
   workflow_call:
     inputs:
       cpu:
         description: 'Amount of CPU units to allocate to a single task'
         type: string
-        required: true
         default: '512'
       memory:
         description: 'Amount of memory (MB) to allocate to a single task'
@@ -15,22 +12,23 @@ on:
       labels:
         description: 'Coma-separated list of labels to assign to the tasks'
         type: string
-        required: true
         default: ''
       count:
         description: 'Number of tasks to run'
         type: string
-        required: true
         default: '1'
+      timeout:
+        description: 'Runner timeout'
+        type: string
+        default: '30m'
     secrets:
       TF_API_TOKEN:
         required: true
       GITHUB_PAT:
         required: true
 
 jobs:
-  run-task:
-    name: Run ECS task
+  setup-runners:
     runs-on: ubuntu-latest
     steps:
       - name: Generate runner token
@@ -71,7 +69,7 @@ jobs:
           echo 'run_task=true' >> .auto.tfvars
           echo 'cpu=${{ inputs.cpu }}' >> .auto.tfvars
           echo 'memory=${{ inputs.memory }}' >> .auto.tfvars
-          echo 'runner_token="${{ steps.gen-token.outputs.result }}"' >> .auto.tfvars
+          echo 'runner_token="::add-mask::${{ steps.gen-token.outputs.result }}"' >> .auto.tfvars
           echo 'repo_url="https://github.com/${{ github.repository }}"' >> .auto.tfvars
           echo 'labels="${{ inputs.labels }}"' >> .auto.tfvars
           echo 'desired_count=${{ inputs.count }}' >> .auto.tfvars

Dockerfile

@@ -3,3 +3,5 @@ FROM myoung34/github-runner:ubuntu-focal
 # modify actions runner binaries to allow custom cache server implementation
 # https://gha-cache-server.falcondev.io/getting-started
 RUN sed -i 's/\x41\x00\x43\x00\x54\x00\x49\x00\x4F\x00\x4E\x00\x53\x00\x5F\x00\x43\x00\x41\x00\x43\x00\x48\x00\x45\x00\x5F\x00\x55\x00\x52\x00\x4C\x00/\x41\x00\x43\x00\x54\x00\x49\x00\x4F\x00\x4E\x00\x53\x00\x5F\x00\x43\x00\x41\x00\x43\x00\x48\x00\x45\x00\x5F\x00\x4F\x00\x52\x00\x4C\x00/g' /actions-runner/bin/Runner.Worker.dll
+
+CMD ["timeout", "$TIMEOUT", "./bin/Runner.Listener", "run", "--startuptype", "service"]

main.tf

@@ -61,6 +61,11 @@ variable "desired_count" {
   default = null
 }
 
+variable "timeout" {
+  type    = string
+  default = null
+}
+
 locals {
   availability_zone = "eu-central-1a"
 }
@@ -121,13 +126,14 @@ module "runner" {
   cache_url         = module.cache-server.url
 }
 
-module "run_task" {
+module "setup-runners" {
   count         = var.run_task ? 1 : 0
-  source        = "./run_task"
+  source        = "./setup-runners"
   cpu           = var.cpu
   memory        = var.memory
   runner_token  = var.runner_token
   repo_url      = var.repo_url
   labels        = var.labels
   desired_count = var.desired_count
+  timeout = var.timeout
 }

run_task/main.tf setup-runners/main.tf

@@ -22,6 +22,10 @@ variable "desired_count" {
   type = number
 }
 
+variable "timeout" {
+  type    = string
+}
+
 data "aws_ecs_cluster" "this" {
   cluster_name = "github-actions-runner"
 }
@@ -76,6 +80,10 @@ data "aws_ecs_task_execution" "this" {
         key   = "START_DOCKER_SERVICE"
         value = true
       }
+      environment {
+        key   = "TIMEOUT"
+        value = var.timeout
+      }
     }
   }
 }