はじめに

AWSのLambdaを用いてDynamoDBにアクセスする場合、簡単な操作はJavaScriptを用いるととても簡単にできます。ただ、複数のテーブルにクエリするような場合、JavaScript特有のCallback地獄に入ってしまい、使いづらさを感じました。またローカル環境での動作確認がしずらく、今回は新しくサポートされた、LambdaのJavaを使用してみようと思います。

準備

LambdaのJavaを記述する場合、Eclipse + AWS Toolkitが非常に楽です。ですので、AWS Toolkitをインストールすることをオススメします。

簡単なサンプルの作成

AWS Toolkitがインストールされていると、以下のように簡単にプロジェクトを作成することができます。"New AWS Lambda Java Project..."を選択します。

LambdaのイベントはS3Eventなど選択できますが、今回はシンプルにCustomを選択します。Input Type, Output Typeを選択できますが、両方Objectを選択します。

これで後はhandleRequestを実装すれば良いだけです。入力を単純に返却しているだけです。

package com.example.lamda;

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

public class MyLambdaFunctionHandler implements RequestHandler<Object, Object> {

    @Override
    public Object handleRequest(Object input, Context context) {
        return input;
    }

}

ユニットテスト

Java版は簡単にEclipseでJUnitかけるのがとても良いですね。事前にローカルで入力値と出力値の簡単なテストができるのはとてもありがたいです。DynamoDBはLocalでも仮に動作させれますので、それを使えば、基本的にローカルで実装をして、最後にデプロイという形がとれます。（個人的にはローカルで環境を整備して、そこでできるだけ実装＆テストするのが開発効率が良いと思っています)

単純に入力したオブジェクトが返却されるかをテストします。

package com.example.lamda;

import static org.junit.Assert.*;

import java.io.IOException;

import org.junit.BeforeClass;
import org.junit.Test;

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

/**
 * A simple test harness for locally invoking your Lambda function handler.
 */
public class MyLambdaFunctionHandlerTest {
    private static class InputData {
        private String key1 = "value1";
    }

    private static InputData sInputData = new InputData();

    private static Object input;

    @BeforeClass
    public static void createInput() throws IOException {
        input = sInputData;
    }

    private Context createContext() {
        TestContext ctx = new TestContext();

        ctx.setFunctionName("Your Function Name");

        return ctx;
    }

    @Test
    public void testMyLambdaFunctionHandler() {
        MyLambdaFunctionHandler handler = new MyLambdaFunctionHandler();
        Context ctx = createContext();

        Object output = handler.handleRequest(input, ctx);

        assertEquals(input, output);
    }
}