001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018
019package org.apache.hadoop.yarn.client;
020
021import java.io.EOFException;
022import java.net.ConnectException;
023import java.net.InetSocketAddress;
024import java.net.NoRouteToHostException;
025import java.net.SocketException;
026import java.net.UnknownHostException;
027import java.security.PrivilegedAction;
028import java.util.HashMap;
029import java.util.Map;
030import java.util.concurrent.TimeUnit;
031
032import org.apache.hadoop.classification.InterfaceAudience.Public;
033import org.apache.hadoop.classification.InterfaceStability.Unstable;
034import org.apache.hadoop.conf.Configuration;
035import org.apache.hadoop.io.retry.RetryPolicies;
036import org.apache.hadoop.io.retry.RetryPolicy;
037import org.apache.hadoop.io.retry.RetryProxy;
038import org.apache.hadoop.ipc.RetriableException;
039import org.apache.hadoop.net.ConnectTimeoutException;
040import org.apache.hadoop.security.UserGroupInformation;
041import org.apache.hadoop.yarn.exceptions.NMNotYetReadyException;
042import org.apache.hadoop.yarn.ipc.YarnRPC;
043
044import com.google.common.base.Preconditions;
045
046@Public
047@Unstable
048public class ServerProxy {
049
050  protected static RetryPolicy createRetryPolicy(Configuration conf,
051      String maxWaitTimeStr, long defMaxWaitTime,
052      String connectRetryIntervalStr, long defRetryInterval) {
053    long maxWaitTime = conf.getLong(maxWaitTimeStr, defMaxWaitTime);
054    long retryIntervalMS =
055        conf.getLong(connectRetryIntervalStr, defRetryInterval);
056
057    Preconditions.checkArgument((maxWaitTime == -1 || maxWaitTime > 0),
058        "Invalid Configuration. " + maxWaitTimeStr + " should be either"
059            + " positive value or -1.");
060    Preconditions.checkArgument(retryIntervalMS > 0, "Invalid Configuration. "
061        + connectRetryIntervalStr + "should be a positive value.");
062
063    RetryPolicy retryPolicy = null;
064    if (maxWaitTime == -1) {
065      // wait forever.
066      retryPolicy = RetryPolicies.retryForeverWithFixedSleep(retryIntervalMS,
067          TimeUnit.MILLISECONDS);
068    } else {
069      retryPolicy =
070          RetryPolicies.retryUpToMaximumTimeWithFixedSleep(maxWaitTime,
071              retryIntervalMS, TimeUnit.MILLISECONDS);
072    }
073
074    Map<Class<? extends Exception>, RetryPolicy> exceptionToPolicyMap =
075        new HashMap<Class<? extends Exception>, RetryPolicy>();
076    exceptionToPolicyMap.put(EOFException.class, retryPolicy);
077    exceptionToPolicyMap.put(ConnectException.class, retryPolicy);
078    exceptionToPolicyMap.put(NoRouteToHostException.class, retryPolicy);
079    exceptionToPolicyMap.put(UnknownHostException.class, retryPolicy);
080    exceptionToPolicyMap.put(ConnectTimeoutException.class, retryPolicy);
081    exceptionToPolicyMap.put(RetriableException.class, retryPolicy);
082    exceptionToPolicyMap.put(SocketException.class, retryPolicy);
083    exceptionToPolicyMap.put(NMNotYetReadyException.class, retryPolicy);
084
085    return RetryPolicies.retryByException(RetryPolicies.TRY_ONCE_THEN_FAIL,
086      exceptionToPolicyMap);
087  }
088
089  @SuppressWarnings("unchecked")
090  protected static <T> T createRetriableProxy(final Configuration conf,
091      final Class<T> protocol, final UserGroupInformation user,
092      final YarnRPC rpc, final InetSocketAddress serverAddress,
093      RetryPolicy retryPolicy) {
094    T proxy = user.doAs(new PrivilegedAction<T>() {
095      @Override
096      public T run() {
097        return (T) rpc.getProxy(protocol, serverAddress, conf);
098      }
099    });
100    return (T) RetryProxy.create(protocol, proxy, retryPolicy);
101  }
102}