Method Subscribe

graphql/admin/http.go:141–212

(
	ctx context.Context,
	document,
	operationName string,
	variableValues map[string]interface{})

}

func (gs *graphqlSubscription) Subscribe(
	ctx context.Context,
	document,
	operationName string,
	variableValues map[string]interface{}) (<-chan interface{}, error) {

	reqHeader := http.Header{}
	// library (graphql-transport-ws) passes the headers which are part of the INIT payload to us
	// in the context. We are extracting those headers and passing them along.
	headerPayload, _ := ctx.Value("Header").(json.RawMessage)
	if len(headerPayload) > 0 {
		headers := make(map[string]interface{})
		if err := json.Unmarshal(headerPayload, &headers); err != nil {
			return nil, err
		}

		for k, v := range headers {
			if vStr, ok := v.(string); ok {
				reqHeader.Set(k, vStr)
			}
		}
	}

	// Earlier the graphql-transport-ws library was ignoring the http headers in the request.
	// The library was relying upon the information present in the request payload. This was
	// blocker for the cloud team because the only control cloud has is over the HTTP headers.
	// This fix ensures that we are setting the request headers if not provided in the payload.
	httpHeaders, _ := ctx.Value("RequestHeader").(http.Header)
	if len(httpHeaders) > 0 {
		for k := range httpHeaders {
			if len(strings.TrimSpace(reqHeader.Get(k))) == 0 {
				reqHeader.Set(k, httpHeaders.Get(k))
			}
		}
	}

	req := &schema.Request{
		OperationName: operationName,
		Query:         document,
		Variables:     variableValues,
		Header:        reqHeader,
	}

	audit.AuditWebSockets(ctx, req)
	namespace := x.ExtractNamespaceHTTP(&http.Request{Header: reqHeader})
	glog.V(2).Infof("namespace: %d. Got GraphQL request over HTTP.", namespace)
	// first load the schema, then do anything else
	if err := LazyLoadSchema(namespace); err != nil {
		return nil, err
	}
	if err := gs.isValid(namespace); err != nil {
		glog.Errorf("namespace: %d. graphqlSubscription not initialized: %s", namespace, err)
		return nil, errors.New(resolve.ErrInternal)
	}

	gs.graphqlHandler.pollerMux.RLock()
	poller := gs.graphqlHandler.poller[namespace]
	gs.graphqlHandler.pollerMux.RUnlock()