YARN-11691. Add HTTP POST method to yarn web proxy servlet.

apache · Apr 29, 2024 · 4a1758b · 4a1758b
1 parent daafc8a
commit 4a1758b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/...er-web-proxy/src/main/java/org/apache/hadoop/yarn/server/webproxy/WebAppProxyServlet.java b/...er-web-proxy/src/main/java/org/apache/hadoop/yarn/server/webproxy/WebAppProxyServlet.java
@@ -69,6 +69,7 @@
 import org.apache.http.client.HttpClient;
 import org.apache.http.client.config.RequestConfig;
 import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
 import org.apache.http.client.methods.HttpPut;
 import org.apache.http.client.methods.HttpRequestBase;
 import org.apache.http.client.utils.URLEncodedUtils;
@@ -295,6 +296,19 @@ private void proxyLink(final HttpServletRequest req,
       }
 
       ((HttpPut) base).setEntity(new StringEntity(sb.toString()));
+    } else if (method.equals(HTTP.POST)){
+      base = new HttpPost(link);
+
+      StringBuilder sb = new StringBuilder();
+      BufferedReader reader =
+              new BufferedReader(
+                      new InputStreamReader(req.getInputStream(), "UTF-8"));
+      String line;
+      while ((line = reader.readLine()) != null) {
+        sb.append(line);
+      }
+
+      ((HttpPost) base).setEntity(new StringEntity(sb.toString()));
     } else {
       resp.setStatus(HttpServletResponse.SC_METHOD_NOT_ALLOWED);
       return;