From cc3af80120a2d0fbd8396d9ef7aadcab929b5e1d Mon Sep 17 00:00:00 2001 From: qydysky Date: Wed, 8 Nov 2023 00:05:00 +0800 Subject: [PATCH] 1 --- ctx/Ctx.go | 12 +++++------- ctx/Ctx_test.go | 6 +++--- 2 files changed, 8 insertions(+), 10 deletions(-) diff --git a/ctx/Ctx.go b/ctx/Ctx.go index dc84069..476632d 100644 --- a/ctx/Ctx.go +++ b/ctx/Ctx.go @@ -105,22 +105,20 @@ type Value[T any] struct { data T } -func (t *Value[T]) get() T { +func (t *Value[T]) Get() T { return t.data } -func (t *Value[T]) set(data T) { +func (t *Value[T]) Set(data T) { t.data = data } -func (t *Value[T]) linkCtx(ctx context.Context) context.Context { +func (t *Value[T]) LinkCtx(ctx context.Context) context.Context { return context.WithValue(ctx, t, t) } -func putVal[T any](ctx context.Context, key *Value[T], v T) { +func PutVal[T any](ctx context.Context, key *Value[T], v T) { if pt, ok := ctx.Value(key).(*Value[T]); ok { - pt.set(v) - } else { - panic("") + pt.Set(v) } } diff --git a/ctx/Ctx_test.go b/ctx/Ctx_test.go index 8b9706a..dd3a24d 100644 --- a/ctx/Ctx_test.go +++ b/ctx/Ctx_test.go @@ -60,9 +60,9 @@ func TestMain2(t *testing.T) { func TestMain3(t *testing.T) { ctx := context.Background() val := Value[error]{} - ctx = val.linkCtx(ctx) - putVal(ctx, &val, errors.New("aaa")) - if val.get().Error() != "aaa" { + ctx = val.LinkCtx(ctx) + PutVal(ctx, &val, errors.New("aaa")) + if val.Get().Error() != "aaa" { t.Fatal() } } -- 2.39.2