Go の Lambda 関数ハンドラーの定義

Lambda 関数ハンドラーは、イベントを処理する関数コード内のメソッドです。関数が呼び出されると、Lambda はハンドラーメソッドを実行します。関数は、ハンドラーが応答を返すか、終了するか、タイムアウトするまで実行されます。

このページでは、プロジェクトのセットアップ、命名規則、ベストプラクティスなど、Go で Lambda 関数ハンドラーを操作する方法について説明します。このページには、注文に関する情報を取得し、テキストファイル受信を生成し、このファイルを Amazon Simple Storage Service (S3) バケットに配置する Go Lambda 関数の例も含まれています。関数を書き込んだ後にデプロイする方法については、「.zip ファイルアーカイブを使用して Go Lambda 関数をデプロイする」または「コンテナイメージを使用して Go Lambda 関数をデプロイする」を参照してください。

Go ハンドラープロジェクトのセットアップ

Go で書き込まれた Lambda 関数は、Go 実行可能ファイルとして作成されます。Go Lambda 関数プロジェクトは、以下の go mod init コマンドを使用して他の Go プロジェクトを初期化するのと同じ方法で初期化できます:

go mod init example-go

ここでは、example-go がモジュール名です。これは何にでも置き換えることができます。このコマンドはプロジェクトを初期化し、プロジェクトの依存関係を一覧表示する go.mod ファイルを生成します。

go get コマンドを使用して、外部依存関係をプロジェクトに追加します。例えば、Go のすべての Lambda 関数には、Go 用の Lambda プログラミングモデルを実装する github.com/aws/aws-lambda-go/lambda パッケージを含める必要があります。以下の go get コマンドでこのパッケージを含めます:

go get github.com/aws/aws-lambda-go

関数コードは Go ファイルに存在する必要があります。以下の例では、この main.go ファイルに名前を付けます。このファイルではハンドラーメソッドに、このハンドラーを呼び出す main() 関数と同様にコア関数ロジックを実装します。

Go Lambda 関数のコードの例

以下の Go Lambda 関数コードの例では、注文に関する情報を取得し、テキストファイル受信を生成し、このファイルを Amazon S3 バケットに配置します。

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/aws/aws-lambda-go/lambda"
	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/s3"
)

type Order struct {
	OrderID string  `json:"order_id"`
	Amount  float64 `json:"amount"`
	Item    string  `json:"item"`
}

var (
	s3Client *s3.Client
)

func init() {
	// Initialize the S3 client outside of the handler, during the init phase
	cfg, err := config.LoadDefaultConfig(context.TODO())
	if err != nil {
		log.Fatalf("unable to load SDK config, %v", err)
	}

	s3Client = s3.NewFromConfig(cfg)
}

func uploadReceiptToS3(ctx context.Context, bucketName, key, receiptContent string) error {
	_, err := s3Client.PutObject(ctx, &s3.PutObjectInput{
		Bucket: &bucketName,
		Key:    &key,
		Body:   strings.NewReader(receiptContent),
	})
	if err != nil {
		log.Printf("Failed to upload receipt to S3: %v", err)
		return err
	}
	return nil
}

func handleRequest(ctx context.Context, event json.RawMessage) error {
	// Parse the input event
	var order Order
	if err := json.Unmarshal(event, &order); err != nil {
		log.Printf("Failed to unmarshal event: %v", err)
		return err
	}

	// Access environment variables
	bucketName := os.Getenv("RECEIPT_BUCKET")
	if bucketName == "" {
		log.Printf("RECEIPT_BUCKET environment variable is not set")
		return fmt.Errorf("missing required environment variable RECEIPT_BUCKET")
	}

	// Create the receipt content and key destination
	receiptContent := fmt.Sprintf("OrderID: %s\nAmount: $%.2f\nItem: %s",
		order.OrderID, order.Amount, order.Item)
	key := "receipts/" + order.OrderID + ".txt"

	// Upload the receipt to S3 using the helper method
	if err := uploadReceiptToS3(ctx, bucketName, key, receiptContent); err != nil {
		return err
	}

	log.Printf("Successfully processed order %s and stored receipt in S3 bucket %s", order.OrderID, bucketName)
	return nil
}

func main() {
	lambda.Start(handleRequest)
}

この main.go ファイルには以下のコードのセクションが含まれます:

この関数が正しく機能するには、実行ロールで s3:PutObject アクションを許可する必要があります。また、必ず RECEIPT_BUCKET 環境変数を定義してください。呼び出しに成功したら、Amazon S3 バケットに受信ファイルが含まれているはずです。

ハンドラーの命名規則

Go の Lambda 関数では、ハンドラーに任意の名前を使用できます。この例では、ハンドラーメソッド名は handleRequest です。コード内でハンドラー値を参照するには、_HANDLER 環境変数を使用できます。

.zip デプロイパッケージを使用してデプロイされた Go 関数の場合、関数コードを含む実行ファイルには bootstrap という名前を付ける必要があります。さらに、bootstrap ファイルは、.zip ファイルのルートに置く必要があります。コンテナイメージを使用してデプロイされた Go 関数の場合、実行ファイルには任意の名前を使用できます。

入力イベントオブジェクトの定義とアクセス

JSON は Lambda 関数の最も一般的な標準入力形式です。この例では、関数は以下のような入力を想定しています:

{
    "order_id": "12345",
    "amount": 199.99,
    "item": "Wireless Headphones"
}

Go で Lambda 関数を使用する場合、予想される入力イベントの形状を Go 構造体として定義できます。この例では、Order を表す構造体を定義します。

type Order struct {
    OrderID string  `json:"order_id"`
    Amount  float64 `json:"amount"`
    Item    string  `json:"item"`
}

この構造体は、予想される入力形状と一致します。構造体を定義したら、encoding/json 標準ライブラリと互換性のある汎用 JSON 型を取り込むハンドラー署名を記述できます。その後、func Unmarshal 関数を使用して、構造体に逆シリアル化できます。これはハンドラーの最初の数行に示されています:

func handleRequest(ctx context.Context, event json.RawMessage) error {
    // Parse the input event
    var order Order
    if err := json.Unmarshal(event, &order); err != nil {
        log.Printf("Failed to unmarshal event: %v", err)
        return err
    ...
}

この逆シリアル化の後、order 変数のフィールドにアクセスできます。例えば、order.OrderID は元の入力から "order_id" の値を取得します。

Lambda コンテキストオブジェクトへのアクセスと使用

Lambda コンテキストオブジェクトには、呼び出し、関数、および実行環境に関する情報が含まれます。この例では、ハンドラー署名でこの変数を ctx として宣言しました:

func handleRequest(ctx context.Context, event json.RawMessage) error {
    ...
}

ctx context.Context 入力は、関数ハンドラーのオプション引数です。受理されたハンドラー署名の詳細については、「Go ハンドラーの有効なハンドラー署名」を参照してください。

AWS SDK を使用して他のサービスを呼び出す場合、コンテキストオブジェクトはいくつかの重要な領域で必要です。例えば、SDK クライアントを適切に初期化するには、以下のようにコンテキストオブジェクトを使用して正しい AWS SDK 設定をロードできます:

// Load AWS SDK configuration using the default credential provider chain
    cfg, err := config.LoadDefaultConfig(ctx)

SDK 呼び出し自体では、コンテキストオブジェクトを入力として必要とする場合があります。例えば、s3Client.PutObject 呼び出しはコンテキストオブジェクトを最初の引数として受け入れます:

// Upload the receipt to S3
    _, err = s3Client.PutObject(ctx, &s3.PutObjectInput{
        ...
    })

AWS SDK リクエスト以外では、関数のモニタリングにコンテキストオブジェクトを使用することもできます。コンテキストオブジェクトの詳細については、「Lambda コンテキストオブジェクトを使用して Go 関数の情報を取得する」を参照してください。

Go ハンドラーの有効なハンドラー署名

Go で Lambda 関数ハンドラーを構築するには複数の方法がありますが、次のルールに従う必要があります。

次のリストは、有効なハンドラー署名の一覧です。 TIn と TOut は、 encoding/json 標準ライブラリと互換性のあるタイプを表しています。詳細については、「func アンマーシャリング」でこれらのタイプが逆シリアル化する方法を参照してください。

ハンドラーでの AWS SDK for Go v2 の使用

多くの場合、Lambda 関数を使用して、他の AWS リソースとやり取りしたり、更新したりします。これらのリソースとインターフェイスする最も簡単な方法は、AWS SDK for Go v2 を使用することです。

SDK 依存関係を関数に追加するには、必要な特定の SDK クライアントに go get コマンドを使用します。前のサンプルコードでは、config ライブラリと s3 ライブラリを使用しました。これらの依存関係を追加するには、go.mod と main.go ファイルが格納されているディレクトリで以下のコマンドを実行します。

go get github.com/aws/aws-sdk-go-v2/config
go get github.com/aws/aws-sdk-go-v2/service/s3

次に、依存関係を関数のインポートブロックにインポートします:

import (
    ...
    "github.com/aws/aws-sdk-go-v2/config"
    "github.com/aws/aws-sdk-go-v2/service/s3"
)

ハンドラーで SDK を使用する場合は、適切な設定でクライアントを設定します。これを行う最も簡単な方法は、デフォルトの認証情報プロバイダーチェーンを使用することです。この例では、この設定をロードする 1 つの方法を示しています:

// Load AWS SDK configuration using the default credential provider chain
    cfg, err := config.LoadDefaultConfig(ctx)
    if err != nil {
        log.Printf("Failed to load AWS SDK config: %v", err)
        return err
    }

この設定を cfg 変数にロードした後、この変数をクライアントインスタンスに渡すことができます。サンプルコードは、以下のように Amazon S3 クライアントをインスタンス化します:

// Create an S3 client
    s3Client := s3.NewFromConfig(cfg)

この例では、関数を呼び出すたびに初期化する必要がないように、init() 関数で Amazon S3 クライアントを初期化しました。問題は、init() 関数で Lambda がコンテキストオブジェクトにアクセスできないことです。回避策として、初期化フェーズ中に context.TODO() のようなプレースホルダーを渡すことができます。後で、クライアントを使用して呼び出しを行うときは、コンテキストオブジェクト全体を渡します。この回避策については、AWS SDK クライアントの初期化と呼び出しでのコンテキストの使用 でも説明されています。

SDK クライアントを設定して初期化したら、SDK クライアントを使用して他の AWS サービスとやり取りできます。サンプルコードは、以下のように Amazon S3 PutObject API を呼び出します。

_, err = s3Client.PutObject(ctx, &s3.PutObjectInput{
    Bucket: &bucketName,
    Key:    &key,
    Body:   strings.NewReader(receiptContent),
})

環境変数にアクセスする

ハンドラーコードでは、os.Getenv() メソッドを使用して任意の環境変数を参照できます。この例では、以下のコード行を使用して、定義された RECEIPT_BUCKET 環境変数を参照します:

// Access environment variables
    bucketName := os.Getenv("RECEIPT_BUCKET")
    if bucketName == "" {
        log.Printf("RECEIPT_BUCKET environment variable is not set")
        return fmt.Errorf("missing required environment variable RECEIPT_BUCKET")
    }

グローバルな状態を使用する

関数を呼び出すたびに新しいリソースが作成されないようにするには、Lambda 関数のハンドラーコードの外部でグローバル変数を宣言および変更できます。これらのグローバル変数は、var ブロックまたはステートメントで定義します。さらに、ハンドラーは初期化フェーズ中に実行される init() 関数を宣言することがあります。この init メソッドは、AWS Lambda でも標準の Go プログラムと同じように作動します。

Lambda 関数のコードのベストプラクティス

Lambda 関数をビルドするときは、次のリストのガイドラインに従って、コーディングのベストプラクティスを実行してください。