Rumah  >  Artikel  >  pembangunan bahagian belakang  >  wget dilaksanakan oleh golang

wget dilaksanakan oleh golang

WBOY
WBOYasal
2023-05-15 09:23:361126semak imbas

1. Latar Belakang

wget ialah alat muat turun yang biasa digunakan dalam sistem Linux, yang boleh memuat turun fail, imej dan sumber lain dengan pantas. Ia mempunyai ciri-ciri kelajuan muat turun yang pantas, kebolehpercayaan yang tinggi dan menyokong muat turun yang disambung semula. Kadang-kadang kita perlu menggunakan wget dalam sistem pengendalian lain, tetapi tidak ada alat yang serupa.

Untuk menyelesaikan masalah ini, saya cuba menggunakan golang untuk menulis alat wget mudah yang dipanggil golang-wget. Ia bukan sahaja menyokong muat turun fail asas, tetapi juga menyokong muat turun resume titik putus, proksi HTTP dan fungsi lain.

Artikel ini akan memperkenalkan prinsip pelaksanaan, struktur kod dan penggunaan golang-wget.

2. Prinsip Pelaksanaan

Pelaksanaan golang-wget terutamanya bergantung pada pakej http golang untuk melaksanakan sambungan protokol HTTP dan muat turun fail.

Prinsip pelaksanaan khusus adalah seperti berikut:

  1. Dapatkan parameter baris arahan

Dapatkan url yang dihantar dari baris arahan, simpan laluan dan sama ada melalui pakej bendera golang Parameter seperti keperluan untuk menyambung semula muat turun dari titik putus dan sama ada untuk menggunakan proksi.

  1. Semak sama ada penyambungan semula titik putus disokong

Jika pengguna perlu melakukan penyambungan titik putus, mula-mula semak sama ada fail sasaran wujud dan jika fail itu wujud, dapatkan fail saiz. Gunakan permintaan Julat HTTP untuk mendapatkan kedudukan permulaan fail sasaran, yang mengelakkan memuat turun keseluruhan fail. Jika fail tidak wujud, keseluruhan fail dimuat turun bermula dari 0.

  1. Konfigurasikan Klien HTTP

Gunakan pakej http golang untuk mencipta Klien HTTP dan konfigurasikan Pengangkutan HTTP mengikut sama ada pengguna perlu menggunakan proksi Untuk kaedah konfigurasi tertentu. sila rujuk dokumentasi rasmi golang.

  1. Muat turun fail

Gunakan Get, Head, Do dan kaedah lain pakej http untuk mendapatkan respons HTTP, dan gunakan kaedah Salin dalam pakej io untuk tulis badan tindak balas ke dalam fail sasaran. Jika anda perlu menyambung semula penghantaran dari titik putus, anda perlu mencari kedudukan permulaan sebelum menulis data.

3. Struktur kod

golang-wget terbahagi terutamanya kepada tiga bahagian: analisis parameter baris arahan, konfigurasi klien HTTP dan muat turun fail.

Penghuraian parameter baris perintah dilaksanakan menggunakan bendera perpustakaan standard golang, yang boleh menghuraikan parameter baris perintah dengan cepat.

Konfigurasi klien HTTP termasuk sama ada untuk menggunakan proksi dan konfigurasi Klien HTTP Menggunakan proksi memerlukan konfigurasi yang lebih kompleks.

Muat turun fail termasuk menyemak sama ada penyambungan titik putus disokong dan menggunakan Klien HTTP untuk memuat turun fail.

Berikut ialah rangka kerja kod utama:

func main() {
    // 解析命令行参数
    flag.Parse()

    // 配置HTTP客户端
    client := http.Client{
        Transport: &http.Transport{
            Proxy:      http.ProxyFromEnvironment,
            TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
        },
    }

    // 下载文件
    download(&client, *url, *filePath, *isResume)
}

func download(client *http.Client, url, filePath string, isResume bool) error {
    // 检查是否支持断点续传
    var startPosition int64 = 0
    var err error
    if isResume {
        if isFileExists(filePath) {
            startPosition, err = getFileSize(filePath)
        }
    }

    // 创建HTTP请求
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        return err
    }
    if isResume {
        req.Header.Set("Range", fmt.Sprintf("bytes=%d-", startPosition))
    }

    // 开始下载文件或续传文件
    res, err := client.Do(req)
    if err != nil {
        return err
    }
    defer res.Body.Close()

    f, err := os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY, os.ModePerm)
    if err != nil {
        return err
    }
    defer f.Close()

    // 如果需要续传,则先将位置定位到起始位置
    if isResume {
        _, err = f.Seek(startPosition, io.SeekStart)
        if err != nil {
            return err
        }
    }
    _, err = io.Copy(f, res.Body)
    if err != nil {
        return err
    }

    return nil
}

4. Penggunaan

Menggunakan golang-wget adalah sangat mudah, cuma masukkan arahan yang sepadan dalam baris arahan.

Format arahan adalah seperti berikut:

golang-wget -url=URL -out=FILE_PATH [-resume=true|false] [-proxy=true|false] [-proxyAddr=PROXY_ADDR] [-debug=true|false]

di mana:

-url: alamat URL muat turun;

-out: laluan fail yang disimpan; >

-resume: sama ada untuk menyokong penyambungan semula titik putus (pilihan, lalai palsu); Alamat (pilihan, hanya perlu ditetapkan apabila -proxy=true);

-debug: sama ada untuk mendayakan mod nyahpepijat (pilihan, lalai palsu).

Sebagai contoh, untuk memuat turun https://golang.org/doc/gopher/pkg.png dan menyimpannya dalam fail /tmp/pkg.png setempat, anda boleh menggunakan arahan berikut:

golang-wget -url=https://golang.org/doc/gopher/pkg.png -out=/tmp/pkg.png

Jika anda perlu menyokong penyambungan semula titik putus, anda boleh menambah parameter -resume=true:

golang-wget -url=https://golang.org/doc/gopher/pkg.png -out=/tmp/pkg.png -resume=true

Jika anda perlu memuat turun dari pelayan proksi, anda boleh menambah -proxy=true dan -proxyAddr parameter:

golang-wget -url=https://golang.org/doc/gopher/pkg.png -out=/tmp/pkg.png -proxy=true -proxyAddr=http://proxyserver:port

5. Ringkasan

Melalui pelaksanaan golang-wget, kita dapat melihat bahawa pakej http golang sangat berkuasa dan boleh melaksanakan sambungan protokol HTTP dan muat turun fail dengan mudah. Melalui contoh ini, kita boleh belajar cara menggunakan golang untuk melaksanakan alat muat turun kita sendiri dan melaksanakan beberapa fungsi lanjutan, seperti muat turun boleh disambung semula dan proksi.

Menulis alat muat turun anda sendiri bukan sahaja boleh meningkatkan keupayaan pelaksanaan teknikal anda sendiri, tetapi juga mengurangkan pergantungan anda pada alatan lain dan meningkatkan kecekapan.

Atas ialah kandungan terperinci wget dilaksanakan oleh golang. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Artikel sebelumnya:golang tunda kaedah tanpa namaArtikel seterusnya:golang tunda kaedah tanpa nama