io.reactivex.netty.protocol.http.server.ContentWriterImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rxnetty-http Show documentation
rxnetty-http
There is a newer version: 0.5.3-rc.2
/*
 * Copyright 2015 Netflix, Inc.
 *
 * 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.reactivex.netty.protocol.http.server;

import io.netty.handler.codec.http.DefaultLastHttpContent;
import io.netty.handler.codec.http.HttpResponse;
import io.netty.handler.codec.http.HttpUtil;
import io.netty.handler.codec.http.LastHttpContent;
import io.reactivex.netty.channel.ChannelOperations;
import io.reactivex.netty.channel.Connection;
import io.reactivex.netty.channel.FlushSelectorOperator;
import io.reactivex.netty.protocol.http.TrailingHeaders;
import io.reactivex.netty.protocol.http.internal.OperatorTrailer;
import rx.Observable;
import rx.Subscriber;
import rx.functions.Func0;
import rx.functions.Func1;
import rx.functions.Func2;

import static io.netty.handler.codec.http.HttpHeaderNames.*;

final class ContentWriterImpl extends ResponseContentWriter {

    @SuppressWarnings("rawtypes")
    private final Connection connection;

    @SuppressWarnings("rawtypes")
    private final Observable headersObservable;
    @SuppressWarnings("rawtypes")
    private final Observable contentObservable;

    private final HttpResponse headers;

    private final Func1 flushOnEachSelector = new Func1() {
        @Override
        public Boolean call(C w) {
            return true;
        }
    };

    ContentWriterImpl(@SuppressWarnings("rawtypes") final Connection connection, final HttpResponse headers) {
        super(new OnSubscribe() {
            @SuppressWarnings("unchecked")
            @Override
            public void call(Subscriber subscriber) {
                /*We are never sending content as the subscription is to the headers only writer.*/
                if (!HttpUtil.isTransferEncodingChunked(headers)) {
                    headers.headers().set(CONTENT_LENGTH, 0);
                }
                connection.write(Observable.just(headers)).unsafeSubscribe(subscriber);
            }
        });
        this.connection = connection;
        this.headers = headers;
        headersObservable = Observable.just(headers);
        contentObservable = null;
    }

    private ContentWriterImpl(final ContentWriterImpl parent,
                              @SuppressWarnings("rawtypes") final Observable content, final boolean appendTrailer) {
        super(new OnSubscribe() {
            @SuppressWarnings("unchecked")
            @Override
            public void call(Subscriber subscriber) {
                parent.connection.write(getHttpStream(parent, content, appendTrailer))
                                 .unsafeSubscribe(subscriber);
            }
        });
        connection = parent.connection;
        headers = parent.headers;
        headersObservable = parent.headersObservable;
        if (null == parent.contentObservable) {
            contentObservable = content;
        } else {
            @SuppressWarnings({"rawtypes", "unchecked"})
            Observable rawMerged = parent.contentObservable.mergeWith(content);
            contentObservable = rawMerged;
        }
    }

    @Override
    public ResponseContentWriter write(Observable msgs) {
        return new ContentWriterImpl<>(this, msgs, true);
    }

    @Override
    public  Observable write(Observable contentSource,
                                                              Func0 trailerFactory,
                                                              Func2 trailerMutator) {
        @SuppressWarnings("rawtypes")
        Observable rawObservable = contentSource;
        return new ContentWriterImpl<>(this, OperatorTrailer.liftFrom(rawObservable, trailerFactory, trailerMutator),
                                       false);
    }

    @Override
    public  Observable write(Observable contentSource, Func0 trailerFactory,
                                                              Func2 trailerMutator,
                                                              Func1 flushSelector) {
        return write(contentSource.lift(new FlushSelectorOperator<>(flushSelector, connection)),
                     trailerFactory, trailerMutator);
    }

    @Override
    public ResponseContentWriter write(Observable msgs, final Func1 flushSelector) {
        return new ContentWriterImpl<>(this, msgs.lift(new FlushSelectorOperator<>(flushSelector, connection)),
                                       true);
    }

    @Override
    public ResponseContentWriter writeAndFlushOnEach(Observable msgs) {
        return write(msgs, flushOnEachSelector);
    }

    @Override
    public ResponseContentWriter writeString(Observable msgs) {
        return new ContentWriterImpl<>(this, msgs, true);
    }

    @Override
    public  Observable writeString(Observable contentSource,
                                                                    Func0 trailerFactory,
                                                                    Func2 trailerMutator) {
        @SuppressWarnings("rawtypes")
        Observable rawObservable = contentSource;
        return new ContentWriterImpl<>(this, OperatorTrailer.liftFrom(rawObservable, trailerFactory, trailerMutator),
                                       false);
    }

    @Override
    public  Observable writeString(Observable contentSource,
                                                                    Func0 trailerFactory,
                                                                    Func2 trailerMutator,
                                                                    Func1 flushSelector) {
        @SuppressWarnings("rawtypes")
        Observable rawObservable = contentSource.lift(new FlushSelectorOperator<>(flushSelector, connection));
        return new ContentWriterImpl<>(this, OperatorTrailer.liftFrom(rawObservable, trailerFactory, trailerMutator),
                                       false);
    }

    @Override
    public ResponseContentWriter writeString(Observable msgs, Func1 flushSelector) {
        return new ContentWriterImpl<>(this, msgs.lift(new FlushSelectorOperator<>(flushSelector, connection)),
                                       true);
    }

    @Override
    public ResponseContentWriter writeStringAndFlushOnEach(Observable msgs) {
        return writeString(msgs, ChannelOperations.FLUSH_ON_EACH_STRING);
    }

    @Override
    public ResponseContentWriter writeBytes(Observable msgs) {
        return new ContentWriterImpl<>(this, msgs, true);
    }

    @Override
    public  Observable writeBytes(Observable contentSource,
                                                                   Func0 trailerFactory,
                                                                   Func2 trailerMutator) {
        @SuppressWarnings("rawtypes")
        Observable rawObservable = contentSource;
        return new ContentWriterImpl<>(this, OperatorTrailer.liftFrom(rawObservable, trailerFactory, trailerMutator),
                                       false);
    }

    @Override
    public  Observable writeBytes(Observable contentSource,
                                                                   Func0 trailerFactory,
                                                                   Func2 trailerMutator,
                                                                   Func1 flushSelector) {
        @SuppressWarnings("rawtypes")
        Observable rawObservable = contentSource.lift(new FlushSelectorOperator<>(flushSelector, connection));
        return new ContentWriterImpl<>(this, OperatorTrailer.liftFrom(rawObservable, trailerFactory, trailerMutator),
                                       false);
    }

    @Override
    public ResponseContentWriter writeBytes(Observable msgs, Func1 flushSelector) {
        return new ContentWriterImpl<>(this, msgs.lift(new FlushSelectorOperator<>(flushSelector, connection)),
                                       true);
    }

    @Override
    public ResponseContentWriter writeBytesAndFlushOnEach(Observable msgs) {
        return writeBytes(msgs, ChannelOperations.FLUSH_ON_EACH_BYTES);
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    private static Observable getHttpStream(final ContentWriterImpl parent, Observable content, boolean appendTrailer) {
        Observable httpStream = parent.headersObservable;
        if (null != parent.contentObservable) {
            httpStream = httpStream.concatWith(parent.contentObservable.mergeWith(content));
        } else {
            httpStream = httpStream.concatWith(content);
        }

        if (appendTrailer) {
            httpStream = httpStream.concatWith(Observable.just(LastHttpContent.EMPTY_LAST_CONTENT));
        }

        return httpStream;
    }
}