DynamoDBの特定テーブルをtruncateするためのシェルスクリプト #aws-cli

概要

DynamoDBの指定したテーブルをtruncateします。

注意事項

batchWriteItemを使用していないためとても遅いです(要改善)。

前提条件

aws cliがcredentialを含めて設定済みである必要があります。
jqがインストール済みである必要があります。

パラメータ

$1: テーブル名
$2: パーティションキーのAttribute name
$3: ソートキーのAttribute name(オプション)

シェルスクリプト

以下を任意のディレクトリにtruncate_dynamodb.shとして保存してください。

#!/bin/sh

TABLE_NAME=$1
P_KEY=$2
S_KEY=$3
REGION=ap-northeast-1
TRUNCATE_LIST=/tmp/truncate.lst_$(date +%s)

if [ -z "$TABLE_NAME" ] || [ -z "$P_KEY" ]; then
        echo "truncate_dynamodb.sh [Table Name] [Partition Key] [Sort Key(Optional)]"
        exit 1;
fi

KEY_LIST="$P_KEY: .$P_KEY"

if [ -n "$S_KEY" ]; then
        KEY_LIST="$KEY_LIST, $S_KEY: .$S_KEY"
fi

echo "Key is '$KEY_LIST'"

aws --region ap-northeast-1 dynamodb scan --table-name $TABLE_NAME | jq -c ".Items[] | { $KEY_LIST  }" > $TRUNCATE_LIST

while read LINE
do
        COMMAND="aws --region "$REGION" dynamodb delete-item --table-name $TABLE_NAME --key "$LINE""
        $COMMAND
done < $TRUNCATE_LIST

rm $TRUNCATE_LIST