modxcms/revolution

<?php
/*
 * Copyright 2010-2011 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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.
 */


/*%******************************************************************************************%*/
// CLASS

/**
 * Contains a set of pre-built Amazon EMR Hadoop Bootstrap Actions.
 *
 * @version 2011.05.03
 * @license See the included NOTICE.md file for more information.
 * @copyright See the included NOTICE.md file for more information.
 * @link http://aws.amazon.com/php/ PHP Developer Center
 * @link http://hadoop.apache.org Apache Hadoop
 */
class CFHadoopBootstrap extends CFHadoopBase
{
    // Config file types
    const CONFIG_SITE = 'S';
    const CONFIG_DEFAULT = 'D';
    const CONFIG_CORE = 'C';
    const CONFIG_HDFS = 'H';
    const CONFIG_MAPREDUCE = 'M';

    // Daemon types
    const DAEMON_NAME_NODE = 'namenode';
    const DAEMON_DATA_NODE = 'datanode';
    const DAEMON_JOB_TRACKER = 'jobtracker';
    const DAEMON_TASK_TRACKER = 'tasktracker';
    const DAEMON_CLIENT = 'client';

    /**
     * Create a new run-if bootstrap action which lets you conditionally run bootstrap actions.
     *
     * @param string $condition (Required) The condition to evaluate. If <code>true</code>, the bootstrap action executes.
     * @param array $args (Optional) An indexed array of arguments to pass to the script.
     * @return array A configuration set to be provided when running a job flow.
     */
    public static function run_if($condition, $args = null)
    {
        if (!$args) $args = array();
        $args = is_array($args) ? $args : array($args);

        return self::script_runner('s3://us-east-1.elasticmapreduce/bootstrap-actions/run-if', $args);
    }

    /**
     * Specify options to merge with Hadoop's default configuration.
     *
     * @param string $file (Required) The Hadoop configuration file to merge with. [Allowed values: <code>CFHadoopBootstrap::CONFIG_SITE</code>, <code>CFHadoopBootstrap::CONFIG_DEFAULT</code>, <code>CFHadoopBootstrap::CONFIG_CORE</code>, <code>CFHadoopBootstrap::CONFIG_HDFS</code>, <code>CFHadoopBootstrap::CONFIG_MAPREDUCE</code>]
     * @param string|array $config (Required) This can either be an XML file in S3 (as <code>s3://bucket/path</code>), or an associative array of key-value pairs.
     * @return array A configuration set to be provided when running a job flow.
     */
    public static function configure($file, $config)
    {
        $args = array();
        $file_arg = '-' . $file;

        if (is_string($config))
        {
            $args[] = $file_arg;
            $args[] = $config;
        }
        elseif (is_array($config))
        {
            foreach ($config as $key => $value)
            {
                $args[] = $file_arg;
                $args[] = $key . '=' . $value;
            }
        }

        return self::script_runner('s3://us-east-1.elasticmapreduce/bootstrap-actions/configure-hadoop', $args);
    }

    /**
     * Create a new bootstrap action which lets you configure Hadoop's daemons. The options are written to
     * the <code>hadoop-user-env.sh</code> file.
     *
     * @param string $daemon_type (Required) The Hadoop daemon to configure.
     * @param array $opt (Optional) An associative array of parameters that can have the following keys: <ul>
     *     <li><code>HeapSize</code> - <code>integer</code> - Optional - The requested heap size of the daemon, in megabytes.</li>
     *     <li><code>CLIOptions</code> - <code>string</code> - Optional - Additional Java command line arguments to pass to the daemon.</li>
     *     <li><code>Replace</code> - <code>boolean</code> - Optional - Whether or not the file should be replaced. A value of <code>true</code> will replace the existing configuration file. A value of <code>false</code> will append the options to the configuration file.</li></ul>
     * @return array A configuration set to be provided when running a job flow.
     */
    public static function daemon($daemon_type, $opt = null)
    {
        if (!$opt) $opt = array();
        $args = array();

        foreach ($opt as $key => $value)
        {
            switch ($key)
            {
                case 'HeapSize':
                    $args[] = '--' . $daemon_type . '-heap-size=' . $value;
                    break;
                case 'CLIOptions':
                    $args[] = '--' . $daemon_type . '-opts="' . $value . '"';
                    break;
                case 'Replace':
                    if ((is_string($value) && $value === 'true') || (is_bool($value) && $value === true))
                    {
                        $args[] = '--replace';
                    }
                    break;
            }
        }

        return self::script_runner('s3://us-east-1.elasticmapreduce/bootstrap-actions/configure-daemons', $args);
    }
}