golang httputil NewSingleHostReverseProxy 代理一个流式响应-Go服务器开发

以下是一个简单的示例，演示如何使用httputil包中的NewSingleHostReverseProxy函数代理一个流式响应。

package main

import (
	"io"
	"net/http"
	"net/http/httptest"
	"net/http/httputil"
	"net/url"
)

func main() {
	// 新建一个测试服务器并返回其URL
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// 发送一个大型响应，以便我们可以在代理中逐步读取它
		for i := 0; i < 100000; i++ {
			w.Write([]byte("chunk\n"))
			w.(http.Flusher).Flush()
		}
	}))
	defer ts.Close()

	// 解析测试服务器的URL，以便我们可以在代理中使用它
	targetURL, _ := url.Parse(ts.URL)

	// 新建一个反向代理，并配置它以将请求转发到测试服务器
	reverseProxy := httputil.NewSingleHostReverseProxy(targetURL)

	// 修改Director方法，以便在请求被发送之前对请求进行修改
	reverseProxy.Director = func(req *http.Request) {
		req.Header.Set("X-Test-Header", "foobar")
	}

	// 创建一个新的HTTP请求
	request, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)

	// 发送请求并接收响应
	response, _ := http.DefaultClient.Do(request)

	// 将响应作为流式响应代理到客户端
	for k, v := range response.Header {
		for _, vv := range v {
			w.Header().Add(k, vv)
		}
	}
	w.WriteHeader(response.StatusCode)

	io.Copy(w, response.Body)
}

在上面的例子中，我们首先创建了一个测试服务器。然后，我们解析这个服务器的URL，并使用它来新建一个反向代理对象。接下来，我们修改了反向代理的Director方法，以便在请求被发送之前向请求添加自定义标头。

然后，我们创建了一个新的HTTP请求并发送它。注意，我们没有直接读取响应的正文，因为我们将要将其作为流式响应代理到客户端。

接着，我们将响应的Header内容添加到代理的响应中，并通过io.Copy函数逐步将响应正文写入代理的响应中。最终，流式响应会被代理到客户端。