Package edu.berkeley.sparrow.daemon

Source Code of edu.berkeley.sparrow.daemon.SparrowDaemon

/*
* Copyright 2013 The Regents of The University California
*
* 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 edu.berkeley.sparrow.daemon;

import joptsimple.OptionParser;
import joptsimple.OptionSet;

import org.apache.commons.configuration.Configuration;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;

import edu.berkeley.sparrow.daemon.nodemonitor.NodeMonitorThrift;
import edu.berkeley.sparrow.daemon.scheduler.SchedulerThrift;
import edu.berkeley.sparrow.daemon.util.Logging;

/**
* A Sparrow Daemon includes both a scheduler and a node monitor.
* The main() method launches a Sparrow daemon.
*/
public class SparrowDaemon {
  // Eventually, we'll want to change this to something higher than debug.
  public final static Level DEFAULT_LOG_LEVEL = Level.DEBUG;

  public void initialize(Configuration conf) throws Exception {
    Level logLevel = Level.toLevel(conf.getString(SparrowConf.LOG_LEVEL, ""),
        DEFAULT_LOG_LEVEL);
    Logger.getRootLogger().setLevel(logLevel);

    // Start as many node monitors as specified in config
    String[] nmPorts = conf.getStringArray(SparrowConf.NM_THRIFT_PORTS);
    String[] inPorts = conf.getStringArray(SparrowConf.INTERNAL_THRIFT_PORTS);

    if (nmPorts.length != inPorts.length) {
      throw new ConfigurationException(SparrowConf.NM_THRIFT_PORTS + " and " +
        SparrowConf.INTERNAL_THRIFT_PORTS + " not of equal length");
    }
    if (nmPorts.length > 1 &&
        (!conf.getString(SparrowConf.DEPLYOMENT_MODE, "").equals("standalone"))) {
      throw new ConfigurationException("Mutliple NodeMonitors only allowed " +
          "in standalone deployment");
    }
    if (nmPorts.length == 0) {
      (new NodeMonitorThrift()).initialize(conf,
          NodeMonitorThrift.DEFAULT_NM_THRIFT_PORT,
          NodeMonitorThrift.DEFAULT_INTERNAL_THRIFT_PORT);
    }
    else {
      for (int i = 0; i < nmPorts.length; i++) {
        (new NodeMonitorThrift()).initialize(conf,
            Integer.parseInt(nmPorts[i]), Integer.parseInt(inPorts[i]));
      }
    }

    SchedulerThrift scheduler = new SchedulerThrift();
    scheduler.initialize(conf);
  }

  public static void main(String[] args) throws Exception {
    OptionParser parser = new OptionParser();
    parser.accepts("c", "configuration file (required)").
      withRequiredArg().ofType(String.class);
    parser.accepts("help", "print help statement");
    OptionSet options = parser.parse(args);

    if (options.has("help") || !options.has("c")) {
      parser.printHelpOn(System.out);
      System.exit(-1);
    }

    // Set up a simple configuration that logs on the console.
    BasicConfigurator.configure();

    Logging.configureAuditLogging();

    String configFile = (String) options.valueOf("c");
    Configuration conf = new PropertiesConfiguration(configFile);
    SparrowDaemon sparrowDaemon = new SparrowDaemon();
    sparrowDaemon.initialize(conf);
  }
}
TOP

Related Classes of edu.berkeley.sparrow.daemon.SparrowDaemon

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.