com.sun.grizzly.standalone.JRuby Maven / Gradle / Ivy

Go to download
/*
 * The contents of this file are subject to the terms 
 * of the Common Development and Distribution License 
 * (the License).  You may not use this file except in
 * compliance with the License.
 * 
 * You can obtain a copy of the license at 
 * https://glassfish.dev.java.net/public/CDDLv1.0.html or
 * glassfish/bootstrap/legal/CDDLv1.0.txt.
 * See the License for the specific language governing 
 * permissions and limitations under the License.
 * 
 * When distributing Covered Code, include this CDDL 
 * Header Notice in each file and include the License file 
 * at glassfish/bootstrap/legal/CDDLv1.0.txt.  
 * If applicable, add the following below the CDDL Header, 
 * with the fields enclosed by brackets [] replaced by
 * you own identifying information: 
 * "Portions Copyrighted [year] [name of copyright owner]"
 * 
 * Copyright 2006 Sun Microsystems, Inc. All rights reserved.
 */
package com.sun.grizzly.standalone;

import java.io.IOException;

import com.sun.grizzly.jruby.RailsSelectorThread;
import com.sun.grizzly.tcp.Adapter;
import com.sun.grizzly.tcp.StaticResourcesAdapter;
import com.sun.grizzly.util.ClassLoaderUtil;
import java.io.File;

public class JRuby {
    
    /**
     * System property for the SelectorThread value.
     */
    private static final String SELECTOR_THREAD = "com.sun.grizzly.selectorThread";
    
    
    private static final String ADAPTER = "com.sun.grizzly.adapterClass";
    
    
    private static int port = 3000;
    
    
    private static String folder = "/public";
    
    
    private static int numberOfRuntime = 1;
   
    
    public static void main(String[] args) throws IOException, InstantiationException {
        if(args.length == 0) {
            printHelpAndExit();
        }
        // parse options
        for (int i = 0; i < args.length - 1; i++) {
            String arg = args[i];
            
            if("-h".equals(arg) || "--help".equals(arg)) {
                printHelpAndExit();
            } else if("-n".equals(arg)) {
                i ++;
                setNumberOfRuntimes(args[i]);
            } else if(arg.startsWith("--num-runtimes=")) {
                String num = arg.substring("--num-runtimes=".length(), arg.length());
                setNumberOfRuntimes(num);
            } else if ("-p".equals(arg)) {
                i ++;
                setPort(args[i]);
            } else if (arg.startsWith("--port=")) {
                String num = arg.substring("--port=".length(), arg.length());
                setPort(num);
            }
        }
        // last argument must be RAILS_ROOT
        String railsRoot = args[args.length - 1];
        if(!new File(railsRoot).isDirectory()) {
            System.err.println("Illegal Rails root");
            printHelpAndExit();
        }

        if (new File("lib").exists()){
            // Load jar under the lib directory
            Thread.currentThread().setContextClassLoader(
                    ClassLoaderUtil.createClassloader(
                        new File("lib"),JRuby.class.getClassLoader()));
        }
        
        RailsSelectorThread selectorThread = null;
        String selectorThreadClassname = System.getProperty(SELECTOR_THREAD);
        if ( selectorThreadClassname != null){
            selectorThread = (RailsSelectorThread)
                loadInstance(selectorThreadClassname);
        } else {
            selectorThread = new RailsSelectorThread();
        }        
        
        String applicationRoot = railsRoot + folder;    
        selectorThread.setDisplayConfiguration(true);
        selectorThread.setRailsRoot(railsRoot);
        selectorThread.setWebAppRootPath(applicationRoot);
        selectorThread.setPort(port);
        selectorThread.setNumberOfRuntime(numberOfRuntime);
        
        String adapterClass =  System.getProperty(ADAPTER);
        Adapter adapter;
        if (adapterClass == null){
            adapter = new StaticResourcesAdapter();
            ((StaticResourcesAdapter)adapter).setRootFolder(folder);
        } else {
            adapter = (Adapter)loadInstance(adapterClass);
        }
        
        selectorThread.setAdapter(adapter);
        selectorThread.initEndpoint();
        selectorThread.startEndpoint();
    }

    /**
     * Util to load classes using reflection.
     */
    private static Object loadInstance(String property){        
        Class className = null;                               
        try{                              
            className = Class.forName(property,true,
                    Thread.currentThread().getContextClassLoader());
            return className.newInstance();
        } catch (Throwable t) {
            t.printStackTrace();
            // Log me 
        }   
        return null;
    }      

    private static void setNumberOfRuntimes(String num) {
        try {
            numberOfRuntime = Integer.parseInt(num);
        } catch (NumberFormatException e) {
            System.err.println("Illegal number -- " + num);
            printHelpAndExit();
        }
    }
   
    private static void setPort(String num) {
        try {
            port = Integer.parseInt(num);
        } catch (NumberFormatException e) {
            System.err.println("Illegal port number -- " + num);
            printHelpAndExit();
        }
    }
    
    private static void printHelpAndExit() {
        System.err.println("Usage: " + JRuby.class.getCanonicalName() + " [options] RAILS_ROOT");
        System.err.println();
        System.err.println("    -p, --port=port                  Runs Rails on the specified port.");
        System.err.println("                                     Default: 3000");
        System.err.println("    -n, --num-runtimes=num           Specifies number of JRuby runtimes.");
        System.err.println("                                     Default: 1");
        System.err.println("    -h, --help                       Show this help message.");
        System.exit(1);
    }
    
}