add consul package

pkg/consul/discovery.go

@@ -0,0 +1,28 @@
+package consul
+
+import (
+	"errors"
+	"fmt"
+
+	consulapi "github.com/hashicorp/consul/api"
+)
+
+// Discovery 服务发现
+func Discovery(addr string, serviceName string) (string, error) {
+	cfg := &consulapi.Config{Address: addr}
+	client, err := consulapi.NewClient(cfg)
+	if err != nil {
+		return "", err
+	}
+
+	services, _, err := client.Health().Service(serviceName, "", true, nil)
+	if err != nil {
+		return "", err
+	}
+
+	if len(services) == 0 {
+		return "", errors.New("没有找到服务")
+	}
+
+	return fmt.Sprintf("%s:%d", services[0].Service.Address, services[0].Service.Port), nil
+}

pkg/consul/registration.go

@@ -0,0 +1,105 @@
+package consul
+
+import (
+	"context"
+	"fmt"
+	"net"
+
+	"github.com/google/uuid"
+	consulapi "github.com/hashicorp/consul/api"
+	"google.golang.org/grpc/health/grpc_health_v1"
+)
+
+type Service struct {
+	client      *consulapi.Client
+	serviceID   string
+	serviceName string
+	serviceHost string
+	servicePort int
+}
+
+// Health struct health
+type Health struct{}
+
+func NewConsul(address string, serviceName string, servicePort int) (*Service, error) {
+	config := consulapi.DefaultConfig()
+	config.Address = address
+	client, err := consulapi.NewClient(config)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Service{
+		client:      client,
+		serviceName: serviceName,
+		serviceHost: localIP(),
+		servicePort: servicePort,
+	}, nil
+}
+
+func (c *Service) Register() error {
+	uid, err := uuid.NewRandom()
+	if err != nil {
+		return err
+	}
+
+	registration := &consulapi.AgentServiceRegistration{
+		ID:      uid.String(),
+		Name:    c.serviceName,
+		Address: c.serviceHost,
+		Port:    c.servicePort,
+		Tags:    []string{"primary"},
+		// http
+		//Check: &consulapi.AgentServiceCheck{
+		//	HTTP:                           fmt.Sprintf("http://%s:%d/health", serverHost, serverPort),
+		//	Timeout:                        "5s",
+		//	Interval:                       "5s",
+		//	DeregisterCriticalServiceAfter: "120s", // 故障检查失败120s后 consul自动将注册服务删除
+		//},
+		// grpc
+		Check: &consulapi.AgentServiceCheck{
+			GRPC:                           fmt.Sprintf("%v:%v/%v", c.serviceHost, c.servicePort, c.serviceName),
+			Interval:                       "5s",
+			Timeout:                        "2s",
+			DeregisterCriticalServiceAfter: "20s", // 故障检查失败120s后 consul自动将注册服务删除
+		},
+	}
+
+	err = c.client.Agent().ServiceRegister(registration)
+	if err != nil {
+		return err
+	}
+
+	c.serviceID = uid.String()
+	return nil
+}
+
+func (c *Service) Deregister() error {
+	return c.client.Agent().ServiceDeregister(c.serviceID)
+}
+
+// Check 实现健康检查接口，这里直接返回健康状态，这里也可以有更复杂的健康检查策略，比如根据服务器负载来返回
+func (h *Health) Check(ctx context.Context, req *grpc_health_v1.HealthCheckRequest) (*grpc_health_v1.HealthCheckResponse, error) {
+	return &grpc_health_v1.HealthCheckResponse{
+		Status: grpc_health_v1.HealthCheckResponse_SERVING,
+	}, nil
+}
+
+func (h *Health) Watch(req *grpc_health_v1.HealthCheckRequest, w grpc_health_v1.Health_WatchServer) error {
+	return nil
+}
+
+func localIP() string {
+	addrs, err := net.InterfaceAddrs()
+	if err != nil {
+		return ""
+	}
+	for _, address := range addrs {
+		if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
+			if ipnet.IP.To4() != nil {
+				return ipnet.IP.String()
+			}
+		}
+	}
+	return ""
+}