Ejemplo n.º 1
0
 protected File copyZipLocally(String fileName, File unzipDir) {
   InputStream is = null;
   FileOutputStream os = null;
   try {
     is = sourceDir.getInputStream(fileName, true);
     if (is != null) {
       File localZipFile = new File(unzipDir, UUID.randomUUID().toString() + ".zip");
       os = new FileOutputStream(localZipFile);
       IOUtils.copy(is, os);
       return localZipFile;
     } else {
       String msg = String.format("Failed to open %s.", fileName);
       throw new IoException(msg);
     }
   } catch (IOException e) {
     throw new IoException(e);
   } finally {
     IOUtils.closeQuietly(is);
     IOUtils.closeQuietly(os);
   }
 }
Ejemplo n.º 2
0
  @Override
  public void handle(
      Message inputMessage, ISendMessageCallback callback, boolean unitOfWorkBoundaryReached) {
    if (inputMessage instanceof TextMessage) {
      List<String> files = ((TextMessage) inputMessage).getPayload();
      ArrayList<String> filePaths = new ArrayList<String>();
      for (String fileName : files) {
        log(LogLevel.INFO, "Preparing to extract file : %s", fileName);
        FileInfo sourceZipFile = sourceDir.listFile(fileName);
        if (mustExist && sourceZipFile == null) {
          throw new IoException(String.format("Could not find file to extract: %s", fileName));
        }
        if (sourceZipFile != null) {
          File unzipDir = new File(LogUtils.getLogDir(), "unzip");
          unzipDir.mkdirs();

          File localZipFile = copyZipLocally(fileName, unzipDir);
          ZipFile zipFile = getNewZipFile(localZipFile);
          InputStream in = null;
          OutputStream out = null;
          try {
            String targetDirNameResolved =
                resolveParamsAndHeaders(targetRelativePath, inputMessage);
            if (targetSubDir) {
              targetDirNameResolved =
                  targetDirNameResolved
                      + "/"
                      + FilenameUtils.removeExtension(
                          new FileInfo(fileName, false, 0, 0).getName());
            }
            for (Enumeration<? extends ZipEntry> e = zipFile.entries(); e.hasMoreElements(); ) {
              ZipEntry entry = e.nextElement();
              if (!entry.isDirectory() && (extractEmptyFiles || entry.getSize() > 0)) {
                String relativePathToEntry = targetDirNameResolved + "/" + entry.getName();
                if (overwrite || targetDir.listFile(relativePathToEntry) == null) {
                  info("Unzipping %s", entry.getName());
                  out = targetDir.getOutputStream(relativePathToEntry, false);
                  in = zipFile.getInputStream(entry);
                  IOUtils.copy(in, out);
                  filePaths.add(relativePathToEntry);
                } else if (!overwrite) {
                  info(
                      "Not unzipping %s.  It already exists and the override property is not enabled",
                      entry.getName());
                }
              }
            }
          } catch (IOException e) {
            throw new IoException(e);
          } finally {
            IOUtils.closeQuietly(in);
            IOUtils.closeQuietly(out);
            IOUtils.closeQuietly(zipFile);
            FileUtils.deleteQuietly(localZipFile);
          }
          if (deleteOnComplete) {
            sourceDir.delete(fileName);
          }

          log(LogLevel.INFO, "Extracted %s", fileName);
          getComponentStatistics().incrementNumberEntitiesProcessed(threadNumber);
        }
      }
      if (filePaths.size() > 0) {
        callback.sendTextMessage(null, filePaths);
      }
    }
  }