はじめに
LambdaでHTTP接続する際、URLのパスで処理を変えたいと思い試した内容を記事にします。
参考
概要
前提として、Pythonを用いています。
以下の2つのやり方を見つけました。
- AWSGIを使う方法
- LambdaでFlaskを使える
- API Gateway経由のeventを処理可能
- Function URLのeventはできない
- AWS Lambda Powertoolsを使う方法
- ユーティリティーのライブラリ
- 今回のイベントハンドラ以外のユーティリティも提供
- Python以外の言語でも提供
- API Gateway経由のeventも、Function URLのeventも可能
- 未検証だが、ALB経由のも可能な模様
- ユーティリティーのライブラリ
やってみた
いずれの方法もAWS SAMを用いています。私はCloud9上で行いました。
AWSGIを使う方法
使い方は以下のページにありますので、そちらをご参考ください。
Flaskを使われていた方には、いいかもしれません。
AWS Lambda Powertoolsを使う方法
以下のページを参考に、コードを書いてみました。
まず使用するフォルダ、ファイルを作ります。
mkdir -p sample-lambda_powertools/functions && cd sample-lambda_powertools
touch functions/app.py
touch functions/requirements.txt
touch template.yaml
それぞれのファイルに記載していきます。
まずはインポートするモジュールを宣言します。
functions/requirements.txt
aws-lambda-powertools
Function URLの場合
Function URLの場合は、LambdaFunctionUrlResolverを用います。
functions/app.py
from aws_lambda_powertools.event_handler import LambdaFunctionUrlResolver
app = LambdaFunctionUrlResolver()
@app.get("/")
def get_home():
return "get_home"
@app.post("/")
def post_home():
return "post_home"
@app.put("/")
def put_home():
return "put_home"
@app.delete("/")
def delete_home():
return "delete_home"
@app.get("/hello")
@app.post("/hello")
def hello():
return "hello"
@app.get("/hello/<inputname>")
@app.post("/hello/<inputname>")
def hello_withname(inputname : str):
return f"hello {inputname}!"
def lambda_handler(event, context):
print('sample function')
return app.resolve(event, context)
Function URLを持つLambdaを作るSAMのテンプレートは以下になります。
template.yaml
AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: SAM Template for Lambda Function
Resources:
TargetFunction:
Type: AWS::Serverless::Function
Properties:
CodeUri: functions/
Handler: app.lambda_handler
Runtime: python3.9
TargetFunctionUrl:
Type: AWS::Lambda::Url
Properties:
AuthType: NONE
TargetFunctionArn: !GetAtt TargetFunction.Arn
TargetFunctionUrlPermission:
Type: AWS::Lambda::Permission
Properties:
Action: lambda:InvokeFunctionUrl
FunctionName: !GetAtt TargetFunction.Arn
FunctionUrlAuthType: NONE
Principal: "*"
FunctionLogGroup:
Type: AWS::Logs::LogGroup
Properties:
LogGroupName: !Sub /aws/lambda/${TargetFunction}
RetentionInDays: 3653
Outputs:
FunctionUrl:
Value: !GetAtt TargetFunctionUrl.FunctionUrl
Export:
Name: TargetFunctionUrl
あとはBuildしてDeployです。
sam build && sam validate
sam deploy --guided
CURLコマンドで確かめられられます。
curl -X POST [関数URL]
curl -X POST [関数URL]/hello
curl -X POST [関数URL]/hello/taro
API Gatewayの場合
API Gatewayを使う場合は、APIGatewayRestResolverを用います。
他の部分はFunction URLのコードと変わらないです。
functions/app.py
from aws_lambda_powertools.event_handler import APIGatewayRestResolver
app = APIGatewayRestResolver()
@app.get("/")
def get_home():
return "get_home"
@app.post("/")
def post_home():
return "post_home"
@app.put("/")
def put_home():
return "put_home"
@app.delete("/")
def delete_home():
return "delete_home"
@app.get("/hello")
@app.post("/hello")
def hello():
return "hello"
@app.get("/hello/<inputname>")
@app.post("/hello/<inputname>")
def hello_withname(inputname : str):
return f"hello {inputname}!"
def lambda_handler(event, context):
print('sample function')
return app.resolve(event, context)
SAMテンプレートは以下になります。
template.yaml
AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: SAM Template for Lambda Function
Resources:
TargetFunction:
Type: AWS::Serverless::Function
Properties:
CodeUri: functions/
Handler: app.lambda_handler
Runtime: python3.9
Events:
ApiRoot:
Type: Api
Properties:
Path: '/'
Method: ANY
ApiProxy:
Type: Api
Properties:
Path: '/{proxy+}'
Method: ANY
FunctionLogGroup:
Type: AWS::Logs::LogGroup
Properties:
LogGroupName: !Sub /aws/lambda/${TargetFunction}
RetentionInDays: 3653
Outputs:
TargetdApi:
Value: !Sub "https://${ServerlessRestApi}.execute-api.${AWS::Region}.amazonaws.com/Prod/"
あとはBuildとDeployで作成し、CURLコマンドで確かめられます。
おわりに
Flaskを勉強していた流れで知った知識であり、色々使いどころがありそうです。
ただ、1つの関数をリッチにしがちになりそうなので、そこは注意が必要かと感じます。