From 04576fcc1d8d70e0253d95e4193614fa24975e56 Mon Sep 17 00:00:00 2001
From: Brandon High <brandon.high@puppet.com>
Date: Fri, 31 Jan 2020 09:16:35 -0800
Subject: [PATCH] Add vmpooler flush script

---
 scripts/vmpooler_flush.sh | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100755 scripts/vmpooler_flush.sh

diff --git a/scripts/vmpooler_flush.sh b/scripts/vmpooler_flush.sh
new file mode 100755
index 0000000..6bb52a6
--- /dev/null
+++ b/scripts/vmpooler_flush.sh
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+if [ -z "${REDIS_PW}" ]; then
+  echo "redis password must be provided by setting \$REDIS_PW"
+  exit 1
+fi
+
+POOLER_URL="${POOLER_URL:-vmpooler.delivery.puppetlabs.net/api/v1/config}"
+REDIS_HOST="${REDIS_HOST:-vmpooler-redis-prod-2.delivery.puppetlabs.net}"
+
+for a in $(curl "${POOLER_URL}" | grep 'name":' | awk '{print "$NF"}' | sed 's/"//g' | sed 's/,//'); do 
+  for i in $(redis-cli --no-auth-warning -h "${REDIS_HOST}" -a "${REDIS_PW}" smembers vmpooler__ready__"${a}") ; do 
+    redis-cli --no-auth-warning -h "${REDIS_HOST}" -a "${REDIS_PW}" smove vmpooler__ready__"${a}" vmpooler__completed__"${a}" "${i}"
+  done
+done