/
githubmirror
/
grafana
Обзор
Документация
Войти
/
githubmirror
/
grafana
Код
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
pkg/services/grpcserver/context/handler.go
47 строк
1 KB
Todd Treece
GRPC Server: Add signedInUser to context (#56059)
04 окт 2022, 23:31
Не верифицирован
04 окт 2022, 23:31
4163f31
Код
Авторство
О чём код?
package grpccontext import ( "context" "github.com/grafana/grafana/pkg/infra/log" "github.com/grafana/grafana/pkg/infra/tracing" "github.com/grafana/grafana/pkg/services/user" ) type ContextHandler interface { SetUser(context.Context, *user.SignedInUser) context.Context GetUser(context.Context) *user.SignedInUser } func ProvideContextHandler(tracer tracing.Tracer) ContextHandler { return &contextHandler{ tracer: tracer, } } type contextHandler struct { tracer tracing.Tracer } func (c *contextHandler) fromContext(ctx context.Context) *GRPCServerContext { grpcContext := FromContext(ctx) if grpcContext != nil { return grpcContext } return &GRPCServerContext{ Tracer: c.tracer, Logger: log.New("grpc-server-context"), } } func (c *contextHandler) SetUser(ctx context.Context, user *user.SignedInUser) context.Context { grpcContext := c.fromContext(ctx) grpcContext.SignedInUser = user return context.WithValue(ctx, grpcContextKey{}, grpcContext) } func (c *contextHandler) GetUser(ctx context.Context) *user.SignedInUser { return c.fromContext(ctx).SignedInUser }