はじめに
AWS DynamoDBは、スケーラブルで柔軟性が高く、パフォーマンスに優れたNoSQLデータベースサービスです。DynamoDB Accelerator(DAX)は、DynamoDBをより高速に扱うことができるサービスです。
しかし、PythonとDAXの組み合わせに関する解説記事は少なく、ハードルが高いかもしれません。
よって本記事ではPythonを使ったDAXの基本的な利用方法をわかりやすく紹介します。
前回の解説記事であるPython×DynamoDBに内容が似ています。
また「Python×AWSのDB」の解説記事は全3つを予定しており、最後はPython×Elasticache Redisを予定しています。
内容
事前準備
まずDynamoDBにアクセスするために、AWSのアカウントのアクセスキーとシークレットキーを環境変数に設定します。
$ export AWS_ACCESS_KEY_ID={アクセスキー}
$ export AWS_SECRET_ACCESS_KEY={シークレットキー}
DAXとの接続
DAXを扱うためにamazon-dax-clientライブラリをインストールします。
(仮想環境)% pip install amazon-dax-client
次にDAXとの接続をします。
DynamoDBのテーブル名とDAXのエンドポイントを準備しておきます。
ポートは基本8111番のポートが開放されていますが、暗号化している場合は9111番のポートが開放されています。
クラスターエンドポイントやポート番号はAWS CLIを用いて以下のように確認することができます。
% aws dax describe-clusters
DAXとの接続のためのプログラムは以下です。
from amazondax import AmazonDaxClient
dax_endpoint = "エンドポイント"
dax_port = 8111 #暗号化している場合は9111
# DAXクライアントの初期化
dax = AmazonDaxClient.resource(
endpoint_url=f'daxs://{dax_endpoint}:{dax_port}',
)
table = dax.Table('テーブル名')
余談ですが、DynamoDBと同様、EC2からの作業だとVPCエンドポイントが設定されていない場合はインターネットゲートウェイを用いてDynamoDBに接続されるので注意が必要です。
本筋とは離れますが、DAXを扱うためには適切な権限とセキュリティーグループの設定が必要です。以下を参考にしてください。（公式ドキュメントでポート番号が8111となっている箇所は、暗号化している場合は9111となるので注意してください。）
データの書き込み&データの読み込み
上記の記事とプログラムや注意事項は同様です。プログラムのみを記載しておきます。
###
# データの書き込み
###
from decimal import Decimal
def convert_floats_to_decimals(obj):
"""
float->Decimalに変換する関数
"""
if isinstance(obj, list):
return [convert_floats_to_decimals(x) for x in obj]
elif isinstance(obj, dict):
return {k: convert_floats_to_decimals(v) for k, v in obj.items()}
elif isinstance(obj, float):
return Decimal(str(obj))
else:
return obj
data = {'Id':1, 'value': 3.5} #保存したいデータ
data = convert_floats_to_decimals(data) #float->Decimalに変換
table.put_item(Item=value) #DynamoDBへ書き込み
###
# データの読み込み
###
def convert_decimals_to_floats(obj):
"""
Decimal->floatに変換する関数
"""
if isinstance(obj, list):
return [convert_decimals_to_floats(x) for x in obj]
elif isinstance(obj, dict):
return {k: convert_decimals_to_floats(v) for k, v in obj.items()}
elif isinstance(obj, Decimal):
return float(obj)
else:
return obj
key ={'Id': '1'} #プライマリーキーはId
response = table.get_item(Key=key) #DynamoDBの読み込み
response = convert_decimals_to_floats(response) #Decimal->float変換
おわりに
いかがだったでしょうか。
上記のようにDAXはDynamoDBと同様にPythonで簡単に扱うことができます！
参考になれば幸いです。
次回はElasticache RedisをPythonで使う方法について解説する予定です。