All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.micronaut.http.netty.NettyHttpResponseBuilder Maven / Gradle / Ivy

There is a newer version: 4.7.5
Show newest version
/*
 * Copyright 2017-2020 original authors
 *
 * 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
 *
 * https://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.micronaut.http.netty;

import io.micronaut.core.annotation.NonNull;
import io.micronaut.core.async.publisher.Publishers;
import io.micronaut.http.HttpResponseWrapper;
import io.micronaut.http.netty.stream.DefaultStreamedHttpResponse;
import io.micronaut.http.netty.stream.StreamedHttpResponse;
import io.netty.buffer.ByteBuf;
import io.netty.handler.codec.http.DefaultFullHttpResponse;
import io.netty.handler.codec.http.DefaultHttpResponse;
import io.netty.handler.codec.http.DefaultLastHttpContent;
import io.netty.handler.codec.http.FullHttpResponse;
import io.netty.handler.codec.http.HttpResponse;
import io.netty.handler.codec.http.HttpResponseStatus;
import io.netty.handler.codec.http.HttpVersion;

import java.util.Objects;

/**
 * Common interface for client and server to implement to construct the Netty versions of the response objects.
 *
 * @author graemerocher
 * @since 2.0.0
 */
public interface NettyHttpResponseBuilder {

    /**
     * Converts this object to a full http response.
     *
     * @return a full http response
     */
    @NonNull FullHttpResponse toFullHttpResponse();

    /**
     * Converts this object to a streamed http response.
     * @return The streamed response
     */
    default @NonNull StreamedHttpResponse toStreamHttpResponse() {
        FullHttpResponse fullHttpResponse = toFullHttpResponse();
        DefaultStreamedHttpResponse streamedHttpResponse = new DefaultStreamedHttpResponse(
                fullHttpResponse.protocolVersion(),
                fullHttpResponse.status(),
                true,
                Publishers.just(new DefaultLastHttpContent(fullHttpResponse.content()))
        );
        streamedHttpResponse.headers().setAll(fullHttpResponse.headers());
        return streamedHttpResponse;
    }

    /**
     * Converts this object to the most appropriate http response type.
     * @return The http response
     */
    default @NonNull HttpResponse toHttpResponse() {
        if (isStream()) {
            return toStreamHttpResponse();
        } else {
            return toFullHttpResponse();
        }
    }

    /**
     * @return Is the response a stream.
     */
    boolean isStream();

    /**
     * Convert the given response to a full http response.
     * @param response The response
     * @return The full response.
     */
    static @NonNull HttpResponse toHttpResponse(@NonNull io.micronaut.http.HttpResponse response) {
        Objects.requireNonNull(response, "The response cannot be null");
        while (response instanceof HttpResponseWrapper wrapper) {
            response = wrapper.getDelegate();
        }
        HttpResponse fullHttpResponse;
        if (response instanceof NettyHttpResponseBuilder builder) {
            fullHttpResponse = builder.toHttpResponse();
        } else {
            // manual conversion
            ByteBuf byteBuf = response.getBody(ByteBuf.class).orElse(null);
            if (byteBuf != null) {
                fullHttpResponse = new DefaultFullHttpResponse(
                    HttpVersion.HTTP_1_1,
                    HttpResponseStatus.valueOf(response.code(), response.reason()),
                    byteBuf
                );
            } else {
                fullHttpResponse = new DefaultHttpResponse(
                    HttpVersion.HTTP_1_1,
                    HttpResponseStatus.valueOf(response.code(), response.reason())
                );
            }

            response.getHeaders()
                .forEach((s, strings) -> fullHttpResponse.headers().add(s, strings));
        }
        return fullHttpResponse;
    }

    /**
     * Convert the given response to a full http response.
     * @param response The response
     * @param byteBuf The byteBuf
     * @return The full response.
     * @since 4.3.0
     */
    static @NonNull FullHttpResponse toFullHttpResponse(@NonNull io.micronaut.http.HttpResponse response, @NonNull ByteBuf byteBuf) {
        while (response instanceof HttpResponseWrapper wrapper) {
            response = wrapper.getDelegate();
        }
        DefaultFullHttpResponse httpResponse = new DefaultFullHttpResponse(
            HttpVersion.HTTP_1_1,
            HttpResponseStatus.valueOf(response.code(), response.reason()),
            byteBuf
        );
        response.getHeaders()
            .forEach((s, strings) -> httpResponse.headers().add(s, strings));
        return httpResponse;
    }

    /**
     * Convert the given response to a full http response.
     * @param response The response
     * @return The full response.
     */
    static @NonNull StreamedHttpResponse toStreamResponse(@NonNull io.micronaut.http.HttpResponse response) {
        Objects.requireNonNull(response, "The response cannot be null");
        while (response instanceof HttpResponseWrapper wrapper) {
            response = wrapper.getDelegate();
        }
        if (response instanceof NettyHttpResponseBuilder builder) {
            if (builder.isStream()) {
                return builder.toStreamHttpResponse();
            } else {
                FullHttpResponse fullHttpResponse = builder.toFullHttpResponse();
                return new DefaultStreamedHttpResponse(
                        HttpVersion.HTTP_1_1,
                        HttpResponseStatus.valueOf(response.code(), response.reason()),
                        Publishers.just(new DefaultLastHttpContent(fullHttpResponse.content()))
                );
            }
        }
        // manual conversion
        StreamedHttpResponse fullHttpResponse;
        ByteBuf byteBuf = response.getBody(ByteBuf.class).orElse(null);
        if (byteBuf != null) {
            fullHttpResponse = new DefaultStreamedHttpResponse(
                    HttpVersion.HTTP_1_1,
                    HttpResponseStatus.valueOf(response.code(), response.reason()),
                    Publishers.just(new DefaultLastHttpContent(byteBuf))
            );
        } else {
            fullHttpResponse = new DefaultStreamedHttpResponse(
                    HttpVersion.HTTP_1_1,
                    HttpResponseStatus.valueOf(response.code(), response.reason()),
                    Publishers.empty()
            );
        }

        response.getHeaders()
                .forEach((s, strings) -> fullHttpResponse.headers().add(s, strings));
        return fullHttpResponse;
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy