Examples with HoodieRestoreException org.apache.hudi.exception.HoodieRestoreException used on opensource projects

Example 1 with HoodieRestoreException

use of org.apache.hudi.exception.HoodieRestoreException in project hudi by apache.

the class BaseHoodieWriteClient method restoreToInstant.

/**
 * NOTE : This action requires all writers (ingest and compact) to a table to be stopped before proceeding. Revert
 * the (inflight/committed) record changes for all commits after the provided instant time.
 *
 * @param instantTime Instant time to which restoration is requested
 */
public HoodieRestoreMetadata restoreToInstant(final String instantTime) throws HoodieRestoreException {
    LOG.info("Begin restore to instant " + instantTime);
    final String restoreInstantTime = HoodieActiveTimeline.createNewInstantTime();
    Timer.Context timerContext = metrics.getRollbackCtx();
    try {
        HoodieTable<T, I, K, O> table = initTable(WriteOperationType.UNKNOWN, Option.empty());
        Option<HoodieRestorePlan> restorePlanOption = table.scheduleRestore(context, restoreInstantTime, instantTime);
        if (restorePlanOption.isPresent()) {
            HoodieRestoreMetadata restoreMetadata = table.restore(context, restoreInstantTime, instantTime);
            if (timerContext != null) {
                final long durationInMs = metrics.getDurationInMs(timerContext.stop());
                final long totalFilesDeleted = restoreMetadata.getHoodieRestoreMetadata().values().stream().flatMap(Collection::stream).mapToLong(HoodieRollbackMetadata::getTotalFilesDeleted).sum();
                metrics.updateRollbackMetrics(durationInMs, totalFilesDeleted);
            }
            return restoreMetadata;
        } else {
            throw new HoodieRestoreException("Failed to restore " + config.getBasePath() + " to commit " + instantTime);
        }
    } catch (Exception e) {
        throw new HoodieRestoreException("Failed to restore to " + instantTime, e);
    }
}

Example 2 with HoodieRestoreException

use of org.apache.hudi.exception.HoodieRestoreException in project hudi by apache.

the class BaseRestoreActionExecutor method execute.

@Override
public HoodieRestoreMetadata execute() {
    HoodieTimer restoreTimer = new HoodieTimer();
    restoreTimer.startTimer();
    Option<HoodieInstant> restoreInstant = table.getRestoreTimeline().filterInflightsAndRequested().filter(instant -> instant.getTimestamp().equals(instantTime)).firstInstant();
    if (!restoreInstant.isPresent()) {
        throw new HoodieRollbackException("No pending restore instants found to execute restore");
    }
    try {
        List<HoodieInstant> instantsToRollback = getInstantsToRollback(restoreInstant.get());
        ValidationUtils.checkArgument(restoreInstant.get().getState().equals(HoodieInstant.State.REQUESTED) || restoreInstant.get().getState().equals(HoodieInstant.State.INFLIGHT));
        Map<String, List<HoodieRollbackMetadata>> instantToMetadata = new HashMap<>();
        if (restoreInstant.get().isRequested()) {
            table.getActiveTimeline().transitionRestoreRequestedToInflight(restoreInstant.get());
        }
        instantsToRollback.forEach(instant -> {
            instantToMetadata.put(instant.getTimestamp(), Collections.singletonList(rollbackInstant(instant)));
            LOG.info("Deleted instant " + instant);
        });
        return finishRestore(instantToMetadata, instantsToRollback, restoreTimer.endTimer());
    } catch (IOException io) {
        throw new HoodieRestoreException("unable to Restore instant " + restoreInstant.get(), io);
    }
}