org.jboss.resteasy.reactive.server.multipart.MultipartFormDataOutput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of resteasy-reactive Show documentation

There is a newer version: 3.17.5

package org.jboss.resteasy.reactive.server.multipart;

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import jakarta.ws.rs.core.MediaType;

/**
 * Used when a Resource method needs to return a multipart output
 */
public final class MultipartFormDataOutput {
    private final Map> parts = new LinkedHashMap<>();

    /**
     * @deprecated use {@link #getAllFormData()} instead
     */
    @Deprecated(forRemoval = true)
    public Map getFormData() {
        Map result = new LinkedHashMap<>();
        for (var entry : parts.entrySet()) {
            if (entry.getValue().isEmpty()) {
                continue;
            }
            // use the last item inserted as this is the old behavior
            int lastIndex = entry.getValue().size() - 1;
            result.put(entry.getKey(), entry.getValue().get(lastIndex));
        }
        return Collections.unmodifiableMap(result);
    }

    public Map> getAllFormData() {
        return Collections.unmodifiableMap(parts);
    }

    public PartItem addFormData(String key, Object entity, MediaType mediaType) {
        return addFormData(key, entity, null, mediaType);
    }

    public PartItem addFormData(String key, Object entity, String genericType, MediaType mediaType) {
        PartItem part = new PartItem(entity, genericType, mediaType);
        return addFormData(key, part);
    }

    public PartItem addFormData(String key, Object entity, MediaType mediaType, String filename) {
        PartItem part = new PartItem(entity, null, mediaType, filename);
        return addFormData(key, part);
    }

    private PartItem addFormData(String key, PartItem part) {
        List items = parts.computeIfAbsent(key, k -> new ArrayList<>());
        items.add(part);
        return part;
    }
}