Examples with ClusterRetrieveException org.apache.flink.client.deployment.ClusterRetrieveException used on opensource projects

Example 1 with ClusterRetrieveException

use of org.apache.flink.client.deployment.ClusterRetrieveException in project flink by apache.

the class YarnClusterDescriptor method retrieve.

// -------------------------------------------------------------
// ClusterClient overrides
// -------------------------------------------------------------
@Override
public ClusterClientProvider<ApplicationId> retrieve(ApplicationId applicationId) throws ClusterRetrieveException {
    try {
        // check if required Hadoop environment variables are set. If not, warn user
        if (System.getenv("HADOOP_CONF_DIR") == null && System.getenv("YARN_CONF_DIR") == null) {
            LOG.warn("Neither the HADOOP_CONF_DIR nor the YARN_CONF_DIR environment variable is set." + "The Flink YARN Client needs one of these to be set to properly load the Hadoop " + "configuration for accessing YARN.");
        }
        final ApplicationReport report = yarnClient.getApplicationReport(applicationId);
        if (report.getFinalApplicationStatus() != FinalApplicationStatus.UNDEFINED) {
            // Flink cluster is not running anymore
            LOG.error("The application {} doesn't run anymore. It has previously completed with final status: {}", applicationId, report.getFinalApplicationStatus());
            throw new RuntimeException("The Yarn application " + applicationId + " doesn't run anymore.");
        }
        setClusterEntrypointInfoToConfig(report);
        return () -> {
            try {
                return new RestClusterClient<>(flinkConfiguration, report.getApplicationId());
            } catch (Exception e) {
                throw new RuntimeException("Couldn't retrieve Yarn cluster", e);
            }
        };
    } catch (Exception e) {
        throw new ClusterRetrieveException("Couldn't retrieve Yarn cluster", e);
    }
}