com.asakusafw.vanilla.client.VanillaDirect.java Source code

Introduction

Here is the source code for com.asakusafw.vanilla.client.VanillaDirect.java

Source

/**
 * Copyright 2011-2017 Asakusa Framework Team.
 *
 * 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 com.asakusafw.vanilla.client;

import java.util.Arrays;

import org.apache.hadoop.conf.Configuration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.asakusafw.bridge.launch.LaunchConfigurationException;
import com.asakusafw.dag.iterative.DirectLaunchConfiguration;
import com.asakusafw.runtime.core.context.RuntimeContext;

/**
 * Direct program entry of Asakusa Vanilla.
 * @since 0.4.1
 * @see VanillaLauncher
 */
public final class VanillaDirect {

    static final Logger LOG = LoggerFactory.getLogger(VanillaDirect.class);

    private VanillaDirect() {
        return;
    }

    /**
     * Program entry.
     * @param args launching configurations
     * @throws LaunchConfigurationException if launching configuration is something wrong
     */
    public static void main(String... args) throws LaunchConfigurationException {
        ClassLoader loader = VanillaDirect.class.getClassLoader();
        RuntimeContext.set(RuntimeContext.DEFAULT.apply(System.getenv()));
        RuntimeContext.get().verifyApplication(loader);

        int status = exec(loader, args);
        if (status != 0) {
            System.exit(status);
        }
    }

    /**
     * Program entry.
     * @param loader the launch class loader
     * @param args launching configurations
     * @return the exit code
     * @throws LaunchConfigurationException if launching configuration is something wrong
     * @see VanillaLauncher#EXEC_SUCCESS
     * @see VanillaLauncher#EXEC_ERROR
     * @see VanillaLauncher#EXEC_INTERRUPTED
     */
    public static int exec(ClassLoader loader, String... args) throws LaunchConfigurationException {
        DirectLaunchConfiguration conf = DirectLaunchConfiguration.parse(loader, Arrays.asList(args));
        Configuration hadoop = new Configuration();
        hadoop.setClassLoader(loader);
        int numberOfRounds = conf.getStageInfo().getRoundCount();
        int currentRound = 0;
        DirectLaunchConfiguration.Cursor cursor = conf.newCursor();
        while (cursor.next()) {
            LOG.info("Round: {}/{}", ++currentRound, numberOfRounds);
            int result = new VanillaLauncher(cursor.get(), hadoop).exec();
            if (result != VanillaLauncher.EXEC_SUCCESS) {
                return result;
            }
        }
        return VanillaLauncher.EXEC_SUCCESS;
    }
}