컨텍스트(context) 패키지 활용

Go code

package main

import (
	"context"
	"fmt"
	"math/rand"
	"time"
)

// 1. 타임아웃 컨텍스트
func fetchData(ctx context.Context, id int) (string, error) {
	// 실제 작업을 시뮬레이션하는 채널
	ch := make(chan string, 1)
	go func() {
		time.Sleep(time.Duration(rand.Intn(300)) * time.Millisecond)
		ch <- fmt.Sprintf("데이터-%d", id)
	}()

	select {
	case data := <-ch:
		return data, nil
	case <-ctx.Done():
		return "", ctx.Err() // context.DeadlineExceeded 또는 Canceled
	}
}

// 2. 취소 전파
func processItems(ctx context.Context, items []int) []string {
	results := make([]string, 0, len(items))
	for _, item := range items {
		select {
		case <-ctx.Done():
			fmt.Println("  작업 취소됨:", ctx.Err())
			return results
		default:
			data, err := fetchData(ctx, item)
			if err != nil {
				fmt.Printf("  항목 %d 실패: %v\n", item, err)
				continue
			}
			results = append(results, data)
			fmt.Printf("  항목 %d 완료: %s\n", item, data)
		}
	}
	return results
}

// 3. 값 전달 (요청 ID 등)
type contextKey string

const requestIDKey contextKey = "requestID"

func withRequestID(ctx context.Context, id string) context.Context {
	return context.WithValue(ctx, requestIDKey, id)
}

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

func main() {
	// 타임아웃 컨텍스트: 200ms 제한
	fmt.Println("=== 타임아웃 테스트 ===")
	ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond)
	defer cancel()
	results := processItems(ctx, []int{1, 2, 3, 4, 5})
	fmt.Printf("완료된 항목: %d개\n\n", len(results))

	// 수동 취소
	fmt.Println("=== 수동 취소 테스트 ===")
	ctx2, cancel2 := context.WithCancel(context.Background())
	go func() {
		time.Sleep(100 * time.Millisecond)
		cancel2() // 100ms 후 취소
	}()
	processItems(ctx2, []int{10, 20, 30})

	// 값 전달
	fmt.Println("\n=== 요청 ID 전달 ===")
	ctx3 := withRequestID(context.Background(), "req-abc-123")
	fmt.Println("요청 ID:", getRequestID(ctx3))
}