upload-etl.sh

#!/bin/bash
#
# This script will use kubectl to copy the ETL backup directory to a temporary 
# location, then tar the contents and remove the tmp directory.
# 

# Accept Optional Namespace -- default to kubecost
namespace=$1
if [ "$namespace" == "" ]; then
  namespace=kubecost
fi

# Accept Optional ETL Store Directory -- default to /var/configs/db/etl
etlDir=$2
if [ "$etlDir" == "" ]; then
  etlDir=/var/configs/db/etl
fi

# Accept etl .tar file to upload
etlFile=$3
if [ "$etlFile" == "" ]; then
  etlFile=kubecost-etl.tar.gz
fi

# Grab the Current Context for Prompt
currentContext=`kubectl config current-context`

echo "This script will delete the Kubecost ETL storage and replace it with ETL files using the following:"
echo "  Kubectl Context: $currentContext"
echo "  Namespace: $namespace"
echo "  ETL File (source): $etlFile"
echo "  ETL Directory (destination): $etlDir"
echo -n "Would you like to continue [Y/n]? "
read r

if [ "$r" == "${r#[Y]}" ]; then
  echo "Exiting..."
  exit 0
fi


# Grab the Pod Name of the cost-analyzer pod
podName=`kubectl get pods -n $namespace -l app=cost-analyzer -o jsonpath='{.items[0].metadata.name}'`

# Grab the Deployment name of the cost-analyzer pod
deployName=`kubectl get deploy -n $namespace -l app=cost-analyzer -o jsonpath='{.items[0].metadata.name}'`

# Copy the Files to tmp directory
echo "Copying ETL Files from $etlFile to $podName..."
kubectl cp -c cost-model $etlFile $namespace/$podName:/var/configs/kubecost-etl.tar.gz

# Exec into the pod and replace the ETL
echo "Execing into the pod and replacing $etlDir "
set -e
kubectl exec -n $namespace pod/$podName -- ash -c " \
  rm -rf /var/configs/kc-etl-tmp && \
  tar xzf /var/configs/kubecost-etl.tar.gz --directory /var/configs && \
  [ -d /var/configs/kc-etl-tmp ] && rm -rf $etlDir \
    || (echo '$etlFile is invalid' && exit 1) && \
  mv /var/configs/kc-etl-tmp $etlDir"

# Restart the application to pull ETL data into memory
echo "Restarting the application"
kubectl -n $namespace rollout restart deployment/$deployName