// Couldn't find an Active RM
/// <exception cref="System.Exception"/>
private static HAServiceProtocol.HAServiceState GetHAState(YarnConfiguration yarnConf
)
{
HAServiceTarget haServiceTarget;
int rpcTimeoutForChecks = yarnConf.GetInt(CommonConfigurationKeys.HaFcCliCheckTimeoutKey
, CommonConfigurationKeys.HaFcCliCheckTimeoutDefault);
yarnConf.Set(CommonConfigurationKeys.HadoopSecurityServiceUserNameKey, yarnConf.Get
(YarnConfiguration.RmPrincipal, string.Empty));
haServiceTarget = new RMHAServiceTarget(yarnConf);
HAServiceProtocol proto = haServiceTarget.GetProxy(yarnConf, rpcTimeoutForChecks);
HAServiceProtocol.HAServiceState haState = proto.GetServiceStatus().GetState();
return(haState);
}
/// <exception cref="Com.Google.Protobuf.ServiceException"/>
public virtual HAServiceProtocolProtos.GetServiceStatusResponseProto GetServiceStatus
(RpcController controller, HAServiceProtocolProtos.GetServiceStatusRequestProto
request)
{
HAServiceStatus s;
try
{
s = server.GetServiceStatus();
}
catch (IOException e)
{
throw new ServiceException(e);
}
HAServiceProtocolProtos.HAServiceStateProto retState;
switch (s.GetState())
{
case HAServiceProtocol.HAServiceState.Active:
{
retState = HAServiceProtocolProtos.HAServiceStateProto.Active;
break;
}
case HAServiceProtocol.HAServiceState.Standby:
{
retState = HAServiceProtocolProtos.HAServiceStateProto.Standby;
break;
}
case HAServiceProtocol.HAServiceState.Initializing:
default:
{
retState = HAServiceProtocolProtos.HAServiceStateProto.Initializing;
break;
}
}
HAServiceProtocolProtos.GetServiceStatusResponseProto.Builder ret = HAServiceProtocolProtos.GetServiceStatusResponseProto
.NewBuilder().SetState(retState).SetReadyToBecomeActive(s.IsReadyToBecomeActive(
));
if (!s.IsReadyToBecomeActive())
{
ret.SetNotReadyReason(s.GetNotReadyReason());
}
return((HAServiceProtocolProtos.GetServiceStatusResponseProto)ret.Build());
}
public virtual void Configure()
{
remoteAdminServiceAccessed = false;
admin = Org.Mockito.Mockito.Mock <ResourceManagerAdministrationProtocol>();
Org.Mockito.Mockito.When(admin.AddToClusterNodeLabels(Matchers.Any <AddToClusterNodeLabelsRequest
>())).ThenAnswer(new _Answer_79(this));
haadmin = Org.Mockito.Mockito.Mock <HAServiceProtocol>();
Org.Mockito.Mockito.When(haadmin.GetServiceStatus()).ThenReturn(new HAServiceStatus
(HAServiceProtocol.HAServiceState.Initializing));
HAServiceTarget haServiceTarget = Org.Mockito.Mockito.Mock <HAServiceTarget>();
Org.Mockito.Mockito.When(haServiceTarget.GetProxy(Matchers.Any <Configuration>(),
Matchers.AnyInt())).ThenReturn(haadmin);
rmAdminCLI = new _RMAdminCLI_96(this, haServiceTarget, new Configuration());
InitDummyNodeLabelsManager();
rmAdminCLI.localNodeLabelsManager = dummyNodeLabelsManager;
YarnConfiguration conf = new YarnConfiguration();
conf.SetBoolean(YarnConfiguration.RmHaEnabled, true);
rmAdminCLIWithHAEnabled = new _RMAdminCLI_113(this, haServiceTarget, conf);
}