io.atrato.pubsubserver.PubsubServer.java Source code

Introduction

Here is the source code for io.atrato.pubsubserver.PubsubServer.java

Source

/**
 * Copyright (c) 2017 Atrato, Inc. 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.
 * 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 io.atrato.pubsubserver;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.servlet.DefaultServlet;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.eclipse.jetty.websocket.servlet.WebSocketServlet;
import org.eclipse.jetty.websocket.servlet.WebSocketServletFactory;
import org.glassfish.jersey.servlet.ServletContainer;

import java.net.InetSocketAddress;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Created by david on 1/23/17.
 */
public class PubsubServer {
    private static final String DEFAULT_HOST = "0.0.0.0";
    private static final int DEFAULT_PORT = 8890;

    private static final String CMD_OPTION_LISTEN_ADDRESS = "listenAddress";
    private static final String CMD_OPTION_HTDOCS = "htdocs";

    private String host = DEFAULT_HOST;
    private int port = DEFAULT_PORT;
    private String htdocsDir;

    private static PubsubBroker broker = new PubsubBroker();

    public static class PubsubServlet extends WebSocketServlet {
        @Override
        public void configure(WebSocketServletFactory factory) {
            factory.getPolicy().setIdleTimeout(1000000);
            factory.register(PubsubSocket.class);
        }
    }

    public static PubsubBroker getBroker() {
        return broker;
    }

    void init(String[] args) throws ParseException {
        Options options = new Options();
        options.addOption(CMD_OPTION_LISTEN_ADDRESS, true,
                "Address to listen to. Default is " + DEFAULT_HOST + ":" + DEFAULT_PORT);
        options.addOption(CMD_OPTION_HTDOCS, true, "The local directory where static files should be served.");

        CommandLineParser parser = new DefaultParser();
        CommandLine cmd = parser.parse(options, args);

        String listenAddress = cmd.getOptionValue(CMD_OPTION_LISTEN_ADDRESS);

        if (listenAddress != null) {
            Pattern pattern = Pattern.compile("(.+:)?(\\d+)");
            Matcher matcher = pattern.matcher(listenAddress);

            if (matcher.find()) {
                String hostString = matcher.group(1);
                if (hostString != null) {
                    host = hostString.substring(0, hostString.length() - 1);
                }
                port = Integer.valueOf(matcher.group(2));
            } else {
                throw new ParseException("listenAddress must be in this format: [host:]port");
            }
        }

        htdocsDir = cmd.getOptionValue(CMD_OPTION_HTDOCS, null);

    }

    void run() throws Exception {
        ServletContextHandler context = new ServletContextHandler(ServletContextHandler.SESSIONS);
        context.setContextPath("/");

        ServletHolder jerseyServlet = context.addServlet(ServletContainer.class, "/ws/v1/*");

        InetSocketAddress address = new InetSocketAddress(host, port);
        Server jettyServer = new Server(address);
        jettyServer.setHandler(context);

        jerseyServlet.setInitOrder(50);

        jerseyServlet.setInitParameter("jersey.config.server.provider.classnames",
                io.atrato.pubsubserver.PubsubRestProvider.class.getCanonicalName());

        ServletHolder wsServlet = new ServletHolder(PubsubServlet.class);
        context.addServlet(wsServlet, "/pubsub");

        if (htdocsDir != null) {
            context.setResourceBase(htdocsDir);
            ServletHolder staticFilesServlet = context.addServlet(DefaultServlet.class, "/");
            staticFilesServlet.setInitOrder(10);
        }

        jettyServer.setHandler(context);

        try {
            jettyServer.start();
            jettyServer.join();
        } finally {
            jettyServer.destroy();
        }
    }

    public static void main(String[] args) throws Exception {
        PubsubServer server = new PubsubServer();
        server.init(args);
        server.run();
    }
}