VertxSimplePushServer.java

/**
 * JBoss, Home of Professional Open Source
 * Copyright Red Hat, Inc., and individual contributors.
 *
 * 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 org.jboss.aerogear.simplepush.vertx;

import org.jboss.aerogear.simplepush.server.DefaultSimplePushConfig;
import org.jboss.aerogear.simplepush.server.DefaultSimplePushServer;
import org.jboss.aerogear.simplepush.server.SimplePushServer;
import org.jboss.aerogear.simplepush.server.SimplePushServerConfig;
import org.jboss.aerogear.simplepush.server.datastore.DataStore;
import org.jboss.aerogear.simplepush.server.datastore.InMemoryDataStore;
import org.vertx.java.core.Handler;
import org.vertx.java.core.eventbus.Message;
import org.vertx.java.core.http.HttpServer;
import org.vertx.java.core.http.RouteMatcher;
import org.vertx.java.core.json.JsonObject;
import org.vertx.java.core.logging.Logger;
import org.vertx.java.core.sockjs.SockJSServer;
import org.vertx.java.platform.Verticle;

/**
 * Verticle for the SimplePush Server.
 */
public class VertxSimplePushServer extends Verticle {

    public static final String WRITE_HANDLER_MAP = "simplepush.writehandler.map";
    public static final String LAST_ACCESSED_MAP = "simplepush.lastaccessed.map";
    public static final String USER_AGENT_REMOVER = "simplepush.useragent.remover";
    public static final String DEFAULT_HOST = "localhost";
    public static final int DEFAULT_PORT = 7777;

    @Override
    public void start() {
        final SimplePushServerConfig config = fromConfig(container.config());
        final DataStore datastore = new InMemoryDataStore();
        final byte[] privateKey = DefaultSimplePushServer.generateAndStorePrivateKey(datastore, config);
        final SimplePushServer simplePushServer = new DefaultSimplePushServer(datastore, config, privateKey);
        final HttpServer httpServer = vertx.createHttpServer();
        setupHttpNotificationHandler(httpServer, simplePushServer);
        setupSimplePushSockJSServer(httpServer, simplePushServer);
        startHttpServer(httpServer);
        setupUserAgentReaperJob(simplePushServer);
    }

    private SimplePushServerConfig fromConfig(JsonObject config) {
        return DefaultSimplePushConfig.create()
                .ackInterval(config.getLong("ackInterval"))
                .endpointPrefix(config.getString("endpointUrlPrefix"))
                .password(config.getString("password", "changeme!!!"))
                .userAgentReaperTimeout(config.getLong("userAgentReaperTimeout")).build();
    }

    private void startHttpServer(final HttpServer httpServer) {
        final String host = container.config().getString("host", DEFAULT_HOST);
        final int port = container.config().getInteger("port", DEFAULT_PORT);
        httpServer.listen(port, host);
        container.logger().info("Started VertxSimplePushServer on host [" + host + "] port [" + port + "]");
    }

    private void setupHttpNotificationHandler(final HttpServer httpServer, final SimplePushServer simplePushServer) {
        final RouteMatcher rm = new RouteMatcher();
        final String endpointUrlPrefix = simplePushServer.config().endpointPrefix();
        rm.put(endpointUrlPrefix + "/:endpoint", new HttpNotificationHandler(simplePushServer, vertx, container));
        httpServer.requestHandler(rm);
    }

    private void setupSimplePushSockJSServer(final HttpServer httpServer, final SimplePushServer simplePushServer) {
        final JsonObject appConfig = new JsonObject().putString("prefix", "/simplepush");
        final SockJSServer sockJSServer = vertx.createSockJSServer(httpServer);
        sockJSServer.installApp(appConfig, new SimplePushServerHandler(simplePushServer, vertx, container));
    }

    private void setupUserAgentReaperJob(final SimplePushServer simplePushServer) {
        final Logger logger = container.logger();
        vertx.eventBus().registerHandler(USER_AGENT_REMOVER, new Handler<Message<String>>() {
            @Override
            public void handle(final Message<String> msg) {
                final String uaid = msg.body();
                simplePushServer.removeAllChannels(uaid);
                logger.info("Removed all channels for [" + uaid + "] due to inactivity");
            }
        });
        container.deployWorkerVerticle(UserAgentReaper.class.getName(), container.config());
    }

}