trunk/oak-run/src/main/java/org/apache/jackrabbit/oak/run/CheckCommand.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.jackrabbit.oak.run;

import static org.apache.jackrabbit.oak.segment.FileStoreHelper.isValidFileStoreOrFail;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.LinkedHashSet;
import java.util.Set;

import joptsimple.ArgumentAcceptingOptionSpec;
import joptsimple.OptionParser;
import joptsimple.OptionSet;
import joptsimple.OptionSpec;
import org.apache.jackrabbit.oak.run.commons.Command;
import org.apache.jackrabbit.oak.segment.tool.Check;

class CheckCommand implements Command {

    @Override
    public void execute(String... args) throws Exception {
        OptionParser parser = new OptionParser();
        ArgumentAcceptingOptionSpec<String> journal = parser.accepts(
                "journal", "journal file")
                .withRequiredArg().ofType(String.class).defaultsTo("journal.log");
        ArgumentAcceptingOptionSpec<Long> notify = parser.accepts(
                "notify", "number of seconds between progress notifications")
                .withRequiredArg().ofType(Long.class).defaultsTo(Long.MAX_VALUE);
        OptionSpec<?> bin = parser.accepts("bin", "read the content of binary properties");
        ArgumentAcceptingOptionSpec<String> filter = parser.accepts(
                "filter", "comma separated content paths to be checked")
                .withRequiredArg().ofType(String.class).withValuesSeparatedBy(',').defaultsTo("/");
        OptionSpec<?> head = parser.accepts("head", "checks only latest /root (i.e without checkpoints)");
        ArgumentAcceptingOptionSpec<String> cp = parser.accepts(
                "checkpoints", "checks only specified checkpoints (comma separated); use --checkpoints all to check all checkpoints")
                .withOptionalArg().ofType(String.class).withValuesSeparatedBy(',').defaultsTo("all");
        OptionSpec<?> ioStatistics = parser.accepts("io-stats", "Print I/O statistics (only for oak-segment-tar)");

        OptionSet options = parser.parse(args);
        
        PrintWriter out = new PrintWriter(System.out, true);
        PrintWriter err = new PrintWriter(System.err, true);

        if (options.nonOptionArguments().size() != 1) {
            printUsage(parser, err);
        }

        File dir = isValidFileStoreOrFail(new File(options.nonOptionArguments().get(0).toString()));
        String journalFileName = journal.value(options);
        long debugLevel = notify.value(options);
        Set<String> filterPaths = new LinkedHashSet<String>(filter.values(options));
        Set<String> checkpoints = new LinkedHashSet<>();
        if (options.has(cp) || !options.has(head)) {
            checkpoints.addAll(cp.values(options));
        }

        boolean checkHead = !options.has(cp) || options.has(head);

        int statusCode = Check.builder()
            .withPath(dir)
            .withJournal(journalFileName)
            .withDebugInterval(debugLevel)
            .withCheckBinaries(options.has(bin))
            .withCheckHead(checkHead)
            .withCheckpoints(checkpoints)
            .withFilterPaths(filterPaths)
            .withIOStatistics(options.has(ioStatistics))
            .withOutWriter(out)
            .withErrWriter(err)
            .build()
            .run();
        System.exit(statusCode);
    }

    private void printUsage(OptionParser parser, PrintWriter err, String... messages) throws IOException {
        for (String message : messages) {
            err.println(message);
        }
        
        err.println("usage: check path/to/segmentstore <options>");
        parser.printHelpOn(err);
        System.exit(1);
    }

}