go-kit微服务：一个简单的API网关-CSDN博客

概述

在上一篇文章《go-kit微服务：服务注册与发现》中，算术服务注册至consul，发现服务使用go-kit工具集实现了对算术服务的发现功能。通过查看源码可以发现，其中仅仅实现了一个接口/calculate的处理逻辑，那如果算术服务的接口很多怎么办呢？

这几天这个问题一直困扰着我，我认为go-kit会提供一种合理的解决方案，可是始终没有找到，对于sd.Factory的设计始终不理解，若哪位朋友了解还请指点迷津。

为了解决这个问题，我决定另辟蹊径：根据客户端HTTP请求，动态查询注册中心的服务实例，通过反向代理实现对后台服务的调用。

这样就相当于实现一个简单的网关，凡是符合规则的请求都可以通过此网关调用后端服务。这里的规则是指http请求的资源路径，规则为：/{serviceName}/#。即：路径第一部分为注册中心服务实例名称，其余部分为服务实例的REST路径。如：

/arithmetic/calculate/Add/10/2
复制代码

arithmetic为服务名称;
/calculate/Add/10/2为算术服务的接口。

实现网关

Step-1：实现思路

客户端向网关发起请求，网关解析请求资源路径中的信息，根据服务名称查询注册中心的服务实例，然后使用反向代理技术把客户端请求转发至后端真实的服务实例，请求执行完毕后，再把响应信息返回客户端。

HTTP请求的规则遵循：/{serviceName}/#，否则不予通过。
使用golang提供的反向代理包httputil.ReverseProxy实现一个简单的反向代理，它能够对请求实现负载均衡，随机地把请求发送给服务实例。
使用consul客户端api动态查询服务实例。

Step-2：编写反向代理方法

在arithmetic_consul_demo下创建目录gateway，然后新建go文件main.go。NewReverseProxy方法接受两个参数：consul客户端对象和日志记录工具，返回反向代理对象。该方法的实现过程如下所述：

获取请求路径，检查是否符合规则，不符合规则直接返回；
解析请求路径，获取服务名称（第一个部分为服务名称）；
使用consul客户端查询服务实例，若查询到结果，则随机选择一个作为目标实例；
根据选定的目标实例，设置反向代理参数：Schema、Host、Path。

完整代码如下：

// NewReverseProxy 创建反向代理处理方法
func NewReverseProxy(client *api.Client, logger log.Logger) *httputil.ReverseProxy {

	//创建Director
	director := func(req *http.Request) {

		//查询原始请求路径，如：/arithmetic/calculate/10/5
		reqPath := req.URL.Path
		if reqPath == "" {
			return
		}
		//按照分隔符'/'对路径进行分解，获取服务名称serviceName
		pathArray := strings.Split(reqPath, "/")
		serviceName := pathArray[1]

		//调用consul api查询serviceName的服务实例列表
		result, _, err := client.Catalog().Service(serviceName, "", nil)
		if err != nil {
			logger.Log("ReverseProxy failed", "query service instace error", err.Error())
			return
		}

		if len(result) == 0 {
			logger.Log("ReverseProxy failed", "no such service instance", serviceName)
			return
		}

		//重新组织请求路径，去掉服务名称部分
		destPath := strings.Join(pathArray[2:], "/")

		//随机选择一个服务实例
		tgt := result[rand.Int()%len(result)]
		logger.Log("service id", tgt.ServiceID)

		//设置代理服务地址信息
		req.URL.Scheme = "http"
		req.URL.Host = fmt.Sprintf("%s:%d", tgt.ServiceAddress, tgt.ServicePort)
		req.URL.Path = "/" + destPath
	}
	return &httputil.ReverseProxy{Director: director}

}
复制代码

Step-3：编写main方法

main方法的主要任务是创建consul连接对象、创建日志记录对象、开启反向代理HTTP服务。整个过程与前面几个示例类似，直接贴代码（为了测试方便，我直接指定了consul服务地址信息）：

func main() {

	// 创建环境变量
	var (
		consulHost = flag.String("consul.host", "192.168.192.146", "consul server ip address")
		consulPort = flag.String("consul.port", "8500", "consul server port")
	)
	flag.Parse()

	//创建日志组件
	var logger log.Logger
	{
		logger = log.NewLogfmtLogger(os.Stderr)
		logger = log.With(logger, "ts", log.DefaultTimestampUTC)
		logger = log.With(logger, "caller", log.DefaultCaller)
	}

	// 创建consul api客户端
	consulConfig := api.DefaultConfig()
	consulConfig.Address = "http://" + *consulHost + ":" + *consulPort
	consulClient, err := api.NewClient(consulConfig)
	if err != nil {
		logger.Log("err", err)
		os.Exit(1)
	}

	//创建反向代理
	proxy := NewReverseProxy(consulClient, logger)

	errc := make(chan error)
	go func() {
		c := make(chan os.Signal)
		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
		errc <- fmt.Errorf("%s", <-c)
	}()

	//开始监听
	go func() {
		logger.Log("transport", "HTTP", "addr", "9090")
		errc <- http.ListenAndServe(":9090", proxy)
	}()

	// 开始运行，等待结束
	logger.Log("exit", <-errc)
}
复制代码

Step-4：运行

使用docker启动consul。终端切换至arithmetic_consul_demo目录，执行以下命令：

sudo docker-compose -f docker/docker-compose.yml up
复制代码

启动算术运算服务。为了测试负载均衡效果，我启动了两个实例。注意需要使用不同的端口

./register/register -consul.host localhost -consul.port 8500 -service.host 192.168.192.146 -service.port 9000

./register/register -consul.host localhost -consul.port 8500 -service.host 192.168.192.146 -service.port 9002
复制代码

在浏览器输入http://localhost:8500可看到arithmetic实例有两个，如下图：

在终端cd至目录gateway，执行go build完成编译，然后启动网关服务。

> ./gateway -consul.host localhost -consul.port 8500

> ts=2019-02-26T07:49:39.0468058Z caller=main.go:54 transport=HTTP addr=9090
复制代码

Step-5：测试

使用postman按下图方式执行请求测试，会发现服务调用成功。

同时，在终端可以看到如下输出，说明多次请求访问了不同的服务实例：

ts=2019-02-26T07:49:39.0468058Z caller=main.go:54 transport=HTTP addr=9090
ts=2019-02-26T07:49:46.8559985Z caller=main.go:94 serviceid=arithmetic82460623-ccdc-4192-a042-c0603ef18888
ts=2019-02-26T07:50:00.1249302Z caller=main.go:94 serviceid=arithmetic65153818-27b3-4f19-8fd1-d7d698168f20
ts=2019-02-26T09:04:09.0470362Z caller=main.go:94 serviceid=arithmetic65153818-27b3-4f19-8fd1-d7d698168f20
ts=2019-02-26T09:04:10.176327Z caller=main.go:94 serviceid=arithmetic65153818-27b3-4f19-8fd1-d7d698168f20
复制代码