Method serveQuery

engine/server/session.go:1326–1433

(w http.ResponseWriter, r *http.Request, client *daggerClient)

}

func (srv *Server) serveQuery(w http.ResponseWriter, r *http.Request, client *daggerClient) (rerr error) {
	sess := client.daggerSession
	sess.dagqlMu.Lock()
	if sess.dagqlClosing {
		sess.dagqlMu.Unlock()
		return gqlErr(errSessionClosing, http.StatusServiceUnavailable)
	}
	sess.dagqlInFlight++
	sess.dagqlMu.Unlock()
	defer func() {
		sess.dagqlMu.Lock()
		sess.dagqlInFlight--
		if sess.dagqlInFlight == 0 {
			sess.dagqlCond.Broadcast()
		}
		sess.dagqlMu.Unlock()
	}()

	ctx := sess.withClosingCancel(r.Context())

	// turn panics into graphql errors — must be set up before any code that
	// could panic (including ensureExtraModulesLoaded and schema loading).
	defer func() {
		if v := recover(); v != nil {
			bklog.G(ctx).Errorf("panic serving schema: %v %s", v, string(debug.Stack()))
			switch v := v.(type) {
			case error:
				rerr = gqlErr(v, http.StatusInternalServerError)
			case string:
				rerr = gqlErr(errors.New(v), http.StatusInternalServerError)
			default:
				rerr = gqlErr(errors.New("internal server error"), http.StatusInternalServerError)
			}
		}
	}()

	// only record telemetry if the request is traced, otherwise
	// we end up with orphaned spans in their own separate traces from tests etc.
	if trace.SpanContextFromContext(ctx).IsValid() {
		// create a span to record telemetry into the client's DB
		//
		// downstream components must use otel.SpanFromContext(ctx).TracerProvider()
		clientTracer := client.tracerProvider.Tracer(InstrumentationLibrary)
		var span trace.Span
		attrs := []attribute.KeyValue{
			attribute.Bool(telemetry.UIPassthroughAttr, true),
		}
		if engineID := client.clientMetadata.CloudScaleOutEngineID; engineID != "" {
			attrs = append(attrs, attribute.String(telemetry.EngineIDAttr, engineID))
		}
		ctx, span = clientTracer.Start(ctx,
			fmt.Sprintf("%s %s", r.Method, r.URL.Path),
			trace.WithAttributes(attrs...),
		)
		defer telemetry.EndWithCause(span, &rerr)
	}

	// install a logger+meter provider that records to the client's DB