feat(server): 添加反向代理网关功能并升级依赖
- 新增 gateway.go 文件实现 HTTP 反向代理功能 - 添加 BuildRequest 函数用于创建反向代理处理器 - 实现自定义 Director 保留原始路径和查询参数 - 集成错误处理机制返回适当的 HTTP 状态码 - 升级 gf-common 依赖从 v1.0.1010 到 v1.0.1011 - 更新 go.mod 和 go.sum 文件依赖信息 - 更新 IDE 缓存配置文件中的路径映射信息
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/http/httputil"
|
||||
"net/url"
|
||||
|
||||
"git.magicany.cc/black1552/gin-base/log"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// BuildRequest 创建 HTTP 反向代理 Handler
|
||||
// @Param host 目标服务器地址,例如 "http://127.0.0.1:8081"
|
||||
func BuildRequest(host string) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
targetURL, err := url.Parse(host)
|
||||
if err != nil {
|
||||
log.Error("parse target host error:", err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "invalid target host"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
// 创建反向代理
|
||||
proxy := httputil.NewSingleHostReverseProxy(targetURL)
|
||||
|
||||
// 自定义 Director 修改请求,保留原始路径和查询参数
|
||||
originalDirector := proxy.Director
|
||||
proxy.Director = func(req *http.Request) {
|
||||
originalDirector(req)
|
||||
// 保留原始请求的路径和查询参数
|
||||
req.URL.Path = c.Request.URL.Path
|
||||
req.URL.RawQuery = c.Request.URL.RawQuery
|
||||
req.Host = targetURL.Host
|
||||
}
|
||||
|
||||
// 处理代理错误
|
||||
proxy.ErrorHandler = func(w http.ResponseWriter, r *http.Request, err error) {
|
||||
log.Error("proxy error:", err)
|
||||
c.JSON(http.StatusBadGateway, gin.H{"error": "proxy error: " + err.Error()})
|
||||
}
|
||||
|
||||
proxy.ServeHTTP(c.Writer, c.Request)
|
||||
c.Abort()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user