【AWS】Lamdba関数からParamter Storeの値を取得する【Python】

この記事では
　・Lamdba関数からParamter Storeの値を取得
する手順を紹介します！

手順は
　・Paramter Storeのパラメータの作成
　・IAMポリシーの作成
　・IAMロールの作成
　・Lambda関数の作成
となります！

※今回は、Lambda関数はPythonで作成します。

Paramter Storeのパラメータの作成

Paramter Storeのページを開いて「パラメータの作成」をクリックして、画面に従ってパラメータを作成します。

今回は以下で作成します。

IAMポリシーの作成

Lamdba関数がParamter Storeから値を取得できるように、「Systems Manager」のアクション「GetParameter」を許可する権限を持つIAMポリシーを作成します。

今回は「get-key-parameterstore-string-text」という名前で作成します。
当該アクションの許可を持つJSONは以下となります。

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": "ssm:GetParameter",
            "Resource": "*"
        }
    ]
}

IAMロールの作成

以下をアタッチしたIAMロールを作成します。

今回は「get-key-parameterstore-Role」という名前で作成します。

Lambda関数の作成

●標準ライブラリ「urllib」を使用する方法

import urllib.request
import os
import json

aws_session_token = os.environ.get("AWS_SESSION_TOKEN")
key_name = "sample-password"

def lambda_handler(event, context):
    req = urllib.request.Request("http://localhost:2773/systemsmanager/parameters/get/?name={}".format(key_name))
    req.add_header("X-Aws-Parameters-Secrets-Token", aws_session_token)
    config = urllib.request.urlopen(req).read()
    
    print("パスワード = " + json.loads(config)["Parameter"]["Value"])

{
	"Parameter": 
		{
		 	"ARN": "arn:aws:ssm:ap-northeast-1:031752030502:parameter/sample-password", 
		 	"DataType": "text", 
		 	"LastModifiedDate": "2024-02-24T00:10:25.539Z", 
		 	"Name": "sample-password", 
		 	"Selector": None, 
		 	"SourceResult": None, 
		 	"Type": "String", 
		 	"Value": "password12345", 
		 	"Version": 1
		 },
	 "ResultMetadata": 
	 	{
	 	}
}

●サードパーティのライブラリ「requests」を使用する方法

import requests
import os
import json

headers = {"X-Aws-Parameters-Secrets-Token": os.environ.get("AWS_SESSION_TOKEN")}
key_name = "sample-password"

def lambda_handler(event, context):
    endpoint = "http://localhost:2773/systemsmanager/parameters/get/?name={}".format(key_name)
    r = requests.get(endpoint, headers=headers)
    
    print("パスワード = " + r.json()["Parameter"]["Value"])

Lambda関数でサードパーティのライブラリを使用するには、当該ライブラリをレイヤーに追加する必要があります。
レイヤーの詳細は以下の記事をご確認ください。

【AWS】Lamdba関数のレイヤーにPythonライブラリを追加する

【AWS】Lamdba関数のレイヤーにPythonライブラリを追加する手順です！

aws-guide.work

2024.02.25

実行結果

Lamdba関数からParamter Storeの値を取得できました。

CloudWatch Logsでも、問題無く実行できたことを確認できます。

参考①

上記に記載したAWSレイヤー「AWS-Parameters-and-Secrets-Lambda-Extension」を使用してParamter Storeの値を取得すると、値が一定期間キャッシュされます。
キャッシュされることでコスト削減が期待できます。

参考②

Lamdba関数からParamter Storeの値を取得する手順の詳細は、公式サイトをご確認ください。

AWS Lambda 関数での Parameter Store パラメーターの使用 - AWS Systems Manager

AWS パラメータとシークレットの Lambda 拡張機能を Parameter Store で使用する方法について説明します。

docs.aws.amazon.com