Example usage for org.apache.commons.fileupload.servlet ServletFileUpload getItemIterator

List of usage examples for org.apache.commons.fileupload.servlet ServletFileUpload getItemIterator

  1. HOME
  2. Java
  3. org.apache.commons
  4. org.apache.commons.fileupload.servlet.*
  5. ServletFileUpload
  6. getItemIterator

Introduction

In this page you can find the example usage for org.apache.commons.fileupload.servlet ServletFileUpload getItemIterator.

Prototype

public FileItemIterator getItemIterator(HttpServletRequest request) throws FileUploadException, IOException

Source Link

Document

Processes an <a href="http://www.ietf.org/rfc/rfc1867.txt">RFC 1867</a> compliant <code>multipart/form-data</code> stream.

Usage

From source file:org.bonitasoft.engine.api.internal.servlet.ServletCall.java

/**
 * Default constructor.//w w  w  . java  2s . c o m
 * 
 * @param request
 *        The request made to access this servletCall.
 * @param response
 *        The response to return.
 * @throws IOException
 * @throws FileUploadException
 */
public ServletCall(final HttpServletRequest request, final HttpServletResponse response)
        throws FileUploadException, IOException {
    super();
    this.request = request;
    this.response = response;
    parameters = new HashMap<String, String>();
    binaryParameters = new ArrayList<byte[]>();
    if (ServletFileUpload.isMultipartContent(request)) {
        final ServletFileUpload upload = new ServletFileUpload();
        // Parse the request
        final FileItemIterator iter = upload.getItemIterator(request);
        while (iter.hasNext()) {
            try {
                final FileItemStream item = iter.next();
                final InputStream stream = item.openStream();
                String fieldName = item.getFieldName();
                if (fieldName.startsWith(BINARY_PARAMETER)) {
                    binaryParameters.add(IOUtil.getAllContentFrom(stream));
                } else {
                    String read = IOUtil.read(stream);
                    parameters.put(fieldName, read);
                }
                stream.close();
            } catch (final Exception t) {
                throw new IOException(t);
            }
        }
    } else {
        final Map<String, String[]> parameterMap = this.request.getParameterMap();
        final Set<Entry<String, String[]>> entrySet = parameterMap.entrySet();
        for (final Entry<String, String[]> entry : entrySet) {
            parameters.put(entry.getKey(), entry.getValue()[0]);
        }
    }
}

From source file:org.brutusin.rpc.http.RpcServlet.java

/**
 *
 * @param req//  w w w.  jav  a 2  s  . com
 * @return
 * @throws IOException
 */
private static Map<String, String[]> parseMultipartParameters(HttpServletRequest req) throws IOException {
    if (isMultipartContent(req)) {
        Map<String, String[]> multipartParameters = new HashMap();
        Map<String, List<String>> map = new HashMap();
        try {
            ServletFileUpload upload = new ServletFileUpload();
            FileItemIterator iter = upload.getItemIterator(req);
            req.setAttribute(REQ_ATT_MULTIPART_ITERATOR, iter);
            while (iter.hasNext()) {
                FileItemStream item = iter.next();
                if (!item.isFormField()) {
                    req.setAttribute(REQ_ATT_MULTIPART_CURRENT_ITEM, item);
                    break;
                }
                List<String> list = map.get(item.getFieldName());
                if (list == null) {
                    list = new ArrayList();
                    map.put(item.getFieldName(), list);
                }
                String encoding = req.getCharacterEncoding();
                if (encoding == null) {
                    encoding = "UTF-8";
                }
                list.add(Miscellaneous.toString(item.openStream(), encoding));
            }
        } catch (FileUploadException ex) {
            throw new RuntimeException(ex);
        }
        for (Map.Entry<String, List<String>> entrySet : map.entrySet()) {
            String key = entrySet.getKey();
            List<String> value = entrySet.getValue();
            multipartParameters.put(key, value.toArray(new String[value.size()]));
        }
        return multipartParameters;
    }
    return null;
}

From source file:org.celstec.arlearn2.upload.BlobStoreServlet.java

protected void doPost(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
    try {//from w ww .  ja  v a  2 s.  co  m
        Long runId = null;
        String account = null;
        ServletFileUpload upload = new ServletFileUpload();
        res.setContentType("text/plain");

        FileItemIterator iterator = upload.getItemIterator(req);

        while (iterator.hasNext()) {
            FileItemStream item = iterator.next();
            InputStream stream = item.openStream();

            if (item.isFormField()) {
                if ("runId".equals(item.getFieldName())) {
                    runId = Long.parseLong(Streams.asString(stream));
                }
                if ("account".equals(item.getFieldName())) {
                    account = Streams.asString(stream);
                }
            } else {
                BlobKey blobkey = storeBlob(item.getContentType(), item.getName(), stream);
                if (blobkey != null) {
                    System.out.println(blobkey);
                    // File exists
                    BlobKey oldkey = FilePathManager.getBlobKey(account, runId, item.getName());
                    if (oldkey != null) {
                        FilePathManager.delete(oldkey);
                        blobstoreService.delete(oldkey);
                    }
                    FilePathManager.addFile(runId, account, item.getName(), blobkey);

                } else {
                    blobkey.toString();
                }

            }
        }
    } catch (Exception ex) {
        throw new ServletException(ex);
    }
}

From source file:org.celstec.arlearn2.upload.BlobStoreServletIncremental.java

protected void doPost(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
    try {//from   ww  w .  j  a v  a2s. co m
        Long runId = null;
        String account = null;
        String serverPath = null;
        boolean last = false;
        ServletFileUpload upload = new ServletFileUpload();
        res.setContentType("text/plain");

        FileItemIterator iterator = upload.getItemIterator(req);
        System.out.println("before while");
        while (iterator.hasNext()) {
            System.out.println("in while");
            FileItemStream item = iterator.next();
            InputStream stream = item.openStream();

            if (item.isFormField()) {
                if ("runId".equals(item.getFieldName())) {
                    runId = Long.parseLong(Streams.asString(stream));
                    System.out.println("runid is " + runId);
                }

                if ("account".equals(item.getFieldName())) {
                    account = Streams.asString(stream);
                    System.out.println("account is " + account);
                }
                if ("last".equals(item.getFieldName())) {
                    last = Boolean.parseBoolean(Streams.asString(stream));
                    System.out.println("last is " + last);
                }
                if ("serverPath".equals(item.getFieldName())) {
                    serverPath = Streams.asString(stream);
                    System.out.println("serverPath is " + serverPath);
                }

            } else {
                log.warning("Got an uploaded file: " + item.getFieldName() + ", name = " + item.getName());
                AppEngineFile file = storeBlob(item.getContentType(), item.getName(), stream, last, serverPath);

                BlobKey blobkey = fileService.getBlobKey(file);
                if (blobkey != null) {
                    // File exists
                    BlobKey oldkey = FilePathManager.getBlobKey(account, runId, item.getName());
                    if (oldkey != null) {
                        FilePathManager.delete(oldkey);
                        blobstoreService.delete(oldkey);
                    }
                    FilePathManager.addFile(runId, account, item.getName(), blobkey);
                    System.out.println(blobkey.toString());
                }
                res.getWriter().write(file.getFullPath());
                //               else {
                //                  blobkey.toString();
                //               }

            }
        }
    } catch (Exception ex) {
        throw new ServletException(ex);
    }
}

From source file:org.celstec.arlearn2.upload.UploadGameServlet.java

protected void doPost(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
    long gameId = 0l;
    String auth = null;/*from w w w.  j  av  a  2s  .  com*/
    try {
        ServletFileUpload upload = new ServletFileUpload();
        FileItemIterator iter = upload.getItemIterator(req);

        String json = "";
        while (iter.hasNext()) {
            FileItemStream item = iter.next();
            String name = item.getFieldName();
            InputStream stream = item.openStream();
            if (item.isFormField()) {
                String value = Streams.asString(stream);
                if ("gameId".equals(name))
                    gameId = Long.parseLong(value);
                if ("auth".equals(name))
                    auth = value;

            } else {
                json = Streams.asString(stream);

            }
        }

        res.setContentType("text/plain");
        JSONObject jObject = new JSONObject(json);
        Object deserialized = JsonBeanDeserializer.deserialize(json);

        if (deserialized instanceof GamePackage && ((GamePackage) deserialized).getGame() != null)
            unpackGame((GamePackage) deserialized, req, auth);
        if (deserialized instanceof RunPackage && ((RunPackage) deserialized).getRun() != null)
            unpackRun((RunPackage) deserialized, req, gameId, auth);

    } catch (Exception ex) {
        throw new ServletException(ex);
    }
}

From source file:org.codehaus.enunciate.modules.rest.StreamingMultipartRequestHandler.java

/**
 * Parses the parts.  The resulting collection has an undefined size. Any attempts to access the size of the
 * collection (including toArray()) will result in an UnsupportedOperationException.<br/><br/>
 *
 * The DataHandler instances each will have a DataSource that is an instance of
 * {@link org.codehaus.enunciate.modules.rest.StreamingFileItemDataSource}
 *
 * @param request The request.//from w  w w .j av  a2  s.c om
 * @return The parts.
 */
public Collection<DataHandler> parseParts(HttpServletRequest request) throws IOException, FileUploadException {
    ServletFileUpload fileUpload = new ServletFileUpload();
    FileItemIterator itemIterator = fileUpload.getItemIterator(request);
    ProgressListener progressListener = null;
    if (getProgressListenerFactory() != null) {
        progressListener = getProgressListenerFactory().newProgressListener(request);
    }
    return new StreamingMultipartCollection(itemIterator, progressListener);
}

From source file:org.codelabor.system.file.web.servlet.FileUploadStreamServlet.java

@Override
protected void upload(HttpServletRequest request, HttpServletResponse response) throws Exception {
    WebApplicationContext ctx = WebApplicationContextUtils
            .getRequiredWebApplicationContext(this.getServletContext());
    FileManager fileManager = (FileManager) ctx.getBean("fileManager");

    boolean isMultipart = ServletFileUpload.isMultipartContent(request);
    Map<String, Object> paramMap = RequestUtils.getParameterMap(request);
    if (logger.isDebugEnabled()) {
        logger.debug(paramMap.toString());
    }//from ww  w.  j  a  v a  2  s  . com

    String mapId = (String) paramMap.get("mapId");
    RepositoryType acceptedRepositoryType = repositoryType;
    String requestedRepositoryType = (String) paramMap.get("repositoryType");
    if (StringUtils.isNotEmpty(requestedRepositoryType)) {
        acceptedRepositoryType = RepositoryType.valueOf(requestedRepositoryType);
    }

    if (isMultipart) {
        ServletFileUpload upload = new ServletFileUpload();
        upload.setFileSizeMax(fileSizeMax);
        upload.setSizeMax(requestSizeMax);
        upload.setHeaderEncoding(characterEncoding);
        upload.setProgressListener(new FileUploadProgressListener());
        try {
            FileItemIterator iter = upload.getItemIterator(request);

            while (iter.hasNext()) {
                FileItemStream fileItemSteam = iter.next();
                if (logger.isDebugEnabled()) {
                    logger.debug(fileItemSteam.toString());
                }
                FileDTO fileDTO = null;
                if (fileItemSteam.isFormField()) {
                    paramMap.put(fileItemSteam.getFieldName(),
                            Streams.asString(fileItemSteam.openStream(), characterEncoding));
                } else {
                    if (fileItemSteam.getName() == null || fileItemSteam.getName().length() == 0)
                        continue;

                    // set DTO
                    fileDTO = new FileDTO();
                    fileDTO.setMapId(mapId);
                    fileDTO.setRealFilename(FilenameUtils.getName(fileItemSteam.getName()));
                    if (acceptedRepositoryType == RepositoryType.FILE_SYSTEM) {
                        fileDTO.setUniqueFilename(getUniqueFilename());
                    }
                    fileDTO.setContentType(fileItemSteam.getContentType());
                    fileDTO.setRepositoryPath(realRepositoryPath);
                    if (logger.isDebugEnabled()) {
                        logger.debug(fileDTO.toString());
                    }
                    UploadUtils.processFile(acceptedRepositoryType, fileItemSteam.openStream(), fileDTO);
                }
                if (fileDTO != null)
                    fileManager.insertFile(fileDTO);
            }
        } catch (FileUploadException e) {
            e.printStackTrace();
            logger.error(e.getMessage());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error(e.getMessage());
        }
    } else {
        paramMap = RequestUtils.getParameterMap(request);
    }
    try {
        processParameters(paramMap);
    } catch (Exception e) {
        e.printStackTrace();
        logger.error(e.getMessage());
    }
    dispatch(request, response, forwardPathUpload);

}

From source file:org.collectionspace.chain.controller.WebUIRequest.java

private void initRequest(UIUmbrella umbrella, HttpServletRequest request, HttpServletResponse response,
        List<String> p) throws IOException, UIException {
    this.request = request;
    this.response = response;
    boolean isMultipart = ServletFileUpload.isMultipartContent(request);
    if (isMultipart) {
        // Create a new file upload handler
        ServletFileUpload upload = new ServletFileUpload();

        // Parse the request
        FileItemIterator iter;/*  ww  w .  j a v  a2 s .  com*/
        try {
            iter = upload.getItemIterator(request);
            while (iter.hasNext()) {
                FileItemStream item = iter.next();
                String name = item.getFieldName();
                //InputStream stream = item.openStream();
                if (item.isFormField()) {
                    //   System.out.println("Form field " + name + " with value "
                    //    + Streams.asString(stream) + " detected.");
                } else {
                    //   System.out.println("File field " + name + " with file name "
                    //    + item.getName() + " detected.");
                    // Process the input stream
                    contentHeaders = item.getHeaders();
                    uploadName = item.getName();

                    ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
                    if (item != null) {
                        InputStream stream = item.openStream();
                        IOUtils.copy(stream, byteOut);
                        new TeeInputStream(stream, byteOut);

                    }
                    bytebody = byteOut.toByteArray();
                }
            }
        } catch (FileUploadException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    } else {
        body = IOUtils.toString(request.getInputStream(), "UTF-8");
    }

    this.ppath = p.toArray(new String[0]);
    if (!(umbrella instanceof WebUIUmbrella))
        throw new UIException("Bad umbrella");
    this.umbrella = (WebUIUmbrella) umbrella;
    session = calculateSessionId();

}

From source file:org.csi.yucca.storage.datamanagementapi.service.MetadataService.java

@POST
@Path("/{tenant}")
@Produces(MediaType.APPLICATION_JSON)// w  w  w.j  ava 2  s .  c o  m
public String createMetadata(@PathParam("tenant") String tenant, @Context HttpServletRequest request)
        throws NumberFormatException, UnknownHostException {
    log.debug("[MetadataService::createMetadata] - START");

    String datasetMetadata = null;
    String encoding = null;
    String formatType = null;
    String csvSeparator = null;
    boolean skipFirstRow = false;
    String csvData = null;
    String fileName = null;

    try {
        ServletFileUpload upload = new ServletFileUpload();
        FileItemIterator iterator = upload.getItemIterator(request);
        while (iterator.hasNext()) {
            FileItemStream item = iterator.next();
            if (IMPORT_BULKDATASET_METADATA_REQ_KEY.equals(item.getFieldName()))
                datasetMetadata = read(item.openStream());
            else if (IMPORT_BULKDATASET_ENCODING_REQ_KEY.equals(item.getFieldName()))
                encoding = read(item.openStream());
            else if (IMPORT_BULKDATASET_FORMAT_TYPE_REQ_KEY.equals(item.getFieldName()))
                formatType = read(item.openStream());
            else if (IMPORT_BULKDATASET_CSV_SEP_REQ_KEY.equals(item.getFieldName()))
                csvSeparator = read(item.openStream());
            else if (IMPORT_BULKDATASET_CSV_SKIP_FIRS_ROW_REQ_KEY.equals(item.getFieldName()))
                skipFirstRow = new Boolean(read(item.openStream()));
            else if (IMPORT_BULKDATASET_FILE_REQ_KEY.equals(item.getFieldName())) {
                csvData = readFileRows(item.openStream(), encoding);
                fileName = item.getName();
            }
        }

    } catch (Exception e) {
        e.printStackTrace();
    }

    log.debug("[MetadataService::createMetadata] - encoding: " + encoding + ", formatType: " + formatType
            + ", csvSeparator: " + csvSeparator);
    Metadata metadata = Metadata.fromJson(datasetMetadata);
    if (fileName != null)
        metadata.getInfo().addFilename(fileName);

    CreateDatasetResponse createDatasetResponse = new CreateDatasetResponse();

    metadata.setDatasetVersion(1);
    if (metadata.getConfigData() == null)
        metadata.setConfigData(new ConfigData());
    metadata.getConfigData().setType(Metadata.CONFIG_DATA_TYPE_DATASET);
    metadata.getConfigData().setSubtype(Metadata.CONFIG_DATA_SUBTYPE_BULK_DATASET);
    metadata.getConfigData().setCurrent(1);
    if (metadata.getInfo() == null)
        metadata.setInfo(new Info());

    if (metadata.getInfo().getFields() != null) {
        for (Field field : metadata.getInfo().getFields()) {
            field.setFieldName(Util.cleanStringCamelCase(field.getFieldName()));
            if (field != null && field.getDataType() == null)
                field.setDataType("string");
        }
    }
    metadata.getInfo().setRegistrationDate(new Date());
    try {

        List<SDPBulkInsertException> checkFileToWriteErrors = null;
        MongoDBDataUpload dataUpload = new MongoDBDataUpload();
        if (csvData != null) {
            checkFileToWriteErrors = dataUpload.checkFileToWrite(csvData, csvSeparator, metadata, skipFirstRow);
        }

        if (checkFileToWriteErrors != null && checkFileToWriteErrors.size() > 0) {
            for (SDPBulkInsertException error : checkFileToWriteErrors) {
                createDatasetResponse.addErrorMessage(new ErrorMessage(error.getErrorCode(),
                        error.getErrorMessage(), error.getErrorDetail()));
            }
        } else {
            MongoClient mongo = MongoSingleton.getMongoClient();
            String supportDb = Config.getInstance().getDbSupport();
            String supportDatasetCollection = Config.getInstance().getCollectionSupportDataset();

            MongoDBMetadataDAO metadataDAO = new MongoDBMetadataDAO(mongo, supportDb, supportDatasetCollection);

            String supportApiCollection = Config.getInstance().getCollectionSupportApi();
            MongoDBApiDAO apiDAO = new MongoDBApiDAO(mongo, supportDb, supportApiCollection);

            BasicDBObject searchTenantQuery = new BasicDBObject();
            searchTenantQuery.put("tenantCode", tenant);
            DBCollection tenantCollection = mongo.getDB(supportDb).getCollection("tenant");
            DBObject tenantData = tenantCollection.find(searchTenantQuery).one();
            Long idTenant = ((Number) tenantData.get("idTenant")).longValue();

            int maxDatasetNum = ((Number) tenantData.get("maxDatasetNum")).intValue();

            if (maxDatasetNum > 0) {

                int numCurrentDataset = metadataDAO.countAllMetadata(tenant, true);
                log.info("[MetadataService::createMetadata] -  tenant=" + tenant + "     maxDatasetNum="
                        + maxDatasetNum + "        numCurrentDataset=" + numCurrentDataset);
                //TODO
                if (numCurrentDataset >= maxDatasetNum)
                    throw new MaxDatasetNumException("too many dataset");

            }

            metadata.getConfigData().setIdTenant(idTenant);

            // binary metadata: create a metadata record specific for attachment
            Metadata binaryMetadata = null;
            if (metadata.getInfo().getFields() != null) {
                for (Field field : metadata.getInfo().getFields()) {
                    if (field.getDataType().equals("binary")) {
                        binaryMetadata = Metadata.createBinaryMetadata(metadata);
                        break;
                    }
                }
            }
            if (binaryMetadata != null) {
                Metadata binaryMetadataCreated = metadataDAO.createMetadata(binaryMetadata, null);
                metadata.getInfo().setBinaryDatasetVersion(binaryMetadataCreated.getDatasetVersion());
                metadata.getInfo().setBinaryIdDataset(binaryMetadataCreated.getIdDataset());
            }

            List<Tenantsharing> lista = new ArrayList<Tenantsharing>();
            if (metadata.getInfo().getTenantssharing() != null) {
                Set<String> tenantSet = new TreeSet<String>();
                for (Tenantsharing tenantInList : metadata.getInfo().getTenantssharing().getTenantsharing()) {
                    if (!tenantInList.getTenantCode().equals(metadata.getConfigData().getTenantCode())
                            && !tenantSet.contains(metadata.getConfigData().getTenantCode())
                            && tenantInList.getIsOwner() != 1) {
                        lista.add(tenantInList);
                        tenantSet.add(tenantInList.getTenantCode());
                    }
                }
            }
            Tenantsharing owner = new Tenantsharing();
            owner.setIdTenant(metadata.getConfigData().getIdTenant());
            owner.setIsOwner(1);
            owner.setTenantCode(metadata.getConfigData().getTenantCode());
            owner.setTenantName(metadata.getConfigData().getTenantCode());
            // owner.setTenantDescription(metadata.getConfigData().get);

            lista.add(owner);
            Tenantsharing arrayTenant[] = new Tenantsharing[lista.size()];
            arrayTenant = lista.toArray(arrayTenant);
            if (metadata.getInfo().getTenantssharing() == null) {
                Tenantssharing tenantssharing = new Tenantssharing();
                metadata.getInfo().setTenantssharing(tenantssharing);
            }
            metadata.getInfo().getTenantssharing().setTenantsharing(arrayTenant);

            // opendata
            if (!"public".equals(metadata.getInfo().getVisibility())) {
                metadata.setOpendata(null);
            }

            Metadata metadataCreated = metadataDAO.createMetadata(metadata, null);

            MyApi api = MyApi.createFromMetadataDataset(metadataCreated);
            api.getConfigData().setType(Metadata.CONFIG_DATA_TYPE_API);
            api.getConfigData().setSubtype(Metadata.CONFIG_DATA_SUBTYPE_API_MULTI_BULK);

            MyApi apiCreated = apiDAO.createApi(api);

            createDatasetResponse.setMetadata(metadataCreated);
            createDatasetResponse.setApi(apiCreated);

            /*
             * Create api in the store
             */
            String apiName = "";
            try {
                apiName = StoreService.createApiforBulk(metadata, false, datasetMetadata);
            } catch (Exception duplicate) {
                if (duplicate.getMessage().toLowerCase().contains("duplicate")) {
                    try {
                        apiName = StoreService.createApiforBulk(metadata, true, datasetMetadata);
                    } catch (Exception e) {
                        log.error(
                                "[MetadataService::createMetadata] - ERROR to update API in Store for Bulk. Message: "
                                        + duplicate.getMessage());
                    }
                } else {
                    log.error(
                            "[MetadataService::createMetadata] -  ERROR in create or update API in Store for Bulk. Message: "
                                    + duplicate.getMessage());
                }
            }
            try {

                StoreService.publishStore("1.0", apiName, "admin");
                Set<String> tenantSet = new TreeSet<String>();
                if (metadata.getInfo().getTenantssharing() != null) {
                    for (Tenantsharing tenantSh : metadata.getInfo().getTenantssharing().getTenantsharing()) {
                        tenantSet.add(tenantSh.getTenantCode());
                        String appName = "userportal_" + tenantSh.getTenantCode();
                        StoreService.addSubscriptionForTenant(apiName, appName);
                    }
                }
                if (!tenantSet.contains(metadata.getConfigData().getTenantCode())) {
                    String appName = "userportal_" + metadata.getConfigData().getTenantCode();
                    StoreService.addSubscriptionForTenant(apiName, appName);
                }

            } catch (Exception e) {
                log.error("[MetadataService::createMetadata] - ERROR in publish Api in store - message: "
                        + e.getMessage());
            }

            if (csvData != null) {
                try {
                    dataUpload.writeFileToMongo(mongo, "DB_" + tenant, "data", metadataCreated);
                } catch (Exception e) {
                    log.error("[MetadataService::createMetadata] - writeFileToMongo ERROR: " + e.getMessage());
                    createDatasetResponse.addErrorMessage(new ErrorMessage(e));
                    e.printStackTrace();
                }
            }
        }
    } catch (MaxDatasetNumException ex) {
        log.error("[MetadataService::createMetadata] - MaxDatasetNumException ERROR: ", ex);
        createDatasetResponse.addErrorMessage(new ErrorMessage(ex));

    }
    return createDatasetResponse.toJson();
}

From source file:org.csi.yucca.storage.datamanagementapi.service.MetadataService.java

@POST
@Path("/add/{tenant}/{datasetCode}")
@Produces(MediaType.APPLICATION_JSON)//from  w  w  w. ja  va  2  s. co m
public String addData(@PathParam("tenant") String tenant, @PathParam("datasetCode") String datasetCode,
        @Context HttpServletRequest request) throws NumberFormatException, UnknownHostException {
    log.debug("[MetadataService::addData] - START");

    String encoding = null;
    String formatType = null;
    String csvSeparator = null;
    boolean skipFirstRow = false;
    String csvData = null;
    String fileName = null;

    try {
        ServletFileUpload upload = new ServletFileUpload();
        FileItemIterator iterator = upload.getItemIterator(request);
        while (iterator.hasNext()) {
            FileItemStream item = iterator.next();
            if (IMPORT_BULKDATASET_ENCODING_REQ_KEY.equals(item.getFieldName()))
                encoding = read(item.openStream());
            else if (IMPORT_BULKDATASET_FORMAT_TYPE_REQ_KEY.equals(item.getFieldName()))
                formatType = read(item.openStream());
            else if (IMPORT_BULKDATASET_CSV_SEP_REQ_KEY.equals(item.getFieldName()))
                csvSeparator = read(item.openStream());
            else if (IMPORT_BULKDATASET_CSV_SKIP_FIRS_ROW_REQ_KEY.equals(item.getFieldName()))
                skipFirstRow = new Boolean(read(item.openStream()));
            else if (IMPORT_BULKDATASET_FILE_REQ_KEY.equals(item.getFieldName())) {
                csvData = readFileRows(item.openStream(), encoding);
                fileName = item.getName();
            }

        }

    } catch (Exception e) {
        e.printStackTrace();
    }

    log.debug("[MetadataService::addData] - encoding: " + encoding + ", formatType: " + formatType
            + ", csvSeparator: " + csvSeparator);
    MongoClient mongo = MongoSingleton.getMongoClient();
    String supportDb = Config.getInstance().getDbSupport();
    String supportDatasetCollection = Config.getInstance().getCollectionSupportDataset();
    MongoDBMetadataDAO metadataDAO = new MongoDBMetadataDAO(mongo, supportDb, supportDatasetCollection);

    Metadata existingMetadata = metadataDAO.readCurrentMetadataByCode(datasetCode);
    existingMetadata.getInfo().addFilename(fileName);
    metadataDAO.updateMetadata(existingMetadata);

    UpdateDatasetResponse updateDatasetResponse = new UpdateDatasetResponse();

    MongoDBDataUpload dataUpload = new MongoDBDataUpload();

    List<SDPBulkInsertException> checkFileToWriteErrors = dataUpload.checkFileToWrite(csvData, csvSeparator,
            existingMetadata, skipFirstRow);
    if (checkFileToWriteErrors != null && checkFileToWriteErrors.size() > 0) {
        for (SDPBulkInsertException error : checkFileToWriteErrors) {
            updateDatasetResponse.addErrorMessage(
                    new ErrorMessage(error.getErrorCode(), error.getErrorMessage(), error.getErrorDetail()));
        }
    } else {

        try {
            dataUpload.writeFileToMongo(mongo, "DB_" + tenant, "data", existingMetadata);
        } catch (Exception e) {
            log.error("[MetadataService::addData] - writeFileToMongo ERROR: " + e.getMessage());
            updateDatasetResponse.addErrorMessage(new ErrorMessage(e));
            e.printStackTrace();
        }
    }

    return updateDatasetResponse.toJson();
}