gateway/core/proxy/proxyServer.go

package proxy

import (
	. "bp.jydev.jianyu360.cn/BaseService/gateway/common/gatecode"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/node"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/proxy/broker"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/proxy/broker/outServer"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/proxy/middleware"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/proxy/proxyClient"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/router"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/gogf/gf/v2/os/gcfg"
	"github.com/gogf/gf/v2/os/gctx"
	"log"
	"net/http"
	"net/url"
)

var bManager = broker.InitBroker()
var routerManager *router.Manager

const errTryTime = 10 //错误尝试

// InitGateWayServer 初始化网关服务
func InitGateWayServer() *ghttp.Server {
	initCtx := gctx.New()
	//创建节点,并持续观察节点变化
	watchNode := node.NewNode(gcfg.Instance().MustGet(initCtx, "system.etcdListen", nil).Strings()...)
	go watchNode.NewWatcher(initCtx, bManager)

	//初始化可访问路由
	var err error
	routerManager, err = router.InitRouterManager()
	if err != nil {
		g.Log().Error(initCtx, err)
	}
	//初始化外部服务
	sussBiServer := outServer.InitSussBi(gcfg.Instance().MustGet(initCtx, "outServer.sussbi.addr", nil).String(),
		gcfg.Instance().MustGet(initCtx, "outServer.sussbi.user", nil).String(),
		gcfg.Instance().MustGet(initCtx, "outServer.sussbi.password", nil).String())
	bManager.RegisterOutServer(sussBiServer.Url, sussBiServer)

	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.FilterPolyHandler) //反爬虫策略
	gateWayServer.Use(middleware.FilterHandler)     //权限过滤

	//加载代理客户端
	proxyClient.ReLoadClient()
	//注册代理
	gateWayServer.BindHandler("POST:/*", proxyHandler)
	gateWayServer.BindHandler("GET:/*", proxyHandler)
	return gateWayServer
}

// proxyHandler 网关代理Handler处理
// 完成所有前置校验后，请求代理服务逻辑
var proxyHandler = func(r *ghttp.Request) {
	if r.GetError() != nil {
		return
	}
	// 获取请求上下文内容
	gCtx := router.GetGContext(r.GetCtx())
	// 请求重试，防止某个服务中断不可用，导致接口不可用。
	for i := 0; i < errTryTime; i++ {
		var proxyAddr *url.URL
		var err error
		if gCtx.RouterRule.IsOutServer {
			//外部服务直接获取
			proxyAddr, err = bManager.GetOutSeverAutoLogin(gCtx.RouterRule.MiddleCode, r)
		} else {
			// 从etcd注册的地址，根据负载规则获取服务地址
			proxyAddr, err = bManager.GetServerAddr(gCtx.RouterRule.MiddleCode, r.GetClientIp())
		}
		if err != nil {
			r.SetError(err)
			return
		}

		// 代理地址存入上下文ctx中
		gCtx.ServerAddr = proxyAddr.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())))
			}
			g.Log().Error(r.Context(), "ErrorHandler ", err)
		}
		log.Println("xxxx", r.Request.Header.Get("Cookie"))
		// 代理请求
		proxyClient.CreateCustomProxyClient(proxyAddr, errHandel).ServeHTTP(r.Response.ResponseWriter, r.Request)

		//if gCtx.RouterRule.IsOutServer {
		//	if r.Response.Status == 401 {
		//		hasErr = true
		//		log.Println("do login Again")
		//		r.Request.Header.Set("Cookie", "JSESSIONID=A4745CB36FA67EE15BA162033B0343F5")
		//		r.Response.Flush()
		//	}
		//}
		//log.Println("qqqqq", r.Request.RequestURI, r.Response.Status)
		// 未捕获到请求，标识请求成功
		if !hasErr {
			return
		}
	}
}