Lambda関数でDockerコンテナ化されたPythonのFastAPIサービスを動かす

• 参考にさせていただいたブログ : https://reffect.co.jp/aws/lambda-docker-fastapi

ディレクトリを作成する

docker -v
mkdir lambda_fastapi
cd lambda_fastapi

fastAPIサービスを実装する

touch app.py

• ~/aws/lambda_fastapi/app.py

from fastapi import FastAPI
from pydantic import BaseModel
from mangum import Mangum

class Item(BaseModel):
    name: str
    price: float

app = FastAPI()

@app.get('/')
async def root():
  return {"message": "Hello from lambda"}

@app.get("/uppercase")
async def uppercase(text: str):
    return {"message": text.upper()}

@app.post("/items")
async def create_item(item: Item):
    return item

handler = Mangum(app)

ライブラリ記述ファイルを作る

touch requirements.txt

• ~/aws/lambda_fastapi/requirements.txt

fastapi
mangum

Dockerfileを作る

touch Dockerfile

• ~/aws/lambda_fastapi/Dockerfile

FROM public.ecr.aws/lambda/python:3.12

COPY app.py ${LAMBDA_TASK_ROOT}
COPY requirements.txt ${LAMBDA_TASK_ROOT}

RUN pip install -r ./requirements.txt

CMD ["app.handler"]

Dockerイメージをビルドする

docker build -t lambda_fastapi .
docker images

動作確認する

docker run -p 9000:8080 lambda_fastapi
docker ps

サービスを確認する

• GETリクエスト

curl -XPOST "http://localhost:9000/2015-03-31/functions/function/invocations" -d '{"resource": "/", "path": "/", "httpMethod": "GET", "requestContext": {}}'

• POSTリクエスト

curl -XPOST "http://localhost:9000/2015-03-31/functions/function/invocations" \
-d '{
  "resource": "/items",
  "path": "/items",
  "httpMethod": "POST",
  "requestContext": {},
  "body": "{\"name\": \"Keyboard\",\"price\":10000}"
}'

ECRリポジトリの作成

• AWSのprodプロファイルに切り替え

aws ecr ls --profile prod
export AWS_PROFILE=prod

• 認証トークンを取得し、レジストリに対して Docker クライアントを認証

aws ecr get-login-password --region ap-northeast-1 --profile prod | docker login --username AWS --password-stdin 11111111111111111.dkr.ecr.ap-northeast-1.amazonaws.com

• ECRリポジトリの作成

aws ecr create-repository --repository-name lambda_fastapi_repo --profile prod --image-scanning-configuration scanOnPush=true --image-tag-mutability MUTABLE

• Dockerイメージをタグ付け

docker tag lambda_fastapi:latest 11111111111111111.dkr.ecr.ap-northeast-1.amazonaws.com/lambda_fastapi_repo:latest
docker images

• 実行結果

REPOSITORY                                                              TAG       IMAGE ID       CREATED          SIZE
11111111111111111.dkr.ecr.ap-northeast-1.amazonaws.com/lambda_fastapi_repo   latest    ee65f52d4fed   44 minutes ago   543MB
lambda_fastapi                                                          latest    ee65f52d4fed   44 minutes ago   543MB

• イメージのアップロード

docker push 11111111111111111.dkr.ecr.ap-northeast-1.amazonaws.com/lambda_fastapi_repo:latest

• 実行結果

The push refers to repository [211125541523.dkr.ecr.ap-northeast-1.amazonaws.com/lambda_fastapi_repo]
3dded92ad458: Pushed 
・・・・省略・・・・
922a9a710c9e: Pushed 
latest: digest: sha256:dkspdksokoskodksokdoskodskodksodksokdosa size: 2203

• リポジトリ情報の確認

aws ecr describe-repositories

• 実行結果

{
  "repositoryArn": "arn:aws:ecr:ap-northeast-1:11111111111111111:repository/lambda_fastapi_repo",
  "registryId": "11111111111111111",
  "repositoryName": "lambda_fastapi_repo",
  "repositoryUri": "11111111111111111.dkr.ecr.ap-northeast-1.amazonaws.com/lambda_fastapi_repo",
  "createdAt": "2024-03-30T10:42:52.223000+09:00",
  "imageTagMutability": "MUTABLE",
  "imageScanningConfiguration": {
      "scanOnPush": true
  },
  "encryptionConfiguration": {
      "encryptionType": "AES256"
  }
}

Lambda関数の作成

• URL : https://ap-northeast-1.console.aws.amazon.com/lambda/home?region=ap-northeast-1#/functions
  • [関数の作成]
• 関数の作成
  • コンテナイメージ : 選択
  • 関数名 : lambdaFastAPI
  • コンテナイメージ URI
    • [イメージを参照]
      • Amazon ECR イメージリポジトリ : lambda_fastapi_repo
      • イメージ : latest選択
      • [イメージを選択]
  • アーキテクチャ : arm64
  • [関数の作成]
    

関数URLの設定

• URL : https://ap-northeast-1.console.aws.amazon.com/lambda/home?region=ap-northeast-1#/functions/lambdaFastAPI?newFunction=true&tab=image
  • [設定]
    • 一般設定
      • 関数URL
        • [関数 URL を作成]
  • 関数 URL を設定
    • 認証タイプ : NONE
    • [保存]
  • 関数URL
    • 関数URL : https://sskofeokfkeofkoekfoekofkeo.lambda-url.ap-northeast-1.on.aws

関数URLの確認

• サービスURL : https://sskofeokfkeofkoekfoekofkeo.lambda-url.ap-northeast-1.on.aws
  • リターン値 : {"message":"Hello from lambda"}
• ドキュメントURL : https://sskofeokfkeofkoekfoekofkeo.lambda-url.ap-northeast-1.on.aws/docs