57 lines
1.9 KiB
Go
57 lines
1.9 KiB
Go
package proxy
|
||
|
||
import (
|
||
"net/http"
|
||
"net/http/httputil"
|
||
"net/url"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// NewProxy 创建反向代理。
|
||
// 路径改写：去除 /api 前缀，保留 /v1 下游 controller 前缀。
|
||
// 适用于下游 controller 路径含 /v1 前缀的服务（iam/core-edu/content/msg/ai/data-ana）：
|
||
// /api/v1/{prefix}/* → 剥离 /api → /v1/{prefix}/* 转发下游。
|
||
func NewProxy(targetURL string) (*httputil.ReverseProxy, error) {
|
||
target, err := url.Parse(targetURL)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
proxy := httputil.NewSingleHostReverseProxy(target)
|
||
originalDirector := proxy.Director
|
||
proxy.Director = func(req *http.Request) {
|
||
originalDirector(req)
|
||
req.URL.Path = strings.TrimPrefix(req.URL.Path, "/api")
|
||
req.Host = target.Host
|
||
}
|
||
return proxy, nil
|
||
}
|
||
|
||
// NewProxyRewrite 创建带自定义路径重写的反向代理。
|
||
// 用于 BFF 路由（teacher/student/parent）和 admin 路由：
|
||
// - BFF：/api/v1/{bff}/graphql → 剥离 /api/v1/{bff} → /graphql（teacher-bff/student-bff）
|
||
// - BFF：/api/v1/{bff}/v1/graphql → 剥离 /api/v1/{bff} → /v1/graphql（parent-bff，ARB-022 §24.4 ISSUE-003 方案 A）
|
||
// - admin：/api/admin/graphql → 剥离 /api/admin → /graphql（teacher-bff admin 命名空间）
|
||
func NewProxyRewrite(targetURL string, pathRewriter func(string) string) (*httputil.ReverseProxy, error) {
|
||
target, err := url.Parse(targetURL)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
proxy := httputil.NewSingleHostReverseProxy(target)
|
||
originalDirector := proxy.Director
|
||
proxy.Director = func(req *http.Request) {
|
||
originalDirector(req)
|
||
req.URL.Path = pathRewriter(req.URL.Path)
|
||
req.Host = target.Host
|
||
}
|
||
return proxy, nil
|
||
}
|
||
|
||
// ProxyHandler 返回 Gin 处理函数
|
||
func ProxyHandler(proxy *httputil.ReverseProxy) gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
proxy.ServeHTTP(c.Writer, c.Request)
|
||
}
|
||
}
|