|
@@ -0,0 +1,141 @@
|
|
|
|
+/**
|
|
|
|
+* Licensed to the Apache Software Foundation (ASF) under one
|
|
|
|
+* or more contributor license agreements. See the NOTICE file
|
|
|
|
+* distributed with this work for additional information
|
|
|
|
+* regarding copyright ownership. The ASF licenses this file
|
|
|
|
+* to you under the Apache License, Version 2.0 (the
|
|
|
|
+* "License"); you may not use this file except in compliance
|
|
|
|
+* with the License. You may obtain a copy of the License at
|
|
|
|
+*
|
|
|
|
+* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
+*
|
|
|
|
+* Unless required by applicable law or agreed to in writing, software
|
|
|
|
+* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
+* See the License for the specific language governing permissions and
|
|
|
|
+* limitations under the License.
|
|
|
|
+*/
|
|
|
|
+package org.apache.hadoop.yarn.webapp.util;
|
|
|
|
+
|
|
|
|
+import java.net.InetAddress;
|
|
|
|
+import java.net.InetSocketAddress;
|
|
|
|
+import java.net.UnknownHostException;
|
|
|
|
+
|
|
|
|
+import org.apache.hadoop.classification.InterfaceAudience.Private;
|
|
|
|
+import org.apache.hadoop.classification.InterfaceStability.Evolving;
|
|
|
|
+import org.apache.hadoop.conf.Configuration;
|
|
|
|
+import org.apache.hadoop.http.HttpConfig;
|
|
|
|
+import org.apache.hadoop.net.NetUtils;
|
|
|
|
+import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
|
|
+
|
|
|
|
+import com.google.common.base.Joiner;
|
|
|
|
+
|
|
|
|
+@Private
|
|
|
|
+@Evolving
|
|
|
|
+public class WebAppUtils {
|
|
|
|
+ private static final Joiner JOINER = Joiner.on("");
|
|
|
|
+
|
|
|
|
+ public static void setRMWebAppPort(Configuration conf, int port) {
|
|
|
|
+ String hostname = getRMWebAppURLWithoutScheme(conf);
|
|
|
|
+ hostname =
|
|
|
|
+ (hostname.contains(":")) ? hostname.substring(0, hostname.indexOf(":"))
|
|
|
|
+ : hostname;
|
|
|
|
+ setRMWebAppHostnameAndPort(conf, hostname, port);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static void setRMWebAppHostnameAndPort(Configuration conf,
|
|
|
|
+ String hostname, int port) {
|
|
|
|
+ String resolvedAddress = hostname + ":" + port;
|
|
|
|
+ if (HttpConfig.isSecure()) {
|
|
|
|
+ conf.set(YarnConfiguration.RM_WEBAPP_HTTPS_ADDRESS, resolvedAddress);
|
|
|
|
+ } else {
|
|
|
|
+ conf.set(YarnConfiguration.RM_WEBAPP_ADDRESS, resolvedAddress);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static void setNMWebAppHostNameAndPort(Configuration conf,
|
|
|
|
+ String hostName, int port) {
|
|
|
|
+ if (HttpConfig.isSecure()) {
|
|
|
|
+ conf.set(YarnConfiguration.NM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ hostName + ":" + port);
|
|
|
|
+ } else {
|
|
|
|
+ conf.set(YarnConfiguration.NM_WEBAPP_ADDRESS,
|
|
|
|
+ hostName + ":" + port);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getRMWebAppURLWithScheme(Configuration conf) {
|
|
|
|
+ return JOINER.join(HttpConfig.getSchemePrefix(),
|
|
|
|
+ HttpConfig.isSecure() ? conf.get(
|
|
|
|
+ YarnConfiguration.RM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_HTTPS_ADDRESS) : conf.get(
|
|
|
|
+ YarnConfiguration.RM_WEBAPP_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_ADDRESS));
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getRMWebAppURLWithoutScheme(Configuration conf) {
|
|
|
|
+ if (HttpConfig.isSecure()) {
|
|
|
|
+ return conf.get(YarnConfiguration.RM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_HTTPS_ADDRESS);
|
|
|
|
+ }else {
|
|
|
|
+ return conf.get(YarnConfiguration.RM_WEBAPP_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_ADDRESS);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getProxyHostAndPort(Configuration conf) {
|
|
|
|
+ String addr = conf.get(YarnConfiguration.PROXY_ADDRESS);
|
|
|
|
+ if(addr == null || addr.isEmpty()) {
|
|
|
|
+ addr = getResolvedRMWebAppURLWithoutScheme(conf);
|
|
|
|
+ }
|
|
|
|
+ return addr;
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getResolvedRMWebAppURLWithScheme(Configuration conf) {
|
|
|
|
+ return HttpConfig.getSchemePrefix()
|
|
|
|
+ + getResolvedRMWebAppURLWithoutScheme(conf);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getResolvedRMWebAppURLWithoutScheme(Configuration conf) {
|
|
|
|
+ InetSocketAddress address = null;
|
|
|
|
+ if (HttpConfig.isSecure()) {
|
|
|
|
+ address =
|
|
|
|
+ conf.getSocketAddr(YarnConfiguration.RM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_HTTPS_PORT);
|
|
|
|
+ } else {
|
|
|
|
+ address =
|
|
|
|
+ conf.getSocketAddr(YarnConfiguration.RM_WEBAPP_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_RM_WEBAPP_PORT);
|
|
|
|
+ }
|
|
|
|
+ address = NetUtils.getConnectAddress(address);
|
|
|
|
+ StringBuffer sb = new StringBuffer();
|
|
|
|
+ InetAddress resolved = address.getAddress();
|
|
|
|
+ if (resolved == null || resolved.isAnyLocalAddress() ||
|
|
|
|
+ resolved.isLoopbackAddress()) {
|
|
|
|
+ String lh = address.getHostName();
|
|
|
|
+ try {
|
|
|
|
+ lh = InetAddress.getLocalHost().getCanonicalHostName();
|
|
|
|
+ } catch (UnknownHostException e) {
|
|
|
|
+ //Ignore and fallback.
|
|
|
|
+ }
|
|
|
|
+ sb.append(lh);
|
|
|
|
+ } else {
|
|
|
|
+ sb.append(address.getHostName());
|
|
|
|
+ }
|
|
|
|
+ sb.append(":").append(address.getPort());
|
|
|
|
+ return sb.toString();
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ public static String getNMWebAppURLWithoutScheme(Configuration conf) {
|
|
|
|
+ if (HttpConfig.isSecure()) {
|
|
|
|
+ return conf.get(YarnConfiguration.NM_WEBAPP_HTTPS_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_NM_WEBAPP_HTTPS_ADDRESS);
|
|
|
|
+ } else {
|
|
|
|
+ return conf.get(YarnConfiguration.NM_WEBAPP_ADDRESS,
|
|
|
|
+ YarnConfiguration.DEFAULT_NM_WEBAPP_ADDRESS);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ }
|
|
|
|
+}
|