org.glassfish.tyrus.platform.main.Server Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of websocket-impl Show documentation
The newest version!
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2011 - 2012 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * http://glassfish.java.net/public/CDDL+GPL_1_1.html
 * or packager/legal/LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at packager/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */
package org.glassfish.tyrus.platform.main;

import org.glassfish.grizzly.http.server.HttpServer;
import org.glassfish.grizzly.websockets.WebSocketAddOn;
import org.glassfish.tyrus.platform.BeanServer;

import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;
import java.util.logging.Logger;

/**
 * Implementation of the WebSocket Server.
 *
 * @author Stepan Kopriva (stepan.kopriva at oracle.com)
 */
public class Server {
    private HttpServer server;
    private BeanServer beanServer;
    private final Set> beans;
    private final String hostName;
    private final int port;
    private final String rootPath;

    private static final String ENGINE_PROVIDER_CLASSNAME = "org.glassfish.tyrus.spi.grizzlyprovider.GrizzlyEngine";
    private static final Logger LOGGER = Logger.getLogger(Server.class.getClass().getName());
    private static final int DEFAULT_PORT = 8025;
    private static final String DEFAULT_HOST_NAME = "localhost";
    private static final String DEFAULT_ROOT_PATH = "/websockets/tests";

    /**
     * Create new server instance.
     *
     * @param beans to be registered with the server
     */
    public Server(Class... beans) {
        this(null, 0, null, beans);
    }

    /**
     * Create new server instance.
     *
     * @param beanNames beans to be registered with the server
     */
    public Server(String... beanNames) {
        this(null, 0, null, new HashSet>());
        for (String beanName : beanNames) {
            try {
                beans.add(Class.forName(beanName));
            } catch (ClassNotFoundException e) {
                throw new RuntimeException(e);
            }
        }
    }

    /**
     * Construct new server.
     *
     * @param hostName hostName of the server.
     * @param port     port of the server.
     * @param rootPath root path to the server App.
     * @param beans    registered application classes.
     */
    public Server(String hostName, int port, String rootPath, Class... beans) {
        this(hostName, port, rootPath, new HashSet>(Arrays.asList(beans)));
    }

    /**
     * Construct new server.
     *
     * @param hostName hostName of the server.
     * @param port     port of the server.
     * @param rootPath root path to the server App.
     * @param beans    registered application classes.
     */
    public Server(String hostName, int port, String rootPath, Set> beans) {
        this.hostName = hostName == null ? DEFAULT_HOST_NAME : hostName;
        this.port = port == 0 ? DEFAULT_PORT : port;
        this.rootPath = rootPath == null ? DEFAULT_ROOT_PATH : rootPath;
        this.beans = beans;
    }


    /**
     * Start the server.
     */
    public synchronized void start(){
        try {
            if(server == null){
                server = HttpServer.createSimpleServer(this.rootPath, this.port);
                server.getListener("grizzly").registerAddOn(new WebSocketAddOn());
                server.start();
                beanServer = new BeanServer(ENGINE_PROVIDER_CLASSNAME);
                try {
                    beanServer.initWebSocketServer(this.rootPath, this.port, beans);
                } catch (Exception e) {
                    e.printStackTrace();
                }
                LOGGER.info("WebSocket Registered apps: URLs all start with ws://" + this.hostName + ":" + this.port);
                LOGGER.info("WebSocket server started.");
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * Stop the server.
     */
    public synchronized void stop(){
        if(server != null){
            server.stop();
            server = null;
            beanServer = null;
            LOGGER.info("Websocket Server stopped.");
        }
    }

    public static void main(String[] args) {
        if (args.length < 4) {
            System.out.println("Please provide: (, , , <;-sep fully qualfied classnames of your bean>) in the command line");
            System.out.println("e.g. localhost 8021 /websockets/myapp myapp.Bean1;myapp.Bean2");
            System.exit(1);
        }
        Set> beanClasses = getClassesFromString(args[3]);
        int port = Integer.parseInt(args[1]);
        String hostname = args[0];
        String wsroot = args[2];

        Server server = new Server(hostname, port, wsroot, beanClasses);

        try {
            server.start();
            System.out.println("Press any key to stop the WebSocket server...");
            System.in.read();
        } catch (IOException ioe) {
            System.err.println("IOException during server run");
            ioe.printStackTrace();
        } finally {
            server.stop();
        }
    }

    private static Set> getClassesFromString(String rawString) {
        Set> beanClasses = new HashSet>();
        StringTokenizer st = new StringTokenizer(rawString, ";");
        while (st.hasMoreTokens()) {
            String nextClassname = st.nextToken().trim();
            if (!"".equals(nextClassname)) {
                try {
                    beanClasses.add(Class.forName(nextClassname));
                } catch (ClassNotFoundException cnfe) {
                    throw new RuntimeException("Stop: cannot load class: " + nextClassname);
                }
            }
        }
        return beanClasses;
    }
}