tripartite_gateway/core/proxy/proxyServer.go

package proxy

import (
	log "app.yhyue.com/moapp/jylog"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/gogf/gf/v2/net/gtrace"
	"github.com/gogf/gf/v2/os/gctx"
	. "jygit.jydev.jianyu360.cn/dataservice/tripartite_gateway/common/gatecode"
	"jygit.jydev.jianyu360.cn/dataservice/tripartite_gateway/core/proxy/middleware"
	"jygit.jydev.jianyu360.cn/dataservice/tripartite_gateway/core/proxy/proxyClient"
	"jygit.jydev.jianyu360.cn/dataservice/tripartite_gateway/core/router"
	"net/http"
	"net/url"
)

var routerManager *router.Manager

const errTryTime = 10 //错误尝试

// InitGateWayServer 初始化网关服务
func InitGateWayServer(initCtx gctx.Ctx) *ghttp.Server {
	initCtx, span := gtrace.NewSpan(initCtx, "InitGateWayServer")
	defer span.End()

	//初始化可访问路由
	var err error
	routerManager, err = router.InitRouterManager()
	if err != nil {
		log.WithContext(initCtx).Errorln(err)
	}

	gateWayServer := g.Server()
	//关闭系统自带请求日志
	gateWayServer.SetLogger(g.Log())
	gateWayServer.SetErrorLogEnabled(false)

	//注册中间件
	gateWayServer.Use(middleware.ErrorHandler) //错误拦截
	gateWayServer.Use(func(r *ghttp.Request) {
		r.SetError(routerManager.InfusionContext(r)) //context注入全局信息
		r.Middleware.Next()
	})
	gateWayServer.Use(middleware.FilterHandler) //权限过滤

	//注册代理
	gateWayServer.BindHandler("POST:/*", proxyHandler)             //接口代理
	gateWayServer.BindHandler("GET:/*", proxyHandler)              //页面代理
	gateWayServer.AddStaticPath("/antiRes", "./resources/antiRes") //静态资源
	return gateWayServer
}

// proxyHandler 网关代理Handler处理
// 完成所有前置校验后，请求代理服务逻辑
var proxyHandler = func(r *ghttp.Request) {
	_, span := gtrace.NewSpan(r.Context(), "proxyHandler")
	defer span.End()
	if r.GetError() != nil {
		return
	}
	// 获取请求上下文内容
	gCtx := router.GetGContext(r.GetCtx())
	// 请求重试，防止某个服务中断不可用，导致接口不可用。
	for i := 0; i < errTryTime; i++ {
		// 代理地址存入上下文ctx中
		serverUrl, _ := url.Parse(gCtx.RouterRule.Server)
		gCtx.ServerAddr = serverUrl.String()
		router.UpdateGContext(r, gCtx)
		// 捕获异常，若代理出错，则进行重试
		var hasErr bool
		errHandel := func(hw http.ResponseWriter, hr *http.Request, err error) {
			hasErr = true
			if err.Error() == "context canceled" {
				return
			}
			if i == (errTryTime - 1) {
				r.SetError(NewErrorWithCode(GATEWAY_PROXY_ERR, fmt.Sprintf("代理异常:%s err:%v \n", gCtx.ServerAddr, err.Error())))
			}
			log.WithContext(r.Context()).Error("ErrorHandler ", err)
		}
		// 代理请求
		proxyClient.CreateCustomProxyClient(serverUrl, errHandel, nil).ServeHTTP(r.Response.ResponseWriter, r.Request)

		// 未捕获到请求，标识请求成功
		if !hasErr {
			return
		}
	}
}