php yield初体验,递归遍历文件夹并压缩

gyts62

php遍历一个文件夹并压缩到zip

private function zip($path,$zipFile){
$zip=new ZipArchive();
$zip->open($zipFile,ZipArchive::CREATE);//创建一个空的zip文件
$this->addFileToZip($path,$zip);
}
private function addFileToZip($path,ZipArchive $zip,$root=''){
if(!is_dir($path)){
return false;
}
if(!$root){
$root= $path;
}
if(strpos($path,$root)!==0){
$root= $path;
}
$handler=opendir($path); //打开当前文件夹由$path指定。
while(($filename=readdir($handler))!==false){
if($filename != "." && $filename != ".."){//文件夹文件名字为'.'和‘..’，不要对他们进行操作
if(is_dir($path."/".$filename)){// 如果读取的某个对象是文件夹，则递归
$this->addFileToZip($path."/".$filename, $zip,$root);
}else{ //将文件加入zip对象
$filenameWithPath = $path."/".$filename;
$localFileName = substr($filenameWithPath,strlen($root));
$zip->addFile($filenameWithPath,$localFileName);
}
}
}
@closedir($handler);
}

使用yield重构代码

private function zipFolder($folder,$zipFile){
$zip=new ZipArchive();
$zip->open($zipFile,ZipArchive::CREATE);//创建一个空的zip文件
foreach($this->yieldFile($folder) as $file){
$localFileName = substr($file,strlen($folder));
$zip->addFile($file,$localFileName);
}
}
private function yieldFile($path){
if(is_dir($path)){
$handler = opendir($path);
while(($filename=readdir($handler))!==false){
if($filename != "." && $filename != "..") {//文件夹文件名字为'.'和‘..’，不要对他们进行操作
if(is_dir($path."/".$filename)){// 如果读取的某个对象是文件夹，则递归
foreach($this->yieldFile($path."/".$filename) as $file){
yield $file;
}
}else{ //将文件加入zip对象
$file = $path."/".$filename;
yield $file;
}
}
}
closedir($handler);
}
}

代码执行

public function anyZip(){
for($i=0;$i<6;$i++) {
$start = microtime(true);
$toPath = 'D:/unzip';
$this->zipFolder($toPath, 'd:/zip/123.zip');
$end = microtime(true);
echo '|zipFolder-delay:' . ($end - $start);
$start = microtime(true);
$toPath = 'D:/unzip';
$this->zip($toPath, 'd:/zip/124.zip');
$end = microtime(true);
echo '|zip-delay:' . ($end - $start) . '<br>';
}
}

|zipFolder-delay:1.6427090167999|zip-delay:1.6077039241791
|zipFolder-delay:1.6132049560547|zip-delay:1.6287071704865
|zipFolder-delay:1.6342070102692|zip-delay:1.6152048110962
|zipFolder-delay:1.6917150020599|zip-delay:1.6022040843964
|zipFolder-delay:1.6297070980072|zip-delay:1.7262189388275
|zipFolder-delay:1.5997030735016|zip-delay:1.5892019271851
使用yield递归和正常的递归执行时间差距不大，主要好处是将数据获取和数据处理拆分开，更易理解和复用