diff --git a/http/graph/resolver/process.resolvers.go b/http/graph/resolver/process.resolvers.go index deab6797..36edeca9 100644 --- a/http/graph/resolver/process.resolvers.go +++ b/http/graph/resolver/process.resolvers.go @@ -11,7 +11,7 @@ import ( // Processes is the resolver for the processes field. func (r *queryResolver) Processes(ctx context.Context, idpattern *string, refpattern *string, group *string) ([]*models.Process, error) { - user, _ := ctx.Value("user").(string) + user, _ := ctx.Value(GraphKey("user")).(string) ids := r.Restream.GetProcessIDs(*idpattern, *refpattern, user, *group) procs := []*models.Process{} @@ -30,14 +30,14 @@ func (r *queryResolver) Processes(ctx context.Context, idpattern *string, refpat // Process is the resolver for the process field. func (r *queryResolver) Process(ctx context.Context, id string, group *string) (*models.Process, error) { - user, _ := ctx.Value("user").(string) + user, _ := ctx.Value(GraphKey("user")).(string) return r.getProcess(id, user, *group) } // Probe is the resolver for the probe field. func (r *queryResolver) Probe(ctx context.Context, id string, group *string) (*models.Probe, error) { - user, _ := ctx.Value("user").(string) + user, _ := ctx.Value(GraphKey("user")).(string) probe := r.Restream.Probe(id, user, *group) diff --git a/http/graph/resolver/resolver.go b/http/graph/resolver/resolver.go index c833a2cb..342b768f 100644 --- a/http/graph/resolver/resolver.go +++ b/http/graph/resolver/resolver.go @@ -86,3 +86,5 @@ func (r *queryResolver) playoutRequest(method, addr, path, contentType string, d return data, nil } + +type GraphKey string diff --git a/http/handler/api/graph.go b/http/handler/api/graph.go index 64bd5440..35013e2d 100644 --- a/http/handler/api/graph.go +++ b/http/handler/api/graph.go @@ -47,7 +47,7 @@ func (g *GraphHandler) Query(c echo.Context) error { user, _ := c.Get("user").(string) r := c.Request() - ctx := context.WithValue(r.Context(), "user", user) + ctx := context.WithValue(r.Context(), resolver.GraphKey("user"), user) g.queryHandler.ServeHTTP(c.Response(), r.WithContext(ctx))