Function writeFormattedContentForRequest

modules/querier/http.go:405–446

(w http.ResponseWriter, req *http.Request, m proto.Message, span oteltrace.Span)

}

func writeFormattedContentForRequest(w http.ResponseWriter, req *http.Request, m proto.Message, span oteltrace.Span) {
	// Check for both explicit nil and typed nil pointers (e.g., (*tempopb.SearchResponse)(nil))
	// A typed nil pointer is not equal to nil when passed as an interface, so we need reflection
	if m == nil || (reflect.ValueOf(m).Kind() == reflect.Ptr && reflect.ValueOf(m).IsNil()) {
		http.Error(w, "internal error: nil response", http.StatusInternalServerError)
		if span != nil {
			span.RecordError(fmt.Errorf("nil response in writeFormattedContentForRequest"))
		}
		return
	}

	switch req.Header.Get(api.HeaderAccept) {
	case api.HeaderAcceptProtobuf:
		b, err := proto.Marshal(m)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}

		w.Header().Set(api.HeaderContentType, api.HeaderAcceptProtobuf)
		_, err = w.Write(b)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		if span != nil {
			span.SetAttributes(attribute.String("contentType", api.HeaderAcceptProtobuf))
		}

	default:
		w.Header().Set(api.HeaderContentType, api.HeaderAcceptJSON)
		err := new(jsonpb.Marshaler).Marshal(w, m)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		if span != nil {
			span.SetAttributes(attribute.String("contentType", api.HeaderAcceptJSON))
		}

	}
}