Examples with HoodieRestorePlan org.apache.hudi.avro.model.HoodieRestorePlan used on opensource projects

Example 1 with HoodieRestorePlan

use of org.apache.hudi.avro.model.HoodieRestorePlan 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 HoodieRestorePlan

use of org.apache.hudi.avro.model.HoodieRestorePlan in project hudi by apache.

the class BaseRestoreActionExecutor method getInstantsToRollback.

private List<HoodieInstant> getInstantsToRollback(HoodieInstant restoreInstant) throws IOException {
    List<HoodieInstant> instantsToRollback = new ArrayList<>();
    HoodieRestorePlan restorePlan = RestoreUtils.getRestorePlan(table.getMetaClient(), restoreInstant);
    for (HoodieInstantInfo instantInfo : restorePlan.getInstantsToRollback()) {
        // If restore crashed mid-way, there are chances that some commits are already rolled back,
        // but some are not. so, we can ignore those commits which are fully rolledback in previous attempt if any.
        Option<HoodieInstant> rollbackInstantOpt = table.getActiveTimeline().getWriteTimeline().filter(instant -> instant.getTimestamp().equals(instantInfo.getCommitTime()) && instant.getAction().equals(instantInfo.getAction())).firstInstant();
        if (rollbackInstantOpt.isPresent()) {
            instantsToRollback.add(rollbackInstantOpt.get());
        } else {
            LOG.warn("Ignoring already rolledback instant " + instantInfo.toString());
        }
    }
    return instantsToRollback;
}

Example 3 with HoodieRestorePlan

use of org.apache.hudi.avro.model.HoodieRestorePlan in project hudi by apache.

the class RestorePlanActionExecutor method execute.

@Override
public Option<HoodieRestorePlan> execute() {
    final HoodieInstant restoreInstant = new HoodieInstant(HoodieInstant.State.REQUESTED, HoodieTimeline.RESTORE_ACTION, instantTime);
    try {
        // Get all the commits on the timeline after the provided commit time
        // rollback pending clustering instants first before other instants (See HUDI-3362)
        List<HoodieInstant> pendingClusteringInstantsToRollback = table.getActiveTimeline().filterPendingReplaceTimeline().filter(instant -> ClusteringUtils.isPendingClusteringInstant(table.getMetaClient(), instant)).getReverseOrderedInstants().filter(instant -> HoodieActiveTimeline.GREATER_THAN.test(instant.getTimestamp(), restoreInstantTime)).collect(Collectors.toList());
        // Get all the commits on the timeline after the provided commit time
        List<HoodieInstant> commitInstantsToRollback = table.getActiveTimeline().getWriteTimeline().getReverseOrderedInstants().filter(instant -> HoodieActiveTimeline.GREATER_THAN.test(instant.getTimestamp(), restoreInstantTime)).filter(instant -> !pendingClusteringInstantsToRollback.contains(instant)).collect(Collectors.toList());
        // Combine both lists - first rollback pending clustering and then rollback all other commits
        List<HoodieInstantInfo> instantsToRollback = Stream.concat(pendingClusteringInstantsToRollback.stream(), commitInstantsToRollback.stream()).map(entry -> new HoodieInstantInfo(entry.getTimestamp(), entry.getAction())).collect(Collectors.toList());
        HoodieRestorePlan restorePlan = new HoodieRestorePlan(instantsToRollback, LATEST_RESTORE_PLAN_VERSION);
        table.getActiveTimeline().saveToRestoreRequested(restoreInstant, TimelineMetadataUtils.serializeRestorePlan(restorePlan));
        table.getMetaClient().reloadActiveTimeline();
        LOG.info("Requesting Restore with instant time " + restoreInstant);
        return Option.of(restorePlan);
    } catch (IOException e) {
        LOG.error("Got exception when saving restore requested file", e);
        throw new HoodieIOException(e.getMessage(), e);
    }
}