Go言語製LambdaのCold Start、2.5秒から519msへ！僕が実践した最適化の全記録

// 改善前: パッケージが多すぎる
import (
    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-lambda-go/events"
    "github.com/aws/aws-sdk-go/aws"
    // ... 合計20以上のパッケージ
)

// 改善前: リクエストごとに毎回初期化（非効率）
func main() {
    lambda.Start(func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
        // 毎回AWSへの接続設定（重い処理）
        sess := session.Must(session.NewSession())
        cognitoClient := cognitoidentityprovider.New(sess)
        
        // ...実際の処理
    })
}

// 改善後: グローバル変数で接続情報をキャッシュ
var (
    // cognitoClientCache: AWSの認証サービスへの接続情報を保存する場所
    cognitoClientCache *cognito.Client
    // clientCacheOnce: この処理を「一度だけ」実行するための仕組み
    clientCacheOnce sync.Once
)

// getCognitoClient は、キャッシュされた接続情報を返す関数
func getCognitoClient() (*cognito.Client, error) {
    var initErr error
    // clientCacheOnce.Doの中の処理は、プログラム全体で本当に一度しか実行されないことを保証します。
    clientCacheOnce.Do(func() {
        // 初回実行時のみ、重い初期化処理を行う
        cognitoClientCache, initErr = cognito.NewClient()
    })
    return cognitoClientCache, initErr
}

func main() {
    lambda.Start(func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
        // 2回目以降はキャッシュされた接続情報を瞬時に取得できる
        client, err := getCognitoClient()
        if err != nil {
            return utils.CreateErrorResponse(500, "Failed to initialize client")
        }
        
        // ...実際の処理
    })
}

// 改善後: 必要最小限のパッケージに整理
import (
    // パッケージ数を20以上から12個に削減（40%削減）
    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-lambda-go/events"
    "poc-cognite/pkg/cognito" // 自作のパッケージに機能をまとめる
    "poc-cognite/pkg/monitoring"
    "poc-cognite/pkg/utils"
    "context"
    "encoding/json"
)

# template.yaml (AWSの設定ファイル)
LoginFunction:
  Type: AWS::Serverless::Function
  Properties:
    MemorySize: 256  # メモリを128MBから256MBに増量
    Environment:
      Variables:
        # Go言語のガベージコレクション（不要なメモリを掃除する機能）を最適化する設定
        GOMEMLIMIT: 240MiB

# 最終的な最適設定
LoginFunction:
  Properties:
    MemorySize: 128     # コストを重視し128MBに戻す
    Timeout: 15         # タイムアウトを30秒から15秒に短縮
    Environment:
      Variables:
        # いわゆるアウトオブメモリ回避のために値を変更します。
        # GOMEMLIMITは、Goのガベージコレクションが使用するメモリの上限を設定します。
        # Lambdaに割り当てられたメモリ（128MB）の約88%（112MiB）に設定することで、
        # Goのランタイムが効率的に動作しつつ、Lambdaの実行環境が使用するメモリも確保できます。        
        # これにより、メモリ不足によるエラーを防ぎながら、コストを抑えることができます。
        GOMEMLIMIT: 112MiB

# template.yaml
LoginFunction:
  Type: AWS::Serverless::Function
  Properties:
    # この設定を追加するだけで有効になる
    ProvisionedConcurrencyConfig:
      ProvisionedConcurrencyEnabled: true
  # これで指定した数の実行環境が常に待機状態になる

// 改善後の実測データ
{
  "operation": "login",
  "total_runs": 14,
  "success_count": 14,
  "success_rate": 1.0,
  "avg_duration_ms": 452,    // 452ms平均 ⚡
  "min_duration_ms": 406,    // 406ms最小 ⚡
  "max_duration_ms": 519,    // 519ms最大 ⚡
  "requests_per_second": 1.83, // 1秒あたりの処理数
  "errors_by_type": {}         // エラーなし
}

// パフォーマンス測定の簡単な実装例
func main() {
    lambda.Start(func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
        // 処理の開始時間を記録
        start := time.Now()
        
        // ここで実際のビジネスロジックを呼び出す
        result, err := businessLogic(ctx, request)
        
        // 処理の終了時間との差分を計算してログに出力
        duration := time.Since(start)
        log.Printf("処理時間: %d ms", duration.Milliseconds())
        
        return result, err
    })
}

# 選択肢の比較
  128MB + 最適化:
    - 性能: 良好
    - コスト: 最小
    - Cold Start: 対策が必要

  256MB + 最適化:
    - 性能: 良好
    - コスト: 中程度
    - Cold Start: 軽微
    
  Provisioned Concurrency:
    - 性能: 優秀（Cold Startなし）
    - コスト: 高い
    - 安定性: 最高

// 効果的なパターン
var (
    // 重い初期化処理はグローバル変数で一度だけ実行する
    expensiveClient *SomeClient
    clientOnce      sync.Once
)

func getClient() *SomeClient {
    clientOnce.Do(func() {
        expensiveClient = NewExpensiveClient()
    })
    return expensiveClient
}

// GOMEMLIMIT環境変数でメモリ管理を最適化
// メモリの88%を目安に設定
// 128MB → GOMEMLIMIT=112MiB
// 256MB → GOMEMLIMIT=240MiB