Java I/O - 如何按文件扩展名获取所有xml文件

 


import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class Main {
  public static void main(String[] argv) {
    getXMLFiles(new File("c:\\a"));
  }

  public static File[] getXMLFiles(File folder) {
    List<File> aList = new ArrayList<File>();

    File[] files = folder.listFiles();
    for (File pf : files) {

      if (pf.isFile() && getFileExtensionName(pf).indexOf("xml") != -1) {
        aList.add(pf);
      }
    }
    return aList.toArray(new File[aList.size()]);
  }

  public static String getFileExtensionName(File f) {
    if (f.getName().indexOf(".") == -1) {
      return "";
    } else {
      return f.getName().substring(f.getName().length() - 3, f.getName().length());
    }
  }
}