Go语言文件下载：小文件与大文件处理的最佳实践

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "os"
)

// downloadSmallFile 下载小文件
func downloadSmallFile(url, filename string) error {
    // 1. 发起HTTP GET请求
    resp, err := http.Get(url)
    if err != nil {
        return fmt.Errorf("发送HTTP请求失败: %w", err)
    }
    // 确保响应体在使用完毕后关闭，避免资源泄露
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        return fmt.Errorf("服务器返回非200状态码: %d %s", resp.StatusCode, resp.Status)
    }

    // 2. 将响应体全部读入内存
    bodyBytes, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        return fmt.Errorf("读取响应体失败: %w", err)
    }

    // 3. 将内存中的字节写入文件
    // 0666 是文件权限，表示所有用户可读写
    err = ioutil.WriteFile(filename, bodyBytes, 0666)
    if err != nil {
        return fmt.Errorf("写入文件失败: %w", err)
    }

    fmt.Printf("小文件 '%s' 下载成功！\n", filename)
    return nil
}

func main() {
    // 替换为实际的图片或小文件URL
    fileURL := "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"
    fileName := "google_logo.png"

    if err := downloadSmallFile(fileURL, fileName); err != nil {
        fmt.Printf("下载文件失败: %v\n", err)
        os.Exit(1)
    }
}

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
)

// downloadLargeFile 下载大文件
func downloadLargeFile(url, filename string) error {
    // 1. 发起HTTP GET请求
    resp, err := http.Get(url)
    if err != nil {
        return fmt.Errorf("发送HTTP请求失败: %w", err)
    }
    defer resp.Body.Close() // 确保响应体在使用完毕后关闭

    if resp.StatusCode != http.StatusOK {
        return fmt.Errorf("服务器返回非200状态码: %d %s", resp.StatusCode, resp.Status)
    }

    // 2. 创建本地文件用于写入
    file, err := os.Create(filename)
    if err != nil {
        return fmt.Errorf("创建文件失败: %w", err)
    }
    defer file.Close() // 确保文件在使用完毕后关闭

    // 3. 使用io.Copy将响应体内容流式写入文件
    // io.Copy 会从 resp.Body (io.Reader) 读取数据，并写入到 file (io.Writer)
    bytesWritten, err := io.Copy(file, resp.Body)
    if err != nil {
        return fmt.Errorf("写入文件失败: %w", err)
    }

    fmt.Printf("大文件 '%s' 下载成功！写入字节数: %d\n", filename, bytesWritten)
    return nil
}

func main() {
    // 演示下载小文件
    smallFileURL := "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"
    smallFileName := "google_logo_small.png"

    fmt.Println("--- 演示小文件下载 ---")
    if err := downloadSmallFile(smallFileURL, smallFileName); err != nil {
        fmt.Printf("下载小文件失败: %v\n", err)
    }

    fmt.Println("\n--- 演示大文件下载 (请替换为实际大文件URL进行测试) ---")
    // 注意：以下URL仅为示例，可能无法直接访问或文件不存在
    // 实际使用时请替换为可访问的大文件URL进行测试。
    // largeFileURL := "http://example.com/large_file.zip" // 假设这是一个大文件
    // largeFileName := "downloaded_large_file.zip"
    // if err := downloadLargeFile(largeFileURL, largeFileName); err != nil {
    //  fmt.Printf("下载大文件失败: %v\n", err)
    // }
    fmt.Println("大文件下载代码已展示，请自行替换URL并测试。")
}