go-micro使用Consul做服务发现的方法和原理解析

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

go-micro v4默认使用mdns做服务发现。不过也支持采用其它的服务发现中间件，因为多年来一直使用Consul做服务发现，为了方便和其它服务集成，所以还是选择了Consul。这篇文章将介绍go-micro使用Consul做服务发现的方法。关于Consul的使用方式请参考我的另一篇文章：搭建Consul服务发现与服务网格。

安装Consul

如果你已经安装Consul，或者对Consul很熟悉了，按照自己的方式处理Consul就行了。

这里提供一个通过docker快速安装Consul的方式，当然前提是你得安装了docker。

执行命令：

docker run --name consul1 -p 8500:8500 -p 8300:8300 -p 8301:8301 -p 8302:8302  -d consul:latest

这会在docker容器中启动一个最新版本的Consul服务，并将相关端口开放给主机。

安装Consul插件

使用Consul作为服务注册和服务发现，需要先安装go-micro的consul插件：

go get github.com/go-micro/plugins/v4/registry/consul

服务端使用Consul

服务注册

为了使用Consul做服务注册，需要为go-micro server显式的指定Consul Registry。直接看代码吧：

func main() {

	registry := consul.NewRegistry()
	rpcServer := server.NewServer(
		server.Name("registry-consul.service"),
		server.Address("0.0.0.0:8001"),
		server.Registry(registry),
	)
	proto.RegisterHelloHandler(rpcServer, &Hello{})
	service := micro.NewService(
		micro.Server(rpcServer),
	service.Init()
	// Run server
	if err := service.Run(); err != nil {
		log.Fatal(err)
	}
}

通过 consul.NewRegistry() 创建一个Consul 注册中心，然后使用 server.NewServer 创建Server的时候把它设置进去；同时我们需要指定服务的名称，这里设置的是 registry-consul.service；另外这里不使用随机端口，指定了一个服务的监听地址。这样基本就OK了。

这里并没有指定Consul的连接地址，因为按照推荐的Consul部署方式，服务所在机器或者容器中应该部署一个Consul的客户端，程序可以直接通过 127.0.0.1:8500 访问到它。如果要显示指定，可以在NewRegistry时设置：

registry := consul.NewRegistry(
		registry.Addrs("127.0.0.1:8500"),
	)

注册过程

通过一张图来看一下，go-micro注册服务到Consul时都做了什么。

服务注册关键是实现两个动作：

1、注册： rpcServer启动的时候，会调用到自身的Register方法，Register方法会调用consul插件的Register方法，然后调用到consul自身SDK提供的Agent.ServiceRegister方法，将服务注册到Consul中。注册的服务名称就是NewServer时的server.Name。

2、健康上报：即刷新TTL，服务注册成功后，会启动一个定时器定时调用consul插件的Register方法，这个方法内部判断服务注册过，则会调用consul自身SDK提供的Agent.PassTTL方法，刷新Consul中对应服务的TTL。

健康检查

go-micro服务的健康状态是通过TTL维护的，服务需要定时去刷新TTL，如果TTL超过指定的时间没有被刷新，则服务会被认为是不健康的。默认情况下有三个设置会涉及到TTL，还是先来看代码：

registry := consul.NewRegistry()

	regCheckFunc := func(ctx context.Context) error {
		fmt.Println(time.Now().Format("2006-01-02 15:04:05") + " do register check")
		if 1+1 == 2 {
			return nil
		}
		return errors.New("this not earth")
	}
	rpcServer := server.NewServer(
		server.Name("registry-consul.service"),
		server.Address("0.0.0.0:8001"),
		server.Registry(registry),
		server.RegisterCheck(regCheckFunc),
		server.RegisterInterval(10*time.Second),
		server.RegisterTTL(20*time.Second),
	)
	proto.RegisterHelloHandler(rpcServer, &Hello{})
	service := micro.NewService(
		micro.Server(rpcServer),
	//service.Init()
	if err := service.Run(); err != nil {
		log.Fatal(err)

关于这三个设置，这里简单介绍下：

1、server.RegisterCheck(regCheckFunc) 服务刷新TTL之前，会调用一个函数检查服务的状态，这个函数的返回值是error类型。默认的函数不进行任何检查，直接返回nil，代表服务状态正常；我们可以自己写一个函数，进行一些检查逻辑，比如是否要下线维护。如果返回的error不是nil，go-micro会尝试在Consul中注销服务，则调用方将不会再访问到这个服务节点。

2、server.RegisterInterval(10*time.Second) 这个设置指定程序去刷新TLL的频率。

3、server.RegisterTTL(20*time.Second) 这个设置指定TTL的生存周期，如果超过这个时间没有刷新TTL，则Consul会认为服务是不健康。

另外需要注意不要使用service.Init()，因为这里边会覆盖 RegisterInterval 和 RegisterTTL 的设置，除非你不关心这两个参数。关于这个问题可以参考：https://github.com/asim/go-micro/issues/2488

客户端使用Consul

调用服务

为了使用Consul做服务发现，需要为go-micro service显式的指定Consul Registry。还是直接看代码：

registry := consul.NewRegistry()

	service := micro.NewService(
		micro.Client(client.NewClient()),
		micro.Registry(registry),
	)
	service.Init()
	client := proto.NewHelloService("registry-consul.service", service.Client())
	rsp, err := client.Say(context.TODO(), &proto.SayRequest{Name: "BOSSMA"})
	if err != nil {
		fmt.Println(err)
	}
	fmt.Println(rsp)

代码很简单，指定consul作为服务发现组件后，调用服务的时候传递的服务名称就会使用consul进行解析，获取到IP、端口后，再进行实际调用。

发现过程

还是先来看张图，从调用 XXXService 的方法 YYY 开始：

整个过程分为两个大的步骤，第一步获取要调用服务的地址，第二步通过http请求调用服务。我们重点看第一步通过Consul获取服务这块。

首先进入一个Selector，就是选择器的意思。Selector首先看缓存中有没有缓存请求服务的地址信息，如果没有就去Consul查询，查询到之后再通过指定的选择策略选出来一个地址，用于后续http请求。这里默认的选择策略是随机选择，比如查询到这个服务有三个部署节点，随机策略会随机返回其中某一个地址。

除了随机策略，go-micro还提供了一个轮询策略，这时候需要自己创建一个Selector：

registry := consul.NewRegistry()
	selector := selector.NewSelector(
		selector.SetStrategy(selector.RoundRobin),
		selector.Registry(registry),
	)

	service := micro.NewService(
		micro.Client(client.NewClient()),
		micro.Selector(selector),
		//micro.Registry(registry),
	)

另外从上面的示意图中，我们可以看到Selector调用了Registry，所以这里创建Selector的时候，我们把Registry设置了进去，然后再把创建的Selector设置到Service中。其实还有另一种方式，把设置Registry放到设置Selector后边，同样可以把Registry注册到Selector中，但是这样比较隐晦，还需要注意设置顺序，我不推荐。不过如果在Selector之外需要使用Registry的时候，还是需要使用micro.Registry(registry)进行注册，这个例子中并没有相关场景。