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

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

Php代码  

  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重构代码 

Php代码  

    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);
        }
    }

代码执行 

ARC Lab

腾讯旗下ARC实验室推出的AI人像修复、抠图和增强工具

下载

立即学习“PHP免费学习笔记（深入）”；

Php代码  

  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/124124.zip');
            $end = microtime(true);
            echo '|zip-delay:' . ($end - $start) . '
';
        }
    }

|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递归和正常的递归执行时间差距不大，主要好处是将数据获取和数据处理拆分开，更易理解和复用