[CLI] 使用 Curl 下载文件实时进度条显示

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

已注册用户请登录

这是一个创建于 2214 天前的主题，其中的信息可能已经有所发展或是发生改变。

最近在捣鼓命令行下的编程，下载文件总是一个难熬的过程，如果有进度条就好很多了！！！

先上一个进度条的扩展包，还是不错的https://github.com/dariuszp/cli-progress-bar 还是挺好看的！

curl做为 PHP 一个很常用的下载方式，这里简单的使用方式；

// 初始化一个 curl
$ch = curl_init();
// 设置请求的 url
curl_setopt($ch, CURLOPT_URL, $url);
// 
curl_setopt($ch, CURLOPT_HEADER, 0);
// 不直接输出，而是通过 curl_exec 返回
curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);

if (false === ($stream = curl_exec($ch))) {
    throw new \Exception(curl_errno($ch));
}

curl_close($ch);

return $stream;

上面是一个很简单的例子，如果一个文件很大，那么用户就需要等待很长的时间，这时候我们就应该加上进度条的效果：

class Request
{
    protected $bar;
    // 是否下载完成
    protected $downloaded = false;

    public function __construct()
    {
        // 初始化一个进度条
        $this->bar = new CliProgressBar(100);
        $this->bar->display();
        $this->bar->setColorToRed();
    }
    
    function download($url)
    {
        $ch = curl_init();
    
        // 从配置文件中获取根路径
        curl_setopt($ch, CURLOPT_URL, $url);
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, 0);
        // 开启进度条
        curl_setopt($ch, CURLOPT_NOPROGRESS, 0);
        // 进度条的触发函数
        curl_setopt($ch, CURLOPT_PROGRESSFUNCTION, 'progress');
        // ps: 如果目标网页跳转，也跟着跳转
        // curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
    
        if (false === ($stream = curl_exec($ch))) {
            throw new \Exception(curl_errno($ch));
        }
    
        curl_close($ch);
    
        return $stream;
    }
    
    /**
     * 进度条下载.
     *
     * @param $ch
     * @param $countDownloadSize    总下载量
     * @param $currentDownloadSize  当前下载量
     * @param $countUploadSize      
     * @param $currentUploadSize
     */
    public function progress($ch, $countDownloadSize, $currentDownloadSize, $countUploadSize, $currentUploadSize)
    {
        
         // 等于 0 的时候，应该是预读资源不等于 0 的时候即开始下载
         // 这里的每一个判断都是坑，多试试就知道了
        if (0 === $countDownloadSize) {
            return false;
        }
        // 有时候会下载两次，第一次很小，应该是重定向下载
        if ($countDownloadSize > $currentDownloadSize) {
            $this->downloaded = false;
            // 继续显示进度条
        }
        // 已经下载完成还会再发三次请求
        elseif ($this->downloaded) {
            return false;
        }
        // 两边相等下载完成并不一定结束，
        elseif ($currentDownloadSize === $countDownloadSize) {
            return false;
        }
        
        // 开始计算
        $bar = $currentDownloadSize / $countDownloadSize * 100;
        $this->bar->progress($bar);
    }
}
(new Request)->download('http://www.shiguopeng.cn/database.sql');

千万千万注意下载回调的判断那里的坑！！！