3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-06-25 22:18:23 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix context keys

Browse source
This commit is contained in:
Caleb Doxsey 2022-07-26 13:34:43 -06:00
parent 7adda15c75
commit e6fe3b0ceb
3 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pkg/storage/querier.go
View file

@ -33,11 +33,11 @@ func (nilQuerier) Query(ctx context.Context, in *databroker.QueryRequest, opts .
return nil, status.Error(codes.NotFound, "not found")
}
var querierKey struct{}
type querierKey struct{}
// GetQuerier gets the databroker Querier from the context.
func GetQuerier(ctx context.Context) Querier {
q, ok := ctx.Value(querierKey).(Querier)
q, ok := ctx.Value(querierKey{}).(Querier)
if !ok {
q = nilQuerier{}
}
@ -46,7 +46,7 @@ func GetQuerier(ctx context.Context) Querier {
// WithQuerier sets the databroker Querier on a context.
func WithQuerier(ctx context.Context, querier Querier) context.Context {
return context.WithValue(ctx, querierKey, querier)
return context.WithValue(ctx, querierKey{}, querier)
}
type staticQuerier struct {