```
package gctx

import (
        "context"
        "fmt"
        "reflect"
)

type body[T any] struct {
        Value T
}

type typeKey[T any] struct {
}

func Put[T any](ctx context.Context, value T) context.Context {
        key := typeKey[body[T]]{}
        return context.WithValue(ctx, key, body[T]{Value: value})
}

func Get[T any](ctx context.Context) (T, error) {
        key := typeKey[body[T]]{}
        value, ok := ctx.Value(key).(body[T])
        if !ok {
                var zero T
                return zero, errors.New("not found")
        }
        return value.Value, nil
}

```
举报· 31 次点击
登录 注册 站外分享
快来抢沙发
0 条回复  
返回顶部