-
Notifications
You must be signed in to change notification settings - Fork 343
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #15717 from cdapio/CDAP-21070-wrap-output-format
[CDAP-21070] Wrap input/output format methods to throw wrapped stage exception
- Loading branch information
Showing
12 changed files
with
502 additions
and
95 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
105 changes: 105 additions & 0 deletions
105
...spark-core-base/src/main/java/io/cdap/cdap/etl/spark/io/StageTrackingOutputCommitter.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,105 @@ | ||
| /* | ||
| * Copyright © 2024 Cask Data, Inc. | ||
| * | ||
| * Licensed 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 io.cdap.cdap.etl.spark.io; | ||
|
|
||
| import io.cdap.cdap.api.exception.WrappedStageException; | ||
| import org.apache.hadoop.mapreduce.JobContext; | ||
| import org.apache.hadoop.mapreduce.OutputCommitter; | ||
| import org.apache.hadoop.mapreduce.TaskAttemptContext; | ||
|
|
||
| /** | ||
| * A delegating output format that catches exceptions thrown during execution of a call | ||
| * and wraps them in a {@link WrappedStageException}. | ||
| * This class is primarily used to associate the exception with a specific stage name in a pipeline, | ||
| * helping in better debugging and error tracking. | ||
| * | ||
| * <p> | ||
| * The class delegates the actual calling operation to another | ||
| * {@link TrackingOutputCommitter} instance and ensures that any exceptions thrown are caught and | ||
| * rethrown as a {@link WrappedStageException},which includes the | ||
| * stage name where the error occurred. | ||
| * </p> | ||
| */ | ||
| public class StageTrackingOutputCommitter extends OutputCommitter { | ||
|
|
||
| private final OutputCommitter delegate; | ||
| private final String stageName; | ||
|
|
||
| public StageTrackingOutputCommitter(OutputCommitter delegate, String stageName) { | ||
| this.delegate = delegate; | ||
| this.stageName = stageName; | ||
| } | ||
|
|
||
| @Override | ||
| public void setupJob(JobContext jobContext) { | ||
| try { | ||
| delegate.setupJob(jobContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void setupTask(TaskAttemptContext taskAttemptContext) { | ||
| try { | ||
| delegate.setupTask(taskAttemptContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public boolean needsTaskCommit(TaskAttemptContext taskAttemptContext) { | ||
| try { | ||
| return delegate.needsTaskCommit(taskAttemptContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void commitTask(TaskAttemptContext taskAttemptContext) { | ||
| try { | ||
| delegate.commitTask(taskAttemptContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void abortTask(TaskAttemptContext taskAttemptContext) { | ||
| try { | ||
| delegate.abortTask(taskAttemptContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public boolean isRecoverySupported() { | ||
| return delegate.isRecoverySupported(); | ||
| } | ||
|
|
||
| @Override | ||
| public void recoverTask(TaskAttemptContext taskContext) { | ||
| try { | ||
| delegate.recoverTask(taskContext); | ||
| } catch (Exception e) { | ||
| throw new WrappedStageException(e, stageName); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.