io.bdeploy.bhive.op.FsckOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Public API including dependencies, ready to be used for integrations and plugins.
The newest version!
package io.bdeploy.bhive.op;

import java.util.Collection;
import java.util.Collections;
import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;

import io.bdeploy.bhive.BHive;
import io.bdeploy.bhive.audit.AuditParameterExtractor.AuditStrategy;
import io.bdeploy.bhive.audit.AuditParameterExtractor.AuditWith;
import io.bdeploy.bhive.model.Manifest;
import io.bdeploy.bhive.objects.view.ElementView;
import io.bdeploy.common.ActivityReporter.Activity;

/**
 * The {@link FsckOperation} checks the hive for consistency problems.
 * 
 * The returned set contains all {@link ElementView}s which are problematic (damaged, missing)
 */
public class FsckOperation extends BHive.Operation> {

    @AuditWith(AuditStrategy.COLLECTION_PEEK)
    private final SortedSet manifests = new TreeSet<>();
    private boolean repair;

    @Override
    public Set call() throws Exception {
        getObjectManager().invalidateCaches();
        getManifestDatabase().invalidateCaches();

        try (Activity activity = getActivityReporter().start("Checking", -1)) {
            if (manifests.isEmpty()) {
                Set localManifests = execute(new ManifestListOperation());
                if (localManifests.isEmpty()) {
                    return Collections.emptySet();
                }

                manifests.addAll(localManifests);
            }

            ManifestConsistencyCheckOperation mfCheck = new ManifestConsistencyCheckOperation().setDryRun(!repair);
            ObjectConsistencyCheckOperation objCheck = new ObjectConsistencyCheckOperation().setDryRun(!repair);

            manifests.forEach(k -> {
                mfCheck.addRoot(k);
                objCheck.addRoot(k);
            });

            Set problematic = new TreeSet<>();

            // check whether all manifests are still valid, objects might have been removed.
            problematic.addAll(execute(mfCheck));

            // scan and re-hash all objects...
            problematic.addAll(execute(objCheck));

            return problematic;
        } finally {
            getObjectManager().invalidateCaches();
            getManifestDatabase().invalidateCaches();
        }
    }

    /**
     * Add a {@link Manifest} to check.
     */
    public FsckOperation addManifest(Manifest.Key key) {
        manifests.add(key);
        return this;
    }

    /**
     * Add multiple {@link Manifest} to check.
     */
    public FsckOperation addManifests(Collection keys) {
        manifests.addAll(keys);
        return this;
    }

    /**
     * Enabling repair will remove all broken objects from the hive.
     */
    public FsckOperation setRepair(boolean repair) {
        this.repair = repair;
        return this;
    }

}