Importations Python relatives dans une fonction lambda Dockerisée

Les importations relatives de Python peuvent être délicates pour les fonctions lambda. J'ai écrit un blog à ce sujet il y a 3 ans. Mais récemment, j'ai rencontré le même problème avec les fonctions lambda Dockerisées. Alors, j'ai pensé qu'il était temps de créer un nouveau blog !

Vous pouvez suivre les étapes ou regarder le résultat directement sur GitHub.

Configuration du projet

Assurez-vous d'avoir installé le cli AWS CDK.

brew install aws-cdk

Initialiser le projet :

cdk init app --language=typescript

Configuration Lambda

Nous devrons d’abord créer la structure des fichiers et des dossiers :

mkdir -p lib/functions/hello-world/hello_world
touch lib/functions/hello-world/hello_world/__init__.py
touch lib/functions/hello-world/requirements.txt
touch lib/functions/hello-world/Dockerfile

Vous devrez maintenant remplir le Dockerfile, comme ceci :

FROM public.ecr.aws/lambda/python:3.12
COPY requirements.txt .
COPY hello_world ${LAMBDA_TASK_ROOT}/hello_world
RUN pip install --no-cache-dir -r requirements.txt
CMD ["hello_world.handler"]

Nous utilisons une image de base Python basée sur Python 3.12. Ensuite, nous copierons dans le fichier exigences.txt et le code source. Nous installerons toutes les dépendances répertoriées dans le fichier exigences.txt et nous assurerons que la méthode du gestionnaire est définie comme CMD.

Ensuite, nous devrons remplir nos fichiers Python avec du code. Dans le fichier __init__.py, vous pouvez placer le contenu suivant :

from typing import Dict, Any


def handler(event: Dict[str, Any], context: Any) -> Dict[str, str]:
    name = event.get("name", "World")

    return {
        "Name": name,
        "Message": f"Hello {name}!",
    }


__all__ = [
    "handler"
]

REMARQUE : Le code utilisé ici pourrait utiliser des importations relatives. Ceci est possible car il se trouve dans un package séparé. Cet exemple montre uniquement le code dans le fichier __init__.py. Cependant, vous pouvez utiliser plusieurs fichiers ici pour améliorer la maintenabilité de votre projet.

Pour cet exemple, je n'ai besoin d'aucune dépendance, nous pouvons donc garder le fichier exigences.txt vide. Je l'ai inclus dans cet exemple pour illustrer comment vous pouvez également inclure des dépendances.

Créer la fonction Lambda à l'aide d'IaC

Nos dossiers et fichiers sont en place, il est donc temps d'ajouter la fonction Lambda à la construction CDK. Vous pouvez simplement l'ajouter comme ceci :

    new lambda.Function(this, 'Function', {
      functionName: "hello-world",
      code: lambda.Code.fromAssetImage("lib/functions/hello-world", {
        platform: ecr_assets.Platform.LINUX_ARM64,
      }),
      runtime: lambda.Runtime.FROM_IMAGE,
      handler: lambda.Handler.FROM_IMAGE,
      architecture: lambda.Architecture.ARM_64,
      timeout: cdk.Duration.seconds(15),
      memorySize: 128,
    });

Pour que cela fonctionne, vous avez également besoin des importations suivantes :

import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as ecr_assets from 'aws-cdk-lib/aws-ecr-assets';

Notez que nous nous assurons que le répertoire de code pointe vers le répertoire qui contient le Dockerfile et que nous sélectionnons la plateforme ARM à la fois pour le code et pour la fonction elle-même.

Tester la fonction lambda localement

Un retour rapide est important, il peut donc y avoir des cas où vous devrez exécuter le conteneur localement. Pour cela, vous devez d'abord construire le conteneur :

docker build --platform linux/arm64 \
  -t hello-world:latest \
  -f ./lib/functions/hello-world/Dockerfile \
  ./lib/functions/hello-world

Notez que cette commande peut être exécutée depuis la racine du projet. Ensuite, nous devons nous assurer qu'il fonctionne avant de pouvoir l'invoquer :

docker run --platform linux/arm64 -p 9000:8080 hello-world:latest

Ensuite, vous pouvez appeler la fonction comme suit :

curl http://localhost:9000/2015-03-31/functions/function/invocations -d '{"name": "Joris"}'

Conclusion

Les importations relatives peuvent être délicates ! Vous devez placer votre code dans un package. Cela vous permet d'effectuer des importations relatives au sein de votre propre package. Cela permettra d'obtenir un code plus propre, car vous pourrez diviser les responsabilités en plusieurs fichiers, ce qui facilitera la gestion et la maintenance.

Photo de Kaique Rocha

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!