Example usage for org.apache.commons.fileupload FileItemIterator next

List of usage examples for org.apache.commons.fileupload FileItemIterator next

Introduction

In this page you can find the example usage for org.apache.commons.fileupload FileItemIterator next.

Prototype

FileItemStream next() throws FileUploadException, IOException;

Source Link

Document

Returns the next available FileItemStream .

Usage

From source file:org.sigmah.server.endpoint.export.sigmah.ExportModelServlet.java

@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

    if (ServletFileUpload.isMultipartContent(req)) {

        final Authentication authentication = retrieveAuthentication(req);
        boolean hasPermission = false;

        if (authentication != null) {
            final User user = authentication.getUser();
            ProfileDTO profile = SigmahAuthDictionaryServlet.aggregateProfiles(user, null, injector);
            hasPermission = ProfileUtils.isGranted(profile, GlobalPermissionEnum.VIEW_ADMIN);
        }// w w w  .  ja  v  a 2s. c o  m

        if (hasPermission) {

            final ServletFileUpload fileUpload = new ServletFileUpload();

            final HashMap<String, String> properties = new HashMap<String, String>();
            byte[] data = null;

            try {
                final FileItemIterator iterator = fileUpload.getItemIterator(req);

                // Iterating on the fields sent into the request
                while (iterator.hasNext()) {

                    final FileItemStream item = iterator.next();
                    final String name = item.getFieldName();

                    final InputStream stream = item.openStream();

                    if (item.isFormField()) {
                        final String value = Streams.asString(stream);
                        LOG.debug("field '" + name + "' = '" + value + '\'');

                        // The current field is a property
                        properties.put(name, value);

                    } else {
                        // The current field is a file
                        LOG.debug("field '" + name + "' (FILE)");

                        final ByteArrayOutputStream serializedData = new ByteArrayOutputStream();
                        long dataSize = 0L;

                        int b = stream.read();

                        while (b != -1 && dataSize < MAXIMUM_FILE_SIZE) {
                            serializedData.write(b);

                            dataSize++;
                            b = stream.read();
                        }

                        stream.close();

                        data = serializedData.toByteArray();
                    }
                }

            } catch (FileUploadException ex) {
                LOG.warn("Error while receiving a serialized model.", ex);
            }

            if (data != null) {
                // A file has been received

                final String type = properties.get("type");
                final ModelHandler handler = handlers.get(type);

                if (handler != null) {

                    if (handler instanceof ProjectModelHandler) {
                        final String projectModelTypeAsString = properties.get("project-model-type");
                        try {
                            final ProjectModelType projectModelType = ProjectModelType
                                    .valueOf(projectModelTypeAsString);
                            ((ProjectModelHandler) handler).setProjectModelType(projectModelType);
                        } catch (IllegalArgumentException e) {
                            LOG.debug("Bad value for project model type: " + projectModelTypeAsString, e);
                        }
                    }

                    final ByteArrayInputStream inputStream = new ByteArrayInputStream(data);

                    try {
                        handler.importModel(inputStream, injector.getInstance(EntityManager.class),
                                authentication);

                    } catch (ExportException ex) {
                        LOG.error("Model import error, type: " + type, ex);
                        resp.sendError(500);
                    }

                } else {
                    LOG.warn("The asked model type (" + type + ") doesn't have any handler registered.");
                    resp.sendError(501);
                }
            } else {
                LOG.warn("No file has been received.");
                resp.sendError(400);
            }
        } else {
            LOG.warn("Unauthorized access to the import service from user " + authentication);
            resp.sendError(401);
        }

    } else {
        LOG.warn("The request doesn't have the correct enctype.");
        resp.sendError(400);
    }
}

From source file:org.sigmah.server.file.util.MultipartRequest.java

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);
    }//from  ww  w  .ja  va 2  s .  c  o m

    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());
        }
    }
}

From source file:org.sigmah.server.servlet.ImportServlet.java

private byte[] readFileAndProperties(final HttpServletRequest request, final Map<String, String> properties)
        throws FileUploadException, IOException {
    byte[] data = null;
    final ServletFileUpload fileUpload = new ServletFileUpload();

    final FileItemIterator iterator = fileUpload.getItemIterator(request);

    // Iterating on the fields sent into the request
    while (iterator.hasNext()) {

        final FileItemStream item = iterator.next();
        final String name = item.getFieldName();

        final InputStream stream = item.openStream();

        if (item.isFormField()) {

            final String value = Streams.asString(stream);

            LOG.debug("field '" + name + "' = '" + value + '\'');

            // The current field is a property
            properties.put(name, value);

        } else {/*from   w w w.j av  a  2  s. c om*/
            // The current field is a file
            LOG.debug("field '" + name + "' (FILE)");

            final ByteArrayOutputStream serializedData = new ByteArrayOutputStream();
            long dataSize = 0L;

            int b = stream.read();

            while (b != -1 && dataSize < MAXIMUM_FILE_SIZE) {
                serializedData.write(b);

                dataSize++;
                b = stream.read();
            }

            stream.close();

            data = serializedData.toByteArray();
        }
    }

    return data;
}

From source file:org.sosy_lab.cpachecker.appengine.server.resource.TasksServerResource.java

@Override
public Representation createTaskFromHtml(Representation input) throws IOException {
    List<String> errors = new LinkedList<>();
    Map<String, Object> settings = new HashMap<>();
    Map<String, String> options = new HashMap<>();

    ServletFileUpload upload = new ServletFileUpload();
    try {/*from  w  ww.  j  av  a 2s  .  c om*/
        FileItemIterator iter = upload.getItemIterator(ServletUtils.getRequest(getRequest()));
        while (iter.hasNext()) {
            FileItemStream item = iter.next();
            InputStream stream = item.openStream();
            if (item.isFormField()) {
                String value = Streams.asString(stream);
                switch (item.getFieldName()) {
                case "specification":
                    value = (value.equals("")) ? null : value;
                    settings.put("specification", value);
                    break;
                case "configuration":
                    value = (value.equals("")) ? null : value;
                    settings.put("configuration", value);
                    break;
                case "disableOutput":
                    options.put("output.disable", "true");
                    break;
                case "disableExportStatistics":
                    options.put("statistics.export", "false");
                    break;
                case "dumpConfig":
                    options.put("configuration.dumpFile", "UsedConfiguration.properties");
                    break;
                case "logLevel":
                    options.put("log.level", value);
                    break;
                case "machineModel":
                    options.put("analysis.machineModel", value);
                    break;
                case "wallTime":
                    options.put("limits.time.wall", value);
                    break;
                case "instanceType":
                    options.put("gae.instanceType", value);
                    break;
                case "programText":
                    if (!value.isEmpty()) {
                        settings.put("programName", "program.c");
                        settings.put("programText", value);
                    }
                    break;
                }
            } else {
                if (settings.get("programText") == null) {
                    settings.put("programName", item.getName());
                    settings.put("programText", IOUtils.toString(stream));
                }
            }
        }
    } catch (FileUploadException | IOException e) {
        getLogger().log(Level.WARNING, "Could not upload program file.", e);
        errors.add("task.program.CouldNotUpload");
    }

    settings.put("options", options);

    Task task = null;
    if (errors.isEmpty()) {
        TaskBuilder taskBuilder = new TaskBuilder();
        task = taskBuilder.fromMap(settings);
        errors = taskBuilder.getErrors();
    }

    if (errors.isEmpty()) {
        try {
            Configuration config = Configuration.builder().setOptions(task.getOptions()).build();
            new GAETaskQueueTaskExecutor(config).execute(task);
        } catch (InvalidConfigurationException e) {
            errors.add("error.invalidConfiguration");
        }
    }

    if (errors.isEmpty()) {
        getResponse().setStatus(Status.SUCCESS_CREATED);
        redirectSeeOther("/tasks/" + task.getKey());
        return getResponseEntity();
    }

    getResponse().setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
    return FreemarkerUtil.templateBuilder().context(getContext()).addData("task", task)
            .addData("errors", errors).addData("allowedOptions", DefaultOptions.getDefaultOptions())
            .addData("defaultOptions", DefaultOptions.getImmutableOptions())
            .addData("specifications", DefaultOptions.getSpecifications())
            .addData("configurations", DefaultOptions.getConfigurations())
            .addData("cpacheckerVersion", CPAchecker.getCPAcheckerVersion()).templateName("root.ftl").build();
}

From source file:org.synchronoss.cloud.nio.multipart.example.web.MultipartController.java

/**
 * <p> Example of parsing the multipart request using commons file upload. In this case the parsing happens in blocking io.
 *
 * @param request The {@code HttpServletRequest}
 * @return The {@code VerificationItems}
 * @throws Exception if an exception happens during the parsing
 *//*from   ww  w.  j a  v a  2  s.c  o m*/
@RequestMapping(value = "/blockingio/fileupload/multipart", method = RequestMethod.POST)
public @ResponseBody VerificationItems blockingIoMultipart(final HttpServletRequest request) throws Exception {

    assertRequestIsMultipart(request);

    final ServletFileUpload servletFileUpload = new ServletFileUpload();
    final FileItemIterator fileItemIterator = servletFileUpload.getItemIterator(request);

    final VerificationItems verificationItems = new VerificationItems();
    Metadata metadata = null;
    while (fileItemIterator.hasNext()) {
        FileItemStream fileItemStream = fileItemIterator.next();
        if (METADATA_FIELD_NAME.equals(fileItemStream.getFieldName())) {
            if (metadata != null) {
                throw new IllegalStateException("Found more than one metadata field");
            }
            metadata = unmarshalMetadata(fileItemStream.openStream());
        } else {
            VerificationItem verificationItem = buildVerificationItem(fileItemStream.openStream(),
                    fileItemStream.getFieldName());
            verificationItems.getVerificationItems().add(verificationItem);
        }
    }
    processVerificationItems(verificationItems, metadata, false);
    return verificationItems;
}

From source file:org.synchronoss.cloud.nio.multipart.FunctionalTest.java

@Test
public void blockingIOAdapterFunctionalTest() throws Exception {

    log.info("BLOCKING IO ADAPTER FUNCTIONAL TEST [ " + testCase.getDescription() + " ]");

    if (log.isDebugEnabled()) {
        log.debug("Request body\n" + IOUtils.toString(testCase.getBodyInputStream()));
    }//from   w w w .j  a va 2 s . c om

    final FileUpload fileUpload = new FileUpload();
    final FileItemIterator fileItemIterator = fileUpload.getItemIterator(testCase.getRequestContext());

    try (final CloseableIterator<PartItem> parts = Multipart.multipart(testCase.getMultipartContext())
            .forBlockingIO(testCase.getBodyInputStream())) {

        while (parts.hasNext()) {

            BlockingIOAdapter.PartItem partItem = parts.next();
            BlockingIOAdapter.PartItem.Type partItemType = partItem.getType();
            if (BlockingIOAdapter.PartItem.Type.NESTED_END.equals(partItemType)
                    || BlockingIOAdapter.PartItem.Type.NESTED_START.equals(partItemType)) {
                // Commons file upload is not returning an item representing the start/end of a nested multipart.
                continue;
            }
            assertTrue(fileItemIterator.hasNext());
            FileItemStream fileItemStream = fileItemIterator.next();
            assertEquals(partItem, fileItemStream);
        }
    }

}

From source file:org.synchronoss.cloud.nio.multipart.FunctionalTest.java

NioMultipartParserListener nioMultipartParserListenerVerifier(final FileItemIterator fileItemIterator,
        final AtomicBoolean finished) {

    return new NioMultipartParserListener() {

        AtomicInteger partIndex = new AtomicInteger(0);

        @Override/* w ww .  j  a  v a  2s. c om*/
        public void onPartFinished(StreamStorage partBodyStreamStorage,
                Map<String, List<String>> headersFromPart) {
            log.info("<<<<< On part complete [" + (partIndex.addAndGet(1)) + "] >>>>>>");
            assertFileItemIteratorHasNext(true);
            final FileItemStream fileItemStream = fileItemIteratorNext();
            assertHeadersAreEqual(fileItemStream.getHeaders(), headersFromPart);
            assertInputStreamsAreEqual(fileItemStreamInputStream(fileItemStream),
                    partBodyStreamStorage.getInputStream());
        }

        @Override
        public void onFormFieldPartFinished(String fieldName, String fieldValue,
                Map<String, List<String>> headersFromPart) {
            log.info("<<<<< On form field complete [" + (partIndex.addAndGet(1)) + "] >>>>>>");
            assertFileItemIteratorHasNext(true);
            final FileItemStream fileItemStream = fileItemIteratorNext();
            assertTrue(fileItemStream.isFormField());
            Assert.assertEquals(fieldName, fileItemStream.getFieldName());
            try {
                Assert.assertEquals(fieldValue, IOUtils.toString(fileItemStream.openStream()));
            } catch (Exception e) {
                throw new IllegalStateException("Unable to assert field value", e);
            }
        }

        @Override
        public void onNestedPartStarted(Map<String, List<String>> headersFromParentPart) {
            log.info("<<<<< On form field complete [" + (partIndex) + "] >>>>>>");
        }

        @Override
        public void onAllPartsFinished() {
            log.info("<<<<< On all parts read: Number of parts [" + partIndex.get() + "] >>>>>>");
            assertFileItemIteratorHasNext(false);
            finished.set(true);
        }

        @Override
        public void onNestedPartFinished() {
            log.info("<<<<< On form field complete [" + (partIndex) + "] >>>>>>");
        }

        @Override
        public void onError(String message, Throwable cause) {
            log.info("<<<<< On error. Part " + partIndex.get() + "] >>>>>>");
            finished.set(true);
            log.error(message, cause);
            fail("Got an error from the parser");
        }

        InputStream fileItemStreamInputStream(final FileItemStream fileItemStream) {
            try {
                return fileItemStream.openStream();
            } catch (Exception e) {
                throw new IllegalStateException("Unable to open the file item inputstream", e);
            }
        }

        void assertFileItemIteratorHasNext(boolean hasNext) {
            try {
                assertTrue("File iterator has next is not " + hasNext, hasNext == fileItemIterator.hasNext());
            } catch (Exception e) {
                throw new IllegalStateException("Unable to verify if the FileItemIterator has a next", e);
            }
        }

        FileItemStream fileItemIteratorNext() {
            try {
                return fileItemIterator.next();
            } catch (Exception e) {
                throw new IllegalStateException("Unable to retrieve the next FileItemStream", e);
            }
        }

    };
}

From source file:org.synchronoss.cloud.nio.multipart.FunctionalTest.java

void dumpFileIterator(final FileItemIterator fileItemIterator) {

    int partIndex = 0;

    try {/*from  ww  w  . ja va2  s .  com*/
        log.info("-- COMMONS FILE UPLOAD --");
        while (fileItemIterator.hasNext()) {
            log.info("-- Part " + partIndex++);
            FileItemStream fileItemStream = fileItemIterator.next();

            FileItemHeaders fileItemHeaders = fileItemStream.getHeaders();
            Iterator<String> headerNames = fileItemHeaders.getHeaderNames();
            while (headerNames.hasNext()) {
                String headerName = headerNames.next();
                log.info("Header: " + headerName + ": "
                        + Joiner.on(',').join(fileItemHeaders.getHeaders(headerName)));
            }
            log.info("Body:\n" + IOUtils.toString(fileItemStream.openStream()));
        }
        log.info("-- ------------------- --");
    } catch (Exception e) {
        log.error("Error dumping the FileItemIterator", e);
    }

}

From source file:org.tangram.spring.StreamingMultipartResolver.java

@Override
public MultipartHttpServletRequest resolveMultipart(HttpServletRequest request) throws MultipartException {
    ServletFileUpload upload = new ServletFileUpload();
    upload.setFileSizeMax(maxUploadSize);
    String encoding = determineEncoding(request);
    Map<String, String[]> multipartParameters = new HashMap<>();
    MultiValueMap<String, MultipartFile> multipartFiles = new LinkedMultiValueMap<>();
    Map<String, String> multipartFileContentTypes = new HashMap<>();

    try {/*from  w w  w.  ja  v a  2 s .c o  m*/
        FileItemIterator iter = upload.getItemIterator(request);
        while (iter.hasNext()) {
            FileItemStream item = iter.next();

            String name = item.getFieldName();
            InputStream stream = item.openStream();
            if (item.isFormField()) {
                String value = Streams.asString(stream, encoding);
                String[] curParam = multipartParameters.get(name);
                if (curParam == null) {
                    // simple form field
                    multipartParameters.put(name, new String[] { value });
                } else {
                    // array of simple form fields
                    String[] newParam = StringUtils.addStringToArray(curParam, value);
                    multipartParameters.put(name, newParam);
                }
            } else {
                try {
                    MultipartFile file = new StreamingMultipartFile(item);
                    multipartFiles.add(name, file);
                    multipartFileContentTypes.put(name, file.getContentType());
                } catch (final IOException e) {
                    LOG.warn("({})", e.getCause().getMessage(), e);
                    MultipartFile file = new MultipartFile() {

                        @Override
                        public String getName() {
                            return "";
                        }

                        @Override
                        public String getOriginalFilename() {
                            return e.getCause().getMessage();
                        }

                        @Override
                        public String getContentType() {
                            return ERROR;
                        }

                        @Override
                        public boolean isEmpty() {
                            return true;
                        }

                        @Override
                        public long getSize() {
                            return 0L;
                        }

                        @Override
                        public byte[] getBytes() throws IOException {
                            return new byte[0];
                        }

                        @Override
                        public InputStream getInputStream() throws IOException {
                            return null;
                        }

                        @Override
                        public void transferTo(File file) throws IOException, IllegalStateException {
                            throw new UnsupportedOperationException("NYI", e);
                        }
                    };
                    multipartFiles.add(name, file);
                    multipartFileContentTypes.put(name, file.getContentType());
                } // try/catch
            } // if
        } // while
    } catch (IOException | FileUploadException e) {
        throw new MultipartException("Error uploading a file", e);
    } // try/catch

    return new DefaultMultipartHttpServletRequest(request, multipartFiles, multipartParameters,
            multipartFileContentTypes);
}

From source file:org.ut.biolab.medsavant.MedSavantServlet.java

private Upload[] handleUploads(FileItemIterator iter)
        throws FileUploadException, IOException, InterruptedException {
    List<Upload> uploads = new ArrayList<Upload>();

    FileItemStream streamToUpload = null;
    long filesize = -1;

    String sn = null;/*from   w w w  . j a  v a  2  s.c  om*/
    String fn = null;

    while (iter.hasNext()) {

        FileItemStream item = iter.next();
        String name = item.getFieldName();
        InputStream stream = item.openStream();
        // System.out.println("Got file " + name);
        if (item.isFormField()) {
            if (name.startsWith("size_")) {
                sn = name.split("_")[1];
                filesize = Long.parseLong(Streams.asString(stream));
            }
        } else if (name.startsWith("file_")) {
            streamToUpload = item;
        } else {
            throw new IllegalArgumentException("Unrecognized file detected with field name " + name);
        }
        if (streamToUpload != null) {
            // Do the upload               
            int streamId = copyStreamToServer(streamToUpload.openStream(), streamToUpload.getName(),
                    (sn != null && fn != null && sn.equals(fn)) ? filesize : -1);
            if (streamId >= 0) {
                uploads.add(new Upload(name, streamId));
            }
        }

    }

    return uploads.toArray(new Upload[uploads.size()]);
}