Intégrelo AWS Lambda en una máquina de estados de Step Functions con Amazon SQS y Amazon SNS

Este proyecto de ejemplo muestra cómo integrar AWS Lambda funciones en máquinas de estados de Step Functions.

En este proyecto, Step Functions utiliza funciones de Lambda para consultar el precio de una acción y determinar una recomendación de compra o venta. A continuación, el usuario recibe esta recomendación y puede decidir si compra o vende las acciones. El resultado de la operación se devuelve utilizando un SNS tema.

Para obtener más información acerca de las integraciones de servicios de Step Functions, consulte los temas siguientes.

Paso 1: Crear la máquina de estado y aprovisionar recursos

Paso 2: Ejecutar la máquina de estado

Una vez aprovisionados e implementados todos los recursos, aparece el cuadro de diálogo Iniciar ejecución.

Acerca de la máquina de estado y su ejecución

La máquina de estados de este proyecto de ejemplo se integra AWS Lambda pasando los parámetros directamente a esos recursos, utiliza una SQS cola de Amazon para gestionar la solicitud de aprobación humana y utiliza un SNS tema de Amazon para devolver los resultados de la consulta.

Una ejecución de Step Functions recibe un JSON texto como entrada y pasa esa entrada al primer estado del flujo de trabajo. Los estados individuales reciben JSON datos como entrada y, por lo general, JSON los pasan como salida al siguiente estado. En este proyecto de muestra, la salida de cada paso se transfiere como entrada al siguiente paso del flujo de trabajo. Por ejemplo, el paso Generar recomendación de compra/venta recibe como la salida del paso Consultar el precio de una acción como entrada. Además, la salida del paso Generar recomendación de compra/venta se transfiere como entrada al siguiente paso, Solicitar aprobación humana, que imita un paso de aprobación humana.

Para implementar un paso de aprobación humana, normalmente se pausa la ejecución del flujo de trabajo hasta que se devuelva un token de tarea. En este proyecto de ejemplo, se pasa un mensaje a una SQS cola de Amazon, que se utiliza como activador de la función Lambda definida para gestionar la funcionalidad de devolución de llamadas. El mensaje contiene un token de tarea y la salida devuelta en el paso anterior. La función de Lambda se invoca con la carga del mensaje. La ejecución del flujo de trabajo se detiene hasta que recibe el token de la tarea con una llamada. SendTaskSuccessAPI Para obtener más información acerca de los tokens de tarea, consulte Cómo esperar una devolución de llamada con el token de tarea.

El siguiente código de la StepFunctionsSample-HelloLambda-ApproveSqsLambda función muestra cómo se define para aprobar automáticamente cualquier tarea enviada por la SQS cola de Amazon a través de la máquina de estados Step Functions.

exports.lambdaHandler = (event, context, callback) => {
  const stepfunctions = new aws.StepFunctions();

  // For every record in sqs queue
  for (const record of event.Records) {
      const messageBody = JSON.parse(record.body);
      const taskToken = messageBody.TaskToken;

      const params = {
          output: "\"approved\"",
          taskToken: taskToken
      };

      console.log(`Calling Step Functions to complete callback task with params ${JSON.stringify(params)}`);

      // Approve
      stepfunctions.sendTaskSuccess(params, (err, data) => {
          if (err) {
              console.error(err.message);
              callback(err.message);
              return;
          }
          console.log(data);
          callback(null);
      });
  }
};

Eche un vistazo a este ejemplo de máquina de estados para ver cómo Step Functions controla Lambda y Amazon. SQS

Para obtener más información sobre cómo AWS Step Functions puede controlar otros AWS servicios, consulteIntegración de otros servicios con Step Functions.

{
    "StartAt": "Check Stock Price",
    "States": {
        "Check Stock Price": {
            "Type": "Task",
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLam-CheckStockPriceLambda-444455556666",
            "Next": "Generate Buy/Sell recommendation"
        },
        "Generate Buy/Sell recommendation": {
            "Type": "Task",
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-Hello-GenerateBuySellRecommend-123456789012",
            "ResultPath": "$.recommended_type",
            "Next": "Request Human Approval"
        },
        "Request Human Approval": {
            "Type": "Task",
            "Resource": "arn:aws:states:::sqs:sendMessage.waitForTaskToken",
            "Parameters": {
                "QueueUrl": "https://sqs.us-west-1.amazonaws.com/111122223333/StepFunctionsSample-HelloLambda4444-5555-6666-RequestHumanApprovalSqs-777788889999",
                "MessageBody": {
                    "Input.$": "$",
                    "TaskToken.$": "$$.Task.Token"
                }
            },
            "ResultPath": null,
            "Next": "Buy or Sell?"
        },
        "Buy or Sell?": {
            "Type": "Choice",
            "Choices": [
                {
                    "Variable": "$.recommended_type",
                    "StringEquals": "buy",
                    "Next": "Buy Stock"
                },
                {
                    "Variable": "$.recommended_type",
                    "StringEquals": "sell",
                    "Next": "Sell Stock"
                }
            ]
        },
        "Buy Stock": {
            "Type": "Task",
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLambda-BuyStockLambda-000000000000",
            "Next": "Report Result"
        },
        "Sell Stock": {
            "Type": "Task",
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLambda-SellStockLambda-111111111111",
            "Next": "Report Result"
        },
        "Report Result": {
            "Type": "Task",
            "Resource": "arn:aws:states:::sns:publish",
            "Parameters": {
                "TopicArn": "arn:aws:sns:us-west-1:111122223333:StepFunctionsSample-HelloLambda1111-2222-3333-ReportResultSnsTopic-222222222222",
                "Message": {
                    "Input.$": "$"
                }
            },
            "End": true
        }
    }
}

Para obtener información sobre cómo configurar el IAM uso de Step Functions con otros AWS servicios, consulteCómo Step Functions genera IAM políticas para servicios integrados.

IAMEjemplos

Estas políticas de ejemplo AWS Identity and Access Management (IAM) generadas por el proyecto de muestra incluyen los privilegios mínimos necesarios para ejecutar la máquina de estados y los recursos relacionados. Le recomendamos que incluya solo los permisos que sean necesarios en sus IAM políticas.

{
    "Statement": [
        {
            "Action": [
                "lambda:InvokeFunction"
            ],
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLam-CheckStockPriceLambda-444455556666",
            "Effect": "Allow"
        }
    ]
}

{
    "Statement": [
        {
            "Action": [
                "lambda:InvokeFunction"
            ],
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-Hello-GenerateBuySellRecommend-123456789012",
            "Effect": "Allow"
        }
    ]
}

{
    "Statement": [
        {
            "Action": [
                "lambda:InvokeFunction"
            ],
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLambda-BuyStockLambda-777788889999",
            "Effect": "Allow"
        }
    ]
}

{
    "Statement": [
        {
            "Action": [
                "lambda:InvokeFunction"
            ],
            "Resource": "arn:aws:lambda:us-west-1:111122223333:function:StepFunctionsSample-HelloLambda-SellStockLambda-000000000000",
            "Effect": "Allow"
        }
    ]
}

{
    "Statement": [
        {
            "Action": [
                "sqs:SendMessage*"
            ],
            "Resource": "arn:aws:sqs:us-west-1:111122223333:StepFunctionsSample-HelloLambda4444-5555-6666-RequestHumanApprovalSqs-111111111111",
            "Effect": "Allow"
        }
    ]
}

{
    "Statement": [
        {
            "Action": [
                "sns:Publish"
            ],
            "Resource": "arn:aws:sns:us-west-1:111122223333:StepFunctionsSample-HelloLambda1111-2222-3333-ReportResultSnsTopic-222222222222",
            "Effect": "Allow"
        }
    ]
}

Para obtener información sobre cómo configurar el IAM uso de Step Functions con otros AWS servicios, consulteCómo Step Functions genera IAM políticas para servicios integrados.