org/objectweb/proactive/core/node/StartNode.java

/* 
 * ################################################################
 * 
 * ProActive: The Java(TM) library for Parallel, Distributed, 
 *            Concurrent computing with Security and Mobility
 * 
 * Copyright (C) 1997-2007 INRIA/University of Nice-Sophia Antipolis
 * Contact: proactive@objectweb.org
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *  
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 * USA
 *  
 *  Initial developer(s):               The ProActive Team
 *                        http://www.inria.fr/oasis/ProActive/contacts.html
 *  Contributor(s): 
 * 
 * ################################################################
 */ 
package org.objectweb.proactive.core.node;

import java.rmi.AlreadyBoundException;

import org.apache.log4j.Logger;
import org.objectweb.proactive.core.Constants;
import org.objectweb.proactive.core.UniqueID;
import org.objectweb.proactive.core.config.ProActiveConfiguration;
import org.objectweb.proactive.core.runtime.jini.JiniRuntimeFactory;
import org.objectweb.proactive.core.runtime.rmi.RmiRuntimeFactory;
import org.objectweb.proactive.core.util.UrlBuilder;
import org.objectweb.proactive.core.util.log.Loggers;
import org.objectweb.proactive.core.util.log.ProActiveLogger;
import org.objectweb.proactive.core.util.profiling.PAProfilerEngine;
import org.objectweb.proactive.core.util.profiling.Profiling;
import org.objectweb.proactive.core.util.timer.AverageMicroTimer;


public class StartNode {
    public static final int DEFAULT_CLASSFILE_SERVER_PORT = 2001;
    static Logger logger;
    protected static final int DEFAULT_PORT = 1099;
    protected static final int MAX_RETRY = 3;
    protected static final String NO_REBIND_OPTION_NAME = "-noRebind";
    protected static final String NO_CLASS_SERVER_OPTION_NAME = "-noClassServer";
    protected static final String NO_REGISTRY_OPTION_NAME = "-noRegistry";
    protected static final String MULTICAST_LOCATOR_NAME = "-multicastLocator";

    static {
        ProActiveConfiguration.load();
        logger = ProActiveLogger.getLogger(Loggers.DEPLOYMENT);

        if (logger.isDebugEnabled()) {
            logger.debug("Loading ProActive class");
        }

        try {
            Class.forName("org.objectweb.proactive.ProActive");
        } catch (ClassNotFoundException e) {
            if (logger.isDebugEnabled()) {
                logger.fatal("Loading of ProActive class FAILED");
            }

            e.printStackTrace();
            System.exit(1);
        }
    }

    protected boolean noClassServer = false;
    protected boolean noRebind = false;
    protected boolean noRegistry = false;
    protected boolean multicastLocator = false;
    protected int registryPortNumber = DEFAULT_PORT;
    protected String classpath;
    protected String nodeURL;

    //
    // -- CONSTRUCTORS -----------------------------------------------
    //
    protected StartNode() {
    }

    private StartNode(String[] args) {
        if (args.length == 0) {
            nodeURL = null;
        } else {
            nodeURL = args[0];
            registryPortNumber = UrlBuilder.getPortFromUrl(nodeURL);
            checkOptions(args, 1);
            readClassPath(args, 1);
        }

        /*
           // debug
           System.out.println("Node name = "+nodeURL);
           if (noRebind)
             System.out.println(" - NoRebind");
           if (noClassServer)
             System.out.println(" - No ClassServer");
           else System.out.println("ClassServer classpath = "+classpath);
         */
    }

    //
    // -- PUBLIC METHODS -----------------------------------------------
    //
    public static void main(String[] args) {
        try {
            new StartNode(args).run();
        } catch (Exception e) {
            e.printStackTrace();
            logger.fatal(e.toString());
        }
    }

    //
    // -- PROTECTED METHODS -----------------------------------------------
    //
    protected void checkOptions(String[] args, int start) {
        for (int i = start; i < args.length; i++)
            checkOption(args[i]);
    }

    protected void readClassPath(String[] args, int start) {
        if (noClassServer) {
            return;
        }

        // look for classpath
        for (int i = start; i < args.length; i++) {
            String s = args[i];

            if (s.charAt(0) != '-') {
                classpath = s;

                break;
            }
        }
    }

    //    /**
    //     * <i><font size="-1" color="#FF0000">**For internal use only** </font></i>
    //     * sets the properties needed for the node creation
    //     */
    //    protected void setProperties() {
    //        //  System.setProperty("sun.rmi.dgc.checkInterval","400");
    //        //  System.setProperty("java.rmi.dgc.leaseValue","800");
    //        //  System.setProperty("sun.rmi.dgc.cleanInterval","400");
    //        //  System.setProperty("sun.rmi.dgc.client.gcInterval","400");
    //    }
    protected void createNode(String nodeURL, boolean noRebind)
        throws NodeException, AlreadyBoundException {
        int exceptionCount = 0;

        while (true) {
            try {
                Node node = null;

                if (nodeURL == null) {
                    node = NodeFactory.getDefaultNode();
                } else {
                    //TODO allow start alone node with security parameters 
                    node = NodeFactory.createNode(nodeURL, !noRebind, null, null);
                }

                logger.info("OK. Node " + node.getNodeInformation().getName() +
                    " is created in VM id=" + UniqueID.getCurrentVMID());

                break;
            } catch (NodeException e) {
                exceptionCount++;

                if (exceptionCount == MAX_RETRY) {
                    throw e;
                } else {
                    logger.error("Error, retrying (" + exceptionCount + ")");

                    try {
                        Thread.sleep(1000);
                    } catch (InterruptedException e2) {
                    }
                }

                // end if
            }

            // try
        }

        // end while
    }

    protected void run()
        throws java.io.IOException, NodeException, AlreadyBoundException {
        //setProperties();
        // set options on node factory
        RmiRuntimeFactory.setShouldCreateClassServer(!noClassServer);
        RmiRuntimeFactory.setShouldCreateRegistry(!noRegistry);
        RmiRuntimeFactory.setRegistryPortNumber(registryPortNumber);

        if (multicastLocator) {
            JiniRuntimeFactory.setMulticastLocator(multicastLocator);
        }

        AverageMicroTimer mt = null;

        if (Profiling.STARTNODE) {
            mt = new AverageMicroTimer("StartNode");
            PAProfilerEngine.registerTimer(mt);
            mt.start();
        }

        // create node
        createNode(nodeURL, noRebind);

        if (Profiling.STARTNODE) {
            mt.stop();

            //  mt.dump();
        }
    }

    protected void checkOption(String option) {
        if (NO_REBIND_OPTION_NAME.equals(option)) {
            noRebind = true;
        } else if (NO_CLASS_SERVER_OPTION_NAME.equals(option)) {
            noClassServer = true;
        } else if (NO_REGISTRY_OPTION_NAME.equals(option)) {
            noRegistry = true;
        } else if (MULTICAST_LOCATOR_NAME.equals(option)) {
            multicastLocator = true;
        } else {
                printUsage();
        }
    }

    //
    // -- PRIVATE METHODS -----------------------------------------------
    //
    private void printUsage() {
        String localhost = "localhost";

        try {
            localhost = UrlBuilder.getHostNameorIP(java.net.InetAddress.getLocalHost());
        } catch (java.net.UnknownHostException e) {
            logger.error("InetAddress failed: " + e.getMessage());
            e.printStackTrace();
        } catch (java.lang.SecurityException e) {
            logger.error("InetAddress failed: " + e.getMessage());
            e.printStackTrace();
        }

        logger.info("usage: java " + this.getClass().getName() +
            " <node URL> [options]");
        logger.info(" - options");
        logger.info("     " + NO_CLASS_SERVER_OPTION_NAME +
            " : indicates not to create a ClassServer for JINI and RMI.");
        logger.info(
            "                      By default a ClassServer is automatically created");
        logger.info("                      to serve class files on demand.");
        logger.info("     " + NO_REBIND_OPTION_NAME +
            "      : indicates not to use rebind when registering the");
        logger.info(
            "                      node to the RMIRegistry. If a node of the same name");
        logger.info(
            "                      already exists, the creation of the new node will fail.");
        logger.info("  for instance: java " + StartNode.class.getName() + " " +
            Constants.RMI_PROTOCOL_IDENTIFIER + "//" + localhost + "/node1");
        logger.info("                java " + StartNode.class.getName() + " " +
            Constants.RMI_PROTOCOL_IDENTIFIER + "://" + localhost + "/node2  " +
            NO_CLASS_SERVER_OPTION_NAME + " " + NO_REBIND_OPTION_NAME);
        logger.info("                java " + StartNode.class.getName() + " " +
            Constants.JINI_PROTOCOL_IDENTIFIER + "://" + localhost + "/node3");
        logger.info("                java " + StartNode.class.getName() + " " +
            Constants.JINI_PROTOCOL_IDENTIFIER + "://" + localhost + "/node4 " +
            MULTICAST_LOCATOR_NAME);
    }
}

---