AWS Lambda JavaでLambdaFunctionで発生した例外の処理について

はじめに

AWS Lambdaで、呼び出し元/呼び出し先ともにJavaを使った際の例外処理についてまとめてみました。

テスト用のクラス

Lambda Functionクラス

AWS Lambdaに登録するためのクラスです。

package lambdatest;

import java.time.LocalDateTime;

import com.amazonaws.services.lambda.runtime.Context;

public class Lambda {

    public String ok(Object o, Context context) {
        context.getLogger().log("OK " + LocalDateTime.now());
        return "OK";
    }

    public String timeout(Object o, Context context) {
        try{
            Thread.sleep(TimeUnit.SECONDS.toMillis(100));
        }catch(Exception e){

        }
        context.getLogger().log("Timeout " + LocalDateTime.now());
        return "Timeout";
    }

    public String runtimeException(Object o, Context context) {
        context.getLogger().log("RuntimeException " + LocalDateTime.now());
        throw new RuntimeException("RuntimeException");
    }

    public String exception(Object o, Context context) throws Exception {
        context.getLogger().log("exception " + LocalDateTime.now());
        throw new Exception("Exception");
    }
}

Lambdaの呼び出しコード

登録したLambdaを呼び出すクライアントクラスです。

package lambdatest;

import com.amazonaws.regions.Region;
import com.amazonaws.regions.Regions;
import com.amazonaws.services.lambda.AWSLambdaClient;
import com.amazonaws.services.lambda.model.InvokeRequest;
import com.amazonaws.services.lambda.model.InvokeResult;

public class InvokeLambda {

    public static void main(String[] args) {
        AWSLambdaClient lambda = new AWSLambdaClient();
        lambda.setRegion(Region.getRegion(Regions.AP_NORTHEAST_1));
        InvokeRequest req = new InvokeRequest();
        req.setFunctionName("LambdaTest");
        req.setPayload("{\"hoo\":\"bar\"}");
        try {
            InvokeResult invoke = lambda.invoke(req);
            String res = new String(invoke.getPayload().array());
            System.out.println(res);
            System.out.println(invoke.getFunctionError());
            System.out.println(invoke.getStatusCode());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

呼び出し時の挙動確認

上記のテストクラスを使って、呼び出すLambdaのメソッドごとの挙動を確認した結果が以下となります。

OKのパターン

正常パターン。lambdatest.Lambda::ok を呼び出す。処理は成功して、"OK"という文字列が返って来ます。

Payload:"OK"
FunctionError:null
StatusCode:200

Timeoutのパターン

lambdatest.Lambda::timeoutを呼び出すと処理がタイムアウトする。
呼び出し元では例外は発生せず、タイムアウト時に以下のPayload,FunctionError,StatusCodeが返ってきます。

Payload:{"errorMessage":"2016-02-19T10:10:00.159Z e59b6744-d6f0-11e5-9eae-c56145704801 Task timed out after 15.00 seconds"}
FunctionError:Unhandled
StatusCode:200

RuntimeExceptionのパターン

lambdatest.Lambda::runtimeExceptionを呼び出すとRuntimeExceptionがすろーされる。
呼び出し元では例外は発生せず、タイムアウト時に以下のPayload,FunctionError,StatusCodeが返ってきます。

Payload:{"errorMessage":"RuntimeException","errorType":"java.lang.RuntimeException","stackTrace":["lambdatest.Lambda.runtimeException(Lambda.java:28)","sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)","sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)","sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)","java.lang.reflect.Method.invoke(Method.java:497)"]}
FunctionError:Unhandled
StatusCode:200

Exceptionのパターン

Payloadに入っている例外クラス名がExceptionとなっていだけで、あとはRuntimeExceptionと同じです。

LambdaのHandler設定が間違っていた時

AWS LambdaのHandler設定が誤っており、呼び出し先クラスやメソッドがない場合。
この場合でも呼び出し元では例外は発生せず、タイムアウト時に以下のPayload,FunctionError,StatusCodeが返ってきます。

Payload:{"errorMessage":"No public method named ERRORHANDLER with appropriate method signature found on class class lambdatest.Lambda"}
FunctionError:Unhandled
StatusCode:200

例外処理の方法

結論として、AWSクレデンシャル/リージョン/FunctionNameが正しければ、Lambda側で例外が発生してもクライアント側では例外がスローされないため、InvokeResultの中身を見て、エラーかどうかを判断する必要があります。
InvokeResult#getFunctionError()の戻り値がnullでない場合はエラーが発生しているとみなしてもよさそうなので、この場合はPayloadからエラー内容を取り出し、適宜メッセージを取り出してログを吐いたりリトライする形になります。

なおAWSクレデンシャル間違いの場合はcom.amazonaws.AmazonServiceException、リージョンやFunctionNameが違う場合はcom.amazonaws.services.lambda.model.ResourceNotFoundExceptionが発生するため、これは別途catchして例外処理を行う必要があります。