Flume-ng spooling source的方式增加了對目錄的遞歸檢測

 

 

原有的spooling source的方式只支持一級目錄的文件變動的檢測,修改了一下源碼支持對目錄的遞歸檢測,可以支持多級目錄下的文件變動

ReliableSpoolingFileEventReader類增加以下下方法

 /**
  * add by 
  *
  * @param files
  * @param dir
  * @param filter
  */
 private void listDirFiles(List<File> files, File dir, FileFilter filter) {
  File[] childs = dir.listFiles(filter);// 列出目錄下所有的文件
  for (int i = 0; i < childs.length; i++) {
   if (childs[i].isFile()) {
    files.add(childs[i]);
   } else {
    if (childs[i].isDirectory()) {
     listDirFiles(files, childs[i], filter);// 如果是個目錄,就繼續遞歸列出其下面的文件.
    }
   }
  }
 }

 

修改getNextFile方法

 private Optional<FileInfo> getNextFile() {
    List<File> candidateFiles = Collections.emptyList();

    if (consumeOrder != ConsumeOrder.RANDOM ||
      candidateFileIter == null ||
      !candidateFileIter.hasNext()) {
      /* Filter to exclude finished or hidden files */
      FileFilter filter = new FileFilter() {
        public boolean accept(File candidate) {
          String fileName = candidate.getName();
          if ((candidate.isDirectory()) ||
            (fileName.endsWith(completedSuffix)) ||
            (fileName.startsWith(".")) ||
            ignorePattern.matcher(fileName).matches()) {
            return false;
          }
          return true;
        }
      };
      //candidateFiles = Arrays.asList(spoolDirectory.listFiles(filter));
	  candidateFiles = new ArrayList<File>();//新增的
	  listDirFiles(candidateFiles, directory, filter);//新增的

 

發佈了287 篇原創文章 · 獲贊 12 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章