package main

import (
	"context"
	"fmt"
)

// context.WithValue() - 值传递

// 类型安全的key定义（重要！）
type contextKey string

const (
	requestIDKey contextKey = "requestID"
	userIDKey    contextKey = "userID"
	authTokenKey contextKey = "authToken"
)

func withValueExample() {
	fmt.Println("=== WithValue 示例 ===")

	// 创建带值的Context链
	ctx := context.Background()
	ctx = context.WithValue(ctx, requestIDKey, "req-12345")
	ctx = context.WithValue(ctx, userIDKey, "user-67890")
	ctx = context.WithValue(ctx, authTokenKey, "token-abcde")

	// 传递Context给多个处理函数
	processRequest(ctx)
}

func processRequest(ctx context.Context) {
	// 从Context中获取值
	requestID := ctx.Value(requestIDKey).(string)
	userID := ctx.Value(userIDKey).(string)

	fmt.Printf("处理请求: requestID=%s, userID=%s\n", requestID, userID)

	// 调用子函数
	callDatabase(ctx)
}

func callDatabase(ctx context.Context) {
	// 在调用链中传递Context
	authToken := ctx.Value(authTokenKey).(string)
	fmt.Printf("数据库调用使用token: %s\n", authToken)
}

func main() {
	withValueExample()
}