31 lines
884 B
Go
31 lines
884 B
Go
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
|
||
} |