【Go言語】TDD実践入門！「止められないgoroutine」を安全に改善する方法

// ❌ 問題のあるコード
func NewEnhancedPerformanceMonitor(...) *EnhancedPerformanceMonitor {
    // 内部で勝手に「終わらない」Contextを作っている
    ctx, cancel := context.WithCancel(context.Background())

    monitor := &EnhancedPerformanceMonitor{
        ctx:    ctx,
        cancel: cancel,
        // ...
    }

    // バックグラウンドで、ずっと動き続けるgoroutineを開始
    go monitor.processAlerts()

    return monitor
}

// Lambda関数で使った場合
func lambdaHandler(ctx context.Context, event Event) error {
    // モニターを開始！
    monitor := NewEnhancedPerformanceMonitor(...)

    // 何かの処理をする
    return processEvent(event)
} // ⬅️ Lambda関数はここで終了する。でも…

// monitor.processAlerts() のgoroutineは、誰にも止められず、
// サーバーの裏側で、永遠に動き続けてしまうのです！
// これが積み重なると、メモリを食いつぶす「メモリリーク」になります。

func TestSomething(t *testing.T) {
    monitor := NewEnhancedPerformanceMonitor(...)

    // テストが終わっても、monitorは裏で動き続ける…
    // これが、他のテストに悪影響を与える可能性があります。
}

// 親の処理
func ParentProcess(ctx context.Context) {
    // 子の処理に「この伝言ゲームに参加してね」とContextを渡す
    ChildProcess(ctx)
}

// 子の処理
func ChildProcess(ctx context.Context) {
    for {
        select {
        case <-ctx.Done(): // 親から「やめて！」という伝言が来たら
            fmt.Println("親が終了したので、私も終了します。")
            return // 子も素直に処理をやめる
        default:
            // 通常の処理を続ける
        }
    }
}

// 問題のコードはコレ
func ChildProcess() {
    // 親を無視して、自分で新しい伝言ゲームを始めてしまう
    ctx := context.Background()

    for {
        select {
        case <-ctx.Done(): // このキャンセル信号は、永遠に来ない！
            return
        default:
            // 止まらない無限ループの完成…
        }
    }
}

// TDD Red Phase: TestContextControl
func TestContextControl(t *testing.T) {

    // テスト1: 「外部からContextを受け取れる新しい関数がほしい！」
    t.Run("外部Contextを受け取れるべき", func(t *testing.T) {
        ctx := context.Background()
        // ⚠️ まだ存在しない、理想の関数を呼んでみる
        monitor := NewEnhancedPerformanceMonitorWithContext(ctx, ...)
        assert.NotNil(t, monitor)
    })

    // テスト2: 「Contextがキャンセルされたら、ちゃんと停止してほしい！」
    t.Run("Contextのキャンセルでgoroutineが停止するべき", func(t *testing.T) {
        // キャンセル可能なContextを用意
        ctx, cancel := context.WithCancel(context.Background())

        monitor := NewEnhancedPerformanceMonitorWithContext(ctx, ...)

        // 「やめて！」というキャンセル信号を送る
        cancel()

        // goroutineが止まるのを少し待つ
        time.Sleep(100 * time.Millisecond)

        // 💡 本当はここで「goroutineが本当に止まったか」を
        //    チェックする仕組みが必要ですが、今回はログ出力で確認します。
    })
}

$ go test -v ./pkg/monitoring
# ...
undefined: NewEnhancedPerformanceMonitorWithContext
FAIL    build failed

// ✅ 新しい関数：外部からContextを受け取れる！
func NewEnhancedPerformanceMonitorWithContext(
    parentCtx context.Context, // 👈 親からContextを受け取る！
    ...
) *EnhancedPerformanceMonitor {

    // 🔑 ポイント：受け取ったContextを親として、子Contextを作る
    ctx, cancel := context.WithCancel(parentCtx)

    monitor := &EnhancedPerformanceMonitor{
        // ... 他の設定は同じ ...
        // 新しいContextを設定
        ctx:    ctx,    // 👈 親と連動するContext
        cancel: cancel,
    }

    // このgoroutineは、親がキャンセルされると一緒に止まる！
    go monitor.processAlerts()

    return monitor
}

// ✅ 古い関数：既存のコードを壊さないための配慮
func NewEnhancedPerformanceMonitor(...) *EnhancedPerformanceMonitor {
    // 内部で、新しい関数をデフォルトのContextで呼び出すだけ
    return NewEnhancedPerformanceMonitorWithContext(context.Background(), ...)
}

$ go test -v ./pkg/monitoring
=== RUN   TestContextControl
=== RUN   TestContextControl/外部Contextを受け取れるべき
=== RUN   TestContextControl/Contextのキャンセルでgoroutineが停止するべき
{"level":"INFO","message":"Shutting down alert processing"} // ⬅️ 注目！
--- PASS: TestContextControl (0.15s)
PASS ✅

$ go test ./...
PASS ✅

// NewEnhancedPerformanceMonitorWithContext は、外部からのContext制御に対応した
// パフォーマンスモニターを作成します。
// AWS Lambdaのようなライフサイクルを持つ環境では、必ずこの関数を使用してください。
//
// 💡 使い方 (Lambdaでの例):
// func lambdaHandler(ctx context.Context, ...) (..., error) {
//     // LambdaのContextを渡すことで、Lambda終了時にモニターも自動で停止します
//     monitor := NewEnhancedPerformanceMonitorWithContext(ctx, ...)
//     defer monitor.Shutdown() // 念のためクリーンアップ
//     ...
// }
func NewEnhancedPerformanceMonitorWithContext(
    parentCtx context.Context,
    ...
) *EnhancedPerformanceMonitor {
    // ... 実装 ...
}

func lambdaHandler(ctx context.Context, request ...) (..., error) {

    // 🔑 ポイント：Lambdaから渡される`ctx`をそのまま渡すだけ！
    monitor := monitoring.NewEnhancedPerformanceMonitorWithContext(
        ctx, // 👈 これでLambdaとモニターの運命は一心同体！
        ...
    )

    // ... 普通に処理 ...

    return successResponse(...), nil
} // ⬅️ Lambdaが終了すると、monitorも自動的にクリーンアップされる！