conn

pkg/discoveryregistry/discovery_register.go

@@ -4,13 +4,16 @@ import (
 	"google.golang.org/grpc"
 )
 
-type SvcDiscoveryRegistry interface {
-	Register(serviceName, host string, port int, opts ...grpc.DialOption) error
-	UnRegister() error
+type Conn interface {
 	GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error)
 	GetConn(serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error)
 	AddOption(opts ...grpc.DialOption)
+}
 
+type SvcDiscoveryRegistry interface {
+	Conn
+	Register(serviceName, host string, port int, opts ...grpc.DialOption) error
+	UnRegister() error
 	RegisterConf2Registry(key string, conf []byte) error
 	GetConfFromRegistry(key string) ([]byte, error)
 }

pkg/discoveryregistry/dns.go

@@ -0,0 +1,54 @@
+package discoveryregistry
+
+import (
+	"context"
+	"fmt"
+	"net"
+
+	"google.golang.org/grpc"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/rest"
+)
+
+type DnsDiscoveryRegistry struct {
+	opts      []grpc.DialOption
+	namespace string
+}
+
+func (d DnsDiscoveryRegistry) GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) {
+	config, err := rest.InClusterConfig()
+	if err != nil {
+		return nil, err
+	}
+	clientset, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		return nil, err
+	}
+	endpoints, err := clientset.CoreV1().Endpoints(d.namespace).Get(context.TODO(), serviceName, metav1.GetOptions{})
+	if err != nil {
+		return nil, err
+	}
+	var conns []*grpc.ClientConn
+	for _, subset := range endpoints.Subsets {
+		for _, address := range subset.Addresses {
+			for _, port := range subset.Ports {
+				conn, err := grpc.Dial(net.JoinHostPort(address.IP, string(port.Port)), opts...)
+				if err != nil {
+					return nil, err
+				}
+				conns = append(conns, conn)
+			}
+		}
+	}
+	return conns, nil
+}
+
+func (d DnsDiscoveryRegistry) GetConn(serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
+	return grpc.Dial(fmt.Sprintf("%s.%s.svc.cluster.local", serviceName, d.namespace), opts...)
+}
+
+func (d *DnsDiscoveryRegistry) AddOption(opts ...grpc.DialOption) {
+	d.opts = append(d.opts, opts...)
+}