Go使用consul做服务发现

开发 前端
通过使用 consul api 我们可以简单的实现基于 consul 的服务发现,在通过结合 http rpc 就可简单的实现服务的调用,下面一章来简单讲下 go 如何发起 http 请求,为我们做 rpc 做个铺垫

一、目标

二、使用步骤

1. 安装 consul

我们可以直接使用官方提供的二进制文件来进行安装部署,其官网地址为 https://www.consul.io/downloads

 

Go 使用 consul 做服务发现

下载后为可执行文件,在我们开发试验过程中,可以直接使用 consul agent -dev 命令来启动一个单节点的 consul

在启动的打印日志中可以看到 agent: Started HTTP server on 127.0.0.1:8500 (tcp), 我们可以在浏览器直接访问 127.0.0.1:8500 即可看到如下

 

Go 使用 consul 做服务发现

这里我们的 consul 就启动成功了

2. 服务注册

在网络编程中,一般会提供项目的 IP、PORT、PROTOCOL,在服务治理中,我们还需要知道对应的服务名、实例名以及一些自定义的扩展信息

在这里使用 ServiceInstance 接口来规定注册服务时必须的一些信息,同时用 DefaultServiceInstance 实现

  1. type ServiceInstance interface { 
  2.     // return The unique instance ID as registered. 
  3.     GetInstanceId() string 
  4.     // return The service ID as registered. 
  5.     GetServiceId() string 
  6.     // return The hostname of the registered service instance. 
  7.     GetHost() string 
  8.     // return The port of the registered service instance. 
  9.     GetPort() int    // return Whether the port of the registered service instance uses HTTPS. 
  10.     IsSecure() bool    // return The key / value pair metadata associated with the service instance. 
  11.     GetMetadata() map[string]string 
  12. }type DefaultServiceInstance struct { 
  13.     InstanceId string 
  14.     ServiceId  string 
  15.     Host       string 
  16.     Port       int    Secure     bool    Metadata   map[string]string 
  17. }func NewDefaultServiceInstance(serviceId string, host string, port int, secure bool, 
  18.     metadata map[string]string, instanceId string) (*DefaultServiceInstance, error) { 
  19.     // 如果没有传入 IP 则获取一下,这个方法在多网卡的情况下,并不好用    if len(host) == 0 { 
  20.         localIP, err := util.GetLocalIP()        if err != nil { 
  21.             return nil, err 
  22.         }        host = localIP    }    if len(instanceId) == 0 { 
  23.         instanceId = serviceId + "-" + strconv.FormatInt(time.Now().Unix(), 10) + "-" + strconv.Itoa(rand.Intn(9000)+1000) 
  24.     }    return &DefaultServiceInstance{InstanceId: instanceId, ServiceId: serviceId, Host: host, Port: port, Secure: secure, Metadata: metadata}, nil 
  25. }func (serviceInstance DefaultServiceInstance) GetInstanceId() string { 
  26.     return serviceInstance.InstanceId 
  27. }func (serviceInstance DefaultServiceInstance) GetServiceId() string { 
  28.     return serviceInstance.ServiceId 
  29. }func (serviceInstance DefaultServiceInstance) GetHost() string { 
  30.     return serviceInstance.Host 
  31. }func (serviceInstance DefaultServiceInstance) GetPort() int {    return serviceInstance.Port 
  32. }func (serviceInstance DefaultServiceInstance) IsSecure() bool {    return serviceInstance.Secure 
  33. }func (serviceInstance DefaultServiceInstance) GetMetadata() map[string]string { 
  34.     return serviceInstance.Metadata 

定义接口

在上面规定了需要注册的服务的必要信息,下面定义下服务注册和剔除的方法

  1. type ServiceRegistry interface { 
  2.     Register(serviceInstance cloud.ServiceInstance) bool 
  3.     Deregister() 

具体实现

因为 consul 提供了 http 接口来对 consul 进行操作,我们也可以使用 http 请求方式进行注册和剔除操作,具体 http 接口文档见 https://www.consul.io/api-docs, consul 默认提供了go 语言的实现,这里直接使用 github.com/hashicorp/consul/api

  1. import ( 
  2.     "errors" 
  3.     "fmt" 
  4.     "github.com/hashicorp/consul/api" 
  5.     "strconv" 
  6.     "unsafe" 
  7. )type consulServiceRegistry struct { 
  8.     serviceInstances     map[string]map[string]cloud.ServiceInstance 
  9.     client               api.Client    localServiceInstance cloud.ServiceInstance}func (c consulServiceRegistry) Register(serviceInstance cloud.ServiceInstance) bool {    // 创建注册到consul的服务到    registration := new(api.AgentServiceRegistration)    registration.ID = serviceInstance.GetInstanceId()    registration.Name = serviceInstance.GetServiceId()    registration.Port = serviceInstance.GetPort()    var tags []string 
  10.     if serviceInstance.IsSecure() { 
  11.         tags = append(tags, "secure=true"
  12.     } else { 
  13.         tags = append(tags, "secure=false"
  14.     }    if serviceInstance.GetMetadata() != nil { 
  15.         var tags []string 
  16.         for key, value := range serviceInstance.GetMetadata() { 
  17.             tags = append(tags, key+"="+value) 
  18.         }        registration.Tags = tags    }    registration.Tags = tags    registration.Address = serviceInstance.GetHost()    // 增加consul健康检查回调函数    check := new(api.AgentServiceCheck)    schema := "http" 
  19.     if serviceInstance.IsSecure() { 
  20.         schema = "https" 
  21.     }    check.HTTP = fmt.Sprintf("%s://%s:%d/actuator/health"schema, registration.Address, registration.Port) 
  22.     check.Timeout = "5s" 
  23.     check.Interval = "5s" 
  24.     check.DeregisterCriticalServiceAfter = "20s" // 故障检查失败30s后 consul自动将注册服务删除 
  25.     registration.Check = check    // 注册服务到consul    err := c.client.Agent().ServiceRegister(registration)    if err != nil { 
  26.         fmt.Println(err)        return false 
  27.     }    if c.serviceInstances == nil { 
  28.         c.serviceInstances = map[string]map[string]cloud.ServiceInstance{} 
  29.     }    services := c.serviceInstances[serviceInstance.GetServiceId()]    if services == nil { 
  30.         services = map[string]cloud.ServiceInstance{} 
  31.     }    services[serviceInstance.GetInstanceId()] = serviceInstance    c.serviceInstances[serviceInstance.GetServiceId()] = services    c.localServiceInstance = serviceInstance    return true 
  32. }// deregister a servicefunc (c consulServiceRegistry) Deregister() {    if c.serviceInstances == nil { 
  33.         return 
  34.     }    services := c.serviceInstances[c.localServiceInstance.GetServiceId()]    if services == nil { 
  35.         return 
  36.     }    delete(services, c.localServiceInstance.GetInstanceId())    if len(services) == 0 { 
  37.         delete(c.serviceInstances, c.localServiceInstance.GetServiceId())    }    _ = c.client.Agent().ServiceDeregister(c.localServiceInstance.GetInstanceId())    c.localServiceInstance = nil 
  38. }// new a consulServiceRegistry instance// token is optionalfunc NewConsulServiceRegistry(host string, port int, token string) (*consulServiceRegistry, error) { 
  39.     if len(host) < 3 { 
  40.         return nil, errors.New("check host"
  41.     }    if port <= 0 || port > 65535 { 
  42.         return nil, errors.New("check port, port should between 1 and 65535"
  43.     }    config := api.DefaultConfig() 
  44.     config.Address = host + ":" + strconv.Itoa(port) 
  45.     config.Token = token 
  46.     client, err := api.NewClient(config) 
  47.     if err != nil { 
  48.         return nil, err 
  49.     }    return &consulServiceRegistry{client: *client}, nil 

测试用例

注册服务的代码基本完成,来测试一下

  1. func TestConsulServiceRegistry(t *testing.T) { 
  2.     host := "127.0.0.1" 
  3.     port := 8500 
  4.     registryDiscoveryClient, _ := extension.NewConsulServiceRegistry(host, port, ""
  5.     ip, err := util.GetLocalIP()    if err != nil { 
  6.         t.Error(err)    }    serviceInstanceInfo, _ := cloud.NewDefaultServiceInstance("go-user-server""", 8090, 
  7.         false, map[string]string{"user":"zyn"}, ""
  8.     registryDiscoveryClient.Register(serviceInstanceInfo)    r := gin.Default()    // 健康检测接口,其实只要是 200 就认为成功了 
  9.     r.GET("/actuator/health", func(c *gin.Context) { 
  10.         c.JSON(200, gin.H{ 
  11.             "message""pong"
  12.         }) 
  13.     }) 
  14.     err = r.Run(":8090"
  15.     if err != nil{ 
  16.         registryDiscoveryClient.Deregister() 
  17.     } 

如果成功,则会在 consul 看到 go-user-server 这个服务

3. 服务发现

在服务发现中,一般会需要两个方法

  • 获取所有的服务列表
  • 获取指定的服务的所有实例信息

接口定义

  1. type DiscoveryClient interface { 
  2.     /** 
  3.      * Gets all ServiceInstances associated with a particular serviceId. 
  4.      * @param serviceId The serviceId to query. 
  5.      * @return A List of ServiceInstance. 
  6.      */ 
  7.     GetInstances(serviceId string) ([]cloud.ServiceInstance, error)    /** 
  8.      * @return All known service IDs. 
  9.      */ 
  10.     GetServices() ([]string, error)} 

具体实现

来实现一下

  1. type consulServiceRegistry struct { 
  2.     serviceInstances     map[string]map[string]cloud.ServiceInstance 
  3.     client               api.Client    localServiceInstance cloud.ServiceInstance}func (c consulServiceRegistry) GetInstances(serviceId string) ([]cloud.ServiceInstance, error) { 
  4.     catalogService, _, _ := c.client.Catalog().Service(serviceId, "", nil) 
  5.     if len(catalogService) > 0 { 
  6.         result := make([]cloud.ServiceInstance, len(catalogService)) 
  7.         for index, sever := range catalogService { 
  8.             s := cloud.DefaultServiceInstance{                InstanceId: sever.ServiceID,                ServiceId:  sever.ServiceName,                Host:       sever.Address,                Port:       sever.ServicePort,                Metadata:   sever.ServiceMeta,            }            result[index] = s        }        return result, nil 
  9.     }    return nil, nil 
  10. }func (c consulServiceRegistry) GetServices() ([]string, error) { 
  11.     services, _, _ := c.client.Catalog().Services(nil) 
  12.     result := make([]string, unsafe.Sizeof(services)) 
  13.     index := 0 
  14.     for serviceName, _ := range services { 
  15.         result[index] = serviceName        index++    }    return result, nil 
  16. }// new a consulServiceRegistry instance 
  17. // token is optional 
  18. func NewConsulServiceRegistry(host string, port int, token string) (*consulServiceRegistry, error) { 
  19.     if len(host) < 3 { 
  20.         return nil, errors.New("check host"
  21.     } 
  22.     if port <= 0 || port > 65535 { 
  23.         return nil, errors.New("check port, port should between 1 and 65535"
  24.     } 
  25.     config := api.DefaultConfig() 
  26.     config.Address = host + ":" + strconv.Itoa(port) 
  27.     config.Token = token 
  28.     client, err := api.NewClient(config) 
  29.     if err != nil { 
  30.         return nil, err 
  31.     } 
  32.     return &consulServiceRegistry{client: *client}, nil 

测试用例

  1. func TestConsulServiceDiscovery(t *testing.T) { 
  2.     host := "127.0.0.1" 
  3.     port := 8500 
  4.     token := "" 
  5.     registryDiscoveryClient, err := extension.NewConsulServiceRegistry(host, port, token) 
  6.     if err != nil {        panic(err) 
  7.     }    t.Log(registryDiscoveryClient.GetServices()) 
  8.     t.Log(registryDiscoveryClient.GetInstances("go-user-server")) 

结果

  1. consul_service_registry_test.go:57: [consul go-user-server      ] <nil> 
  2. consul_service_registry_test.go:59: [{go-user-server-1602590661-56179 go-user-server 127.0.0.1 8090 false map[user:zyn]}] <nil> 

总结

通过使用 consul api 我们可以简单的实现基于 consul 的服务发现,在通过结合 http rpc 就可简单的实现服务的调用,下面一章来简单讲下 go 如何发起 http 请求,为我们做 rpc 做个铺垫

责任编辑:未丽燕 来源: 今日头条
相关推荐

2023-06-02 08:33:43

微服务架构服务注册

2022-01-16 23:10:40

语言服务注册

2021-07-07 05:46:46

运维监控Prometheus

2017-06-25 13:33:25

Spring Clou微服务架构

2022-01-26 09:36:53

Consul语言微服务

2018-12-27 09:28:08

Consul服务Server

2021-09-30 08:54:58

prometheus监控远端服务

2023-07-04 07:45:11

gogRPC服务

2023-04-03 07:17:34

CP集群AP

2023-09-07 23:25:34

微服务服务发现

2015-06-03 10:01:56

云平台发现服务ZooKeeper

2015-04-28 15:14:53

云平台发现服务ZooKeeper

2021-01-08 13:52:15

Consul微服务服务注册中心

2018-11-22 15:07:17

代码github程序

2020-06-29 07:58:18

ZooKeeperConsul 注册中心

2021-01-21 10:02:45

Consul架构安装

2022-08-14 07:04:44

微服务架构设计模式

2019-12-24 09:39:06

Kubernetes工具微服务

2022-06-17 12:05:25

微服务注册

2021-02-18 22:21:20

ASM服务组件化
点赞
收藏

51CTO技术栈公众号