初始化2

bj_power_wms_client/internal/proxy/proxy.go

@@ -0,0 +1,31 @@
+package proxy
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/httputil"
+	"net/url"
+)
+
+// New 创建指向 WMS 后端的反向代理。
+// 请求转发时统一注入内部 API Token（X-API-TOKEN），由 WMS 依此鉴权；
+// 响应头默认全量透传（含滑动续签的 X-Renewed-Token，由 WMS 写入）。
+func New(upstream, apiToken string) *httputil.ReverseProxy {
+	target, err := url.Parse(upstream)
+	if err != nil {
+		panic(fmt.Sprintf("WmsAddr 配置非法: %v", err))
+	}
+	rp := httputil.NewSingleHostReverseProxy(target)
+	origDirector := rp.Director
+	rp.Director = func(req *http.Request) {
+		origDirector(req)
+		req.Host = target.Host
+		if apiToken != "" {
+			req.Header.Set("X-API-TOKEN", apiToken)
+		}
+	}
+	rp.ErrorHandler = func(w http.ResponseWriter, r *http.Request, err error) {
+		http.Error(w, "WMS 服务不可达: "+err.Error(), http.StatusBadGateway)
+	}
+	return rp
+}