feat(liveliness): adding liveliness probe (#7)

Author: brandonrjacobs

Dockerfile

@@ -50,6 +50,11 @@ ADD entrypoint.sh /entrypoint.sh
 ADD create_ca_cert.sh /create_ca_cert.sh
 RUN chmod +x /create_ca_cert.sh /entrypoint.sh
 
+# Add Liveliness Probe script for CoreWeave
+RUN apk --no-cache add curl
+ADD liveliness.sh /liveliness.sh
+RUN chmod +x /liveliness.sh
+
 # Clients should only use 3128, not anything else.
 EXPOSE 3128
 

liveliness.sh

@@ -0,0 +1,15 @@
+#!/bin/bash
+
+IFS=', '
+
+read -r -a repos <<< "$LIVELINESS_REPOS"
+
+for i in "${repos[@]}"
+do
+  echo "repo $i liveliness check..."
+  response=$(curl --head -w '%{http_code}' -o /dev/null -s -k -x http://127.0.0.1:3128 "$i")
+  if [ "$response" -ne 200 ]; then
+    echo "failed curl for repo $i"
+    exit 1
+  fi
+done