me.desair.tus.server.util.TusServletResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tus-java-server Show documentation

Server-side implementation of the open file upload protocol tus (https://tus.io/) that supports resumable file uploads for small and very large files

There is a newer version: 1.0.0-3.0

Show newest version

package me.desair.tus.server.util;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;

import org.apache.commons.lang3.StringUtils;

/**
 * {@link HttpServletResponseWrapper} to capture header values set on the current {@link HttpServletResponse}
 */
public class TusServletResponse extends HttpServletResponseWrapper {

    private Map> headers = new HashMap<>();

    /**
     * Constructs a response adaptor wrapping the given response.
     *
     * @param response The response that has to be wrapped
     * @throws IllegalArgumentException if the response is null
     */
    public TusServletResponse(HttpServletResponse response) {
        super(response);
    }

    @Override
    public void setDateHeader(String name, long date) {
        super.setDateHeader(name, date);
        overwriteHeader(name, Objects.toString(date));
    }

    @Override
    public void addDateHeader(String name, long date) {
        super.addDateHeader(name, date);
        recordHeader(name, Objects.toString(date));
    }

    @Override
    public void setHeader(String name, String value) {
        super.setHeader(name, value);
        overwriteHeader(name, value);
    }

    @Override
    public void addHeader(String name, String value) {
        super.addHeader(name, value);
        recordHeader(name, value);
    }

    @Override
    public void setIntHeader(String name, int value) {
        super.setIntHeader(name, value);
        overwriteHeader(name, Objects.toString(value));
    }

    @Override
    public void addIntHeader(String name, int value) {
        super.addIntHeader(name, value);
        recordHeader(name, Objects.toString(value));
    }

    @Override
    public String getHeader(String name) {
        String value;
        if (headers.containsKey(name)) {
            value = headers.get(name).get(0);
        } else {
            value = super.getHeader(name);
        }
        return StringUtils.trimToNull(value);
    }

    private void recordHeader(String name, String value) {
        List values = headers.get(name);
        if (values == null) {
            values = new LinkedList<>();
            headers.put(name, values);
        }

        values.add(value);
    }

    private void overwriteHeader(String name, String value) {
        List values = new LinkedList<>();
        values.add(value);
        headers.put(name, values);
    }

}