想要获取某个目录下指定后缀的所有文件(包含子目录中的文件),这是一个很常见的小需求,使用 PHP
应该怎么实现呢?
代码
class Dir
{
/**
* 获取文件夹内指定后缀的所有文件
* @param array $result 结果集
* @param string $dir 指定目录
* @param array $filter 后缀过滤,为空即全部文件
*/
public static function getFiles(&$result, $dir, $filter = [])
{
$files = array_diff(scandir($dir), array('.', '..', '__MACOSX'));
if (is_array($files)) {
foreach ($files as $value) {
if (is_dir($dir . '/' . $value)) {
self::getFiles($result, $dir . '/' . $value, $filter);
} else {
$path_info = pathinfo($dir . '/' . $value);
$extension = array_key_exists('extension', $path_info) ? $path_info['extension'] : '';
if (empty($filter) || (!empty($filter) && in_array($extension, $filter))) {
$result[] = $dir . '/' . $value;
}
}
}
}
}
}
使用
获取目录 /opt/data/uploads
下所有后缀为 'pdf', 'doc', 'docx'
的文件,目录下包含多层子目录,各种格式的上传文件,结果存在数组 $files
里面。
$files = [];
$path = '/opt/data/uploads';
$extensions = ['pdf', 'doc', 'docx'];
Dir::getFiles($files, $path, $extensions);
非常方便,代码为项目中自用,如有 bug
望留言告知。
感谢分享,正需要这个。
最关键的递归那一行,self::getFilesFromDir,函数名写错了= =
多谢指出来,项目中用了个长名字,放这里我说把函数名简化一下,忘记里面还用到了= =