org.restexpress.pipeline.PipelineInitializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of RestExpress Show documentation
Internet scale, high-performance RESTful Services in Java
There is a newer version: 0.11.3
/*
 * Copyright 2010, eCollege, Inc.  All rights reserved.
 */
package org.restexpress.pipeline;

import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpContentCompressor;
import io.netty.handler.codec.http.HttpContentDecompressor;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpRequestDecoder;
import io.netty.handler.codec.http.HttpResponseEncoder;
import io.netty.handler.ssl.SslHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLEngine;
import java.util.ArrayList;
import java.util.List;

/**
 * Provides a tiny DSL to define the pipeline features.
 *
 * @author toddf
 * @since Aug 27, 2010
 */
public class PipelineInitializer
        extends ChannelInitializer {
    // SECTION: CONSTANTS

    private static final int DEFAULT_MAX_CONTENT_LENGTH = 20480;


    // SECTION: INSTANCE VARIABLES

    private List requestHandlers = new ArrayList();
    private int maxContentLength = DEFAULT_MAX_CONTENT_LENGTH;
    private SSLContext sslContext = null;


    // SECTION: CONSTRUCTORS

    public PipelineInitializer() {
        super();
    }


    // SECTION: BUILDER METHODS

    public PipelineInitializer addRequestHandler(ChannelHandler handler) {
        if (!requestHandlers.contains(handler)) {
            requestHandlers.add(handler);
        }

        return this;
    }

    /**
     * Set the maximum length of the aggregated (chunked) content. If the length of the
     * aggregated content exceeds this value, a TooLongFrameException will be raised during
     * the request, which can be mapped in the RestExpress server to return a
     * BadRequestException, if desired.
     *
     * @param value
     * @return this PipelineBuilder for method chaining.
     */
    public PipelineInitializer setMaxContentLength(int value) {
        this.maxContentLength = value;
        return this;
    }

    public PipelineInitializer setSSLContext(SSLContext sslContext) {
        this.sslContext = sslContext;
        return this;
    }

    public SSLContext getSSLContext() {
        return sslContext;
    }

    // SECTION: CHANNEL PIPELINE FACTORY

    @Override
    public void initChannel(SocketChannel ch)
            throws Exception {
        ChannelPipeline pipeline = ch.pipeline();

        if (null != sslContext) {
            SSLEngine sslEngine = sslContext.createSSLEngine();
            sslEngine.setUseClientMode(false);
            SslHandler sslHandler = new SslHandler(sslEngine);
            pipeline.addLast("ssl", sslHandler);
        }

        // Inbound handlers
        pipeline.addLast("decoder", new HttpRequestDecoder());
        pipeline.addLast("inflater", new HttpContentDecompressor());

        // Outbound handlers
        pipeline.addLast("encoder", new HttpResponseEncoder());
        pipeline.addLast("chunkWriter", new ChunkedWriteHandler());
        pipeline.addLast("deflater", new HttpContentCompressor());

        // Aggregator MUST be added last, otherwise results are not correct
        pipeline.addLast("aggregator", new HttpObjectAggregator(maxContentLength));

        for (ChannelHandler handler : requestHandlers) {
            pipeline.addLast(handler.getClass().getSimpleName(), handler);
        }
    }
}