org.sigmah.server.file.util.MultipartRequest.java Source code

Introduction

Here is the source code for org.sigmah.server.file.util.MultipartRequest.java

Source

package org.sigmah.server.file.util;

/*
 * #%L
 * Sigmah
 * %%
 * Copyright (C) 2010 - 2016 URD
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import com.google.gwt.http.client.Response;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import org.apache.commons.fileupload.FileItemIterator;
import org.apache.commons.fileupload.FileItemStream;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.util.Streams;
import org.sigmah.server.servlet.base.StatusServletException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Parse a multipart request and allow access to its data as a map of <code>String</code> and <code>InputStream</code>.
 * 
 * @author Raphal Calabro (rcalabro@ideia.fr)
 */
public class MultipartRequest {
    private static final Logger LOGGER = LoggerFactory.getLogger(MultipartRequest.class);

    private final Map<String, String> properties;
    private final HttpServletRequest request;

    public MultipartRequest(HttpServletRequest request) {
        this.request = request;
        this.properties = new HashMap<>();
    }

    public void parse(MultipartRequestCallback callback)
            throws IOException, FileUploadException, StatusServletException {
        if (!ServletFileUpload.isMultipartContent(request)) {
            LOGGER.error("Request content is not multipart.");
            throw new StatusServletException(Response.SC_PRECONDITION_FAILED);
        }

        final FileItemIterator iterator = new ServletFileUpload(new DiskFileItemFactory()).getItemIterator(request);
        while (iterator.hasNext()) {
            // Gets the first HTTP request element.
            final FileItemStream item = iterator.next();

            if (item.isFormField()) {
                final String value = Streams.asString(item.openStream(), "UTF-8");
                properties.put(item.getFieldName(), value);

            } else if (callback != null) {
                callback.onInputStream(item.openStream(), item.getFieldName(), item.getContentType());
            }
        }
    }

    public String getProperty(String property) {
        return properties.get(property);
    }

    public Map<String, String> getProperties() {
        return properties;
    }
}