io.micronaut.http.server.netty.body.ImmediateMultiObjectBody Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017-2023 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.server.netty.body;

import io.micronaut.core.annotation.Internal;
import io.micronaut.core.util.CollectionUtils;
import io.micronaut.http.server.exceptions.InternalServerException;
import io.micronaut.http.server.netty.FormRouteCompleter;
import io.micronaut.http.server.netty.MicronautHttpData;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.buffer.ByteBufHolder;
import io.netty.buffer.ByteBufInputStream;
import io.netty.buffer.CompositeByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.util.ReferenceCounted;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;

import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;

/**
 * Immediate {@link MultiObjectBody}, all operations are eager.
 *
 * @since 4.0.0
 * @author Jonas Konrad
 */
@Internal
public final class ImmediateMultiObjectBody extends ManagedBody> implements MultiObjectBody {
    ImmediateMultiObjectBody(List objects) {
        super(objects);
    }

    @Override
    void release(List value) {
        value.forEach(ImmediateSingleObjectBody::release0);
    }

    public ImmediateSingleObjectBody single(Charset defaultCharset, ByteBufAllocator alloc) {
        List objects = prepareClaim();
        if (objects.isEmpty()) {
            return next(new ImmediateSingleObjectBody(null));
        }
        boolean allFormData = true;
        for (Object object : objects) {
            if (!(object instanceof MicronautHttpData)) {
                allFormData = false;
                break;
            }
        }
        if (allFormData) {
            //noinspection unchecked
            List> data = (List>) objects;
            Map map = toMap(defaultCharset, data);
            for (MicronautHttpData datum : data) {
                datum.release();
            }
            return next(new ImmediateSingleObjectBody(map));
        }
        if (objects.size() == 1) {
            Object o = objects.get(0);
            return next(new ImmediateSingleObjectBody(o instanceof ByteBufHolder bbh ? bbh.content() : o));
        }
        return next(new ImmediateSingleObjectBody(coerceToComposite(objects, alloc)));
    }

    private static CompositeByteBuf coerceToComposite(List objects, ByteBufAllocator alloc) {
        CompositeByteBuf composite = alloc.compositeBuffer();
        for (Object object : objects) {
            composite.addComponent(true, (ByteBuf) object);
        }
        return composite;
    }

    public static Map toMap(Charset charset, Collection> dataList) {
        Map singleMap = CollectionUtils.newLinkedHashMap(dataList.size());
        Map> multiMap = new LinkedHashMap<>();
        for (MicronautHttpData data : dataList) {
            String key = data.getName();
            String newValue;
            try {
                newValue = data.getString(charset);
            } catch (IOException e) {
                throw new InternalServerException("Error retrieving or decoding the value for: " + data.getName());
            }
            List