【CLI】使用 Curl 下載檔案實時進度條顯示

WaitMoonMan發表於2018-03-24

最近在搗鼓命令列下的程式設計，下載檔案總是一個難熬的過程，如果有進度條就好很多了！！！

先上一個進度條的擴充套件包，還是不錯的github.com/dariuszp/cl…

https://user-gold-cdn.xitu.io/2018/3/24/1625889b9e6c6bf2?w=722&h=434&f=gif&s=37169
還是挺好看的！

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');
複製程式碼

千萬千萬注意下載回撥的判斷那裡的坑！！！

  • 還有一個問題：如果跳轉下載的，設定了curl也跟著跳轉，返回的檔案將會出問題，
  • 我下載的是zip檔案，會導致檔案頭有第一此請求的HTTP響應頭的內容，
  • 所以看自己需要curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);

相關文章