org.meta.plugin.webservice.MetaWebServer.java Source code

Introduction

Here is the source code for org.meta.plugin.webservice.MetaWebServer.java

Source

/*
 *
 * JMeta - Meta's java implementation
 *
 * Copyright (C) 2013-2015 Pablo Joubert
 * Copyright (C) 2013-2015 Thomas Lavocat
 * Copyright (C) 2013-2015 Nicolas Michon
 *
 * This file is part of JMeta.
 *
 * JMeta is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * JMeta 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package org.meta.plugin.webservice;

import com.mongodb.util.JSONSerializers;
import com.mongodb.util.ObjectSerializer;
import java.util.HashMap;
import java.util.Iterator;
import org.bson.types.BasicBSONList;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.server.handler.HandlerList;
import org.eclipse.jetty.server.handler.ResourceHandler;
import org.eclipse.jetty.util.thread.QueuedThreadPool;
import org.meta.api.ws.AbstractPluginWebServiceController;
import org.meta.configuration.WSConfigurationImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * The web server part.
 *
 * @author faquin
 * @version $Id: $
 */
public class MetaWebServer {

    private HashMap<String, AbstractPluginWebServiceController> mapPlugins = null;
    private Server server = null;
    private final Logger logger = LoggerFactory.getLogger(MetaWebServer.class);

    private final WSConfigurationImpl configuration;

    /**
     * <p>Constructor for MetaWebServer.</p>
     *
     * @param config the web service configuration
     */
    public MetaWebServer(final WSConfigurationImpl config) {
        this.configuration = config;
        mapPlugins = new HashMap<>();
    }

    /**
     * return the plugin pointed by the given parameters.
     *
     * @param pluginName the plugin name
     * @return the plugin if found, null otherwise
     */
    public AbstractPluginWebServiceController getPlugin(final String pluginName) {
        return mapPlugins.get(pluginName);
    }

    /**
     * <p>start</p>
     */
    //@Override
    public void start() {
        //Launch the server on the right port
        //TODO get addresses from configuration and bind them all
        logger.info("Web server listening on port " + this.configuration.getWsPort());
        //server = new Server(this.configuration.getWsPort()).;
        server = new Server(new QueuedThreadPool(3, 2));
        ServerConnector connector = new ServerConnector(server, 1, 1);
        connector.setPort(this.configuration.getWsPort());
        server.setConnectors(new Connector[] { connector });

        // serve statics files within 'static' directory
        ResourceHandler resourceHandler = new ResourceHandler();
        resourceHandler.setDirectoriesListed(true);
        //give a way to serve the web site part
        resourceHandler.setResourceBase("static");

        HandlerList handlers = new HandlerList();
        handlers.setHandlers(new Handler[] { resourceHandler, new WebRequestHandler(this) });

        //Give the webrequestHandler to the server
        server.setHandler(handlers);
        //start and join the server
        try {
            server.start();
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
            //TODO throw MetaWSException here
        }
    }

    /**
     * Register a plugin pointed by the given plugin name.
     *
     * @param pluginName plugin name
     * @param abstractPluginWebServiceControler plugin webservice controler
     */
    public void registerPlugin(final String pluginName,
            final AbstractPluginWebServiceController abstractPluginWebServiceControler) {
        mapPlugins.put(pluginName, abstractPluginWebServiceControler);
    }

    /**
     * <p>getPluginListAsJson</p>
     *
     * @return a plugin list as JSON
     */
    public String getPluginListAsJson() {
        BasicBSONList list = new BasicBSONList();
        for (Iterator<String> i = mapPlugins.keySet().iterator(); i.hasNext();) {
            String key = (String) i.next();
            list.add(key);
        }

        // Serialize BasicBSONList in JSON
        ObjectSerializer jsonSerializer = JSONSerializers.getStrict();
        return jsonSerializer.serialize(list);
    }

    /**
     * terminate the web server.
     */
    public void close() {
        try {
            if (server != null) {
                server.stop();
            }
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
        }
    }
}