Zenn
📌

【API Gateway・StepFunctions・Lambda】API Gatewayでリダイレクトさせる

に公開
AWS
Amazon API Gateway
Step Functions
tech

はじめに

API Gateway でStepFunctionsと統合しているときに、APIのレスポンスによってリダイレクトをしたり処理を分ける方法は、フロント側で分岐させる方法もありますが、今回はバックエンドのみで実装してみます。
統合バックエンドでリダイレクトヘッダーを生成してAPI Gatewayでは単純にパススルーだけする方法ももちろん出来ますが、何らかの理由でAPI Gateway側で制御したい場合を考えて試してみます。

StepFunctions使用の場合

StepFunctions

下記のように最終的なステートでstatusCodeを出力するようにしておきます。
こちらはLambdaでも可能です。

{
  "statusCode": 302,
  "headers": {
    "Location": "https://example.com/"
  },
  "body": ""
}

API Gateway

マッピングテンプレートでcontext.responseOverrideを使用していきます。

統合レスポンスのマッピングテンプレートでcontext.responseOverrideを使用して以下のように記述します。
StepFunctionsから受け取ったデータの中のstatusCodeを見て、context.responseOverrideで強制的にHTTPステータスコードを301に変更します。

VTL
#set($inputRoot = $input.path('$'))
#set($output = $util.parseJson($inputRoot.output))
#if($output.statusCode == 302)
  #set($context.responseOverride.status = 302)
  #set($context.responseOverride.header.Location = "$output.headers.Location")
  $output.body
#else
{
  "statusCode": 200,
  "body": $output
}
#end

Lambda使用の場合

Lambda関数内で処理することも可能ですが、今回はAPI Gateway側で制御していきます。

Lambda

python
import json

def lambda_handler(event, context):
    return {
        'result': 200
    }

API Gateway

マッピングを行いたいので、プロキシ統合はオフにしておきます。

VTL
#set($inputRoot = $input.path('$'))
#if($inputRoot.result == 200)
#set($context.responseOverride.status = 302)
#set($context.responseOverride.header.Location = "https://example.com/")
#else
#set($context.responseOverride.status = 302)
#set($context.responseOverride.header.Location = "https://example.com/error")
#end

Discussion