feat(cloud-init): add pre/post-scripts to retrieve the job repository (#17)

* feat(cloud-init): add pre/post-scripts to retrieve the job repository

KAG-7656

* use heredoc syntax

* remove some tags

Author: Water-Melon

cloud-init.sh.tmpl

@@ -96,6 +96,26 @@ if [[ $ARCH == "x86_64" ]]; then ARCH=x64
 else ARCH=arm64
 fi
 
+cat << 'EOF' > start.sh
+#!/bin/bash
+date +%s%3N > /tmp/job_start.ts
+EOF
+
+cat << EOF > stop.sh
+#!/bin/bash
+now_ts=\$(date +%s%3N)
+start_ts=\$(cat /tmp/job_start.ts)
+job_duration_ms=\$((now_ts - start_ts))
+function send_metrics() {
+  metrics="github.actions.\$1:\$${2:-1}|\$${3:-c}|#runner_group:${RUNNERGROUP}"
+  metrics=\$metrics,#repo:\$GITHUB_REPOSITORY,#workflow:\$GITHUB_WORKFLOW,#job:\$GITHUB_JOB,#arch:$ARCH
+  echo "Send metrics \$metrics to $dd_host"
+  echo -n "\$metrics" > /dev/udp/$dd_host/8125
+}
+send_metrics jobs.repo
+send_metrics jobs.duration \$job_duration_ms "ms|@1"
+EOF
+
 curl -o actions-runner-linux-$${ARCH}-${RUNNER_VER}.tar.gz -L https://github.com/actions/runner/releases/download/v${RUNNER_VER}/actions-runner-linux-$${ARCH}-${RUNNER_VER}.tar.gz
 
 tar xzf ./actions-runner-linux-$${ARCH}-${RUNNER_VER}.tar.gz
@@ -159,16 +179,13 @@ provision_duration_ms=$(awk '{print $1*1000}' /proc/uptime)
 send_metrics runner.provision_duration $provision_duration_ms "ms|@1"
 
 IFS=$'\r\n'
+export ACTIONS_RUNNER_HOOK_JOB_STARTED=/home/ubuntu/actions-runner/start.sh
+export ACTIONS_RUNNER_HOOK_JOB_COMPLETED=/home/ubuntu/actions-runner/stop.sh
 ./run.sh | grep -vE "^[ \t|/_\\()',\.\-]*$"| grep -vF 'Self-hosted runner registration' | while read line; do
 	echo $line
 	if echo $line | grep -q "Running job: "; then
 		ts=$(echo $line | grep -oE '[0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2}Z')
 		start_time=$(date -u -d "$ts" +%s)
-	elif echo $line | grep -q "completed with result: "; then
-		ts=$(echo $line | grep -oE '[0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2}Z')
-		job_duration_ms=$(( ( $(date -u -d "$ts" +%s) - start_time ) * 1000 ))
-		send_metrics jobs.duration $job_duration_ms "ms|@1"
-		send_logs job_duration $job_duration_ms
 	fi
 	send_logs runner_log $line
 done