【CLI】使用Curl下載文件實時進度條顯示

最近在搗鼓命令行下的編程,下載文件老是一個難熬的過程,若是有進度條就好不少了!!!php


先上一個進度條的擴展包,仍是不錯的https://github.com/dariuszp/cli-progress-bar
https://github.com/dariuszp/cli-progress-bar/raw/master/examples/img/terminal.gif
仍是挺好看的!git


curl作爲 PHP 一個很經常使用的下載方式,這裏簡單的使用方式;github

// 初始化一個 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;

上面是一個很簡單的例子,若是一個文件很大,那麼用戶就須要等待很長的時間,這時候咱們就應該加上進度條的效果:sql

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

千萬千萬注意下載回調的判斷那裏的坑!!!編程

  • 還有一個問題:若是跳轉下載的,設置了curl也跟着跳轉,返回的文件將會出問題,
  • 我下載的是zip文件,會致使文件頭有第一此請求的HTTP響應頭的內容,
  • 因此看本身須要curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
相關文章
相關標籤/搜索