Search in sources :

Example 1 with Info

use of com.microsoft.azure.hdinsight.common.MessageInfoType.Info in project azure-tools-for-java by Microsoft.

the class Session method deploy.

/*
     * Observable APIs, all IO operations
     */
public Observable<Session> deploy() {
    final Deployable deployDelegate = getDeploy();
    if (deployDelegate == null) {
        return Observable.just(this);
    }
    return Observable.from(getArtifactsToDeploy()).doOnNext(artifactPath -> ctrlSubject.onNext(new SparkLogLine(TOOL, Info, "Start uploading artifact " + artifactPath))).flatMap(artifactPath -> deployDelegate.deploy(new File(artifactPath), ctrlSubject)).doOnNext(uri -> ctrlSubject.onNext(new SparkLogLine(TOOL, Info, "Uploaded to " + uri))).toList().onErrorResumeNext(err -> {
        ctrlSubject.onNext(new SparkLogLine(TOOL, Warning, "Failed to upload artifact: " + err));
        ctrlSubject.onNext(new SparkLogLine(TOOL, Warning, "Try to start interactive session without those artifacts dependency..."));
        return Observable.empty();
    }).map(uploadedUris -> {
        this.createParameters.uploadedArtifactsUris.addAll(uploadedUris);
        return this;
    }).defaultIfEmpty(this);
}
Also used : java.util(java.util) Warning(com.microsoft.azure.hdinsight.common.MessageInfoType.Warning) SessionState(com.microsoft.azure.hdinsight.sdk.rest.livy.interactive.SessionState) SessionKind(com.microsoft.azure.hdinsight.sdk.rest.livy.interactive.SessionKind) SimpleImmutableEntry(java.util.AbstractMap.SimpleImmutableEntry) StringUtils(org.apache.commons.lang3.StringUtils) HDInsightLoader(com.microsoft.azure.hdinsight.common.HDInsightLoader) ILogger(com.microsoft.azure.hdinsight.common.logger.ILogger) Observable(rx.Observable) Info(com.microsoft.azure.hdinsight.common.MessageInfoType.Info) ImmutableList(com.google.common.collect.ImmutableList) ByteArrayInputStream(java.io.ByteArrayInputStream) Schedulers(rx.schedulers.Schedulers) Thread.sleep(java.lang.Thread.sleep) URI(java.net.URI) TOOL(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine.TOOL) PostSessions(com.microsoft.azure.hdinsight.sdk.rest.livy.interactive.api.PostSessions) Debug(com.microsoft.azure.hdinsight.common.MessageInfoType.Debug) MemorySize(com.microsoft.azure.hdinsight.sdk.common.livy.MemorySize) Nullable(com.microsoft.azuretools.azurecommons.helpers.Nullable) ImmutableMap(com.google.common.collect.ImmutableMap) Exceptions.propagate(rx.exceptions.Exceptions.propagate) HttpResponse(com.microsoft.azure.hdinsight.sdk.common.HttpResponse) StringEntity(org.apache.http.entity.StringEntity) SessionNotStartException(com.microsoft.azure.hdinsight.sdk.common.livy.interactive.exceptions.SessionNotStartException) Scheduler(rx.Scheduler) StatementExecutionError(com.microsoft.azure.hdinsight.sdk.common.livy.interactive.exceptions.StatementExecutionError) Collectors(java.util.stream.Collectors) ImmutablePair(org.apache.commons.lang3.tuple.ImmutablePair) File(java.io.File) StandardCharsets(java.nio.charset.StandardCharsets) HttpObservable(com.microsoft.azure.hdinsight.sdk.common.HttpObservable) TimeUnit(java.util.concurrent.TimeUnit) Stream(java.util.stream.Stream) AppInsightsClient(com.microsoft.azuretools.telemetry.AppInsightsClient) Closeable(java.io.Closeable) ApplicationNotStartException(com.microsoft.azure.hdinsight.sdk.common.livy.interactive.exceptions.ApplicationNotStartException) Deployable(com.microsoft.azure.hdinsight.spark.common.Deployable) SparkLogLine(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine) PublishSubject(rx.subjects.PublishSubject) ExceptionUtils(org.apache.commons.lang3.exception.ExceptionUtils) Deployable(com.microsoft.azure.hdinsight.spark.common.Deployable) File(java.io.File) SparkLogLine(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine)

Example 2 with Info

use of com.microsoft.azure.hdinsight.common.MessageInfoType.Info in project azure-tools-for-java by Microsoft.

the class CosmosServerlessSparkBatchJob method getSubmissionLog.

@NotNull
@Override
public Observable<SparkLogLine> getSubmissionLog() {
    final ImmutableSet<String> ignoredEmptyLines = ImmutableSet.of("stdout:", "stderr:", "yarn diagnostics:");
    final int GET_LIVY_URL_REPEAT_DELAY_MILLISECONDS = 3000;
    final int MAX_LOG_LINES_PER_REQUEST = 128;
    final int GET_LOG_REPEAT_DELAY_MILLISECONDS = 1000;
    // We need to repeatly call getSparkBatchJobRequest() since "livyServerApi" field does not always exist in response but
    // only appeared for a while and before that we can't get the "livyServerApi" field.
    ctrlInfo("Trying to get livy URL...");
    return getSparkBatchJobRequest().flatMap(batchResp -> getJobSchedulerState(batchResp) == null ? Observable.error(new IOException("Failed to get scheduler state of the job.")) : Observable.just(batchResp)).retryWhen(err -> err.zipWith(Observable.range(1, getRetriesMax()), (n, i) -> i).delay(getDelaySeconds(), TimeUnit.SECONDS)).repeatWhen(ob -> ob.delay(GET_LIVY_URL_REPEAT_DELAY_MILLISECONDS, TimeUnit.MILLISECONDS)).takeUntil(batchResp -> isJobEnded(batchResp) || StringUtils.isNotEmpty(getLivyAPI(batchResp))).filter(batchResp -> isJobEnded(batchResp) || StringUtils.isNotEmpty(getLivyAPI(batchResp))).flatMap(job -> {
        if (isJobEnded(job)) {
            final String jobState = getJobState(job);
            final String schedulerState = getJobSchedulerState(job);
            final String message = String.format("Job scheduler state: %s. Job running state: %s.", schedulerState, jobState);
            return Observable.just(new SparkLogLine(TOOL, Info, message));
        } else {
            return Observable.just(job).doOnNext(batchResp -> {
                ctrlInfo("Successfully get livy URL: " + batchResp.properties().livyServerAPI());
                ctrlInfo("Trying to retrieve livy submission logs...");
                // After test we find batch id won't be provided until the job is in running state
                // However, since only one spark job will be run on the cluster, the batch ID should always be 0
                setBatchId(0);
            }).map(batchResp -> batchResp.properties().livyServerAPI()).flatMap(livyUrl -> Observable.defer(() -> getSubmissionLogRequest(livyUrl, getBatchId(), getLogStartIndex(), MAX_LOG_LINES_PER_REQUEST)).map(sparkJobLog -> Optional.ofNullable(sparkJobLog.getLog()).orElse(Collections.<String>emptyList())).doOnNext(logs -> setLogStartIndex(getLogStartIndex() + logs.size())).map(logs -> logs.stream().filter(logLine -> !ignoredEmptyLines.contains(logLine.trim().toLowerCase())).collect(Collectors.toList())).flatMap(logLines -> {
                if (logLines.size() > 0) {
                    return Observable.just(Triple.of(logLines, SparkBatchJobState.STARTING.toString(), SchedulerState.SCHEDULED.toString()));
                } else {
                    return getSparkBatchJobRequest().map(batchResp -> Triple.of(logLines, getJobState(batchResp), getJobSchedulerState(batchResp)));
                }
            }).onErrorResumeNext(errors -> getSparkBatchJobRequest().delay(getDelaySeconds(), TimeUnit.SECONDS).map(batchResp -> Triple.of(new ArrayList<>(), getJobState(batchResp), getJobSchedulerState(batchResp)))).repeatWhen(ob -> ob.delay(GET_LOG_REPEAT_DELAY_MILLISECONDS, TimeUnit.MILLISECONDS)).takeUntil(logAndStatesTriple -> {
                String jobRunningState = logAndStatesTriple.getMiddle();
                String jobSchedulerState = logAndStatesTriple.getRight();
                return jobRunningState != null && !jobRunningState.equalsIgnoreCase(SparkBatchJobState.STARTING.toString()) || jobSchedulerState != null && jobSchedulerState.equalsIgnoreCase(SchedulerState.ENDED.toString());
            }).flatMap(logAndStatesTriple -> {
                final String jobRunningState = logAndStatesTriple.getMiddle();
                final String jobSchedulerState = logAndStatesTriple.getRight();
                if (jobRunningState != null && !jobRunningState.equalsIgnoreCase(SparkBatchJobState.STARTING.toString()) || jobSchedulerState != null && jobSchedulerState.equalsIgnoreCase(SchedulerState.ENDED.toString())) {
                    final String message = String.format("Job scheduler state: %s. Job running state: %s.", jobSchedulerState, jobRunningState);
                    return Observable.just(new SparkLogLine(TOOL, Info, message));
                } else {
                    return Observable.from(logAndStatesTriple.getLeft()).map(line -> new SparkLogLine(LIVY, Log, line));
                }
            }));
        }
    });
}
Also used : BasicNameValuePair(org.apache.http.message.BasicNameValuePair) AzureSparkServerlessAccount(com.microsoft.azure.hdinsight.sdk.common.azure.serverless.AzureSparkServerlessAccount) java.util(java.util) CreateSparkBatchJobParameters(com.microsoft.azure.hdinsight.sdk.rest.azure.serverless.spark.models.CreateSparkBatchJobParameters) NotNull(com.microsoft.azuretools.azurecommons.helpers.NotNull) ADLStoreClient(com.microsoft.azure.datalake.store.ADLStoreClient) StringUtils(org.apache.commons.lang3.StringUtils) Header(org.apache.http.Header) Observable(rx.Observable) Info(com.microsoft.azure.hdinsight.common.MessageInfoType.Info) URI(java.net.URI) Triple(org.apache.commons.lang3.tuple.Triple) TOOL(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine.TOOL) ImmutableSet(com.google.common.collect.ImmutableSet) Nullable(com.microsoft.azuretools.azurecommons.helpers.Nullable) IOException(java.io.IOException) SchedulerState(com.microsoft.azure.hdinsight.sdk.rest.azure.serverless.spark.models.SchedulerState) Collectors(java.util.stream.Collectors) File(java.io.File) TimeUnit(java.util.concurrent.TimeUnit) Log(com.microsoft.azure.hdinsight.common.MessageInfoType.Log) BasicHeader(org.apache.http.message.BasicHeader) LIVY(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine.LIVY) NameValuePair(org.apache.http.NameValuePair) SparkLogLine(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine) AzureHttpObservable(com.microsoft.azure.hdinsight.sdk.common.AzureHttpObservable) ExceptionUtils(org.apache.commons.lang3.exception.ExceptionUtils) IOException(java.io.IOException) SparkLogLine(com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine) NotNull(com.microsoft.azuretools.azurecommons.helpers.NotNull)

Aggregations

Info (com.microsoft.azure.hdinsight.common.MessageInfoType.Info)2 SparkLogLine (com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine)2 TOOL (com.microsoft.azure.hdinsight.spark.common.log.SparkLogLine.TOOL)2 Nullable (com.microsoft.azuretools.azurecommons.helpers.Nullable)2 File (java.io.File)2 URI (java.net.URI)2 java.util (java.util)2 TimeUnit (java.util.concurrent.TimeUnit)2 Collectors (java.util.stream.Collectors)2 StringUtils (org.apache.commons.lang3.StringUtils)2 ExceptionUtils (org.apache.commons.lang3.exception.ExceptionUtils)2 Observable (rx.Observable)2 ImmutableList (com.google.common.collect.ImmutableList)1 ImmutableMap (com.google.common.collect.ImmutableMap)1 ImmutableSet (com.google.common.collect.ImmutableSet)1 ADLStoreClient (com.microsoft.azure.datalake.store.ADLStoreClient)1 HDInsightLoader (com.microsoft.azure.hdinsight.common.HDInsightLoader)1 Debug (com.microsoft.azure.hdinsight.common.MessageInfoType.Debug)1 Log (com.microsoft.azure.hdinsight.common.MessageInfoType.Log)1 Warning (com.microsoft.azure.hdinsight.common.MessageInfoType.Warning)1