Integrasikan AWS Lambda dalam mesin status Step Functions dengan Amazon SQS dan Amazon SNS

Proyek sampel ini menunjukkan cara mengintegrasikan AWS Lambda fungsi dalam mesin status Step Functions.

Dalam proyek ini, Step Functions menggunakan fungsi Lambda untuk memeriksa harga saham dan menentukan rekomendasi perdagangan beli atau jual. Pengguna kemudian memberikan rekomendasi ini dan dapat memilih apakah akan membeli atau menjual saham. Hasil perdagangan dikembalikan menggunakan SNS topik.

Untuk informasi selengkapnya tentang integrasi layanan Step Functions, lihat hal berikut:

Langkah 1: Buat mesin negara bagian dan sumber daya penyediaan

Langkah 2: Jalankan mesin negara

Setelah semua sumber daya disediakan dan digunakan, kotak dialog Mulai eksekusi ditampilkan.

Tentang mesin negara dan pelaksanaannya

Mesin status dalam proyek sampel ini terintegrasi AWS Lambda dengan meneruskan parameter langsung ke sumber daya tersebut, menggunakan SQS antrian Amazon untuk mengelola permintaan persetujuan manusia, dan menggunakan SNS topik Amazon untuk mengembalikan hasil kueri.

Eksekusi Step Functions menerima JSON teks sebagai input dan meneruskan input tersebut ke status pertama dalam alur kerja. Setiap negara menerima JSON data sebagai input dan biasanya meneruskan JSON data sebagai output ke status berikutnya. Dalam proyek sampel ini, output dari setiap langkah diteruskan sebagai input ke langkah berikutnya dalam alur kerja. Misalnya, langkah rekomendasi Hasilkan Beli/Jual menerima output dari langkah Periksa Harga Saham sebagai masukan. Selanjutnya, output dari langkah rekomendasi Generate Beli/Sell diteruskan sebagai masukan ke langkah berikutnya, Request Human Approval, yang meniru langkah persetujuan manusia.

Untuk menerapkan langkah persetujuan manusia, Anda biasanya menjeda eksekusi alur kerja hingga token tugas dikembalikan. Dalam proyek contoh ini, pesan diteruskan ke SQS antrean Amazon, yang digunakan sebagai pemicu fungsi Lambda yang ditentukan untuk menangani fungsionalitas panggilan balik. Pesan berisi token tugas dan output yang dikembalikan oleh langkah sebelumnya. Fungsi Lambda dipanggil dengan muatan pesan. Eksekusi alur kerja dijeda hingga menerima token tugas kembali dengan panggilan. SendTaskSuccessAPI Untuk informasi selengkapnya tentang token tugas, lihatTunggu Panggilan Balik dengan Token Tugas.

Kode berikut untuk StepFunctionsSample-HelloLambda-ApproveSqsLambda fungsi ini menunjukkan cara didefinisikan untuk secara otomatis menyetujui tugas apa pun yang dikirimkan oleh SQS antrian Amazon melalui mesin status 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);
      });
  }
};

Jelajahi contoh mesin status ini untuk melihat bagaimana Step Functions mengontrol Lambda dan Amazon. SQS

Untuk informasi selengkapnya tentang AWS Step Functions cara mengontrol AWS layanan lain, lihatMengintegrasikan layanan lain dengan 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
        }
    }
}

Untuk informasi tentang cara mengonfigurasi IAM saat menggunakan Step Functions dengan AWS layanan lain, lihatBagaimana Step Functions menghasilkan IAM kebijakan untuk layanan terintegrasi.

IAMContoh

Kebijakan contoh AWS Identity and Access Management (IAM) yang dihasilkan oleh proyek sampel ini mencakup hak istimewa paling sedikit yang diperlukan untuk mengeksekusi mesin negara dan sumber daya terkait. Kami menyarankan Anda hanya menyertakan izin yang diperlukan dalam IAM kebijakan Anda.

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

Untuk informasi tentang cara mengonfigurasi IAM saat menggunakan Step Functions dengan AWS layanan lain, lihatBagaimana Step Functions menghasilkan IAM kebijakan untuk layanan terintegrasi.