go实现http代理服务并下载指定类型文件代码-golang代码-BFW代码库

go实现http代理服务并下载指定类型文件代码

代码语言：golang

所属分类：其他

代码描述：go实现http代理服务并下载指定类型文件代码

下面为部分代码预览，完整代码请点击下载或在bfwstudio webide中打开

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)
        return
    }
    defer resp.Body.Close()

    // 复制响应头
    for key, values := range resp.Header {
        for _, value := range values {
            w.Header().Add(key, value)
        }
    }

    // 检查响应内容类型
    contentType := resp.Header.Get("Content-Type")
    if shouldDownloadFile.........完整代码请登录后点击上方下载按钮下载查看