How to store a value of type context.CancelFunc using context.WithValue()?

huangapple
go评论119阅读模式
英文:

How to store a value of type context.CancelFunc using context.WithValue()?

问题

要在ctx中存储string类型的数据，需要使用类型定义来定义键和值，如下所示：

  1.     // 用于表示追踪ID的类型
  2.     type TraceID string
  3.     
  4.     // 用于键的类型定义。键是特定类型的，只有相同类型的值才能匹配。
  5.     type TraceIDKey int
  6.     
  7.     func main() {
  8.     
  9.     	// 为此请求创建一个追踪ID。
  10.     	traceID := TraceID("f47ac10b-58cc-0372-8567-0e02b2c3d479")
  11.     
  12.     	// 声明一个值为零的键，类型为TraceIDKey。
  13.     	const traceIDKey TraceIDKey = 0
  14.     
  15.     	// 使用键类型的零值将traceID值存储在上下文中。
  16.     	ctx := context.WithValue(context.Background(), traceIDKey, traceID)
  17.     
  18.     	// 从上下文值中检索该追踪ID值。
  19.     	if uuid, ok := ctx.Value(traceIDKey).(TraceID); ok {
  20.     		fmt.Println("TraceID:", uuid)
  21.     	}
  22.     
  23.     	// 不使用正确的键类型从上下文值中检索该追踪ID值。
  24.     	if _, ok := ctx.Value(0).(TraceID); !ok {
  25.     		fmt.Println("TraceID Not Found")
  26.     	}
  27.     }

如何使用context.WithValue() API存储context.CancelFunc类型的值？

英文:

To store string type data in ctx, type definitions need to be used for both key & value, as shown below:

  1.     // Sample program to show how to store and retrieve
  2.     // values from a context.
  3.     package main
  4.     
  5.     import (
  6.     	"context"
  7.     	"fmt"
  8.     )
  9.     
  10.     // TraceID represents the trace id.
  11.     type TraceID string
  12.     
  13.     // TraceIDKey is the type of value to use for the key. The key is
  14.     // type specific and only values of the same type will match.
  15.     type TraceIDKey int
  16.     
  17.     func main() {
  18.     
  19.     	// Create a traceID for this request.
  20.     	traceID := TraceID("f47ac10b-58cc-0372-8567-0e02b2c3d479")
  21.     
  22.     	// Declare a key with the value of zero of type userKey.
  23.     	const traceIDKey TraceIDKey = 0
  24.     
  25.     	// Store the traceID value inside the context with a value of
  26.     	// zero for the key type.
  27.     	ctx := context.WithValue(context.Background(), traceIDKey, traceID)
  28.     
  29.     	// Retrieve that traceID value from the Context value bag.
  30.     	if uuid, ok := ctx.Value(traceIDKey).(TraceID); ok {
  31.     		fmt.Println("TraceID:", uuid)
  32.     	}
  33.     
  34.     	// Retrieve that traceID value from the Context value bag not
  35.     	// using the proper key type.
  36.     	if _, ok := ctx.Value(0).(TraceID); !ok {
  37.     		fmt.Println("TraceID Not Found")
  38.     	}
  39.     }

<hr>

How to store a value of type context.CancelFunc using context.WithValue() api?

答案1

得分: 4

你部分正确。你应该使用自定义类型作为上下文键，而不是使用内置类型，以确保避免冲突。这个类型应该是未导出的，除非你希望其他包能够读取/写入你的上下文键。然而，值可以是任何你喜欢的类型，例如：

  1. package main
  3. import (
  4. 	"context"
  5. 	"fmt"
  6. )
  8. type contextKey int
  10. const (
  11. 	traceIDKey contextKey = iota
  12. 	aFunctionWhyNot
  13. )
  15. func main() {
  17. 	// 为这个请求创建一个 traceID。
  18. 	traceID := "f47ac10b-58cc-0372-8567-0e02b2c3d479"
  20. 	// 使用零值为键类型，在上下文中存储 traceID 值。
  21. 	ctx := context.WithValue(context.Background(), traceIDKey, traceID)
  23. 	// 从上下文值中检索 traceID 值。
  24. 	if uuid, ok := ctx.Value(traceIDKey).(string); ok {
  25. 		fmt.Println("TraceID:", uuid)
  26. 	}
  28. 	// 或者一个函数
  29. 	ctx = context.WithValue(ctx, aFunctionWhyNot, func() { fmt.Println("lol, I'm a function on a context") })
  31. 	// 可以调用它
  32. 	if f, ok := ctx.Value(aFunctionWhyNot).(func()); ok {
  33. 		f()
  34. 	}
  35. }

希望对你有帮助！

英文:

You are partially correct. You should use a bespoke type for your context key rather than the builtin types to make collisions impossible. This type should be unexported unless you want other packages to be able to read/write your context key. However the value can be anything you like, eg:

  1. package main
  3. import (
  4. 	&quot;context&quot;
  5. 	&quot;fmt&quot;
  6. )
  8. type contextKey int
  10. const (
  11. 	traceIDKey contextKey = iota
  12. 	aFunctionWhyNot
  13. )
  15. func main() {
  17. 	// Create a traceID for this request.
  18. 	traceID := &quot;f47ac10b-58cc-0372-8567-0e02b2c3d479&quot;
  20. 	// Store the traceID value inside the context with a value of
  21. 	// zero for the key type.
  22. 	ctx := context.WithValue(context.Background(), traceIDKey, traceID)
  24. 	// Retrieve that traceID value from the Context value bag.
  25. 	if uuid, ok := ctx.Value(traceIDKey).(string); ok {
  26. 		fmt.Println(&quot;TraceID:&quot;, uuid)
  27. 	}
  29. 	// Or a function
  30. 	ctx = context.WithValue(ctx, aFunctionWhyNot, func() { fmt.Println(&quot;lol, I&#39;m a function on a context&quot;) })
  32. 	// Call it maybe
  33. 	if f, ok := ctx.Value(aFunctionWhyNot).(func()); ok {
  34. 		f()
  35. 	}
  36. }

答案2

得分: 2

你可以像存储其他任何值一样，将函数存储在上下文中：

  1. type cancelFuncKeyType struct{}
  2. var cancelFuncKey = cancelFuncKeyType{}
  4. ...
  5. newctx := context.WithValue(oldctx, cancelFuncKey, cancelFunc)
  7. cFunc := newctx.Value(cancelFuncKey).(context.CancelFunc)
英文:

You can store a function in the context the same way you store any other value:

  1. type cancelFuncKeyType struct{}
  2. var cancelFuncKey =cancelFuncKeyType{}
  4. ...
  5. newctx:=context.WithValue(oldctx,cancelFuncKey,cancelFunc)
  8. cFunc:=newctx.Value(cancelFuncKey).(context.CancelFunc)

huangapple
  • 本文由 发表于 2021年10月6日 00:37:16
  • 转载请务必保留本文链接：https://go.coder-hub.com/69454253.html
  • go
  • go-context
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定