support for multipart/form-data (#14771)

Author: adameat

ydb/core/viewer/viewer_query.h

@@ -47,6 +47,7 @@ class TJsonQuery : public TViewerPipeClient {
  NHttp::THttpOutgoingResponsePtr HttpResponse;
  std::vector<bool> ResultSetHasColumns;
  bool ConcurrentResults = false;
+ TString ContentType;
 
 public:
  ESchemaType StringToSchemaType(const TString& schemaStr) {
@@ -129,8 +130,16 @@ class TJsonQuery : public TViewerPipeClient {
  if (Streaming) {
  NHttp::THeaders headers(HttpEvent->Get()->Request->Headers);
  TStringBuf accept = headers["Accept"];
- if (accept.find("multipart/x-mixed-replace") == TString::npos) {
- return TBase::ReplyAndPassAway(GetHTTPBADREQUEST("text/plain", "Multipart request must accept multipart/x-mixed-replace content"), "BadRequest");
+ auto posMixedReplace = accept.find("multipart/x-mixed-replace");
+ auto posFormData = accept.find("multipart/form-data");
+ auto posFirst = std::min(posMixedReplace, posFormData);
+ if (posFirst == TString::npos) {
+ return TBase::ReplyAndPassAway(GetHTTPBADREQUEST("text/plain", "Multipart request must accept multipart content-type"), "BadRequest");
+ }
+ if (posFirst == posMixedReplace) {
+ ContentType = "multipart/x-mixed-replace";
+ } else if (posFirst == posFormData) {
+ ContentType = "multipart/form-data";
  }
  }
  if (Streaming && QueryId.empty()) {
@@ -238,7 +247,7 @@ class TJsonQuery : public TViewerPipeClient {
  }
  CreateSessionResponse = MakeRequest<NKqp::TEvKqp::TEvCreateSessionResponse>(NKqp::MakeKqpProxyID(SelfId().NodeId()), event.release());
  if (Streaming) {
- HttpResponse = HttpEvent->Get()->Request->CreateResponseString(Viewer->GetChunkedHTTPOK(GetRequest(), "multipart/x-mixed-replace;boundary=boundary"));
+ HttpResponse = HttpEvent->Get()->Request->CreateResponseString(Viewer->GetChunkedHTTPOK(GetRequest(), ContentType + ";boundary=boundary"));
  Send(HttpEvent->Sender, new NHttp::TEvHttpProxy::TEvHttpOutgoingResponse(HttpResponse));
  }
  }
@@ -1022,6 +1031,10 @@ class TJsonQuery : public TViewerPipeClient {
  schema:
  type: object
  description: format depends on schema parameter
+ multipart/form-data:
+ schema:
+ type: object
+ description: format depends on schema parameter
  400:
  description: Bad Request
  403: