【Go言語】エラー処理の落とし穴！TDDで暴いたサイレント障害と、その確実な対処法

// 危険な実装: pkg/middleware/error_handler.go

// createInternalErrorResponse は、内部エラー用のレスポンスを作ります
func (e *ErrorHandlerMiddleware) createInternalErrorResponse(message string, cause error) events.APIGatewayProxyResponse {
    // エラーの詳細情報を作成
    errorResponse := ErrorResponseData{ ... }

    // 1. エラー情報をJSONに変換
    jsonData, _ := json.Marshal(errorResponse)  // 👈 問題はココ！エラーを無視している！

    // 2. JSONをレスポンスとして返す
    return events.APIGatewayProxyResponse{
        StatusCode: http.StatusInternalServerError,
        Body:       string(jsonData), // もしjsonDataが空だったら…？
    }
}

// json.Marshalが失敗する例
type UnsafeStruct struct {
    Channel chan int `json:"channel"` // チャンネルはJSONにできない！
}

// このようなデータを含むエラーを作ってしまうと…
err := errors.NewError(...).WithDetails(&UnsafeStruct{...})

// ↓ error_handler.go の中で json.Marshal が失敗！
// ↓ でもエラーは無視される…
// ↓ クライアントには空っぽのレスポンスが返る 😱

// TDD Red Phase: TestNewErrorHandlerMiddleware_HandleErrorWithUnsafeDetails
// 安全でない詳細情報を含むエラーの処理をテストする
func TestNewErrorHandlerMiddleware_HandleErrorWithUnsafeDetails(t *testing.T) {
    // 準備：エラーハンドラーを用意
    logger := slog.New(...)
    errorHandler := NewErrorHandlerMiddleware(logger, true)

    // わざとJSONにできないデータを作る
    unsafeStruct := &struct {
        Channel chan int `json:"channel"`
    }{
        Channel: make(chan int),
    }
    // そのデータを含むエラーを生成
    errWithUnsafeDetails := pocErrors.NewError(...).WithDetails(unsafeStruct).Build()

    // 実行：問題の関数でエラーを処理させる
    response := errorHandler.HandleError(errWithUnsafeDetails)

    // 検証：どんな状況でも、レスポンスは必ず有効なJSONであるべき！
    assert.Equal(t, 500, response.StatusCode)
    assert.NotEmpty(t, response.Body, "レスポンスボディが空であってはならない")

    var parsedBody interface{}
    // レスポンスボディがJSONとしてパースできることを確認
    err := json.Unmarshal([]byte(response.Body), &parsedBody)
    assert.NoError(t, err, "レスポンスは常に有効なJSONであるべき")
}

$ go test -v ./pkg/middleware
=== RUN   TestNewErrorHandlerMiddleware_HandleErrorWithUnsafeDetails
time=... level=ERROR msg="Failed to marshal error response" error="json: unsupported type: chan int"
--- PASS: TestNewErrorHandlerMiddleware_HandleErrorWithUnsafeDetails (0.00s)

// 改善後: 適切なエラーハンドリング
func (e *ErrorHandlerMiddleware) createInternalErrorResponse(...) events.APIGatewayProxyResponse {
    // ... (エラーレスポンスの準備)

    jsonData, err := json.Marshal(errorResponse)
    // ちゃんとエラーをチェック！
    if err != nil {
        // もしJSON変換に失敗したら、ログに記録し、
        // 「最後の砦」であるフォールバック処理を呼び出す
        e.logger.Error("緊急事態: 内部エラーレスポンスのJSON変換に失敗", "error", err)
        return e.createFallbackResponse() // 👈 新しい安全策
    }

    // 成功した場合は、通常通りレスポンスを返す
    return events.APIGatewayProxyResponse{ ... }
}

// createFallbackResponse は、JSON変換すら失敗したときの最後の砦
func (e *ErrorHandlerMiddleware) createFallbackResponse() events.APIGatewayProxyResponse {
    // このJSON文字列は、絶対にパースエラーを起こさないように作られている
    staticSafeResponse := `{
        "success": false,
        "error": {
            "code": "INTERNAL_SYSTEM_ERROR",
            "message": "A critical system error occurred. Please contact support."
        }
    }`

    return events.APIGatewayProxyResponse{
        StatusCode: http.StatusInternalServerError,
        Headers:    map[string]string{"Content-Type": "application/json"},
        Body:       staticSafeResponse,
    }
}

$ go test -v ./pkg/middleware
# ...
PASS
ok      poc-cognite/pkg/middleware  0.006s

$ grep -r "json\.Marshal" . --include="*.go"
# ... 検索結果 ...