fr.meuret.web.HttpServer.java Source code

Introduction

Here is the source code for fr.meuret.web.HttpServer.java

Source

/*
* Copyright 2012 The Netty Project
*
* The Netty Project licenses this file to you 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 fr.meuret.web;

import com.beust.jcommander.JCommander;
import com.beust.jcommander.ParameterException;
import fr.meuret.web.conf.HttpConfiguration;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.Channel;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.SslProvider;
import io.netty.handler.ssl.util.SelfSignedCertificate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * A simple multihreaded and asynchronous file-based web server build on top of Netty framework.
 * <p/>
 * This server handles the HTTP/1.1 keep-alive behavior and it also implements {@code 'If-Modified-Since'} header to
 * take advantage of browser cache, as described in
 * <a href="http://tools.ietf.org/html/rfc2616#section-14.25">RFC 2616</a>.
 *
 * @see HttpStaticFileServerHandler
 */
public class HttpServer {

    private static Logger logger = LoggerFactory.getLogger(HttpServer.class);
    private HttpConfiguration configuration;

    public HttpServer(HttpConfiguration configuration) {
        this.configuration = configuration;

    }

    public HttpServer() {

        this(HttpConfiguration.defaultConfiguration());
    }

    public static void main(String[] args) throws Exception {
        //Initialize http server configuration
        HttpConfiguration.Builder configurationBuilder = new HttpConfiguration.Builder();
        //Initialize jCommander
        final JCommander jCommander = new JCommander(configurationBuilder);
        try {
            //Parse command line arguments and set values in the configuration
            jCommander.parse(args);
            final HttpServer httpServer = new HttpServer(configurationBuilder.build());
            httpServer.start();
        } catch (ParameterException e) {
            logger.error("Invalid configuration for the HTTP server.", e);
            jCommander.usage();
        }

    }

    /**
     * @throws Exception if anything goes wrong when starting the server.
     */

    public void start() throws Exception {
        logger.info("Starting HTTP server on port {}, ssl = {}, rootPath = {}", configuration.getPort(),
                configuration.useSSL(), configuration.getRootPath().toString());
        // Configure SSL.
        final SslContext sslCtx;
        if (configuration.useSSL()) {
            SelfSignedCertificate ssc = new SelfSignedCertificate();
            sslCtx = SslContext.newServerContext(SslProvider.JDK, ssc.certificate(), ssc.privateKey());
        } else {
            sslCtx = null;
        }

        final EventLoopGroup bossGroup = new NioEventLoopGroup(1);
        final EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            ServerBootstrap b = new ServerBootstrap();

            b.group(bossGroup, workerGroup).channel(NioServerSocketChannel.class)
                    .handler(new LoggingHandler(LogLevel.INFO))
                    .childHandler(new HttpStaticFileServerInitializer(sslCtx, configuration.getRootPath()));

            Channel ch = b.bind(configuration.getPort()).sync().channel();

            logger.info("Open your web browser and navigate to " + (configuration.useSSL() ? "https" : "http")
                    + "://127.0.0.1:" + configuration.getPort() + '/');

            ch.closeFuture().sync();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }

    }
}