Integrazione AWS Lambda in una macchina a stati Step Functions con Amazon SQS e Amazon SNS

Questo progetto di esempio dimostra come integrare AWS Lambda le funzioni nelle macchine a stati Step Functions.

In questo progetto, Step Functions utilizza le funzioni Lambda per controllare il prezzo di un'azione e determinare una raccomandazione commerciale di acquisto o vendita. L'utente riceve quindi questa raccomandazione e può scegliere se acquistare o vendere il titolo. Il risultato dello scambio viene restituito utilizzando un SNS argomento.

Per ulteriori informazioni sulle integrazioni dei servizi Step Functions, consulta quanto segue:

Fase 1: Creare la macchina a stati e fornire risorse

Fase 2: Eseguire la macchina a stati

Dopo aver fornito e distribuito tutte le risorse, viene visualizzata la finestra di dialogo Avvia esecuzione.

Informazioni sulla macchina a stati e sulla sua esecuzione

La macchina a stati di questo progetto di esempio si integra AWS Lambda passando i parametri direttamente a tali risorse, utilizza una SQS coda Amazon per gestire la richiesta di approvazione umana e utilizza un SNS argomento Amazon per restituire i risultati della query.

Un'esecuzione di Step Functions riceve un JSON testo come input e lo passa al primo stato del workflow. I singoli stati ricevono JSON i dati come input e in genere JSON li passano come output allo stato successivo. In questo progetto di esempio, l'output di ogni fase viene passato come input alla fase successiva del flusso di lavoro. Ad esempio, la fase di raccomandazione Generate Buy/Sell riceve come input l'output della fase Check Stock Price. Inoltre, l'output della fase di raccomandazione Generate Buy/Sell viene passato come input alla fase successiva, Request Human Approval, che simula una fase di approvazione umana.

Per implementare una fase di approvazione umana, in genere si sospende l'esecuzione del flusso di lavoro fino alla restituzione di un token di attività. In questo progetto di esempio, un messaggio viene passato a una SQS coda Amazon, che viene utilizzata come trigger per la funzione Lambda definita per gestire la funzionalità di callback. Il messaggio contiene un token di attività e l'output restituito dal passaggio precedente. La funzione Lambda viene richiamata con il payload del messaggio. L'esecuzione del flusso di lavoro viene sospesa fino a quando non riceve il token dell'attività con una chiamata. SendTaskSuccessAPI Per ulteriori informazioni sui task token, vedere. Attendere un callback con il token dell’attività

Il codice seguente per la StepFunctionsSample-HelloLambda-ApproveSqsLambda funzione mostra come viene definita l'approvazione automatica di qualsiasi attività inviata dalla SQS coda Amazon tramite la macchina a stati 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);
      });
  }
};

Sfoglia questo esempio di macchina a stati per vedere come Step Functions controlla Lambda e Amazon. SQS

Per ulteriori informazioni su come AWS Step Functions controllare altri AWS servizi, consultaIntegrazione di altri servizi 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
        }
    }
}

Per informazioni su come configurare l'IAMutilizzo di Step Functions con altri AWS servizi, vedereIn che modo Step Functions genera IAM politiche per servizi integrati.

IAMEsempi

Queste politiche di esempio AWS Identity and Access Management (IAM) generate dal progetto di esempio includono il minimo privilegio necessario per eseguire la macchina a stati e le risorse correlate. Ti consigliamo di includere solo le autorizzazioni necessarie nelle tue IAM politiche.

{
    "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"
        }
    ]
}

Per informazioni su come configurare l'IAMutilizzo di Step Functions con altri AWS servizi, vedereIn che modo Step Functions genera IAM politiche per servizi integrati.