2024年11月22日に AWS Step Functions で新機能「変数 (variables)」と「JSONata」がサポートされた❗️
特に「変数」は待望の機能なんじゃないかな〜と思う．今までは AWS Step Functions でタスク結果を引き回すときにどうしてもバケツリレーが必要で，正直 ResultSelector / ResultPath という構文も難しく（何度書いても忘れてしまう😇）ある程度複雑なワークフローを実装するときはデータシュミレーターがないと無理に思えるほどだった．個人的には AWS Step Functions は動くようになったら最高に便利だけどイメージ通りに実装するまでの過程が結構大変〜というイメージがある💨
Step Functions でタスク結果を引き回すときにバケツリレーするの本当に大変だったし ResultSelector / ResultPath とかも正直難しくてデータフローシミュレーターに頼ってたから「変数」機能は待ってました〜❗️って感じのやつだ ＼( 'ω')／https://t.co/a4cJnZ147u
— カック (@kakakakakku) 2024年11月26日
変数と JSONata を試す
さっそく変数と JSONata を試してみる❗️今回は AWS Lambda 関数から出力された数値同士を「掛け算」するサンプルにする．実は今までの AWS Step Functions 組み込み関数では States.MathAdd を使った「足し算」しかサポートされてなく，過去に困った経験があった．
AWS Step Functions ワークフロー
以下のようなワークフローを作った．AWS Lambda 関数 returnX と returnY はランダムな整数を返して結果を「変数」設定し，最後の Pass では「JSONata」を使って returnX と returnY の値を計算（掛け算）する👌
ステートマシンは必要最低限にしていて，ポイントをザッと以下に挙げる📝
QueryLanguageにJSONataを指定するAssignを使って AWS Lambda 関数returnXとreturnYの結果{% $states.result.body %}を変数に設定する- JSONata 式
{% $x * $y %}で計算（掛け算）する
{ "QueryLanguage": "JSONata", "StartAt": "returnX", "States": { "returnX": { "Type": "Task", "Resource": "arn:aws:lambda:ap-northeast-1:000000000000:function:returnX", "Assign": { "x": "{% $states.result.body %}" }, "Next": "returnY" }, "returnY": { "Type": "Task", "Resource": "arn:aws:lambda:ap-northeast-1:000000000000:function:returnY", "Assign": { "y": "{% $states.result.body %}" }, "Next": "multiplication" }, "multiplication": { "Type": "Pass", "Output": { "multiplication": "{% $x * $y %}" }, "End": true } } }
AWS Lambda 関数 returnX と returnY の実装はシンプル💡
import json import random def lambda_handler(event, context): return { 'statusCode': 200, 'body': random.randint(1, 100) }
ちなみに 0 ≤ n < 1 の範囲でランダムな値を取得するなら JSONata の $random() 関数で表現できてしまう．あくまで今回は AWS Lambda 関数で何かしらの処理を実行するサンプルとして見てもらえればと👌
実行結果
AWS Step Functions ワークフローを実行すると期待通りに「変数」と「JSONata」を使えた❗️
注意点
今回のリリースと同時に今まで何度もお世話になったデータフローシュミレーターがメンテナンスされなくなっていた😇
AWS Step Functions で JSONata がサポートされて高度なデータ操作ができるようになったのは最高だけどその引き換えに今まで何度もお世話になった「データフローシミュレーター」がサポートされなくなっていた😇
— カック (@kakakakakku) 2024年11月26日
>データフローシミュレータページはメンテナンスされなくなりました pic.twitter.com/RSSlDI3Rfh
LocalStack で変数と JSONata を試す
今回さらにスゴイのは LocalStack も同時に変数と JSONata をサポートしている点で，リリースノートにも LocalStack の紹介文が載っていた❗️
We have also partnered with LocalStack and Datadog to ensure that their local emulation and observability experiences are updated to support Variables and JSONata.
そして LocalStack 側のブログにも記事が出ている📝
正確には LocalStack v4.0.1 でサポートされているため，今回は最新の LocalStack v4.0.2 で試す．
AWS Step Functions ワークフロー
以下の AWS SAM テンプレートを準備した．マネジメントコンソールで試した構成とほぼ同じ👌
AWSTemplateFormatVersion: 2010-09-09 Transform: AWS::Serverless-2016-10-31 Resources: FunctionReturnX: Type: AWS::Serverless::Function Properties: FunctionName: returnX CodeUri: ./src Handler: app.lambda_handler Runtime: python3.12 Architectures: - x86_64 FunctionReturnY: Type: AWS::Serverless::Function Properties: FunctionName: returnY CodeUri: ./src Handler: app.lambda_handler Runtime: python3.12 Architectures: - x86_64 StateMachine: Type: AWS::Serverless::StateMachine Properties: Name: sandbox DefinitionUri: ./sandbox.asl.json DefinitionSubstitutions: FunctionXArn: !GetAtt FunctionReturnX.Arn FunctionYArn: !GetAtt FunctionReturnY.Arn
ステートマシン sandbox.asl.json もほぼ同じで，AWS Lambda 関数名の ARN を AWS SAM テンプレートから渡すところが違う👌
{ "QueryLanguage": "JSONata", "StartAt": "returnX", "States": { "returnX": { "Type": "Task", "Resource": "${FunctionXArn}", "Assign": { "x": "{% $states.result.body %}" }, "Next": "returnY" }, "returnY": { "Type": "Task", "Resource": "${FunctionYArn}", "Assign": { "y": "{% $states.result.body %}" }, "Next": "multiplication" }, "multiplication": { "Type": "Pass", "Output": { "multiplication": "{% $x * $y %}" }, "End": true } } }
次に LocalStack にデプロイすると LocalStack Resource Browser で確認できる❗️
$ samlocal build $ samlocal deploy
そして LocalStack 上の AWS Step Functions ワークフローを実行すると，{\"multiplication\":1344} と出力されていて期待通りに計算（掛け算）できている👏
最高だ〜 ＼( 'ω')／
$ awslocal stepfunctions start-execution --state-machine-arn arn:aws:states:ap-northeast-1:000000000000:stateMachine:sandbox { "executionArn": "arn:aws:states:ap-northeast-1:000000000000:execution:sandbox:ac69989b-6569-4881-aea7-23de22d124ba", "startDate": "2024-11-28T00:00:00.000000+09:00" } $ awslocal stepfunctions describe-execution --execution-arn arn:aws:states:ap-northeast-1:000000000000:execution:sandbox:ac69989b-6569-4881-aea7-23de22d124ba { "executionArn": "arn:aws:states:ap-northeast-1:000000000000:execution:sandbox:ac69989b-6569-4881-aea7-23de22d124ba", "stateMachineArn": "arn:aws:states:ap-northeast-1:000000000000:stateMachine:sandbox", "name": "ac69989b-6569-4881-aea7-23de22d124ba", "status": "SUCCEEDED", "startDate": "2024-11-28T00:00:00.000000+09:00", "stopDate": "2024-11-28T00:00:00.000000+09:00", "input": "{}", "inputDetails": { "included": true }, "output": "{\"multiplication\":1344}", "outputDetails": { "included": true } }
LocalStack Resource Browser でも実行結果を確認できた❗️（今のところ変数は表示されていなさそうだった）
