golang搭建的http代理服务如何下载指定类型的文件？

golang代码

package main

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

// 代理服务器地址
const proxyAddress = "0.0.0.0:8080"

// 文件下载目录
const downloadDir = "./downloads"

// 要下载的文件类型
var fileTypes = []string{"image/jpeg", "image/png", "application/pdf"}

func main() {
    http.HandleFunc("/", handleRequestAndRedirect)
    fmt.Printf("Starting proxy server on %s\n", proxyAddress)
    if err := http.ListenAndServe(proxyAddress, nil); err != nil {
        fmt.Printf("Failed to start server: %v\n", err)
    }
}

func handleRequestAndRedirect(w http.ResponseWriter, req *http.Request) {
    client := &http.Client{}

    // 创建新的请求
    newReq, err := http.NewRequest(req.Method, req.URL.String(), req.Body)
    if err != nil {
        http.Error(w, "Failed to create request", http.StatusInternalServerError)
        return
    }

    // 复制请求头
    for key, values := range req.Header {
        for _, value := range values {
            newReq.Header.Add(key, value)
        }
    }

    // 发送请求
    resp, err := client.Do(newReq)
    if err != nil {
        http.Error(w, "Failed to get response", http.StatusInternalServerError)
 ...

点击查看剩余70%