gateway/core/proxy/proxyServer.go

package proxy

import (
	"fmt"
	"log"
	"net/http"

	. "bp.jydev.jianyu360.cn/BaseService/gateway/common/gatecode"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/logs"
	"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/middleware"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/proxy/proxyClient"
	"bp.jydev.jianyu360.cn/BaseService/gateway/core/router"
	"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"
)

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 {
		logs.GInfo.Error(initCtx, err)
	}

	gateWayServer := g.Server()
	//关闭系统自带请求日志
	gateWayServer.SetLogger(logs.GInfo)
	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) //权限过滤

	//加载代理客户端
	proxyClient.ReLoadClient()
	//注册代理
	gateWayServer.BindHandler("POST:/*", 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++ {
		// 根据负载规则获取服务地址
		log.Println(r.GetClientIp(), gCtx.RouterRule.MiddleCode)
		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 i == (errTryTime - 1) {
				r.SetError(NewErrorWithCode(GATEWAY_PROXY_ERR, fmt.Sprintf("代理异常:%s err:%v \n", gCtx.ServerAddr, err.Error())))
			}
		}
		// 代理请求
		proxyClient.CreateCustomProxyClient(proxyAddr, errHandel).ServeHTTP(r.Response.ResponseWriter, r.Request)

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