Truncate dynamodb table

gistfile1.txt

#!/usr/bin/env bash

set -euo pipefail

TABLE_NAME="$1"
REGION="${2:-us-east-1}"

if [[ -z "$TABLE_NAME" ]]; then
  echo "Usage: $0 <table-name> [region]"
  exit 1
fi

echo "🔍 Describing table: $TABLE_NAME"

DESC=$(aws dynamodb describe-table \
  --table-name "$TABLE_NAME" \
  --region "$REGION")

echo "💾 Extracting schema..."

ATTRIBUTES=$(echo "$DESC" | jq '.Table.AttributeDefinitions')
KEY_SCHEMA=$(echo "$DESC" | jq '.Table.KeySchema')
BILLING_MODE=$(echo "$DESC" | jq -r '.Table.BillingModeSummary.BillingMode // "PROVISIONED"')

if [[ "$BILLING_MODE" == "PROVISIONED" ]]; then
  READ_CAP=$(echo "$DESC" | jq '.Table.ProvisionedThroughput.ReadCapacityUnits')
  WRITE_CAP=$(echo "$DESC" | jq '.Table.ProvisionedThroughput.WriteCapacityUnits')

  THROUGHPUT="--provisioned-throughput ReadCapacityUnits=$READ_CAP,WriteCapacityUnits=$WRITE_CAP"
else
  THROUGHPUT="--billing-mode PAY_PER_REQUEST"
fi

echo "🗑️ Deleting table..."
aws dynamodb delete-table \
  --table-name "$TABLE_NAME" \
  --region "$REGION"

echo "⏳ Waiting for table deletion..."
aws dynamodb wait table-not-exists \
  --table-name "$TABLE_NAME" \
  --region "$REGION"

echo "🛠️ Recreating table..."

aws dynamodb create-table \
  --table-name "$TABLE_NAME" \
  --attribute-definitions "$ATTRIBUTES" \
  --key-schema "$KEY_SCHEMA" \
  $THROUGHPUT \
  --region "$REGION"

echo "⏳ Waiting for table to become ACTIVE..."
aws dynamodb wait table-exists \
  --table-name "$TABLE_NAME" \
  --region "$REGION"

echo "✅ Table truncated successfully."