Go语言反向代理在视频流导航中的应用与优化实践

package main

import (
    "net/http"
    "log"
)

type proxy struct {
    // ...
}

func (p *proxy) start() {
    http.HandleFunc("/play", p.connection)
    log.Fatal(http.ListenAndServe("localhost:8040", nil))
}

func (p *proxy) connection(w http.ResponseWriter, r *http.Request) {
    // 创建一个用于通知断开连接的通道
    disconnect := make(chan bool, 1)
    go p.send(w, r, disconnect)

    // 阻塞直到发送协程完成
    <-disconnect
}

func (p *proxy) send(rv http.ResponseWriter, rvq *http.Request, disconnect chan bool) {
    // 构建新的请求，目标是后端视频服务器
    rq, err := http.NewRequest("GET", "http://localhost/31285611", rvq.Body)
    if err != nil {
        log.Printf("Error creating request: %v", err)
        http.Error(rv, "Internal Server Error", http.StatusInternalServerError)
        disconnect <- true
        return
    }
    // 复制原始请求的头部，包括Range头部
    rq.Header = rvq.Header

    // 发送请求到后端服务器
    rs, err := http.DefaultClient.Do(rq)
    if err != nil {
        log.Printf("Error sending request to backend: %v", err)
        http.Error(rv, "Internal Server Error", http.StatusInternalServerError)
        disconnect <- true
        return
    }
    defer rs.Body.Close()

    // 复制后端响应的头部到客户端响应
    for k, v := range rs.Header {
        rv.Header().Set(k, v[0])
    }
    rv.WriteHeader(http.StatusOK)

    // 缓冲区，用于读取和写入数据
    buf := make([]byte, 1024)

    // 示例：仅发送前100000个块（这会导致视频无法完整播放和导航失败）
    for i := 0; i < 100000; i++ {
        n, e := rs.Body.Read(buf[0:])
        if n == 0 || e != nil {
            break // 读取完毕或发生错误
        }
        _, err := rv.Write(buf[0:n])
        if err != nil {
            log.Printf("Error writing to client: %v", err)
            break
        }
    }

    disconnect <- true
}

func main() {
    (&proxy{}).start()
}

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
)

func main() {
    // 定义后端视频服务器的URL
    backendURL, err := url.Parse("http://localhost/31285611") // 替换为你的实际后端视频URL
    if err != nil {
        log.Fatalf("Failed to parse backend URL: %v", err)
    }

    // 创建一个SingleHostReverseProxy实例
    // 这个代理会将所有请求转发到指定的backendURL
    proxy := httputil.NewSingleHostReverseProxy(backendURL)

    // 可以选择添加自定义的请求修改逻辑，例如添加或修改头部
    // proxy.Director = func(req *http.Request) {
    //     req.URL.Scheme = backendURL.Scheme
    //     req.URL.Host = backendURL.Host
    //     req.URL.Path = backendURL.Path // 如果需要固定路径，否则保持原样
    //     // req.Host = req.URL.Host // 可选：设置Host头部为后端Host
    //     // 确保原始请求的Host头部被保留或根据需要修改
    //     if _, ok := req.Header["User-Agent"]; !ok {
    //         // explicitly disable User-Agent to make sure requests are not blocked
    //         req.Header.Set("User-Agent", "")
    //     }
    // }

    // 监听8040端口，并将所有请求通过proxy处理
    log.Printf("Starting reverse proxy on :8040, proxying to %s", backendURL.String())
    err = http.ListenAndServe(":8040", proxy)
    if err != nil {
        log.Fatalf("ListenAndServe failed: %v", err)
    }
}