Method serveHTTPToClient

engine/server/session.go:1156–1257

(w http.ResponseWriter, r *http.Request, opts *ClientInitOpts)

const InstrumentationLibrary = "dagger.io/engine.server"

func (srv *Server) serveHTTPToClient(w http.ResponseWriter, r *http.Request, opts *ClientInitOpts) (rerr error) {
	if srv.isShuttingDown() {
		switch r.URL.Path {
		case engine.QueryEndpoint:
			return gqlErr(errServerShuttingDown, http.StatusServiceUnavailable)
		default:
			return httpErr(errServerShuttingDown, http.StatusServiceUnavailable)
		}
	}

	ctx := srv.withShutdownCancel(r.Context())

	ctx, cancel := context.WithCancelCause(ctx)
	defer cancel(fmt.Errorf("http request done for client %q", opts.ClientID))

	clientMetadata := opts.ClientMetadata
	ctx = engine.ContextWithClientMetadata(ctx, clientMetadata)

	// propagate span context and baggage from the client
	ctx = telemetry.Propagator.Extract(ctx, propagation.HeaderCarrier(r.Header))

	// Check if repeated telemetry is requested via baggage
	baggage := baggage.FromContext(ctx)
	if member := baggage.Member("repeat-telemetry"); member.Value() != "" {
		ctx = dagql.WithRepeatedTelemetry(ctx)
	}

	ctx = bklog.WithLogger(ctx, bklog.G(ctx).
		WithField("trace", trace.SpanContextFromContext(ctx).TraceID().String()).
		WithField("span", trace.SpanContextFromContext(ctx).SpanID().String()).
		WithField("client_id", clientMetadata.ClientID).
		WithField("client_hostname", clientMetadata.ClientHostname).
		WithField("session_id", clientMetadata.SessionID))
	ctx = slog.WithLogger(ctx, slog.FromContext(ctx).With(
		"client_id", clientMetadata.ClientID,
		"client_hostname", clientMetadata.ClientHostname,
		"session_id", clientMetadata.SessionID,
		"trace", trace.SpanContextFromContext(ctx).TraceID().String(),
		"span", trace.SpanContextFromContext(ctx).SpanID().String(),
	))
	ctx = dagql.ContextWithCache(ctx, srv.engineCache)

	// Debug https://github.com/dagger/dagger/issues/7592 by logging method and some headers, which
	// are checked by gqlgen's handler
	bklog.G(ctx).WithFields(logrus.Fields{
		"path":          r.URL.Path,
		"method":        r.Method,
		"upgradeHeader": r.Header.Get("Upgrade"),
		"contentType":   r.Header.Get("Content-Type"),
		"trace":         trace.SpanContextFromContext(ctx).TraceID().String(),
		"span":          trace.SpanContextFromContext(ctx).SpanID().String(),
	}).Debug("handling http request")

	mux := http.NewServeMux()
	switch r.URL.Path {
	case "/v1/traces", "/v1/logs", "/v1/metrics":
		// Just get the client if it exists, don't init it.
		client, err := srv.clientFromIDs(clientMetadata.SessionID, clientMetadata.ClientID)