GoでDIを実践してみた！AWSセッション管理を改善し、テストを格段にやりやすくする方法

// ❌ 悪い例：呼び出されるたびにAWSセッションとクライアントを作成
package middleware

import (
	"log"
	"net/http"
	"time"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/cloudwatch"
)

type MetricsMiddleware struct {
	// ...
}

// この関数が呼ばれるたびに、重い初期化処理が走る
func (m *MetricsMiddleware) RecordMetric(metricName string, value float64) error {
	// 🚨 問題点1: 毎回AWSセッションを作成（認証情報読み込み、HTTP通信などが発生）
	sess, err := session.NewSession(&aws.Config{
		Region: aws.String("ap-northeast-1"),
	})
	if err != nil {
		log.Printf("Failed to create AWS session: %v", err)
		return err
	}

	// 🚨 問題点2: 毎回CloudWatchクライアントを作成
	cwClient := cloudwatch.New(sess)

	// メトリクス送信処理...
	_, err = cwClient.PutMetricData(&cloudwatch.PutMetricDataInput{
		// ...
	})
	return err
}

// 1回のリクエストで複数回メトリクスが送信される
func HandleRequest(w http.ResponseWriter, r *http.Request) {
	metrics := &MetricsMiddleware{}
	metrics.RecordMetric("RequestCount", 1)      // 1回目のセッション作成
	metrics.RecordMetric("ProcessingTime", 150)  // 2回目のセッション作成
	metrics.RecordMetric("MemoryUsage", 128)     // 3回目のセッション作成
}

// ✅ 改善案：testableなインターフェースを定義
package middleware

import "github.com/aws/aws-sdk-go/service/cloudwatch"

// CloudWatchClientは、`PutMetricData`メソッドを持つあらゆる型を表すインターフェース
type CloudWatchClient interface {
	PutMetricData(input *cloudwatch.PutMetricDataInput) (*cloudwatch.PutMetricDataOutput, error)
}

// これにより、*cloudwatch.CloudWatch型だけでなく、自作のモックも受け入れ可能になる
var _ CloudWatchClient = (*cloudwatch.CloudWatch)(nil)

// ✅ 改善案：依存性注入（DI）を適用したMetricsMiddleware
type MetricsMiddleware struct {
	functionName string
	namespace    string
	cwClient     CloudWatchClient // 🚨 外部から注入されるクライアント
}

// NewMetricsMiddlewareはコンストラクタで依存性を受け取る
func NewMetricsMiddleware(
	functionName, namespace string,
	client CloudWatchClient, // インターフェース型で受け取る
) *MetricsMiddleware {
	return &MetricsMiddleware{
		functionName: functionName,
		namespace:    namespace,
		cwClient:     client, // 注入されたクライアントを保持
	}
}

func (m *MetricsMiddleware) RecordMetric(metricName string, value float64) error {
	// 保持しているクライアントを再利用するため、セッション作成は不要！
	_, err := m.cwClient.PutMetricData(&cloudwatch.PutMetricDataInput{
		// ...
	})
	// ...
	return err
}

// ✅ 改善案：シングルトンパターンでAWSクライアントを一元管理
package main

import (
	"log"
	"your-project/middleware"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/cloudwatch"
)

var (
	cwClient *cloudwatch.CloudWatch
	// ... sync.Onceなどを使ったより堅牢な実装も可能
)

// init関数でアプリケーション起動時に一度だけクライアントを生成
func init() {
	sess, err := session.NewSession(&aws.Config{
		Region: aws.String("ap-northeast-1"),
	})
	if err != nil {
		log.Fatalf("Failed to create AWS session: %v", err)
	}
	cwClient = cloudwatch.New(sess)
	log.Println("AWS CloudWatch client initialized successfully.")
}

func main() {
	// 生成済みのクライアントをMetricsMiddlewareに注入
	metrics := middleware.NewMetricsMiddleware(
		"MyFunction",
		"MyApp/Lambda",
		cwClient, // 注入！
	)

	// あとはこのmetricsインスタンスを使い回すだけ
	// ...
}

// ✅ テスト用のモッククライアント
type MockCloudWatchClient struct {
	// PutMetricDataが何回、どのような引数で呼ばれたかを記録
	PutMetricDataCalls int
	LastInput        *cloudwatch.PutMetricDataInput
	// テスト用にエラーを返すことも可能
	ReturnError error
}

func (m *MockCloudWatchClient) PutMetricData(input *cloudwatch.PutMetricDataInput) (*cloudwatch.PutMetricDataOutput, error) {
	m.PutMetricDataCalls++
	m.LastInput = input
	if m.ReturnError != nil {
		return nil, m.ReturnError
	}
	return &cloudwatch.PutMetricDataOutput{}, nil
}

// ✅ 改善後：モックによる完全なユニットテスト
func TestMetricsMiddleware_RecordMetric(t *testing.T) {
	// 1. モッククライアントを作成
	mockClient := &MockCloudWatchClient{}

	// 2. モックをMetricsMiddlewareに注入
	metrics := NewMetricsMiddleware("TestFunc", "Test/App", mockClient)

	// 3. テスト対象のメソッドを実行
	err := metrics.RecordMetric("TestMetric", 42.0)

	// 4. 結果を検証
	assert.NoError(t, err) // エラーがないことを確認

	// 5. モックの状態を検証（意図通りに呼び出されたか）
	assert.Equal(t, 1, mockClient.PutMetricDataCalls, "PutMetricData should be called once")
	assert.Equal(t, "TestMetric", *mockClient.LastInput.MetricData[0].MetricName)
	assert.Equal(t, 42.0, *mockClient.LastInput.MetricData[0].Value)
}

func TestMetricsMiddleware_ErrorHandling(t *testing.T) {
	// エラーを返すモックを準備
	mockClient := &MockCloudWatchClient{
		ReturnError: errors.New("mock CloudWatch API error"),
	}
	metrics := NewMetricsMiddleware("TestFunc", "Test/App", mockClient)

	err := metrics.RecordMetric("TestMetric", 1.0)

	// エラーが正しく伝播されることを確認
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "mock CloudWatch API error")
}