Integre-se AWS Lambda em uma máquina de estado Step Functions com Amazon SQS e Amazon SNS

Este projeto de amostra demonstra como integrar AWS Lambda funções em máquinas de estado Step Functions.

Neste projeto, o Step Functions usa funções do Lambda para verificar o preço de uma ação e determinar uma recomendação de negociação de compra ou venda. O usuário então recebe essa recomendação e pode escolher entre comprar ou vender as ações. O resultado da negociação é retornado usando um SNS tópico.

Para obter mais informações sobre integrações de serviços do Step Functions, consulte o seguinte:

Etapa 1: Criar a máquina de estado e provisionar os recursos

Etapa 2: Executar a máquina de estado

Depois que todos os recursos forem provisionados e implantados, a caixa de diálogo Iniciar execução será exibida.

Sobre a máquina de estado e sua execução

A máquina de estado neste projeto de amostra se integra AWS Lambda passando parâmetros diretamente para esses recursos, usa uma SQS fila da Amazon para gerenciar a solicitação de aprovação humana e usa um SNS tópico da Amazon para retornar os resultados da consulta.

Uma execução de Step Functions recebe um JSON texto como entrada e passa essa entrada para o primeiro estado no fluxo de trabalho. Estados individuais recebem JSON dados como entrada e geralmente passam JSON dados como saída para o próximo estado. Neste exemplo de projeto, a saída de cada etapa é passada como entrada para a próxima etapa no fluxo de trabalho. Por exemplo, a etapa Gerar recomendação de compra/venda recebe a saída da etapa Verificar preço da ação como entrada. Além disso, a saída da etapa Gerar recomendação de compra/venda é passada como entrada para a próxima etapa, Solicitar aprovação humana, que imita uma etapa de aprovação humana.

Para implementar uma etapa de aprovação humana, você normalmente pausa a execução do fluxo de trabalho até que um token de tarefa seja retornado. Neste projeto de amostra, uma mensagem é passada para uma SQS fila da Amazon, que é usada como um gatilho para a função Lambda definida para lidar com a funcionalidade de retorno de chamada. A mensagem contém um token de tarefa e a saída retornada pela etapa anterior. A função do Lambda é invocada com a carga da mensagem. A execução do fluxo de trabalho é pausada até receber o token da tarefa de volta com uma SendTaskSuccessAPIchamada. Para mais informações sobre tokens de tarefa, consulte Aguardar um retorno de chamada com um token de tarefa.

O código a seguir para a StepFunctionsSample-HelloLambda-ApproveSqsLambda função mostra como ela é definida para aprovar automaticamente todas as tarefas enviadas pela SQS fila da Amazon por meio da máquina de estado 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);
      });
  }
};

Navegue por esse exemplo de máquina de estado para ver como o Step Functions controla a Lambda e a Amazon. SQS

Para obter mais informações sobre como AWS Step Functions controlar outros AWS serviços, consulteIntegrando outros serviços com 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 obter informações sobre como configurar IAM ao usar Step Functions com outros AWS serviços, consulteComo o Step Functions gera IAM políticas para serviços integrados.

IAMExemplos

Essas políticas example AWS Identity and Access Management (IAM) geradas pelo projeto de amostra incluem o menor privilégio necessário para executar a máquina de estado e os recursos relacionados. Recomendamos que você inclua somente as permissões necessárias em suas 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 obter informações sobre como configurar IAM ao usar Step Functions com outros AWS serviços, consulteComo o Step Functions gera IAM políticas para serviços integrados.