Skip to content

Instantly share code, notes, and snippets.

@ww9
Last active April 10, 2024 14:14
Show Gist options
  • Star 17 You must be signed in to star a gist
  • Fork 3 You must be signed in to fork a gist
  • Save ww9/4ad7b2ddfb94816a30dfdf2218e02f48 to your computer and use it in GitHub Desktop.
Save ww9/4ad7b2ddfb94816a30dfdf2218e02f48 to your computer and use it in GitHub Desktop.
Why use empty struct{} and not int or string for context.Value() key types in #go

Use struct{} as keys for context.Value() in Go

In the other file of this gist I detail why we should use struct{} as context.Value() keys and not int or string. Open gist to see main.go but the TLDR is:

	type key struct{}
	ctx = context.WithValue(ctx, key{}, "my value") // Set value
	myValue, ok := ctx.Value(key{}).(string) // Get value
package main
import (
"fmt"
"unsafe"
)
type StructKey1 struct{}
type StructKey2 struct{}
type IntKey1 int
type IntKey2 int
// Why use empty struct{} and not int as context.Value() key types:
func main() {
// First of all, plain string keys look simple but they do not protect context
// values from being tampered by other packages:
// Also they cause allocation, so you probably don't want to use them.
compareKeys("dontUseString", "dontUseString") // true
// Now let's see how struct{} types behave.
// Same struct{} keys are always equal:
compareKeys(StructKey1{}, StructKey1{}) // true
// Different struct{} keys are never equal even if they appear to be of the same struct{} type:
compareKeys(StructKey1{}, StructKey2{}) // false
// This also applies to int keys. Same type means equal:
compareKeys(IntKey1(0), IntKey1(0)) // true
// And different keys are never equal even though they have 0 value:
compareKeys(IntKey1(0), IntKey2(0)) // false
// However, unlike struct{}, an int typed key allows for mistakes with the value:
compareKeys(IntKey1(0), IntKey1(1)) // false
// To add to why you shouldn't use int typed keys, when seeing that the package define a
// key as int, the user might be tempted to pass a primitive 0 instead. Which doesn't work:
compareKeys(IntKey1(0), 0) // false
// Whereas with struct{} there's no ambiguity or margin for missuse.
// So if for any reason your package wants to allow other packages to set/get a
// context.Value() that's specific to your package, either export a struct{} key
// like `type StructKey1 struct{}` or, even better, export methods that interact
// with these values but keep the key unexported:
// func WithUserIP(ctx context.Context, userIP net.IP) context.Context {
// return context.WithValue(ctx, userIPKey{}, userIP)
// }
// func UserIPFromContext(ctx context.Context) (net.IP, bool) {
// userIP, ok := ctx.Value(userIPKey{}).(net.IP)
// return userIP, ok
// }
}
func compareKeys(key1 interface{}, key2 interface{}) {
type ifaceHdr struct {
T unsafe.Pointer
V unsafe.Pointer
}
fmt.Println("\nkey1 == key2?", key1 == key2)
fmt.Printf("key1 %+v\n", *(*ifaceHdr)(unsafe.Pointer(&key1)))
fmt.Printf("key2 %+v\n", *(*ifaceHdr)(unsafe.Pointer(&key2)))
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment