Day 017 — context パッケージ②（タイムアウト・値の伝播）

📚 背景知識（読んでから問題へ）

Day 016ではcontext.WithCancelを使い、「明示的にcancel()を呼ぶことでキャンセルを親から子へ連鎖させる」パターンを学びました。しかし実務でもっとよく使うのは、「〇〇ミリ秒以内に終わらなければ自動的にキャンセルする」というタイムアウト制御です。DBクエリ・外部API呼び出し・複数リソースの並行取得など、「時間がかかりすぎたら諦めて先に進む」という判断はあらゆる箇所で必要になります。

contextパッケージはこれを2つの関数で提供します。

• context.WithTimeout(parent, duration) — 「今からduration後」に自動キャンセルされるcontextを作る
• context.WithDeadline(parent, time.Time) — 「指定した時刻になったら」自動キャンセルされるcontextを作る

両者は本質的に同じもので、WithTimeout(parent, d)は内部的にWithDeadline(parent, time.Now().Add(d))を呼んでいるだけです。「あと何ミリ秒」という相対時間で考えたいときはWithTimeout、「12:00:00までに」という絶対時刻で考えたいとき（例えば上流のサービスから伝わってきた締切時刻をそのまま使うとき）はWithDeadlineを使う、という使い分けになります。どちらも制限時間に達するとctx.Done()が自動的にcloseされ、ctx.Err()はcontext.DeadlineExceededになります。Day 016のWithCancelのときはctx.Err()がcontext.Canceledでしたが、原因によって値が変わる、という設計です。

もう1つ、contextパッケージにはcontext.WithValue(parent, key, value)という関数があります。これは「リクエストスコープの値（リクエストID・認証済みユーザー情報など）を、呼び出しツリーの奥深くまで明示的な引数を増やさずに伝播させる」ための仕組みです。ただしGoの公式ドキュメントは「関数のオプション引数の代わりに使ってはいけない」と明確に警告しています。WithValueは「その処理全体を貫くリクエストのメタデータ（トレースID・認証情報など）」のためのものであり、ビジネスロジックのパラメータを渡す手段ではありません。乱用するとコードの見た目からは何が渡っているか分からなくなり、型安全性も失われるためです。

WithValueを安全に使うために、Goコミュニティでは「キーには文字列リテラルではなく、パッケージ内だけで通用する非公開の独自型を使う」という規約が定着しています。これは、もし複数のパッケージが同じ文字列（例えば"userID"）をキーとして使ってしまうと、意図せず値が上書き・衝突する可能性があるためです。独自型（例えばtype ctxKey string）を使えば、たとえ値が同じ文字列でも型が異なるため衝突しません。

📝 問題

以下の要件を満たす、go run main.goでそのまま実行できる1つのGoプログラムを実装してください。

1. type ctxKey stringのような非公開の独自型をキーとして定義し、context.WithValueを使って"req-8f3c1a"というリクエストIDをcontextに埋め込んでください。キーとして文字列リテラルを直接使うのは禁止です
2. requestIDFrom(ctx context.Context) stringという関数を実装し、ctxからリクエストIDを安全に取り出してください（型アサーションが失敗した場合は"unknown"を返すこと）
3. fetchResource(ctx context.Context, id int) fetchResultという関数を実装してください。fetchResultはid int・msg string・err errorを持つ構造体とします。この関数は「id * 120ミリ秒だけ時間がかかるリソース取得」をselectで模擬し、時間内に終わればmsgにrequestIDFrom(ctx)で取得したリクエストIDを含む成功メッセージを、ctxが先にキャンセルされたらerrにctx.Err()を含むエラーを設定して返してください
4. context.WithTimeoutを使って全体の制限時間400ミリ秒を設定したcontextを作り、deferで必ずcancel()を呼んでください
5. idが1から5までの5つのfetchResourceをgoroutineとして並行実行し、それぞれの結果をchannel経由で集約し、成功したものと失敗（タイムアウト）したものを区別して出力してください
6. 全ての結果を出力し終えたあと、最終的なctx.Err()を出力してください（context.DeadlineExceededになっているはずです）

🔍 ヒント（段階的開示）

📋 コピーpackage main

import (
	"context"
	"fmt"
	"sync"
	"time"
)

type ctxKey string

const requestIDKey ctxKey = "requestID"

func requestIDFrom(ctx context.Context) string {
	// TODO: ctx.Value(requestIDKey) を安全に string に変換する
}

type fetchResult struct {
	id  int
	msg string
	err error
}

func fetchResource(ctx context.Context, id int) fetchResult {
	reqID := requestIDFrom(ctx)
	delay := time.Duration(id) * 120 * time.Millisecond

	select {
	case <-time.After(delay):
		// TODO: 成功時の fetchResult を返す
	case <-ctx.Done():
		// TODO: 失敗時（タイムアウト）の fetchResult を返す
	}
}

func main() {
	parentCtx := context.WithValue(context.Background(), requestIDKey, "req-8f3c1a")

	ctx, cancel := context.WithTimeout(parentCtx, 400*time.Millisecond)
	defer cancel()

	var wg sync.WaitGroup
	results := make(chan fetchResult, 5)

	for id := 1; id <= 5; id++ {
		wg.Add(1)
		go func(id int) {
			defer wg.Done()
			// TODO: fetchResource を呼び results に送る
		}(id)
	}

	// TODO: 全goroutine完了後に results を close する
	// TODO: results を range で受け取り、成功/失敗を分けて出力する
	// TODO: 最終的な ctx.Err() を出力する
}

✅ 模範解答

📋 コピーpackage main

import (
	"context"
	"fmt"
	"sync"
	"time"
)

// ctxKey はcontext.WithValueのキーに使う非公開の独自型。
// 文字列リテラルを直接キーにすると他パッケージと衝突しうるため、
// パッケージ内だけで通用する型を使うのがGoの規約。
type ctxKey string

const requestIDKey ctxKey = "requestID"

// requestIDFrom はctxからリクエストIDを安全に取り出す。
// 型アサーションが失敗する（キーが設定されていない）場合は "unknown" を返す。
func requestIDFrom(ctx context.Context) string {
	if v, ok := ctx.Value(requestIDKey).(string); ok {
		return v
	}
	return "unknown"
}

// fetchResult はfetchResourceの結果（成功メッセージ or エラー）を表す。
type fetchResult struct {
	id  int
	msg string
	err error
}

// fetchResource はネットワーク越しのリソース取得を模擬する。
// idが大きいほど時間がかかる設計にし、タイムアウトする/しないの両方を再現する。
func fetchResource(ctx context.Context, id int) fetchResult {
	reqID := requestIDFrom(ctx)
	delay := time.Duration(id) * 120 * time.Millisecond

	select {
	case <-time.After(delay):
		return fetchResult{
			id:  id,
			msg: fmt.Sprintf("resource-%d を取得しました（request: %s）", id, reqID),
		}
	case <-ctx.Done():
		return fetchResult{
			id:  id,
			err: fmt.Errorf("resource-%d の取得中止（request: %s, 理由: %w）", id, reqID, ctx.Err()),
		}
	}
}

func main() {
	// 1. リクエストスコープの値（リクエストID）をcontextに埋め込む。
	parentCtx := context.WithValue(context.Background(), requestIDKey, "req-8f3c1a")

	// 2. 全体の制限時間400msを設定する。400ms経過で自動的にDeadlineExceededになる。
	ctx, cancel := context.WithTimeout(parentCtx, 400*time.Millisecond)
	defer cancel()

	var wg sync.WaitGroup
	results := make(chan fetchResult, 5)

	for id := 1; id <= 5; id++ {
		wg.Add(1)
		go func(id int) {
			defer wg.Done()
			results <- fetchResource(ctx, id)
		}(id)
	}

	go func() {
		wg.Wait()
		close(results)
	}()

	for r := range results {
		if r.err != nil {
			fmt.Println("失敗:", r.err)
		} else {
			fmt.Println("成功:", r.msg)
		}
	}

	fmt.Println("最終的なctx.Err():", ctx.Err())
}

成功: resource-1 を取得しました（request: req-8f3c1a）
成功: resource-2 を取得しました（request: req-8f3c1a）
成功: resource-3 を取得しました（request: req-8f3c1a）
失敗: resource-4 の取得中止（request: req-8f3c1a, 理由: context deadline exceeded）
失敗: resource-5 の取得中止（request: req-8f3c1a, 理由: context deadline exceeded）
最終的なctx.Err(): context deadline exceeded

※ id=1,2,3（120ms・240ms・360ms）は制限時間400msより先に終わるため成功し、id=4,5（480ms・600ms）はタイムアウトの方が先に来るため失敗します。プログラム全体の実行時間も約400ms程度で終わり、id=5の600msを律儀に待つことはありません。これがcontextによるタイムアウト制御の効果です。

🪜 Step-by-Step 解説

📋 コピーtype ctxKey string

const requestIDKey ctxKey = "requestID"

func requestIDFrom(ctx context.Context) string {
	if v, ok := ctx.Value(requestIDKey).(string); ok {
		return v
	}
	return "unknown"
}

📋 コピーfunc fetchResource(ctx context.Context, id int) fetchResult {
	reqID := requestIDFrom(ctx)
	delay := time.Duration(id) * 120 * time.Millisecond

	select {
	case <-time.After(delay):
		return fetchResult{id: id, msg: fmt.Sprintf("resource-%d を取得しました（request: %s）", id, reqID)}
	case <-ctx.Done():
		return fetchResult{id: id, err: fmt.Errorf("resource-%d の取得中止（request: %s, 理由: %w）", id, reqID, ctx.Err())}
	}
}

📋 コピーctx, cancel := context.WithTimeout(parentCtx, 400*time.Millisecond)
defer cancel()

📋 コピーvar wg sync.WaitGroup
results := make(chan fetchResult, 5)

for id := 1; id <= 5; id++ {
	wg.Add(1)
	go func(id int) {
		defer wg.Done()
		results <- fetchResource(ctx, id)
	}(id)
}

go func() {
	wg.Wait()
	close(results)
}()

for r := range results {
	if r.err != nil {
		fmt.Println("失敗:", r.err)
	} else {
		fmt.Println("成功:", r.msg)
	}
}

📋 コピーfmt.Println("最終的なctx.Err():", ctx.Err())

💡 設計思想・なぜこう書くのか

🌐 他言語との比較

Goが際立っているのは、キャンセル・タイムアウトも値の伝播も、すべてctxという1つの明示的な引数を通じて行われる点です。他言語のThreadLocalやContextVarのような「暗黙的にどこかに紐づいている状態」を持たないため、ある関数が何に依存しているかは、その関数のシグネチャ（ctx context.Contextを受け取っているかどうか）を見れば分かります。この一貫した明示性が、Goのコードレビューやテストのしやすさに直結しています。

🏆 実務での使いどころ

• 外部API呼び出しの制限時間管理: 複数の外部サービスに並行してリクエストを送る際、context.WithTimeoutで全体の制限時間を1箇所に設定するだけで、遅い一部のサービスを待たずに「間に合った分だけ」で処理を進める、といったSLA遵守の実装が簡潔に書けます
• 分散トレーシングのトレースID伝播: context.WithValueでトレースID・スパンIDをcontextに埋め込み、複数のサービス呼び出し・goroutineをまたいで一貫したトレースID付きログを出力する（OpenTelemetryなどの分散トレーシングライブラリは内部でこのパターンを使っています）
• 認証済みユーザー情報の伝播: HTTPミドルウェアで認証を行った後、認証済みユーザー情報をcontext.WithValueでリクエストのcontextに埋め込み、後段のハンドラ・サービス層でctxから取り出して使う（Phase 2後半のWeb API実装テーマで本格的に扱います）
• バッチ処理・並行取得の部分的タイムアウト: 今回のように「複数のリソースを並行取得し、制限時間内に取れた分だけ使う」というパターンは、検索結果の複数ソース統合・レコメンデーションの複数モデル呼び出しなど、実務のあらゆる箇所に現れます

⚠️ よくある誤解・ミス

🚀 次のステップ

• 発展: context.WithDeadline(parentCtx, time.Now().Add(400*time.Millisecond))に書き換えても、今回のWithTimeout版と全く同じ結果になることを実際に確認してみましょう。さらに、複数のサービスから伝わってきた締切時刻（time.Time）をそのままWithDeadlineに渡すケースを想像すると、WithTimeoutとWithDeadlineの使い分けがより実感できます
• 次回予告: Day 018 — エラーラッピング（errors.Is/As/Unwrap）（実装問題）。今回fmt.Errorfで%wを使ってラップしたctx.Err()を、errors.Isで正確に判定する方法を学びます

🎯 自己評価

自分の回答

気づき・メモ